1 // Copyright 2014 The Chromium Authors. All rights reserved.
2 // Use of this source code is governed by a BSD-style license that can be
3 // found in the LICENSE file.
4
5 #include "base/macros.h"
6 #include "base/strings/string_split.h"
7 #include "base/strings/string_util.h"
8 #include "remoting/host/client_session_details.h"
9 #include "remoting/host/fake_host_extension.h"
10 #include "remoting/host/host_extension_session_manager.h"
11 #include "remoting/host/host_mock_objects.h"
12 #include "remoting/proto/control.pb.h"
13 #include "remoting/protocol/protocol_mock_objects.h"
14 #include "testing/gtest/include/gtest/gtest.h"
15
16 namespace remoting {
17
18 class HostExtensionSessionManagerTest : public testing::Test {
19 public:
HostExtensionSessionManagerTest()20 HostExtensionSessionManagerTest()
21 : extension1_("ext1", "cap1"),
22 extension2_("ext2", std::string()),
23 extension3_("ext3", "cap3") {
24 extensions_.push_back(&extension1_);
25 extensions_.push_back(&extension2_);
26 extensions_.push_back(&extension3_);
27 }
28 ~HostExtensionSessionManagerTest() override = default;
29
30 protected:
31 // Fake HostExtensions for testing.
32 FakeExtension extension1_;
33 FakeExtension extension2_;
34 FakeExtension extension3_;
35 HostExtensionSessionManager::HostExtensions extensions_;
36
37 // Mocks of interfaces provided by ClientSession.
38 MockClientSessionDetails client_session_details_;
39 protocol::MockClientStub client_stub_;
40
41 DISALLOW_COPY_AND_ASSIGN(HostExtensionSessionManagerTest);
42 };
43
44 // Verifies that messages are handled by the correct extension.
TEST_F(HostExtensionSessionManagerTest,ExtensionMessages_MessageHandled)45 TEST_F(HostExtensionSessionManagerTest, ExtensionMessages_MessageHandled) {
46 HostExtensionSessionManager extension_manager(extensions_,
47 &client_session_details_);
48 extension_manager.OnNegotiatedCapabilities(
49 &client_stub_, extension_manager.GetCapabilities());
50
51 protocol::ExtensionMessage message;
52 message.set_type("ext2");
53 extension_manager.OnExtensionMessage(message);
54
55 EXPECT_FALSE(extension1_.has_handled_message());
56 EXPECT_TRUE(extension2_.has_handled_message());
57 EXPECT_FALSE(extension3_.has_handled_message());
58 }
59
60 // Verifies that extension messages not handled by extensions don't result in a
61 // crash.
TEST_F(HostExtensionSessionManagerTest,ExtensionMessages_MessageNotHandled)62 TEST_F(HostExtensionSessionManagerTest, ExtensionMessages_MessageNotHandled) {
63 HostExtensionSessionManager extension_manager(extensions_,
64 &client_session_details_);
65 extension_manager.OnNegotiatedCapabilities(
66 &client_stub_, extension_manager.GetCapabilities());
67
68 protocol::ExtensionMessage message;
69 message.set_type("ext4");
70 extension_manager.OnExtensionMessage(message);
71
72 EXPECT_FALSE(extension1_.has_handled_message());
73 EXPECT_FALSE(extension2_.has_handled_message());
74 EXPECT_FALSE(extension3_.has_handled_message());
75 }
76
77 // Verifies that the correct set of capabilities are reported to the client,
78 // based on the registered extensions.
TEST_F(HostExtensionSessionManagerTest,ExtensionCapabilities_AreReported)79 TEST_F(HostExtensionSessionManagerTest, ExtensionCapabilities_AreReported) {
80 HostExtensionSessionManager extension_manager(extensions_,
81 &client_session_details_);
82
83 std::vector<std::string> reported_caps = base::SplitString(
84 extension_manager.GetCapabilities(), " ", base::KEEP_WHITESPACE,
85 base::SPLIT_WANT_NONEMPTY);
86 std::sort(reported_caps.begin(), reported_caps.end());
87
88 ASSERT_EQ(2U, reported_caps.size());
89 EXPECT_EQ("cap1", reported_caps[0]);
90 EXPECT_EQ("cap3", reported_caps[1]);
91 }
92
93 // Verifies that an extension is not instantiated if the client does not
94 // support its required capability, and that it does not receive messages.
TEST_F(HostExtensionSessionManagerTest,ExtensionCapabilities_AreChecked)95 TEST_F(HostExtensionSessionManagerTest, ExtensionCapabilities_AreChecked) {
96 HostExtensionSessionManager extension_manager(extensions_,
97 &client_session_details_);
98 extension_manager.OnNegotiatedCapabilities(&client_stub_, "cap1");
99
100 protocol::ExtensionMessage message;
101 message.set_type("ext3");
102 extension_manager.OnExtensionMessage(message);
103
104 EXPECT_TRUE(extension1_.was_instantiated());
105 EXPECT_TRUE(extension2_.was_instantiated());
106 EXPECT_FALSE(extension3_.was_instantiated());
107 }
108
109 } // namespace remoting
110