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
|
// 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/enterprise/connectors/analysis/content_analysis_sdk_manager.h"
#include "chrome/browser/enterprise/connectors/test/fake_content_analysis_sdk_client.h"
#include "chrome/browser/enterprise/connectors/test/fake_content_analysis_sdk_manager.h"
#include "content/public/test/browser_task_environment.h"
#include "testing/gtest/include/gtest/gtest.h"
namespace enterprise_connectors {
class ContentAnalysisSdkManagerTest : public testing::Test {
public:
ContentAnalysisSdkManagerTest() = default;
private:
content::BrowserTaskEnvironment task_environment_;
};
TEST_F(ContentAnalysisSdkManagerTest, Create) {
content_analysis::sdk::Client::Config config{"local_test"};
FakeContentAnalysisSdkManager manager;
EXPECT_FALSE(manager.HasClientForTesting(config));
auto wrapped = manager.GetClient(config);
EXPECT_TRUE(manager.HasClientForTesting(config));
EXPECT_TRUE(wrapped->HasAtLeastOneRef());
}
TEST_F(ContentAnalysisSdkManagerTest, ResetClient) {
content_analysis::sdk::Client::Config config{"local_test"};
FakeContentAnalysisSdkManager manager;
auto wrapped = manager.GetClient(config);
manager.ResetClient(config);
EXPECT_FALSE(manager.HasClientForTesting(config));
// Existing refptr should still be valid.
EXPECT_TRUE(wrapped->HasOneRef());
EXPECT_EQ(config.name, wrapped->client()->GetConfig().name);
EXPECT_EQ(config.user_specific, wrapped->client()->GetConfig().user_specific);
}
TEST_F(ContentAnalysisSdkManagerTest, ResetAllClients) {
content_analysis::sdk::Client::Config config1{"local_test1"};
content_analysis::sdk::Client::Config config2{"local_test2"};
FakeContentAnalysisSdkManager manager;
auto wrapped1 = manager.GetClient(config1);
auto wrapped2 = manager.GetClient(config2);
manager.ResetAllClients();
EXPECT_TRUE(manager.NoConnectionEstablished());
EXPECT_FALSE(manager.HasClientForTesting(config1));
EXPECT_FALSE(manager.HasClientForTesting(config2));
// Existing refptrs should still be valid.
EXPECT_TRUE(wrapped1->HasOneRef());
EXPECT_EQ(config1.name, wrapped1->client()->GetConfig().name);
EXPECT_EQ(config1.user_specific,
wrapped1->client()->GetConfig().user_specific);
EXPECT_TRUE(wrapped2->HasOneRef());
EXPECT_EQ(config2.name, wrapped2->client()->GetConfig().name);
EXPECT_EQ(config2.user_specific,
wrapped2->client()->GetConfig().user_specific);
}
} // namespace enterprise_connectors
|
