Adds API test for imageWriterPrivate.writeFromFile

chrome/browser/extensions/api/image_writer_private/test_utils.h

 // Copyright 2013 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 CHROME_BROWSER_EXTENSIONS_API_IMAGE_WRITER_PRIVATE_TEST_UTILS_H_
 #define CHROME_BROWSER_EXTENSIONS_API_IMAGE_WRITER_PRIVATE_TEST_UTILS_H_
 
 #include "base/file_util.h"
 #include "base/files/scoped_temp_dir.h"
 #include "base/message_loop/message_loop.h"
@@ skipping 47 matching lines @@
 // A fake for the DiskMountManager that will successfully call the unmount
 // callback.
 class FakeDiskMountManager : public chromeos::disks::MockDiskMountManager {
  public:
   FakeDiskMountManager();
   virtual ~FakeDiskMountManager();
 
   virtual void UnmountDeviceRecursively(
       const std::string& device_path,
       const UnmountDeviceRecursivelyCallbackType& callback) OVERRIDE;
-  /*
-  MOCK_METHOD1(AddObserver, void(chromeos::disks::DiskMountManager::Observer*));
-  MOCK_METHOD1(RemoveObserver,
-  void(chromeos::disks::DiskMountManager::Observer*));
-  MOCK_CONST_METHOD0(disks, const DiskMap&());
-  MOCK_CONST_METHOD1(FindDiskBySourcePath, const Disk*(const std::string&));
-  MOCK_CONST_METHOD0(mount_points, const MountPointMap&());
-  MOCK_METHOD0(RequestMountInfoRefresh, void());
-  MOCK_METHOD4(MountPath, void(const std::string&, const std::string&, const
-  std::string&, chromeos::MountType));
-  MOCK_METHOD3(UnmountPath, void(const std::string&, chromeos::UnmountOptions,
-  const UnmountPathCallback&));
-  MOCK_METHOD1(FormatMountedDevice, void(const std::string&));
-  */
 
  private:
   DiskMap disks_;
 };
 #endif
 
 class FakeImageWriterClient : public ImageWriterUtilityClient {
  public:
   FakeImageWriterClient();
 
   virtual void Write(const ProgressCallback& progress_callback,
                      const SuccessCallback& success_callback,
                      const ErrorCallback& error_callback,
                      const base::FilePath& source,
                      const base::FilePath& target) OVERRIDE;
 
   virtual void Verify(const ProgressCallback& progress_callback,
                       const SuccessCallback& success_callback,
                       const ErrorCallback& error_callback,
                       const base::FilePath& source,
                       const base::FilePath& target) OVERRIDE;
 
   virtual void Cancel(const CancelCallback& cancel_callback) OVERRIDE;
 
   virtual void Shutdown() OVERRIDE;
 
+  // Sets a callback for when a Write call is made.
+  void SetWriteCallback(const base::Closure& write_callback);
+  // Sets a callback for when a Verify call is made.
+  void SetVerifyCallback(const base::Closure& verify_callback);
+
+  // Triggers the progress callback.
   void Progress(int64 progress);
+  // Triggers the success callback.
   void Success();
+  // Triggers the error callback.
   void Error(const std::string& message);
+  // Triggers the cancel callback.
   void Cancel();
-  static scoped_refptr<FakeImageWriterClient> Create();
 
  private:
   virtual ~FakeImageWriterClient();
 
   ProgressCallback progress_callback_;
   SuccessCallback success_callback_;
   ErrorCallback error_callback_;
   CancelCallback cancel_callback_;
+
+  base::Closure write_callback_;
+  base::Closure verify_callback_;
 };
 
-// Base class for unit tests that manages creating image and device files.
-class ImageWriterUnitTestBase : public testing::Test {
- protected:
-  ImageWriterUnitTestBase();
-  virtual ~ImageWriterUnitTestBase();
-
-  virtual void SetUp() OVERRIDE;
-
-  virtual void TearDown() OVERRIDE;
+class ImageWriterTestUtils {
+ public:
+  ImageWriterTestUtils();
+  virtual ~ImageWriterTestUtils();
 
   // Verifies that the data in image_path was written to the file at
   // device_path.  This is different from base::ContentsEqual because the device
   // may be larger than the image.
-  bool ImageWrittenToDevice(const base::FilePath& image_path,
-                            const base::FilePath& device_path);
+  bool ImageWrittenToDevice();
 
   // Fills |file| with |length| bytes of |pattern|, overwriting any existing
   // data.
   bool FillFile(const base::FilePath& file,
                 const int pattern,
                 const int length);
 
+  // Set up the test utils, creating temporary folders and such.
+  // Note that browser tests should use the alternate form and pass "true" as an
+  // argument.
+  virtual void SetUp();
+  // Set up the test utils, creating temporary folders and such.  If
+  // |is_browser_test| is true then it will use alternate initialization
+  // appropriate for a browser test.  This should be run in
+  // |SetUpInProcessBrowserTestFixture|.
+  virtual void SetUp(bool is_browser_test);
+
+  virtual void TearDown();
+
+  const base::FilePath& GetTempDir();
+  const base::FilePath& GetImagePath();
+  const base::FilePath& GetDevicePath();
+
+#if !defined(OS_CHROMEOS)
+  FakeImageWriterClient* GetUtilityClient();
+#endif
+
+ protected:
   base::ScopedTempDir temp_dir_;
   base::FilePath test_image_path_;
   base::FilePath test_device_path_;
 
+#if !defined(OS_CHROMEOS)
+  scoped_refptr<FakeImageWriterClient> client_;
+#endif
+};
+
+// Base class for unit tests that manages creating image and device files.
+class ImageWriterUnitTestBase : public testing::Test {
+ protected:
+  ImageWriterUnitTestBase();
+  virtual ~ImageWriterUnitTestBase();
+
+  virtual void SetUp() OVERRIDE;
+  virtual void TearDown() OVERRIDE;
+
+  ImageWriterTestUtils test_utils_;
+
  private:
   content::TestBrowserThreadBundle thread_bundle_;
 };
 
 }  // namespace image_writer
 }  // namespace extensions
 
 #endif  // CHROME_BROWSER_EXTENSIONS_API_IMAGE_WRITER_PRIVATE_TEST_UTILS_H_