mojo filesystem: Simplify full file reading/writing.

components/filesystem/directory_impl_unittest.cc

 // Copyright 2015 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.
 
 #include <stddef.h>
 #include <stdint.h>
 
 #include <map>
 #include <string>
 
 #include "base/macros.h"
 #include "components/filesystem/files_test_base.h"
+#include "mojo/common/common_type_converters.h"
 #include "mojo/util/capture_util.h"
 
 using mojo::Capture;
 
 namespace filesystem {
 namespace {
 
 using DirectoryImplTest = FilesTestBase;
 
 TEST_F(DirectoryImplTest, Read) {
@@ skipping 123 matching lines @@
     // Attempt to open that directory as a file. This must fail!
     FilePtr file;
     error = FileError::FAILED;
     directory->OpenFile("my_file", GetProxy(&file), kFlagRead | kFlagOpen,
                         Capture(&error));
     ASSERT_TRUE(directory.WaitForIncomingResponse());
     EXPECT_EQ(FileError::NOT_A_FILE, error);
   }
 }
 
+TEST_F(DirectoryImplTest, WriteFileReadFile) {
+  DirectoryPtr directory;
+  GetTemporaryRoot(&directory);
+  FileError error;
+
+  std::string data("one two three");
+  {
+    directory->WriteFile("data", mojo::Array<uint8_t>::From(data),
+                         Capture(&error));
+    ASSERT_TRUE(directory.WaitForIncomingResponse());
+    EXPECT_EQ(FileError::OK, error);
+  }
+
+  {
+    mojo::Array<uint8_t> file_contents;
+    directory->ReadEntireFile("data", Capture(&error, &file_contents));
+    ASSERT_TRUE(directory.WaitForIncomingResponse());
+    EXPECT_EQ(FileError::OK, error);
+
+    EXPECT_EQ(data, file_contents.To<std::string>());
+  }
+}
+
+TEST_F(DirectoryImplTest, ReadEmptyFileIsNotFoundError) {
+  DirectoryPtr directory;
+  GetTemporaryRoot(&directory);
+  FileError error;
+
+  {
+    mojo::Array<uint8_t> file_contents;
+    directory->ReadEntireFile("doesnt_exist", Capture(&error, &file_contents));
+    ASSERT_TRUE(directory.WaitForIncomingResponse());
+    EXPECT_EQ(FileError::NOT_FOUND, error);
+  }
+}
+
+TEST_F(DirectoryImplTest, CantReadEntireFileOnADirectory) {
+  DirectoryPtr directory;
+  GetTemporaryRoot(&directory);
+  FileError error;
+
+  // Create a directory
+  {
+    DirectoryPtr my_file_directory;
+    error = FileError::FAILED;
+    directory->OpenDirectory(
+        "my_dir", GetProxy(&my_file_directory),
+        kFlagRead | kFlagWrite | kFlagCreate,
+        Capture(&error));
+    ASSERT_TRUE(directory.WaitForIncomingResponse());
+    EXPECT_EQ(FileError::OK, error);
+  }
+
+  // Try to read it as a file
+  {
+    mojo::Array<uint8_t> file_contents;
+    directory->ReadEntireFile("my_dir", Capture(&error, &file_contents));
+    ASSERT_TRUE(directory.WaitForIncomingResponse());
+    EXPECT_EQ(FileError::NOT_A_FILE, error);
+  }
+}
+
+TEST_F(DirectoryImplTest, CantWriteFileOnADirectory) {
+  DirectoryPtr directory;
+  GetTemporaryRoot(&directory);
+  FileError error;
+
+  // Create a directory
+  {
+    DirectoryPtr my_file_directory;
+    error = FileError::FAILED;
+    directory->OpenDirectory(
+        "my_dir", GetProxy(&my_file_directory),
+        kFlagRead | kFlagWrite | kFlagCreate,
+        Capture(&error));
+    ASSERT_TRUE(directory.WaitForIncomingResponse());
+    EXPECT_EQ(FileError::OK, error);
+  }
+
+  {
+    std::string data("one two three");
+    directory->WriteFile("my_dir", mojo::Array<uint8_t>::From(data),
+                         Capture(&error));
+    ASSERT_TRUE(directory.WaitForIncomingResponse());
+    EXPECT_EQ(FileError::NOT_A_FILE, error);
+  }
+}
 
 // TODO(vtl): Test delete flags.
 
 }  // namespace
 }  // namespace filesystem