Clean up ARC file system unit tests.

components/arc/test/fake_file_system_instance.h

Index: components/arc/test/fake_file_system_instance.h
diff --git a/components/arc/test/fake_file_system_instance.h b/components/arc/test/fake_file_system_instance.h
index 98db1026d1f06eaad0925acc3c0ce5646898cae0..3d2f7d09d900815cff3d1e2830d172c3866cb7f5 100644
--- a/components/arc/test/fake_file_system_instance.h
+++ b/components/arc/test/fake_file_system_instance.h
@@ -5,40 +5,126 @@
 #ifndef COMPONENTS_ARC_TEST_FAKE_FILE_SYSTEM_INSTANCE_H_
 #define COMPONENTS_ARC_TEST_FAKE_FILE_SYSTEM_INSTANCE_H_
 
+#include <stdint.h>
+
+#include <map>
 #include <string>
+#include <utility>
 #include <vector>
 
+#include "base/files/scoped_temp_dir.h"
 #include "base/macros.h"
+#include "base/threading/thread_checker.h"
 #include "components/arc/common/file_system.mojom.h"
 
 namespace arc {
 
-// A fake implementation which returns errors for all requests or just ignores
-// them.
+// Fake implementation to operate on documents in memory.
+//
+// ArcFileSystemOperationRunner provides two types of methods: content URL
+// based and documents provider based. According to backend type, you need
+// to setup the fake with different functions.
+//
+// Content URL based functions are:
+// - GetFileSize()
+// - OpenFileToRead()
+// Fake files for those functions can be set up by AddFile().
+//
+// Documents provider based functions are:
+// - GetDocument()
+// - GetChildDocuments()
+// Fake documents for those functions can be set up by AddDocument().
+//
+// Notes:
+// - GetChildDocuments() returns child documents in the same order as they were
+//   added with AddDocument().
+// - All member functions must be called on the same thread.
 class FakeFileSystemInstance : public mojom::FileSystemInstance {
  public:
-  FakeFileSystemInstance();
+  // Specification of a fake file available to content URL based methods.
+  struct FileSpec {
+    enum Seekable {

[hidehiko, 2017/01/27 15:50:00]

Could you use "enum class"?

[Shuhei Takahashi, 2017/01/30 02:27:35]

Done.

+      SEEKABLE,
+      NON_SEEKABLE,
+    };
 
-  // mojom::FileSystemInstance:
+    // Content URL of a file.
+    const char* url;
+
+    // The content of a file. This must be a null-terminated string.
+    const char* content;
+
+    // Whether this file is seekable or not.
+    Seekable seekable;
+  };
+
+  // Specification of a fake document available to documents provider based
+  // methods.
+  struct DocumentSpec {
+    // Authority.
+    const char* authority;
+
+    // ID of this document.
+    const char* document_id;
+
+    // ID of the parent document. Can be nullptr string if this is a root.
+    const char* parent_document_id;
+
+    // File name displayed to users.
+    const char* display_name;
+
+    // MIME type.
+    const char* mime_type;
+
+    // File size in bytes. Set to -1 if size is not available.
+    int64_t size;
+
+    // Last modified time in milliseconds from the UNIX epoch.
+    // TODO(crbug.com/672737): Use base::Time once the corresponding field
+    // in file_system.mojom stops using uint64.
+    uint64_t last_modified;
+  };
+
+  FakeFileSystemInstance();
   ~FakeFileSystemInstance() override;
 
+  // Adds a file accessible by content URL based methods.
+  void AddFile(const FileSpec& spec);
+
+  // Adds a document accessible by document provider based methods.
+  void AddDocument(const DocumentSpec& spec);
+
+  // mojom::FileSystemInstance:
   void GetChildDocuments(const std::string& authority,
                          const std::string& document_id,
                          const GetChildDocumentsCallback& callback) override;
-
   void GetDocument(const std::string& authority,
                    const std::string& document_id,
                    const GetDocumentCallback& callback) override;
-
   void GetFileSize(const std::string& url,
                    const GetFileSizeCallback& callback) override;
-
   void OpenFileToRead(const std::string& url,
                       const OpenFileToReadCallback& callback) override;
-
   void RequestMediaScan(const std::vector<std::string>& paths) override;
 
  private:
+  // A pair of an authority and a document ID which identifies the location
+  // of a document in documents providers.
+  using DocumentKey = std::pair<std::string, std::string>;
+
+  base::ThreadChecker thread_checker_;
+
+  base::ScopedTempDir temp_dir_;
+
+  // Mapping from a content URL to a file.
+  std::map<std::string, FileSpec> files_;
+
+  // Mapping from a document key to a document.
+  std::map<DocumentKey, DocumentSpec> documents_;
+
+  // Mapping from a document key to its child documents.
+  std::map<DocumentKey, std::vector<DocumentKey>> child_documents_;
+
   DISALLOW_COPY_AND_ASSIGN(FakeFileSystemInstance);
 };