1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
|
// Copyright 2013 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#include "chrome/browser/extensions/api/image_writer_private/operation_manager.h"
#include <memory>
#include "base/command_line.h"
#include "base/functional/bind.h"
#include "build/build_config.h"
#include "chrome/browser/extensions/api/image_writer_private/error_constants.h"
#include "chrome/browser/extensions/api/image_writer_private/test_utils.h"
#include "chrome/test/base/testing_profile.h"
#include "extensions/browser/test_event_router.h"
namespace extensions {
namespace image_writer {
namespace {
class ImageWriterOperationManagerTest : public ImageWriterUnitTestBase {
public:
void StartCallback(bool success, const std::string& error) {
started_ = true;
start_success_ = success;
start_error_ = error;
}
void CancelCallback(bool success, const std::string& error) {
cancelled_ = true;
cancel_success_ = true;
cancel_error_ = error;
}
protected:
ImageWriterOperationManagerTest()
: started_(false),
start_success_(false) {
}
void SetUp() override {
ImageWriterUnitTestBase::SetUp();
CreateAndUseTestEventRouter(&test_profile_);
}
bool started_;
bool start_success_;
std::string start_error_;
bool cancelled_;
bool cancel_success_;
std::string cancel_error_;
TestingProfile test_profile_;
};
TEST_F(ImageWriterOperationManagerTest, WriteFromFile) {
OperationManager manager(&test_profile_);
manager.StartWriteFromFile(
kDummyExtensionId, test_utils_.GetImagePath(),
test_utils_.GetDevicePath().AsUTF8Unsafe(),
base::BindOnce(&ImageWriterOperationManagerTest::StartCallback,
base::Unretained(this)));
EXPECT_TRUE(started_);
EXPECT_TRUE(start_success_);
EXPECT_EQ("", start_error_);
manager.CancelWrite(
kDummyExtensionId,
base::BindOnce(&ImageWriterOperationManagerTest::CancelCallback,
base::Unretained(this)));
EXPECT_TRUE(cancelled_);
EXPECT_TRUE(cancel_success_);
EXPECT_EQ("", cancel_error_);
content::RunAllTasksUntilIdle();
}
TEST_F(ImageWriterOperationManagerTest, DestroyPartitions) {
OperationManager manager(&test_profile_);
manager.DestroyPartitions(
kDummyExtensionId, test_utils_.GetDevicePath().AsUTF8Unsafe(),
base::BindOnce(&ImageWriterOperationManagerTest::StartCallback,
base::Unretained(this)));
EXPECT_TRUE(started_);
EXPECT_TRUE(start_success_);
EXPECT_EQ("", start_error_);
manager.CancelWrite(
kDummyExtensionId,
base::BindOnce(&ImageWriterOperationManagerTest::CancelCallback,
base::Unretained(this)));
EXPECT_TRUE(cancelled_);
EXPECT_TRUE(cancel_success_);
EXPECT_EQ("", cancel_error_);
content::RunAllTasksUntilIdle();
}
} // namespace
} // namespace image_writer
} // namespace extensions
|
