Chromium Code Reviews| Index: remoting/host/it2me/it2me_native_messaging_host_unittest.cc |
| diff --git a/remoting/host/it2me/it2me_native_messaging_host_unittest.cc b/remoting/host/it2me/it2me_native_messaging_host_unittest.cc |
| index 072bfb6d2a8ce15b36559c4b8b33b92b0d30f74b..45b9af4b25099c2704fc21c5014bbb00882c6084 100644 |
| --- a/remoting/host/it2me/it2me_native_messaging_host_unittest.cc |
| +++ b/remoting/host/it2me/it2me_native_messaging_host_unittest.cc |
| @@ -125,13 +125,11 @@ void FakePolicyService::RefreshPolicies(const base::Closure& callback) { |
| class MockIt2MeHost : public It2MeHost { |
| public: |
| MockIt2MeHost(std::unique_ptr<ChromotingHostContext> context, |
| - std::unique_ptr<PolicyWatcher> policy_watcher, |
| base::WeakPtr<It2MeHost::Observer> observer, |
| std::unique_ptr<SignalStrategy> signal_strategy, |
| const std::string& username, |
| const std::string& directory_bot_jid) |
| : It2MeHost(std::move(context), |
| - std::move(policy_watcher), |
| /*confirmation_dialog_factory=*/nullptr, |
| observer, |
| std::move(signal_strategy), |
| @@ -229,8 +227,7 @@ scoped_refptr<It2MeHost> MockIt2MeHostFactory::CreateIt2MeHost( |
| std::unique_ptr<SignalStrategy> signal_strategy, |
| const std::string& username, |
| const std::string& directory_bot_jid) { |
| - return new MockIt2MeHost(std::move(context), |
| - /*policy_watcher=*/nullptr, observer, |
| + return new MockIt2MeHost(std::move(context), observer, |
| std::move(signal_strategy), username, |
| directory_bot_jid); |
| } |
| @@ -399,14 +396,16 @@ void It2MeNativeMessagingHostTest::VerifyErrorResponse() { |
| void It2MeNativeMessagingHostTest::VerifyConnectResponses(int request_id) { |
| bool connect_response_received = false; |
| + bool nat_policy_received = false; |
|
rkjnsn
2017/05/03 21:31:47
I see this was added to fix the unit test, but I c
Jamie
2017/05/03 23:41:10
The change to the native messaging host means that
rkjnsn
2017/05/04 20:45:45
Ah, I see. Since MockIt2MeHost overrides Connect,
|
| bool starting_received = false; |
| bool requestedAccessCode_received = false; |
| bool receivedAccessCode_received = false; |
| bool connecting_received = false; |
| bool connected_received = false; |
| - // We expect a total of 6 messages: 1 connectResponse and 5 hostStateChanged. |
| - for (int i = 0; i < 6; ++i) { |
| + // We expect a total of 7 messages: 1 connectResponse, 1 natPolicyChanged, |
| + // and 5 hostStateChanged. |
| + for (int i = 0; i < 7; ++i) { |
| std::unique_ptr<base::DictionaryValue> response = |
| ReadMessageFromOutputPipe(); |
| ASSERT_TRUE(response); |
| @@ -418,6 +417,9 @@ void It2MeNativeMessagingHostTest::VerifyConnectResponses(int request_id) { |
| EXPECT_FALSE(connect_response_received); |
| connect_response_received = true; |
| VerifyId(std::move(response), request_id); |
| + } else if (type == "natPolicyChanged") { |
| + EXPECT_FALSE(nat_policy_received); |
| + nat_policy_received = true; |
| } else if (type == "hostStateChanged") { |
| std::string state; |
| ASSERT_TRUE(response->GetString("state", &state)); |
| @@ -609,7 +611,7 @@ TEST_F(It2MeNativeMessagingHostTest, Id) { |
| EXPECT_EQ("42", value); |
| } |
| -TEST_F(It2MeNativeMessagingHostTest, Connect) { |
| +TEST_F(It2MeNativeMessagingHostTest, ConnectMultiple) { |
| // A new It2MeHost instance is created for every it2me session. The native |
| // messaging host, on the other hand, is long lived. This test verifies |
| // multiple It2Me host startup and shutdowns. |