15821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Copyright (c) 2012 The Chromium Authors. All rights reserved. 25821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Use of this source code is governed by a BSD-style license that can be 35821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// found in the LICENSE file. 45821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 55821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "remoting/host/register_support_host_request.h" 65821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 75821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "base/bind.h" 85821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "base/memory/ref_counted.h" 99ab5563a3196760eb381d102cbb2bc0f7abc6a50Ben Murdoch#include "base/message_loop/message_loop.h" 105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "base/observer_list.h" 115e3f23d412006dc4db4e659864679f29341e113fTorne (Richard Coles)#include "base/strings/string_number_conversions.h" 125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "remoting/base/constants.h" 132a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)#include "remoting/base/rsa_key_pair.h" 142a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)#include "remoting/base/test_rsa_key_pair.h" 155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "remoting/host/in_memory_host_config.h" 165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "remoting/jingle_glue/iq_sender.h" 175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "remoting/jingle_glue/mock_objects.h" 185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "testing/gmock/include/gmock/gmock.h" 195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "testing/gtest/include/gtest/gtest.h" 205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "third_party/libjingle/source/talk/xmllite/xmlelement.h" 215821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "third_party/libjingle/source/talk/xmpp/constants.h" 225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 235821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)using buzz::QName; 245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)using buzz::XmlElement; 255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)using testing::_; 275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)using testing::Invoke; 285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)using testing::NotNull; 295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)using testing::Return; 305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)using testing::SaveArg; 315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 325821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)namespace remoting { 335821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)namespace { 352a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)const char kTestBotJid[] = "remotingunittest@bot.talk.google.com"; 365821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)const char kTestJid[] = "user@gmail.com/chromoting123"; 375821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)const char kSupportId[] = "AB4RF3"; 385821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)const char kSupportIdLifetime[] = "300"; 395821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)const char kStanzaId[] = "123"; 405821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 415821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)ACTION_P(AddListener, list) { 425821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) list->AddObserver(arg0); 435821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 445821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)ACTION_P(RemoveListener, list) { 455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) list->RemoveObserver(arg0); 465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 485821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)class MockCallback { 495821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) public: 505821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) MOCK_METHOD3(OnResponse, void(bool result, const std::string& support_id, 515821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const base::TimeDelta& lifetime)); 525821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}; 535821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 545821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} // namespace 555821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 565821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)class RegisterSupportHostRequestTest : public testing::Test { 575821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) public: 585821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) protected: 595821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) virtual void SetUp() { 602a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) key_pair_ = RsaKeyPair::FromString(kTestRsaKeyPair); 61868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) ASSERT_TRUE(key_pair_.get()); 625821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 635821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_CALL(signal_strategy_, AddListener(NotNull())) 645821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) .WillRepeatedly(AddListener(&signal_strategy_listeners_)); 655821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_CALL(signal_strategy_, RemoveListener(NotNull())) 665821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) .WillRepeatedly(RemoveListener(&signal_strategy_listeners_)); 675821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_CALL(signal_strategy_, GetLocalJid()) 685821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) .WillRepeatedly(Return(kTestJid)); 695821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 705821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 71c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) base::MessageLoop message_loop_; 725821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) MockSignalStrategy signal_strategy_; 735821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ObserverList<SignalStrategy::Listener, true> signal_strategy_listeners_; 742a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) scoped_refptr<RsaKeyPair> key_pair_; 755821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) MockCallback callback_; 765821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}; 775821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 785821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)TEST_F(RegisterSupportHostRequestTest, Send) { 795821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // |iq_request| is freed by RegisterSupportHostRequest. 805821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) int64 start_time = static_cast<int64>(base::Time::Now().ToDoubleT()); 815821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 825821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) scoped_ptr<RegisterSupportHostRequest> request( 832a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) new RegisterSupportHostRequest(&signal_strategy_, key_pair_, 842a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) kTestBotJid, 855821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) base::Bind(&MockCallback::OnResponse, 865821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) base::Unretained(&callback_)))); 875821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 885821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) XmlElement* sent_iq = NULL; 895821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_CALL(signal_strategy_, GetNextId()) 905821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) .WillOnce(Return(kStanzaId)); 915821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_CALL(signal_strategy_, SendStanzaPtr(NotNull())) 925821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) .WillOnce(DoAll(SaveArg<0>(&sent_iq), Return(true))); 935821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 945821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) request->OnSignalStrategyStateChange(SignalStrategy::CONNECTED); 955821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) message_loop_.RunUntilIdle(); 965821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 975821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Verify format of the query. 985821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) scoped_ptr<XmlElement> stanza(sent_iq); 995821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ASSERT_TRUE(stanza != NULL); 1005821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 101c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) EXPECT_EQ(stanza->Attr(buzz::QName(std::string(), "to")), 1022a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) std::string(kTestBotJid)); 103c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) EXPECT_EQ(stanza->Attr(buzz::QName(std::string(), "type")), "set"); 1045821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1055821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_EQ(QName(kChromotingXmlNamespace, "register-support-host"), 1065821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) stanza->FirstElement()->Name()); 1075821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1085821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) QName signature_tag(kChromotingXmlNamespace, "signature"); 1095821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) XmlElement* signature = stanza->FirstElement()->FirstNamed(signature_tag); 1105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ASSERT_TRUE(signature != NULL); 1115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_TRUE(stanza->NextNamed(signature_tag) == NULL); 1125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) std::string time_str = 1145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) signature->Attr(QName(kChromotingXmlNamespace, "time")); 1155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) int64 time; 1165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_TRUE(base::StringToInt64(time_str, &time)); 1175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) int64 now = static_cast<int64>(base::Time::Now().ToDoubleT()); 1185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_LE(start_time, time); 1195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_GE(now, time); 1205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1212a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) scoped_refptr<RsaKeyPair> key_pair = RsaKeyPair::FromString(kTestRsaKeyPair); 122868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) ASSERT_TRUE(key_pair.get()); 1232a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 1245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) std::string expected_signature = 1252a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) key_pair->SignMessage(std::string(kTestJid) + ' ' + time_str); 1265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_EQ(expected_signature, signature->BodyText()); 1275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Generate response and verify that callback is called. 1295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_CALL(callback_, OnResponse(true, kSupportId, 1305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) base::TimeDelta::FromSeconds(300))); 1315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1325821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) scoped_ptr<XmlElement> response(new XmlElement(buzz::QN_IQ)); 133c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) response->AddAttr(QName(std::string(), "from"), kTestBotJid); 134c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) response->AddAttr(QName(std::string(), "type"), "result"); 135c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) response->AddAttr(QName(std::string(), "id"), kStanzaId); 1365821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1375821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) XmlElement* result = new XmlElement( 1385821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) QName(kChromotingXmlNamespace, "register-support-host-result")); 1395821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) response->AddElement(result); 1405821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1415821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) XmlElement* support_id = new XmlElement( 1425821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) QName(kChromotingXmlNamespace, "support-id")); 1435821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) support_id->AddText(kSupportId); 1445821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) result->AddElement(support_id); 1455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) XmlElement* support_id_lifetime = new XmlElement( 1475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) QName(kChromotingXmlNamespace, "support-id-lifetime")); 1485821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) support_id_lifetime->AddText(kSupportIdLifetime); 1495821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) result->AddElement(support_id_lifetime); 1505821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1515821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) int consumed = 0; 1525821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ObserverListBase<SignalStrategy::Listener>::Iterator it( 1535821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) signal_strategy_listeners_); 1545821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) SignalStrategy::Listener* listener; 1555821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) while ((listener = it.GetNext()) != NULL) { 1565821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (listener->OnSignalStrategyIncomingStanza(response.get())) 1575821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) consumed++; 1585821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 1595821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_EQ(1, consumed); 1605821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1615821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) message_loop_.RunUntilIdle(); 1625821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 1635821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1645821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} // namespace remoting 165