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
|
// Copyright 2017 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 "chrome/browser/ui/views/chrome_cleaner_dialog_win.h"
#include <memory>
#include "base/macros.h"
#include "chrome/browser/safe_browsing/chrome_cleaner/chrome_cleaner_controller_win.h"
#include "chrome/browser/safe_browsing/chrome_cleaner/chrome_cleaner_dialog_controller_win.h"
#include "chrome/browser/safe_browsing/chrome_cleaner/mock_chrome_cleaner_controller_win.h"
#include "chrome/browser/ui/browser.h"
#include "chrome/browser/ui/browser_dialogs.h"
#include "chrome/browser/ui/test/test_browser_dialog.h"
#include "content/public/test/browser_test.h"
#include "testing/gmock/include/gmock/gmock.h"
#include "testing/gtest/include/gtest/gtest.h"
using ::testing::NiceMock;
using ::testing::Return;
namespace {
class MockChromeCleanerDialogController
: public safe_browsing::ChromeCleanerDialogController {
public:
MOCK_METHOD0(DialogShown, void());
MOCK_METHOD1(Accept, void(bool));
MOCK_METHOD0(Cancel, void());
MOCK_METHOD0(Close, void());
MOCK_METHOD0(ClosedWithoutUserInteraction, void());
MOCK_METHOD1(DetailsButtonClicked, void(bool));
MOCK_METHOD1(SetLogsEnabled, void(bool));
MOCK_METHOD0(LogsEnabled, bool());
MOCK_METHOD0(LogsManaged, bool());
};
class ChromeCleanerDialogTest : public DialogBrowserTest {
public:
ChromeCleanerDialogTest()
: mock_dialog_controller_(
std::make_unique<NiceMock<MockChromeCleanerDialogController>>()),
mock_cleaner_controller_(
std::make_unique<
NiceMock<safe_browsing::MockChromeCleanerController>>()) {
ON_CALL(*mock_dialog_controller_, LogsEnabled())
.WillByDefault(Return(true));
ON_CALL(*mock_cleaner_controller_, state())
.WillByDefault(
Return(safe_browsing::ChromeCleanerController::State::kInfected));
}
void ShowUi(const std::string& name) override {
chrome::ShowChromeCleanerPrompt(browser(), mock_dialog_controller_.get(),
mock_cleaner_controller_.get());
}
protected:
// Since the DialogBrowserTest can be run interactively, we use NiceMock here
// to suppress warnings about uninteresting calls.
std::unique_ptr<NiceMock<MockChromeCleanerDialogController>>
mock_dialog_controller_;
std::unique_ptr<NiceMock<safe_browsing::MockChromeCleanerController>>
mock_cleaner_controller_;
private:
DISALLOW_COPY_AND_ASSIGN(ChromeCleanerDialogTest);
};
IN_PROC_BROWSER_TEST_F(ChromeCleanerDialogTest, InvokeUi_default) {
ShowAndVerifyUi();
}
} // namespace
|
