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
|
// Copyright 2020 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_ENTERPRISE_CONNECTORS_TEST_DEEP_SCANNING_BROWSERTEST_BASE_H_
#define CHROME_BROWSER_ENTERPRISE_CONNECTORS_TEST_DEEP_SCANNING_BROWSERTEST_BASE_H_
#include "chrome/browser/enterprise/connectors/analysis/content_analysis_delegate.h"
#include "chrome/test/base/in_process_browser_test.h"
#include "components/safe_browsing/core/common/safe_browsing_prefs.h"
namespace enterprise_connectors::test {
// Base test class for deep scanning browser tests. Common utility functions
// used by browser tests should be added to this class.
class DeepScanningBrowserTestBase : public InProcessBrowserTest {
public:
DeepScanningBrowserTestBase();
~DeepScanningBrowserTestBase() override;
void TearDownOnMainThread() override;
// Sets up a FakeContentAnalysisDelegate to use this class's StatusCallback
// and EncryptionStatusCallback. Also sets up a test DM token.
void SetUpDelegate();
// Sets up a FakeContentAnalysisDelegate that never returns responses.
void SetUpUnresponsiveDelegate();
// Set up a quit closure to be called by the test. This is useful to control
// when the test ends.
void SetQuitClosure(base::RepeatingClosure quit_closure);
void CallQuitClosure();
// Set what StatusCallback returns.
void SetStatusCallbackResponse(ContentAnalysisResponse response);
// Callbacks used to set up the fake delegate factory.
ContentAnalysisResponse StatusCallback(const std::string& contents,
const base::FilePath& path);
// Creates temporary files for testing in |temp_dir_|, and add them to |data|.
// Strings in |paths| must not contain path separators.
//
// If |parent| is not the empty string, a subdirectory with this name is
// created, and the the new files are created there. Only a FilePath for
// the subdirectory is added to |data|. Use this option to help test with
// testing directory expansion.
void CreateFilesForTest(const std::vector<std::string>& paths,
const std::vector<std::string>& contents,
ContentAnalysisDelegate::Data* data,
const std::string& parent = std::string());
const std::vector<base::FilePath>& created_file_paths() const;
protected:
base::test::ScopedFeatureList scoped_feature_list_;
private:
base::RepeatingClosure quit_closure_;
ContentAnalysisResponse connector_status_callback_response_;
base::ScopedTempDir temp_dir_;
std::vector<base::FilePath> created_file_paths_;
};
} // namespace enterprise_connectors::test
#endif // CHROME_BROWSER_ENTERPRISE_CONNECTORS_TEST_DEEP_SCANNING_BROWSERTEST_BASE_H_
|
