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 a19a48c6446e151cd809652a178b64b4525510e3..9542bbb3b6b30823503ed67adf8077997431dee2 100644 |
--- a/chrome/browser/extensions/api/image_writer_private/test_utils.h |
+++ b/chrome/browser/extensions/api/image_writer_private/test_utils.h |
@@ -9,8 +9,10 @@ |
#include "base/files/scoped_temp_dir.h" |
#include "base/message_loop/message_loop.h" |
#include "base/run_loop.h" |
+#include "chrome/browser/extensions/api/image_writer_private/image_writer_utility_client.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" |
@@ -47,17 +49,57 @@ class MockOperationManager : public OperationManager { |
const std::string& error_message)); |
}; |
+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; |
+ |
+ void Progress(int64 progress); |
+ void Success(); |
+ void Error(const std::string& message); |
+ void Cancel(); |
+ static scoped_refptr<FakeImageWriterClient> Create(); |
+ |
+ private: |
+ virtual ~FakeImageWriterClient(); |
+ |
+ ProgressCallback progress_callback_; |
+ SuccessCallback success_callback_; |
+ ErrorCallback error_callback_; |
+ CancelCallback cancel_callback_; |
+}; |
+ |
// Base class for unit tests that manages creating image and device files. |
class ImageWriterUnitTestBase : public testing::Test { |
- public: |
+ protected: |
ImageWriterUnitTestBase(); |
virtual ~ImageWriterUnitTestBase(); |
- protected: |
virtual void SetUp() OVERRIDE; |
virtual void TearDown() OVERRIDE; |
+ // 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); |
+ |
// Fills |file| with |length| bytes of |pattern|, overwriting any existing |
// data. |
bool FillFile(const base::FilePath& file, |