OLD | NEW |
1 // Copyright 2014 The Chromium Authors. All rights reserved. | 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 | 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/signaling/iq_sender.h" | 5 #include "remoting/signaling/iq_sender.h" |
6 | 6 |
7 #include <utility> | 7 #include <utility> |
8 | 8 |
9 #include "base/bind.h" | 9 #include "base/bind.h" |
10 #include "base/memory/ref_counted.h" | 10 #include "base/memory/ref_counted.h" |
(...skipping 42 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
53 public: | 53 public: |
54 IqSenderTest() { | 54 IqSenderTest() { |
55 EXPECT_CALL(signal_strategy_, AddListener(NotNull())); | 55 EXPECT_CALL(signal_strategy_, AddListener(NotNull())); |
56 sender_.reset(new IqSender(&signal_strategy_)); | 56 sender_.reset(new IqSender(&signal_strategy_)); |
57 EXPECT_CALL(signal_strategy_, RemoveListener( | 57 EXPECT_CALL(signal_strategy_, RemoveListener( |
58 static_cast<SignalStrategy::Listener*>(sender_.get()))); | 58 static_cast<SignalStrategy::Listener*>(sender_.get()))); |
59 } | 59 } |
60 | 60 |
61 protected: | 61 protected: |
62 void SendTestMessage() { | 62 void SendTestMessage() { |
63 scoped_ptr<XmlElement> iq_body( | 63 std::unique_ptr<XmlElement> iq_body( |
64 new XmlElement(QName(kNamespace, kBodyTag))); | 64 new XmlElement(QName(kNamespace, kBodyTag))); |
65 XmlElement* sent_stanza; | 65 XmlElement* sent_stanza; |
66 EXPECT_CALL(signal_strategy_, GetNextId()) | 66 EXPECT_CALL(signal_strategy_, GetNextId()) |
67 .WillOnce(Return(kStanzaId)); | 67 .WillOnce(Return(kStanzaId)); |
68 EXPECT_CALL(signal_strategy_, SendStanzaPtr(_)) | 68 EXPECT_CALL(signal_strategy_, SendStanzaPtr(_)) |
69 .WillOnce(DoAll(SaveArg<0>(&sent_stanza), Return(true))); | 69 .WillOnce(DoAll(SaveArg<0>(&sent_stanza), Return(true))); |
70 request_ = sender_->SendIq(kType, kTo, std::move(iq_body), base::Bind( | 70 request_ = sender_->SendIq(kType, kTo, std::move(iq_body), base::Bind( |
71 &MockCallback::OnReply, base::Unretained(&callback_))); | 71 &MockCallback::OnReply, base::Unretained(&callback_))); |
72 | 72 |
73 std::string expected_xml_string = | 73 std::string expected_xml_string = |
74 base::StringPrintf( | 74 base::StringPrintf( |
75 "<cli:iq type=\"%s\" to=\"%s\" id=\"%s\" " | 75 "<cli:iq type=\"%s\" to=\"%s\" id=\"%s\" " |
76 "xmlns:cli=\"jabber:client\">" | 76 "xmlns:cli=\"jabber:client\">" |
77 "<%s:%s xmlns:%s=\"%s\"/>" | 77 "<%s:%s xmlns:%s=\"%s\"/>" |
78 "</cli:iq>", | 78 "</cli:iq>", |
79 kType, kTo, kStanzaId, kNamespacePrefix, kBodyTag, | 79 kType, kTo, kStanzaId, kNamespacePrefix, kBodyTag, |
80 kNamespacePrefix, kNamespace); | 80 kNamespacePrefix, kNamespace); |
81 EXPECT_EQ(expected_xml_string, sent_stanza->Str()); | 81 EXPECT_EQ(expected_xml_string, sent_stanza->Str()); |
82 delete sent_stanza; | 82 delete sent_stanza; |
83 } | 83 } |
84 | 84 |
85 bool FormatAndDeliverResponse(const std::string& from, | 85 bool FormatAndDeliverResponse(const std::string& from, |
86 scoped_ptr<XmlElement>* response_out) { | 86 std::unique_ptr<XmlElement>* response_out) { |
87 scoped_ptr<XmlElement> response(new XmlElement(buzz::QN_IQ)); | 87 std::unique_ptr<XmlElement> response(new XmlElement(buzz::QN_IQ)); |
88 response->AddAttr(QName(std::string(), "type"), "result"); | 88 response->AddAttr(QName(std::string(), "type"), "result"); |
89 response->AddAttr(QName(std::string(), "id"), kStanzaId); | 89 response->AddAttr(QName(std::string(), "id"), kStanzaId); |
90 response->AddAttr(QName(std::string(), "from"), from); | 90 response->AddAttr(QName(std::string(), "from"), from); |
91 | 91 |
92 XmlElement* response_body = new XmlElement( | 92 XmlElement* response_body = new XmlElement( |
93 QName("test:namespace", "response-body")); | 93 QName("test:namespace", "response-body")); |
94 response->AddElement(response_body); | 94 response->AddElement(response_body); |
95 | 95 |
96 bool result = sender_->OnSignalStrategyIncomingStanza(response.get()); | 96 bool result = sender_->OnSignalStrategyIncomingStanza(response.get()); |
97 | 97 |
98 if (response_out) | 98 if (response_out) |
99 *response_out = std::move(response); | 99 *response_out = std::move(response); |
100 | 100 |
101 return result; | 101 return result; |
102 } | 102 } |
103 | 103 |
104 base::MessageLoop message_loop_; | 104 base::MessageLoop message_loop_; |
105 MockSignalStrategy signal_strategy_; | 105 MockSignalStrategy signal_strategy_; |
106 scoped_ptr<IqSender> sender_; | 106 std::unique_ptr<IqSender> sender_; |
107 MockCallback callback_; | 107 MockCallback callback_; |
108 scoped_ptr<IqRequest> request_; | 108 std::unique_ptr<IqRequest> request_; |
109 }; | 109 }; |
110 | 110 |
111 TEST_F(IqSenderTest, SendIq) { | 111 TEST_F(IqSenderTest, SendIq) { |
112 ASSERT_NO_FATAL_FAILURE({ | 112 ASSERT_NO_FATAL_FAILURE({ |
113 SendTestMessage(); | 113 SendTestMessage(); |
114 }); | 114 }); |
115 | 115 |
116 scoped_ptr<XmlElement> response; | 116 std::unique_ptr<XmlElement> response; |
117 EXPECT_TRUE(FormatAndDeliverResponse(kTo, &response)); | 117 EXPECT_TRUE(FormatAndDeliverResponse(kTo, &response)); |
118 | 118 |
119 EXPECT_CALL(callback_, OnReply(request_.get(), XmlEq(response.get()))); | 119 EXPECT_CALL(callback_, OnReply(request_.get(), XmlEq(response.get()))); |
120 base::RunLoop().RunUntilIdle(); | 120 base::RunLoop().RunUntilIdle(); |
121 } | 121 } |
122 | 122 |
123 TEST_F(IqSenderTest, Timeout) { | 123 TEST_F(IqSenderTest, Timeout) { |
124 ASSERT_NO_FATAL_FAILURE({ | 124 ASSERT_NO_FATAL_FAILURE({ |
125 SendTestMessage(); | 125 SendTestMessage(); |
126 }); | 126 }); |
127 | 127 |
128 request_->SetTimeout(base::TimeDelta::FromMilliseconds(2)); | 128 request_->SetTimeout(base::TimeDelta::FromMilliseconds(2)); |
129 | 129 |
130 EXPECT_CALL(callback_, OnReply(request_.get(), nullptr)) | 130 EXPECT_CALL(callback_, OnReply(request_.get(), nullptr)) |
131 .WillOnce( | 131 .WillOnce( |
132 InvokeWithoutArgs(&message_loop_, &base::MessageLoop::QuitWhenIdle)); | 132 InvokeWithoutArgs(&message_loop_, &base::MessageLoop::QuitWhenIdle)); |
133 message_loop_.Run(); | 133 message_loop_.Run(); |
134 } | 134 } |
135 | 135 |
136 TEST_F(IqSenderTest, NotNormalizedJid) { | 136 TEST_F(IqSenderTest, NotNormalizedJid) { |
137 ASSERT_NO_FATAL_FAILURE({ | 137 ASSERT_NO_FATAL_FAILURE({ |
138 SendTestMessage(); | 138 SendTestMessage(); |
139 }); | 139 }); |
140 | 140 |
141 // Set upper-case from value, which is equivalent to kTo in the original | 141 // Set upper-case from value, which is equivalent to kTo in the original |
142 // message. | 142 // message. |
143 scoped_ptr<XmlElement> response; | 143 std::unique_ptr<XmlElement> response; |
144 EXPECT_TRUE(FormatAndDeliverResponse("USER@domain.com", &response)); | 144 EXPECT_TRUE(FormatAndDeliverResponse("USER@domain.com", &response)); |
145 | 145 |
146 EXPECT_CALL(callback_, OnReply(request_.get(), XmlEq(response.get()))); | 146 EXPECT_CALL(callback_, OnReply(request_.get(), XmlEq(response.get()))); |
147 base::RunLoop().RunUntilIdle(); | 147 base::RunLoop().RunUntilIdle(); |
148 } | 148 } |
149 | 149 |
150 TEST_F(IqSenderTest, InvalidFrom) { | 150 TEST_F(IqSenderTest, InvalidFrom) { |
151 ASSERT_NO_FATAL_FAILURE({ | 151 ASSERT_NO_FATAL_FAILURE({ |
152 SendTestMessage(); | 152 SendTestMessage(); |
153 }); | 153 }); |
154 | 154 |
155 EXPECT_FALSE(FormatAndDeliverResponse("different_user@domain.com", nullptr)); | 155 EXPECT_FALSE(FormatAndDeliverResponse("different_user@domain.com", nullptr)); |
156 | 156 |
157 EXPECT_CALL(callback_, OnReply(_, _)).Times(0); | 157 EXPECT_CALL(callback_, OnReply(_, _)).Times(0); |
158 base::RunLoop().RunUntilIdle(); | 158 base::RunLoop().RunUntilIdle(); |
159 } | 159 } |
160 | 160 |
161 } // namespace remoting | 161 } // namespace remoting |
OLD | NEW |