Index: chrome/browser/extensions/api/image_writer_private/test_utils.cc |
diff --git a/chrome/browser/extensions/api/image_writer_private/test_utils.cc b/chrome/browser/extensions/api/image_writer_private/test_utils.cc |
index 999898bd9029da0a3c01d01735977336fe93bf97..7d6fde21bf77ebc06889a79ff6f8881a531e1911 100644 |
--- a/chrome/browser/extensions/api/image_writer_private/test_utils.cc |
+++ b/chrome/browser/extensions/api/image_writer_private/test_utils.cc |
@@ -51,14 +51,34 @@ class ImageWriterFakeImageBurnerClient |
} // namespace |
#endif |
+QuittingMockManager::QuittingMockManager(const base::Closure& quit_closure) |
+ : quit_closure_(quit_closure) {} |
+ |
+void QuittingMockManager::OnComplete(const ExtensionId& extension_id) { |
+ MockOperationManager::OnComplete(extension_id); |
+ quit_closure_.Run(); |
+} |
+ |
+void QuittingMockManager::OnError(const ExtensionId& extension_id, |
+ image_writer_api::Stage stage, |
+ int progress, |
+ const std::string& error_message) { |
+ MockOperationManager::OnError(extension_id, stage, progress, error_message); |
+ quit_closure_.Run(); |
+} |
+ |
MockOperationManager::MockOperationManager() : OperationManager(NULL) {} |
MockOperationManager::~MockOperationManager() {} |
-ImageWriterUnitTestBase::ImageWriterUnitTestBase() {} |
+ImageWriterUnitTestBase::ImageWriterUnitTestBase() |
+ : thread_bundle_(content::TestBrowserThreadBundle::IO_MAINLOOP) {} |
ImageWriterUnitTestBase::~ImageWriterUnitTestBase() {} |
void ImageWriterUnitTestBase::SetUp() { |
testing::Test::SetUp(); |
+ |
+ // thread_bundle_.reset(new content::TestBrowserThreadBundle()); |
+ |
ASSERT_TRUE(temp_dir_.CreateUniqueTempDir()); |
ASSERT_TRUE(base::CreateTemporaryFileInDir(temp_dir_.path(), |
@@ -87,27 +107,18 @@ void ImageWriterUnitTestBase::TearDown() { |
#endif |
} |
-bool ImageWriterUnitTestBase::CompareImageAndDevice() { |
+bool ImageWriterUnitTestBase::ImageWrittenToDevice( |
+ const base::FilePath& image_path, |
+ const base::FilePath& device_path) { |
scoped_ptr<char[]> image_buffer(new char[kTestFileSize]); |
scoped_ptr<char[]> device_buffer(new char[kTestFileSize]); |
- while (true) { |
- int image_bytes_read = ReadFile(test_image_path_, |
- image_buffer.get(), |
- kTestFileSize); |
- int device_bytes_read = ReadFile(test_device_path_, |
- device_buffer.get(), |
- kTestFileSize); |
- |
- if (image_bytes_read != device_bytes_read) |
- return false; |
+ int image_bytes_read = |
+ ReadFile(image_path, image_buffer.get(), kTestFileSize); |
+ ReadFile(device_path, device_buffer.get(), kTestFileSize); |
- if (image_bytes_read == 0) |
- return true; |
- |
- if (memcmp(image_buffer.get(), device_buffer.get(), image_bytes_read) != 0) |
- return false; |
- } |
+ if (memcmp(image_buffer.get(), device_buffer.get(), image_bytes_read) == 0) |
+ return true; |
return false; |
} |