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
|
// Copyright 2015 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#ifndef CHROME_BROWSER_EXTENSIONS_EXTENSION_INSTALL_PROMPT_TEST_HELPER_H_
#define CHROME_BROWSER_EXTENSIONS_EXTENSION_INSTALL_PROMPT_TEST_HELPER_H_
#include "chrome/browser/extensions/extension_install_prompt.h"
// A helper class to be used with ExtensionInstallPrompt that keeps track of the
// payload. Note that this class does no lifetime management.
class ExtensionInstallPromptTestHelper {
public:
ExtensionInstallPromptTestHelper();
explicit ExtensionInstallPromptTestHelper(base::OnceClosure quit_closure);
ExtensionInstallPromptTestHelper(const ExtensionInstallPromptTestHelper&) =
delete;
ExtensionInstallPromptTestHelper& operator=(
const ExtensionInstallPromptTestHelper&) = delete;
~ExtensionInstallPromptTestHelper();
// Returns a callback to be used with the ExtensionInstallPrompt.
ExtensionInstallPrompt::DoneCallback GetCallback();
// Note: This causes `ADD_FAILURE()` if `payload_` has not been set.
ExtensionInstallPrompt::DoneCallbackPayload payload() const;
// Note: This causes `ADD_FAILURE()` if `payload_` has not been set.
ExtensionInstallPrompt::Result result() const;
// Note: This causes `ADD_FAILURE()` if `payload_` has not been set.
std::string justification() const;
bool has_payload() const { return payload_ != nullptr; }
// Clears the payload to re-use this test helper.
// Note: This ADD_FAILURE()s if the payload_ has not been set.
void ClearPayloadForTesting();
private:
void HandlePayload(ExtensionInstallPrompt::DoneCallbackPayload payload);
std::unique_ptr<ExtensionInstallPrompt::DoneCallbackPayload> payload_;
// A closure to run once HandlePayload() has been called; used for exiting
// run loops in tests.
base::OnceClosure quit_closure_;
};
#endif // CHROME_BROWSER_EXTENSIONS_EXTENSION_INSTALL_PROMPT_TEST_HELPER_H_
|
