| OLD | NEW |
| 1 // Copyright 2014 The Chromium Authors. All rights reserved. | 1 // Copyright 2014 The Chromium Authors. All rights reserved. |
| 2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 | 4 |
| 5 #include <stdint.h> |
| 6 #include <string.h> |
| 7 |
| 5 #include <string> | 8 #include <string> |
| 6 | 9 |
| 7 #include "base/files/file_util.h" | 10 #include "base/files/file_util.h" |
| 8 #include "base/files/scoped_temp_dir.h" | 11 #include "base/files/scoped_temp_dir.h" |
| 9 #include "base/run_loop.h" | 12 #include "base/run_loop.h" |
| 10 #include "chrome/utility/image_writer/error_messages.h" | 13 #include "chrome/utility/image_writer/error_messages.h" |
| 11 #include "chrome/utility/image_writer/image_writer.h" | 14 #include "chrome/utility/image_writer/image_writer.h" |
| 12 #include "chrome/utility/image_writer/image_writer_handler.h" | 15 #include "chrome/utility/image_writer/image_writer_handler.h" |
| 13 #include "testing/gmock/include/gmock/gmock.h" | 16 #include "testing/gmock/include/gmock/gmock.h" |
| 14 #include "testing/gtest/include/gtest/gtest.h" | 17 #include "testing/gtest/include/gtest/gtest.h" |
| 15 | 18 |
| 16 namespace image_writer { | 19 namespace image_writer { |
| 17 | 20 |
| 18 using testing::_; | 21 using testing::_; |
| 19 using testing::AnyNumber; | 22 using testing::AnyNumber; |
| 20 using testing::AtLeast; | 23 using testing::AtLeast; |
| 21 using testing::Lt; | 24 using testing::Lt; |
| 22 | 25 |
| 23 namespace { | 26 namespace { |
| 24 | 27 |
| 25 const int64 kTestFileSize = 1 << 15; // 32 kB | 28 const int64_t kTestFileSize = 1 << 15; // 32 kB |
| 26 const int kTestPattern = 0x55555555; | 29 const int kTestPattern = 0x55555555; |
| 27 | 30 |
| 28 class ImageWriterUtilityTest : public testing::Test { | 31 class ImageWriterUtilityTest : public testing::Test { |
| 29 protected: | 32 protected: |
| 30 void SetUp() override { | 33 void SetUp() override { |
| 31 ASSERT_TRUE(temp_dir_.CreateUniqueTempDir()); | 34 ASSERT_TRUE(temp_dir_.CreateUniqueTempDir()); |
| 32 ASSERT_TRUE(base::CreateTemporaryFileInDir(temp_dir_.path(), &image_path_)); | 35 ASSERT_TRUE(base::CreateTemporaryFileInDir(temp_dir_.path(), &image_path_)); |
| 33 ASSERT_TRUE( | 36 ASSERT_TRUE( |
| 34 base::CreateTemporaryFileInDir(temp_dir_.path(), &device_path_)); | 37 base::CreateTemporaryFileInDir(temp_dir_.path(), &device_path_)); |
| 35 } | 38 } |
| (...skipping 12 matching lines...) Expand all Loading... |
| 48 base::FilePath image_path_; | 51 base::FilePath image_path_; |
| 49 base::FilePath device_path_; | 52 base::FilePath device_path_; |
| 50 | 53 |
| 51 private: | 54 private: |
| 52 base::MessageLoop message_loop_; | 55 base::MessageLoop message_loop_; |
| 53 base::ScopedTempDir temp_dir_; | 56 base::ScopedTempDir temp_dir_; |
| 54 }; | 57 }; |
| 55 | 58 |
| 56 class MockHandler : public ImageWriterHandler { | 59 class MockHandler : public ImageWriterHandler { |
| 57 public: | 60 public: |
| 58 MOCK_METHOD1(SendProgress, void(int64)); | 61 MOCK_METHOD1(SendProgress, void(int64_t)); |
| 59 MOCK_METHOD1(SendFailed, void(const std::string& message)); | 62 MOCK_METHOD1(SendFailed, void(const std::string& message)); |
| 60 MOCK_METHOD0(SendSucceeded, void()); | 63 MOCK_METHOD0(SendSucceeded, void()); |
| 61 MOCK_METHOD1(OnMessageReceived, bool(const IPC::Message& message)); | 64 MOCK_METHOD1(OnMessageReceived, bool(const IPC::Message& message)); |
| 62 }; | 65 }; |
| 63 | 66 |
| 64 // This Mock has the additional feature that it will start verification when | 67 // This Mock has the additional feature that it will start verification when |
| 65 // the write completes. | 68 // the write completes. |
| 66 class VerifyingHandler : public MockHandler { | 69 class VerifyingHandler : public MockHandler { |
| 67 public: | 70 public: |
| 68 VerifyingHandler() : image_writer_(NULL), verified_(false) {} | 71 VerifyingHandler() : image_writer_(NULL), verified_(false) {} |
| (...skipping 152 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 221 EXPECT_CALL(mock_handler, SendFailed(_)).Times(0); | 224 EXPECT_CALL(mock_handler, SendFailed(_)).Times(0); |
| 222 | 225 |
| 223 FillDefault(image_path_); | 226 FillDefault(image_path_); |
| 224 | 227 |
| 225 image_writer.Write(); | 228 image_writer.Write(); |
| 226 | 229 |
| 227 base::RunLoop().RunUntilIdle(); | 230 base::RunLoop().RunUntilIdle(); |
| 228 } | 231 } |
| 229 | 232 |
| 230 } // namespace image_writer | 233 } // namespace image_writer |
| OLD | NEW |