| Index: chrome/browser/extensions/api/image_writer_private/test_utils.h
|
| diff --git a/chrome/browser/extensions/api/image_writer_private/test_utils.h b/chrome/browser/extensions/api/image_writer_private/test_utils.h
|
| index b290e881ee87e86dd7096e3d1bfb981b5d9c5517..c61df2f2b6e9675c0474fefe14af90ab4eec2677 100644
|
| --- a/chrome/browser/extensions/api/image_writer_private/test_utils.h
|
| +++ b/chrome/browser/extensions/api/image_writer_private/test_utils.h
|
| @@ -11,6 +11,7 @@
|
| #include "base/run_loop.h"
|
| #include "chrome/browser/extensions/api/image_writer_private/operation_manager.h"
|
| #include "content/public/test/test_browser_thread_bundle.h"
|
| +#include "content/public/test/test_utils.h"
|
| #include "testing/gmock/include/gmock/gmock.h"
|
| #include "testing/gtest/include/gtest/gtest.h"
|
|
|
| @@ -46,6 +47,20 @@ class MockOperationManager : public OperationManager {
|
| const std::string& error_message));
|
| };
|
|
|
| +class QuittingMockManager : public MockOperationManager {
|
| + public:
|
| + explicit QuittingMockManager(const base::Closure& quit_closure);
|
| +
|
| + virtual void OnComplete(const ExtensionId& extension_id);
|
| + virtual void OnError(const ExtensionId& extension_id,
|
| + image_writer_api::Stage stage,
|
| + int progress,
|
| + const std::string& error_message);
|
| +
|
| + private:
|
| + base::Closure quit_closure_;
|
| +};
|
| +
|
| // Base class for unit tests that manages creating image and device files.
|
| class ImageWriterUnitTestBase : public testing::Test {
|
| public:
|
| @@ -57,22 +72,25 @@ class ImageWriterUnitTestBase : public testing::Test {
|
|
|
| virtual void TearDown() OVERRIDE;
|
|
|
| - // Compare the image and device files, returning true if they are the same,
|
| - // false if different.
|
| - bool CompareImageAndDevice();
|
| + // 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);
|
|
|
| base::ScopedTempDir temp_dir_;
|
| base::FilePath test_image_path_;
|
| base::FilePath test_device_path_;
|
|
|
| + content::TestBrowserThreadBundle thread_bundle_;
|
| + content::InProcessUtilityThreadHelper utility_thread_helper_;
|
| +
|
| private:
|
| // Fills |file| with |length| bytes of |pattern|, overwriting any existing
|
| // data.
|
| bool FillFile(const base::FilePath& file,
|
| const int pattern,
|
| const int length);
|
| -
|
| - content::TestBrowserThreadBundle thread_bundle_;
|
| };
|
|
|
| } // namespace image_writer
|
|
|