Adds imageWriterPrivate support for Windows

chrome/browser/image_writer/image_writer_unittest.cc

+// Copyright (c) 2012 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 "base/callback.h"
+#include "base/file_util.h"
+#include "base/run_loop.h"
+#include "chrome/browser/image_writer/image_writer.h"
+#include "chrome/browser/image_writer/image_writer_test_utils.h"
+#include "chrome/common/chrome_utility_messages.h"
+#include "content/public/browser/browser_thread.h"
+#include "content/public/test/test_browser_thread_bundle.h"
+#include "testing/gmock/include/gmock/gmock.h"
+#include "testing/gtest/include/gtest/gtest.h"
+
+namespace image_writer {
+
+using content::BrowserThread;
+using testing::_;
+using testing::AtLeast;
+
+namespace {
+
+class ImageWriterTest : public testing::Test {
+ protected:
+  virtual void SetUp() OVERRIDE {
+    temp_files_.SetUp();
+  }
+
+  TempFilesForTesting temp_files_;
+ private:
+  content::TestBrowserThreadBundle thread_bundle_;
+};
+
+class MockDelegate : public ImageWriter::Delegate {
+ public:
+  MockDelegate() {}
+
+  MOCK_METHOD0(OnWriteImageSucceeded, void());
+  MOCK_METHOD1(OnWriteImageFailed, void(const std::string&));
+  MOCK_METHOD1(OnWriteImageProgress, void(int));
+};
+
+} // namespace
+
+TEST_F(ImageWriterTest, CallsDelegateSucceeded) {
+  MockDelegate delegate;
+
+  scoped_refptr<ImageWriter> image_writer(
+      new ImageWriter(&delegate,
+                      temp_files_.ImagePath(),
+                      temp_files_.DevicePath()));
+
+  EXPECT_CALL(delegate, OnWriteImageFailed(_)).Times(0);
+  EXPECT_CALL(delegate, OnWriteImageSucceeded()).Times(1);
+  EXPECT_CALL(delegate, OnWriteImageProgress(_)).Times(0);
+
+  EXPECT_TRUE(image_writer->OnMessageReceived(
+      ChromeUtilityHostMsg_ImageWriter_Succeeded()));
+}
+
+TEST_F(ImageWriterTest, CallsDelegateFailed) {
+  MockDelegate delegate;
+
+  scoped_refptr<ImageWriter> image_writer(
+      new ImageWriter(&delegate,
+                      temp_files_.ImagePath(),
+                      temp_files_.DevicePath()));
+
+  EXPECT_CALL(delegate, OnWriteImageFailed("fake message")).Times(1);
+  EXPECT_CALL(delegate, OnWriteImageSucceeded()).Times(0);
+  EXPECT_CALL(delegate, OnWriteImageProgress(_)).Times(0);
+
+  EXPECT_TRUE(image_writer->OnMessageReceived(
+      ChromeUtilityHostMsg_ImageWriter_Failed("fake message")));
+}
+
+TEST_F(ImageWriterTest, CallsDelegateProgress) {
+  MockDelegate delegate;
+
+  scoped_refptr<ImageWriter> image_writer(
+      new ImageWriter(&delegate,
+                      temp_files_.ImagePath(),
+                      temp_files_.DevicePath()));
+
+  EXPECT_CALL(delegate, OnWriteImageFailed(_)).Times(0);
+  EXPECT_CALL(delegate, OnWriteImageSucceeded()).Times(0);
+  EXPECT_CALL(delegate, OnWriteImageProgress(_)).Times(0);
+  EXPECT_CALL(delegate, OnWriteImageProgress(100)).Times(1);
+
+  EXPECT_TRUE(image_writer->OnMessageReceived(
+      ChromeUtilityHostMsg_ImageWriter_Progress(100)));
+}
+
+} // namespace image_writer
+
+