| OLD | NEW |
| 1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. |
| 2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 | 4 |
| 5 #include "remoting/host/host_status_logger.h" |
| 6 |
| 5 #include "base/message_loop/message_loop.h" | 7 #include "base/message_loop/message_loop.h" |
| 6 #include "base/message_loop/message_loop_proxy.h" | 8 #include "base/message_loop/message_loop_proxy.h" |
| 7 #include "remoting/host/host_status_monitor_fake.h" | 9 #include "remoting/host/host_status_monitor_fake.h" |
| 8 #include "remoting/host/log_to_server.h" | |
| 9 #include "remoting/jingle_glue/mock_objects.h" | 10 #include "remoting/jingle_glue/mock_objects.h" |
| 10 #include "testing/gmock/include/gmock/gmock.h" | 11 #include "testing/gmock/include/gmock/gmock.h" |
| 11 #include "testing/gmock_mutant.h" | 12 #include "testing/gmock_mutant.h" |
| 12 #include "testing/gtest/include/gtest/gtest.h" | 13 #include "testing/gtest/include/gtest/gtest.h" |
| 13 #include "third_party/libjingle/source/talk/xmllite/xmlelement.h" | 14 #include "third_party/libjingle/source/talk/xmllite/xmlelement.h" |
| 14 | 15 |
| 15 using buzz::XmlElement; | 16 using buzz::XmlElement; |
| 16 using buzz::QName; | 17 using buzz::QName; |
| 17 using testing::_; | 18 using testing::_; |
| 18 using testing::DeleteArg; | 19 using testing::DeleteArg; |
| (...skipping 23 matching lines...) Expand all Loading... |
| 42 node->Attr(QName(std::string(), "mode")) == "me2me" && | 43 node->Attr(QName(std::string(), "mode")) == "me2me" && |
| 43 node->Attr(QName(std::string(), "connection-type")) == | 44 node->Attr(QName(std::string(), "connection-type")) == |
| 44 connection_type); | 45 connection_type); |
| 45 } | 46 } |
| 46 | 47 |
| 47 MATCHER_P(IsClientConnected, connection_type, "") { | 48 MATCHER_P(IsClientConnected, connection_type, "") { |
| 48 if (arg->Name() != QName(kJabberClientNamespace, "iq")) { | 49 if (arg->Name() != QName(kJabberClientNamespace, "iq")) { |
| 49 return false; | 50 return false; |
| 50 } | 51 } |
| 51 buzz::XmlElement* log_stanza = arg->FirstChild()->AsElement(); | 52 buzz::XmlElement* log_stanza = arg->FirstChild()->AsElement(); |
| 52 if (log_stanza->Name() !=QName(kChromotingNamespace, "log")) { | 53 if (log_stanza->Name() != QName(kChromotingNamespace, "log")) { |
| 53 return false; | 54 return false; |
| 54 } | 55 } |
| 55 if (log_stanza->NextChild()) { | 56 if (log_stanza->NextChild()) { |
| 56 return false; | 57 return false; |
| 57 } | 58 } |
| 58 buzz::XmlElement* log_entry = log_stanza->FirstChild()->AsElement(); | 59 buzz::XmlElement* log_entry = log_stanza->FirstChild()->AsElement(); |
| 59 if (!IsLogEntryForConnection(log_entry, connection_type)) { | 60 if (!IsLogEntryForConnection(log_entry, connection_type)) { |
| 60 return false; | 61 return false; |
| 61 } | 62 } |
| 62 if (log_entry->NextChild()) { | 63 if (log_entry->NextChild()) { |
| 63 return false; | 64 return false; |
| 64 } | 65 } |
| 65 return true; | 66 return true; |
| 66 } | 67 } |
| 67 | 68 |
| 68 MATCHER_P2(IsTwoClientsConnected, connection_type1, connection_type2, "") { | 69 MATCHER_P2(IsTwoClientsConnected, connection_type1, connection_type2, "") { |
| 69 if (arg->Name() != QName(kJabberClientNamespace, "iq")) { | 70 if (arg->Name() != QName(kJabberClientNamespace, "iq")) { |
| 70 return false; | 71 return false; |
| 71 } | 72 } |
| 72 buzz::XmlElement* log_stanza = arg->FirstChild()->AsElement(); | 73 buzz::XmlElement* log_stanza = arg->FirstChild()->AsElement(); |
| 73 if (log_stanza->Name() !=QName(kChromotingNamespace, "log")) { | 74 if (log_stanza->Name() != QName(kChromotingNamespace, "log")) { |
| 74 return false; | 75 return false; |
| 75 } | 76 } |
| 76 if (log_stanza->NextChild()) { | 77 if (log_stanza->NextChild()) { |
| 77 return false; | 78 return false; |
| 78 } | 79 } |
| 79 buzz::XmlElement* log_entry = log_stanza->FirstChild()->AsElement(); | 80 buzz::XmlElement* log_entry = log_stanza->FirstChild()->AsElement(); |
| 80 if (!IsLogEntryForConnection(log_entry, connection_type1)) { | 81 if (!IsLogEntryForConnection(log_entry, connection_type1)) { |
| 81 return false; | 82 return false; |
| 82 } | 83 } |
| 83 log_entry = log_entry->NextChild()->AsElement(); | 84 log_entry = log_entry->NextChild()->AsElement(); |
| (...skipping 30 matching lines...) Expand all Loading... |
| 114 return false; | 115 return false; |
| 115 } | 116 } |
| 116 if (log_entry->NextChild()) { | 117 if (log_entry->NextChild()) { |
| 117 return false; | 118 return false; |
| 118 } | 119 } |
| 119 return true; | 120 return true; |
| 120 } | 121 } |
| 121 | 122 |
| 122 } // namespace | 123 } // namespace |
| 123 | 124 |
| 124 class LogToServerTest : public testing::Test { | 125 class HostStatusLoggerTest : public testing::Test { |
| 125 public: | 126 public: |
| 126 LogToServerTest() {} | 127 HostStatusLoggerTest() {} |
| 127 virtual void SetUp() OVERRIDE { | 128 virtual void SetUp() OVERRIDE { |
| 128 message_loop_proxy_ = base::MessageLoopProxy::current(); | 129 message_loop_proxy_ = base::MessageLoopProxy::current(); |
| 129 EXPECT_CALL(signal_strategy_, AddListener(_)); | 130 EXPECT_CALL(signal_strategy_, AddListener(_)); |
| 130 log_to_server_.reset( | 131 host_status_logger_.reset( |
| 131 new LogToServer(host_status_monitor_.AsWeakPtr(), | 132 new HostStatusLogger(host_status_monitor_.AsWeakPtr(), |
| 132 ServerLogEntry::ME2ME, | 133 ServerLogEntry::ME2ME, |
| 133 &signal_strategy_, | 134 &signal_strategy_, |
| 134 kTestBotJid)); | 135 kTestBotJid)); |
| 135 EXPECT_CALL(signal_strategy_, RemoveListener(_)); | 136 EXPECT_CALL(signal_strategy_, RemoveListener(_)); |
| 136 } | 137 } |
| 137 | 138 |
| 138 protected: | 139 protected: |
| 139 base::MessageLoop message_loop_; | 140 base::MessageLoop message_loop_; |
| 140 scoped_refptr<base::MessageLoopProxy> message_loop_proxy_; | 141 scoped_refptr<base::MessageLoopProxy> message_loop_proxy_; |
| 141 MockSignalStrategy signal_strategy_; | 142 MockSignalStrategy signal_strategy_; |
| 142 scoped_ptr<LogToServer> log_to_server_; | 143 scoped_ptr<HostStatusLogger> host_status_logger_; |
| 143 HostStatusMonitorFake host_status_monitor_; | 144 HostStatusMonitorFake host_status_monitor_; |
| 144 }; | 145 }; |
| 145 | 146 |
| 146 TEST_F(LogToServerTest, SendNow) { | 147 TEST_F(HostStatusLoggerTest, SendNow) { |
| 147 { | 148 { |
| 148 InSequence s; | 149 InSequence s; |
| 149 EXPECT_CALL(signal_strategy_, GetLocalJid()) | 150 EXPECT_CALL(signal_strategy_, GetLocalJid()) |
| 150 .WillRepeatedly(Return(kHostJid)); | 151 .WillRepeatedly(Return(kHostJid)); |
| 151 EXPECT_CALL(signal_strategy_, AddListener(_)); | 152 EXPECT_CALL(signal_strategy_, AddListener(_)); |
| 152 EXPECT_CALL(signal_strategy_, GetNextId()); | 153 EXPECT_CALL(signal_strategy_, GetNextId()); |
| 153 EXPECT_CALL(signal_strategy_, SendStanzaPtr(IsClientConnected("direct"))) | 154 EXPECT_CALL(signal_strategy_, SendStanzaPtr(IsClientConnected("direct"))) |
| 154 .WillOnce(DoAll(DeleteArg<0>(), Return(true))); | 155 .WillOnce(DoAll(DeleteArg<0>(), Return(true))); |
| 155 EXPECT_CALL(signal_strategy_, RemoveListener(_)) | 156 EXPECT_CALL(signal_strategy_, RemoveListener(_)) |
| 156 .WillOnce(QuitMainMessageLoop(&message_loop_)) | 157 .WillOnce(QuitMainMessageLoop(&message_loop_)) |
| 157 .RetiresOnSaturation(); | 158 .RetiresOnSaturation(); |
| 158 } | 159 } |
| 159 log_to_server_->OnSignalStrategyStateChange(SignalStrategy::CONNECTED); | 160 host_status_logger_->SetSignalingStateForTest(SignalStrategy::CONNECTED); |
| 160 protocol::TransportRoute route; | 161 protocol::TransportRoute route; |
| 161 route.type = protocol::TransportRoute::DIRECT; | 162 route.type = protocol::TransportRoute::DIRECT; |
| 162 log_to_server_->OnClientRouteChange(kClientJid1, "video", route); | 163 host_status_logger_->OnClientRouteChange(kClientJid1, "video", route); |
| 163 log_to_server_->OnClientAuthenticated(kClientJid1); | 164 host_status_logger_->OnClientAuthenticated(kClientJid1); |
| 164 log_to_server_->OnClientConnected(kClientJid1); | 165 host_status_logger_->OnClientConnected(kClientJid1); |
| 165 log_to_server_->OnSignalStrategyStateChange(SignalStrategy::DISCONNECTED); | 166 host_status_logger_->SetSignalingStateForTest( |
| 167 SignalStrategy::DISCONNECTED); |
| 166 message_loop_.Run(); | 168 message_loop_.Run(); |
| 167 } | 169 } |
| 168 | 170 |
| 169 TEST_F(LogToServerTest, SendLater) { | 171 TEST_F(HostStatusLoggerTest, SendLater) { |
| 170 protocol::TransportRoute route; | 172 protocol::TransportRoute route; |
| 171 route.type = protocol::TransportRoute::DIRECT; | 173 route.type = protocol::TransportRoute::DIRECT; |
| 172 log_to_server_->OnClientRouteChange(kClientJid1, "video", route); | 174 host_status_logger_->OnClientRouteChange(kClientJid1, "video", route); |
| 173 log_to_server_->OnClientAuthenticated(kClientJid1); | 175 host_status_logger_->OnClientAuthenticated(kClientJid1); |
| 174 log_to_server_->OnClientConnected(kClientJid1); | 176 host_status_logger_->OnClientConnected(kClientJid1); |
| 175 { | 177 { |
| 176 InSequence s; | 178 InSequence s; |
| 177 EXPECT_CALL(signal_strategy_, GetLocalJid()) | 179 EXPECT_CALL(signal_strategy_, GetLocalJid()) |
| 178 .WillRepeatedly(Return(kHostJid)); | 180 .WillRepeatedly(Return(kHostJid)); |
| 179 EXPECT_CALL(signal_strategy_, AddListener(_)); | 181 EXPECT_CALL(signal_strategy_, AddListener(_)); |
| 180 EXPECT_CALL(signal_strategy_, GetNextId()); | 182 EXPECT_CALL(signal_strategy_, GetNextId()); |
| 181 EXPECT_CALL(signal_strategy_, SendStanzaPtr(IsClientConnected("direct"))) | 183 EXPECT_CALL(signal_strategy_, SendStanzaPtr(IsClientConnected("direct"))) |
| 182 .WillOnce(DoAll(DeleteArg<0>(), Return(true))); | 184 .WillOnce(DoAll(DeleteArg<0>(), Return(true))); |
| 183 EXPECT_CALL(signal_strategy_, RemoveListener(_)) | 185 EXPECT_CALL(signal_strategy_, RemoveListener(_)) |
| 184 .WillOnce(QuitMainMessageLoop(&message_loop_)) | 186 .WillOnce(QuitMainMessageLoop(&message_loop_)) |
| 185 .RetiresOnSaturation(); | 187 .RetiresOnSaturation(); |
| 186 } | 188 } |
| 187 log_to_server_->OnSignalStrategyStateChange(SignalStrategy::CONNECTED); | 189 host_status_logger_->SetSignalingStateForTest(SignalStrategy::CONNECTED); |
| 188 log_to_server_->OnSignalStrategyStateChange(SignalStrategy::DISCONNECTED); | 190 host_status_logger_->SetSignalingStateForTest(SignalStrategy::DISCONNECTED); |
| 189 message_loop_.Run(); | 191 message_loop_.Run(); |
| 190 } | 192 } |
| 191 | 193 |
| 192 TEST_F(LogToServerTest, SendTwoEntriesLater) { | 194 TEST_F(HostStatusLoggerTest, SendTwoEntriesLater) { |
| 193 protocol::TransportRoute route1; | 195 protocol::TransportRoute route1; |
| 194 route1.type = protocol::TransportRoute::DIRECT; | 196 route1.type = protocol::TransportRoute::DIRECT; |
| 195 log_to_server_->OnClientRouteChange(kClientJid1, "video", route1); | 197 host_status_logger_->OnClientRouteChange(kClientJid1, "video", route1); |
| 196 log_to_server_->OnClientAuthenticated(kClientJid1); | 198 host_status_logger_->OnClientAuthenticated(kClientJid1); |
| 197 log_to_server_->OnClientConnected(kClientJid1); | 199 host_status_logger_->OnClientConnected(kClientJid1); |
| 198 protocol::TransportRoute route2; | 200 protocol::TransportRoute route2; |
| 199 route2.type = protocol::TransportRoute::STUN; | 201 route2.type = protocol::TransportRoute::STUN; |
| 200 log_to_server_->OnClientRouteChange(kClientJid2, "video", route2); | 202 host_status_logger_->OnClientRouteChange(kClientJid2, "video", route2); |
| 201 log_to_server_->OnClientAuthenticated(kClientJid2); | 203 host_status_logger_->OnClientAuthenticated(kClientJid2); |
| 202 log_to_server_->OnClientConnected(kClientJid2); | 204 host_status_logger_->OnClientConnected(kClientJid2); |
| 203 { | 205 { |
| 204 InSequence s; | 206 InSequence s; |
| 205 EXPECT_CALL(signal_strategy_, GetLocalJid()) | 207 EXPECT_CALL(signal_strategy_, GetLocalJid()) |
| 206 .WillRepeatedly(Return(kHostJid)); | 208 .WillRepeatedly(Return(kHostJid)); |
| 207 EXPECT_CALL(signal_strategy_, AddListener(_)); | 209 EXPECT_CALL(signal_strategy_, AddListener(_)); |
| 208 EXPECT_CALL(signal_strategy_, GetNextId()); | 210 EXPECT_CALL(signal_strategy_, GetNextId()); |
| 209 EXPECT_CALL(signal_strategy_, SendStanzaPtr( | 211 EXPECT_CALL(signal_strategy_, |
| 210 IsTwoClientsConnected("direct", "stun"))) | 212 SendStanzaPtr(IsTwoClientsConnected("direct", "stun"))) |
| 211 .WillOnce(DoAll(DeleteArg<0>(), Return(true))); | 213 .WillOnce(DoAll(DeleteArg<0>(), Return(true))); |
| 212 EXPECT_CALL(signal_strategy_, RemoveListener(_)) | 214 EXPECT_CALL(signal_strategy_, RemoveListener(_)) |
| 213 .WillOnce(QuitMainMessageLoop(&message_loop_)) | 215 .WillOnce(QuitMainMessageLoop(&message_loop_)) |
| 214 .RetiresOnSaturation(); | 216 .RetiresOnSaturation(); |
| 215 } | 217 } |
| 216 log_to_server_->OnSignalStrategyStateChange(SignalStrategy::CONNECTED); | 218 host_status_logger_->SetSignalingStateForTest(SignalStrategy::CONNECTED); |
| 217 log_to_server_->OnSignalStrategyStateChange(SignalStrategy::DISCONNECTED); | 219 host_status_logger_->SetSignalingStateForTest(SignalStrategy::DISCONNECTED); |
| 218 message_loop_.Run(); | 220 message_loop_.Run(); |
| 219 } | 221 } |
| 220 | 222 |
| 221 TEST_F(LogToServerTest, HandleRouteChangeInUnusualOrder) { | 223 TEST_F(HostStatusLoggerTest, HandleRouteChangeInUnusualOrder) { |
| 222 { | 224 { |
| 223 InSequence s; | 225 InSequence s; |
| 224 EXPECT_CALL(signal_strategy_, GetLocalJid()) | 226 EXPECT_CALL(signal_strategy_, GetLocalJid()) |
| 225 .WillRepeatedly(Return(kHostJid)); | 227 .WillRepeatedly(Return(kHostJid)); |
| 226 EXPECT_CALL(signal_strategy_, AddListener(_)); | 228 EXPECT_CALL(signal_strategy_, AddListener(_)); |
| 227 EXPECT_CALL(signal_strategy_, GetNextId()); | 229 EXPECT_CALL(signal_strategy_, GetNextId()); |
| 228 EXPECT_CALL(signal_strategy_, SendStanzaPtr(IsClientConnected("direct"))) | 230 EXPECT_CALL(signal_strategy_, SendStanzaPtr(IsClientConnected("direct"))) |
| 229 .WillOnce(DoAll(DeleteArg<0>(), Return(true))); | 231 .WillOnce(DoAll(DeleteArg<0>(), Return(true))); |
| 230 EXPECT_CALL(signal_strategy_, GetNextId()); | 232 EXPECT_CALL(signal_strategy_, GetNextId()); |
| 231 EXPECT_CALL(signal_strategy_, SendStanzaPtr(IsClientDisconnected())) | 233 EXPECT_CALL(signal_strategy_, SendStanzaPtr(IsClientDisconnected())) |
| 232 .WillOnce(DoAll(DeleteArg<0>(), Return(true))); | 234 .WillOnce(DoAll(DeleteArg<0>(), Return(true))); |
| 233 EXPECT_CALL(signal_strategy_, GetNextId()); | 235 EXPECT_CALL(signal_strategy_, GetNextId()); |
| 234 EXPECT_CALL(signal_strategy_, SendStanzaPtr(IsClientConnected("stun"))) | 236 EXPECT_CALL(signal_strategy_, SendStanzaPtr(IsClientConnected("stun"))) |
| 235 .WillOnce(DoAll(DeleteArg<0>(), Return(true))); | 237 .WillOnce(DoAll(DeleteArg<0>(), Return(true))); |
| 236 EXPECT_CALL(signal_strategy_, RemoveListener(_)) | 238 EXPECT_CALL(signal_strategy_, RemoveListener(_)) |
| 237 .WillOnce(QuitMainMessageLoop(&message_loop_)) | 239 .WillOnce(QuitMainMessageLoop(&message_loop_)) |
| 238 .RetiresOnSaturation(); | 240 .RetiresOnSaturation(); |
| 239 } | 241 } |
| 240 log_to_server_->OnSignalStrategyStateChange(SignalStrategy::CONNECTED); | 242 host_status_logger_->SetSignalingStateForTest(SignalStrategy::CONNECTED); |
| 241 protocol::TransportRoute route1; | 243 protocol::TransportRoute route1; |
| 242 route1.type = protocol::TransportRoute::DIRECT; | 244 route1.type = protocol::TransportRoute::DIRECT; |
| 243 log_to_server_->OnClientRouteChange(kClientJid1, "video", route1); | 245 host_status_logger_->OnClientRouteChange(kClientJid1, "video", route1); |
| 244 log_to_server_->OnClientAuthenticated(kClientJid1); | 246 host_status_logger_->OnClientAuthenticated(kClientJid1); |
| 245 log_to_server_->OnClientConnected(kClientJid1); | 247 host_status_logger_->OnClientConnected(kClientJid1); |
| 246 protocol::TransportRoute route2; | 248 protocol::TransportRoute route2; |
| 247 route2.type = protocol::TransportRoute::STUN; | 249 route2.type = protocol::TransportRoute::STUN; |
| 248 log_to_server_->OnClientRouteChange(kClientJid2, "video", route2); | 250 host_status_logger_->OnClientRouteChange(kClientJid2, "video", route2); |
| 249 log_to_server_->OnClientDisconnected(kClientJid1); | 251 host_status_logger_->OnClientDisconnected(kClientJid1); |
| 250 log_to_server_->OnClientAuthenticated(kClientJid2); | 252 host_status_logger_->OnClientAuthenticated(kClientJid2); |
| 251 log_to_server_->OnClientConnected(kClientJid2); | 253 host_status_logger_->OnClientConnected(kClientJid2); |
| 252 log_to_server_->OnSignalStrategyStateChange(SignalStrategy::DISCONNECTED); | 254 host_status_logger_->SetSignalingStateForTest(SignalStrategy::DISCONNECTED); |
| 253 message_loop_.Run(); | 255 message_loop_.Run(); |
| 254 } | 256 } |
| 255 | 257 |
| 256 } // namespace remoting | 258 } // namespace remoting |
| OLD | NEW |