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 2014 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 "base/macros.h"
#include "base/strings/string_split.h"
#include "base/strings/string_util.h"
#include "remoting/host/client_session_details.h"
#include "remoting/host/fake_host_extension.h"
#include "remoting/host/host_extension_session_manager.h"
#include "remoting/host/host_mock_objects.h"
#include "remoting/proto/control.pb.h"
#include "remoting/protocol/protocol_mock_objects.h"
#include "testing/gtest/include/gtest/gtest.h"
namespace remoting {
class HostExtensionSessionManagerTest : public testing::Test {
public:
HostExtensionSessionManagerTest()
: extension1_("ext1", "cap1"),
extension2_("ext2", std::string()),
extension3_("ext3", "cap3") {
extensions_.push_back(&extension1_);
extensions_.push_back(&extension2_);
extensions_.push_back(&extension3_);
}
~HostExtensionSessionManagerTest() override {}
protected:
// Fake HostExtensions for testing.
FakeExtension extension1_;
FakeExtension extension2_;
FakeExtension extension3_;
HostExtensionSessionManager::HostExtensions extensions_;
// Mocks of interfaces provided by ClientSession.
MockClientSessionDetails client_session_details_;
protocol::MockClientStub client_stub_;
DISALLOW_COPY_AND_ASSIGN(HostExtensionSessionManagerTest);
};
// Verifies that messages are handled by the correct extension.
TEST_F(HostExtensionSessionManagerTest, ExtensionMessages_MessageHandled) {
HostExtensionSessionManager extension_manager(extensions_,
&client_session_details_);
extension_manager.OnNegotiatedCapabilities(
&client_stub_, extension_manager.GetCapabilities());
protocol::ExtensionMessage message;
message.set_type("ext2");
extension_manager.OnExtensionMessage(message);
EXPECT_FALSE(extension1_.has_handled_message());
EXPECT_TRUE(extension2_.has_handled_message());
EXPECT_FALSE(extension3_.has_handled_message());
}
// Verifies that extension messages not handled by extensions don't result in a
// crash.
TEST_F(HostExtensionSessionManagerTest, ExtensionMessages_MessageNotHandled) {
HostExtensionSessionManager extension_manager(extensions_,
&client_session_details_);
extension_manager.OnNegotiatedCapabilities(
&client_stub_, extension_manager.GetCapabilities());
protocol::ExtensionMessage message;
message.set_type("ext4");
extension_manager.OnExtensionMessage(message);
EXPECT_FALSE(extension1_.has_handled_message());
EXPECT_FALSE(extension2_.has_handled_message());
EXPECT_FALSE(extension3_.has_handled_message());
}
// Verifies that the correct set of capabilities are reported to the client,
// based on the registered extensions.
TEST_F(HostExtensionSessionManagerTest, ExtensionCapabilities_AreReported) {
HostExtensionSessionManager extension_manager(extensions_,
&client_session_details_);
std::vector<std::string> reported_caps = base::SplitString(
extension_manager.GetCapabilities(), " ", base::KEEP_WHITESPACE,
base::SPLIT_WANT_NONEMPTY);
std::sort(reported_caps.begin(), reported_caps.end());
ASSERT_EQ(2U, reported_caps.size());
EXPECT_EQ("cap1", reported_caps[0]);
EXPECT_EQ("cap3", reported_caps[1]);
}
// Verifies that an extension is not instantiated if the client does not
// support its required capability, and that it does not receive messages.
TEST_F(HostExtensionSessionManagerTest, ExtensionCapabilities_AreChecked) {
HostExtensionSessionManager extension_manager(extensions_,
&client_session_details_);
extension_manager.OnNegotiatedCapabilities(&client_stub_, "cap1");
protocol::ExtensionMessage message;
message.set_type("ext3");
extension_manager.OnExtensionMessage(message);
EXPECT_TRUE(extension1_.was_instantiated());
EXPECT_TRUE(extension2_.was_instantiated());
EXPECT_FALSE(extension3_.was_instantiated());
}
} // namespace remoting
|
