Index: trunk/src/chrome/utility/image_writer/image_writer_unittest.cc |
=================================================================== |
--- trunk/src/chrome/utility/image_writer/image_writer_unittest.cc (revision 252594) |
+++ trunk/src/chrome/utility/image_writer/image_writer_unittest.cc (working copy) |
@@ -1,182 +0,0 @@ |
-// Copyright 2014 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/file_util.h" |
-#include "base/files/scoped_temp_dir.h" |
-#include "base/run_loop.h" |
-#include "chrome/utility/image_writer/error_messages.h" |
-#include "chrome/utility/image_writer/image_writer.h" |
-#include "chrome/utility/image_writer/image_writer_handler.h" |
-#include "testing/gmock/include/gmock/gmock.h" |
-#include "testing/gtest/include/gtest/gtest.h" |
- |
-namespace image_writer { |
- |
-using testing::_; |
-using testing::AnyNumber; |
-using testing::AtLeast; |
-using testing::Lt; |
- |
-namespace { |
- |
-const int64 kTestFileSize = 1 << 15; // 32 kB |
-const int kTestPattern = 0x55555555; |
- |
-class ImageWriterUtilityTest : public testing::Test { |
- protected: |
- virtual void SetUp() OVERRIDE { |
- ASSERT_TRUE(temp_dir_.CreateUniqueTempDir()); |
- ASSERT_TRUE(base::CreateTemporaryFileInDir(temp_dir_.path(), &image_path_)); |
- ASSERT_TRUE( |
- base::CreateTemporaryFileInDir(temp_dir_.path(), &device_path_)); |
- } |
- |
- virtual void TearDown() OVERRIDE {} |
- |
- void FillFile(const base::FilePath& path) { |
- scoped_ptr<char[]> buffer(new char[kTestFileSize]); |
- memset(buffer.get(), kTestPattern, kTestFileSize); |
- |
- ASSERT_TRUE(file_util::WriteFile(path, buffer.get(), kTestFileSize)); |
- } |
- |
- base::FilePath image_path_; |
- base::FilePath device_path_; |
- |
- private: |
- base::MessageLoop message_loop_; |
- base::ScopedTempDir temp_dir_; |
-}; |
- |
-class MockHandler : public ImageWriterHandler { |
- public: |
- MOCK_METHOD1(SendProgress, void(int64)); |
- MOCK_METHOD1(SendFailed, void(const std::string& message)); |
- MOCK_METHOD0(SendSucceeded, void()); |
- MOCK_METHOD1(OnMessageReceived, bool(const IPC::Message& message)); |
-}; |
- |
-} // namespace |
- |
-TEST_F(ImageWriterUtilityTest, WriteSuccessful) { |
- MockHandler mock_handler; |
- ImageWriter image_writer(&mock_handler); |
- |
- EXPECT_CALL(mock_handler, SendProgress(_)).Times(AnyNumber()); |
- EXPECT_CALL(mock_handler, SendProgress(kTestFileSize)).Times(1); |
- EXPECT_CALL(mock_handler, SendProgress(0)).Times(1); |
- EXPECT_CALL(mock_handler, SendSucceeded()).Times(1); |
- EXPECT_CALL(mock_handler, SendFailed(_)).Times(0); |
- |
- FillFile(image_path_); |
- image_writer.Write(image_path_, device_path_); |
- base::RunLoop().RunUntilIdle(); |
-} |
- |
-TEST_F(ImageWriterUtilityTest, WriteInvalidImageFile) { |
- MockHandler mock_handler; |
- ImageWriter image_writer(&mock_handler); |
- |
- EXPECT_CALL(mock_handler, SendProgress(_)).Times(0); |
- EXPECT_CALL(mock_handler, SendSucceeded()).Times(0); |
- EXPECT_CALL(mock_handler, SendFailed(error::kOpenImage)).Times(1); |
- |
- ASSERT_TRUE(base::DeleteFile(image_path_, false)); |
- image_writer.Write(image_path_, device_path_); |
- base::RunLoop().RunUntilIdle(); |
-} |
- |
-TEST_F(ImageWriterUtilityTest, WriteInvalidDeviceFile) { |
- MockHandler mock_handler; |
- ImageWriter image_writer(&mock_handler); |
- |
- EXPECT_CALL(mock_handler, SendProgress(_)).Times(0); |
- EXPECT_CALL(mock_handler, SendSucceeded()).Times(0); |
- EXPECT_CALL(mock_handler, SendFailed(error::kOpenDevice)).Times(1); |
- |
- ASSERT_TRUE(base::DeleteFile(device_path_, false)); |
- image_writer.Write(image_path_, device_path_); |
- base::RunLoop().RunUntilIdle(); |
-} |
- |
-TEST_F(ImageWriterUtilityTest, VerifySuccessful) { |
- MockHandler mock_handler; |
- ImageWriter image_writer(&mock_handler); |
- |
- EXPECT_CALL(mock_handler, SendProgress(_)).Times(AnyNumber()); |
- EXPECT_CALL(mock_handler, SendProgress(kTestFileSize)).Times(1); |
- EXPECT_CALL(mock_handler, SendProgress(0)).Times(1); |
- EXPECT_CALL(mock_handler, SendSucceeded()).Times(1); |
- EXPECT_CALL(mock_handler, SendFailed(_)).Times(0); |
- |
- FillFile(image_path_); |
- FillFile(device_path_); |
- |
- image_writer.Verify(image_path_, device_path_); |
- |
- base::RunLoop().RunUntilIdle(); |
-} |
- |
-TEST_F(ImageWriterUtilityTest, VerifyInvalidImageFile) { |
- MockHandler mock_handler; |
- ImageWriter image_writer(&mock_handler); |
- |
- EXPECT_CALL(mock_handler, SendProgress(_)).Times(0); |
- EXPECT_CALL(mock_handler, SendSucceeded()).Times(0); |
- EXPECT_CALL(mock_handler, SendFailed(error::kOpenImage)).Times(1); |
- |
- ASSERT_TRUE(base::DeleteFile(image_path_, false)); |
- |
- image_writer.Verify(image_path_, device_path_); |
- |
- base::RunLoop().RunUntilIdle(); |
-} |
- |
-TEST_F(ImageWriterUtilityTest, VerifyInvalidDeviceFile) { |
- MockHandler mock_handler; |
- ImageWriter image_writer(&mock_handler); |
- |
- EXPECT_CALL(mock_handler, SendProgress(_)).Times(0); |
- EXPECT_CALL(mock_handler, SendSucceeded()).Times(0); |
- EXPECT_CALL(mock_handler, SendFailed(error::kOpenDevice)).Times(1); |
- |
- ASSERT_TRUE(base::DeleteFile(device_path_, false)); |
- |
- image_writer.Verify(image_path_, device_path_); |
- |
- base::RunLoop().RunUntilIdle(); |
-} |
- |
-TEST_F(ImageWriterUtilityTest, VerifyFailed) { |
- MockHandler mock_handler; |
- ImageWriter image_writer(&mock_handler); |
- |
- EXPECT_CALL(mock_handler, SendProgress(_)).Times(AnyNumber()); |
- EXPECT_CALL(mock_handler, SendSucceeded()).Times(0); |
- EXPECT_CALL(mock_handler, SendFailed(error::kVerificationFailed)).Times(1); |
- |
- FillFile(image_path_); |
- image_writer.Verify(image_path_, device_path_); |
- |
- base::RunLoop().RunUntilIdle(); |
-} |
- |
-TEST_F(ImageWriterUtilityTest, WriteThenVerify) { |
- MockHandler mock_handler; |
- ImageWriter image_writer(&mock_handler); |
- |
- EXPECT_CALL(mock_handler, SendProgress(_)).Times(AnyNumber()); |
- EXPECT_CALL(mock_handler, SendSucceeded()).Times(2); |
- EXPECT_CALL(mock_handler, SendFailed(_)).Times(0); |
- |
- image_writer.Write(image_path_, device_path_); |
- |
- base::RunLoop().RunUntilIdle(); |
- |
- image_writer.Verify(image_path_, device_path_); |
- |
- base::RunLoop().RunUntilIdle(); |
-} |
- |
-} // namespace image_writer |