Keep track of which attachments are referenced by which sync entries.

sync/syncable/directory_unittest.h

 // Copyright 2014 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #ifndef SYNC_SYNCABLE_DIRECTORY_UNITTEST_H_
 #define SYNC_SYNCABLE_DIRECTORY_UNITTEST_H_
 
 #include <string>
 
 #include "base/basictypes.h"
 #include "base/message_loop/message_loop.h"
+#include "sync/internal_api/public/base_node.h"
+#include "sync/internal_api/public/user_share.h"
 #include "sync/syncable/in_memory_directory_backing_store.h"
 #include "sync/syncable/mutable_entry.h"
 #include "sync/syncable/syncable_read_transaction.h"
 #include "sync/syncable/syncable_write_transaction.h"
 #include "sync/test/engine/test_id_factory.h"
 #include "sync/test/fake_encryptor.h"
 #include "sync/test/null_directory_change_delegate.h"
 #include "sync/test/null_transaction_observer.h"
 #include "sync/util/test_unrecoverable_error_handler.h"
 #include "testing/gtest/include/gtest/gtest.h"
@@ skipping 25 matching lines @@
   DirOpenResult ReopenDirectory();
 
   // Creates an empty entry and sets the ID field to a default one.
   void CreateEntry(const std::string& entryname);
 
   // Creates an empty entry and sets the ID field to id.
   void CreateEntry(const std::string& entryname, const int id);
 
   void CreateEntry(const std::string& entryname, Id id);
 
+  // Create an entry with the given |model_type|, |client_tag| and
+  // |attachment_metadata|.
+  void CreateEntryWithAttachmentMetadata(
+      const ModelType& model_type,
+      const std::string& client_tag,
+      const sync_pb::AttachmentMetadata& attachment_metadata);
+
   // When a directory is saved then loaded from disk, it will pass through
   // DropDeletedEntries().  This will remove some entries from the directory.
   // This function is intended to simulate that process.
   //
   // WARNING: The directory will be deleted by this operation.  You should
   // not have any pointers to the directory (open transactions included)
   // when you call this.
   DirOpenResult SimulateSaveAndReloadDir();
 
   // This function will close and re-open the directory without saving any
   // pending changes.  This is intended to simulate the recovery from a crash
   // scenario.  The same warnings for SimulateSaveAndReloadDir apply here.
   DirOpenResult SimulateCrashAndReloadDir();
 
   void GetAllMetaHandles(BaseTransaction* trans, MetahandleSet* result);
   void CheckPurgeEntriesWithTypeInSucceeded(ModelTypeSet types_to_purge,
                                             bool before_reload);
   bool IsInDirtyMetahandles(int64 metahandle);
   bool IsInMetahandlesToPurge(int64 metahandle);
 
+  // Attempts to load the entry specified by |model_type| and |client_tag| and
+  // returns the lookup result code.
+  BaseNode::InitByLookupResult LookupEntryByClientTag(
+      const ModelType& model_type,
+      const std::string& client_tag);
+
+  // Replace the entry specified by |model_type| and |client_tag| with a
+  // tombstone.
+  void ReplaceWithTombstone(const ModelType& model_type,
+                            const std::string& client_tag);
+
   scoped_ptr<Directory>& dir();
+  UserShare* user_share();
   DirectoryChangeDelegate* directory_change_delegate();
   Encryptor* encryptor();
   UnrecoverableErrorHandler* unrecoverable_error_handler();
 
  private:
   void ValidateEntry(BaseTransaction* trans,
                      int64 id,
                      bool check_name,
                      const std::string& name,
                      int64 base_version,
                      int64 server_version,
                      bool is_del);
 
   base::MessageLoop message_loop_;
-  scoped_ptr<Directory> dir_;
+  syncer::UserShare user_share_;
   NullDirectoryChangeDelegate delegate_;
   FakeEncryptor encryptor_;
   TestUnrecoverableErrorHandler handler_;
   sql::Connection connection_;
 };
 
 }  // namespace syncable
 
 }  // namespace syncer
 
 #endif  // SYNC_SYNCABLE_DIRECTORY_UNITTEST_H_