| // Copyright 2015 The Chromium OS Authors. All rights reserved. |
| // Use of this source code is governed by a BSD-style license that can be |
| // found in the LICENSE file. |
| |
| #include "libweave/src/notification/xmpp_channel.h" |
| |
| #include <queue> |
| |
| #include <base/test/simple_test_clock.h> |
| #include <chromeos/bind_lambda.h> |
| #include <chromeos/message_loops/fake_message_loop.h> |
| #include <chromeos/streams/fake_stream.h> |
| #include <gmock/gmock.h> |
| #include <gtest/gtest.h> |
| |
| namespace weave { |
| |
| |
| namespace { |
| |
| constexpr char kAccountName[] = "Account@Name"; |
| constexpr char kAccessToken[] = "AccessToken"; |
| |
| constexpr char kStartStreamResponse[] = |
| "<stream:stream from=\"clouddevices.gserviceaccount.com\" " |
| "id=\"0CCF520913ABA04B\" version=\"1.0\" " |
| "xmlns:stream=\"http://etherx.jabber.org/streams\" " |
| "xmlns=\"jabber:client\">" |
| "<stream:features><starttls xmlns=\"urn:ietf:params:xml:ns:xmpp-tls\">" |
| "<required/></starttls><mechanisms " |
| "xmlns=\"urn:ietf:params:xml:ns:xmpp-sasl\"><mechanism>X-OAUTH2</mechanism>" |
| "<mechanism>X-GOOGLE-TOKEN</mechanism></mechanisms></stream:features>"; |
| constexpr char kTlsStreamResponse[] = |
| "<stream:features><mechanisms xmlns=\"urn:ietf:params:xml:ns:xmpp-sasl\">" |
| "<mechanism>X-OAUTH2</mechanism>" |
| "<mechanism>X-GOOGLE-TOKEN</mechanism></mechanisms></stream:features>"; |
| constexpr char kAuthenticationSucceededResponse[] = |
| "<success xmlns=\"urn:ietf:params:xml:ns:xmpp-sasl\"/>"; |
| constexpr char kAuthenticationFailedResponse[] = |
| "<failure xmlns=\"urn:ietf:params:xml:ns:xmpp-sasl\"><not-authorized/>" |
| "</failure>"; |
| constexpr char kRestartStreamResponse[] = |
| "<stream:features><bind xmlns=\"urn:ietf:params:xml:ns:xmpp-bind\"/>" |
| "<session xmlns=\"urn:ietf:params:xml:ns:xmpp-session\"/>" |
| "</stream:features>"; |
| constexpr char kBindResponse[] = |
| "<iq id=\"1\" type=\"result\">" |
| "<bind xmlns=\"urn:ietf:params:xml:ns:xmpp-bind\">" |
| "<jid>110cc78f78d7032cc7bf2c6e14c1fa7d@clouddevices.gserviceaccount.com" |
| "/19853128</jid></bind></iq>"; |
| constexpr char kSessionResponse[] = "<iq type=\"result\" id=\"2\"/>"; |
| constexpr char kSubscribedResponse[] = |
| "<iq to=\"" |
| "110cc78f78d7032cc7bf2c6e14c1fa7d@clouddevices.gserviceaccount.com/" |
| "19853128\" from=\"" |
| "110cc78f78d7032cc7bf2c6e14c1fa7d@clouddevices.gserviceaccount.com\" " |
| "id=\"3\" type=\"result\"/>"; |
| constexpr char kStartStreamMessage[] = |
| "<stream:stream to='clouddevices.gserviceaccount.com' " |
| "xmlns:stream='http://etherx.jabber.org/streams' xml:lang='*' " |
| "version='1.0' xmlns='jabber:client'>"; |
| constexpr char kStartTlsMessage[] = |
| "<starttls xmlns='urn:ietf:params:xml:ns:xmpp-tls'/>"; |
| constexpr char kAuthenticationMessage[] = |
| "<auth xmlns='urn:ietf:params:xml:ns:xmpp-sasl' mechanism='X-OAUTH2' " |
| "auth:service='oauth2' auth:allow-non-google-login='true' " |
| "auth:client-uses-full-bind-result='true' " |
| "xmlns:auth='http://www.google.com/talk/protocol/auth'>" |
| "AEFjY291bnRATmFtZQBBY2Nlc3NUb2tlbg==</auth>"; |
| constexpr char kBindMessage[] = |
| "<iq id='1' type='set'><bind " |
| "xmlns='urn:ietf:params:xml:ns:xmpp-bind'/></iq>"; |
| constexpr char kSessionMessage[] = |
| "<iq id='2' type='set'><session " |
| "xmlns='urn:ietf:params:xml:ns:xmpp-session'/></iq>"; |
| constexpr char kSubscribeMessage[] = |
| "<iq id='3' type='set' to='Account@Name'>" |
| "<subscribe xmlns='google:push'><item channel='cloud_devices' from=''/>" |
| "</subscribe></iq>"; |
| } // namespace |
| |
| class FakeXmppChannel : public XmppChannel { |
| public: |
| explicit FakeXmppChannel(base::Clock* clock) |
| : XmppChannel{kAccountName, kAccessToken, nullptr}, |
| fake_stream_{chromeos::Stream::AccessMode::READ_WRITE, clock} {} |
| |
| XmppState state() const { return state_; } |
| void set_state(XmppState state) { state_ = state; } |
| |
| void Connect(const std::string& host, |
| uint16_t port, |
| const base::Closure& callback) override { |
| set_state(XmppState::kConnecting); |
| stream_ = &fake_stream_; |
| callback.Run(); |
| } |
| |
| void SchedulePing(base::TimeDelta interval, |
| base::TimeDelta timeout) override {} |
| |
| chromeos::FakeStream fake_stream_; |
| }; |
| |
| class XmppChannelTest : public ::testing::Test { |
| protected: |
| void SetUp() override { |
| fake_loop_.SetAsCurrent(); |
| |
| xmpp_client_.reset(new FakeXmppChannel{&clock_}); |
| clock_.SetNow(base::Time::Now()); |
| } |
| |
| void StartStream() { |
| xmpp_client_->fake_stream_.ExpectWritePacketString({}, kStartStreamMessage); |
| xmpp_client_->fake_stream_.AddReadPacketString({}, kStartStreamResponse); |
| xmpp_client_->fake_stream_.ExpectWritePacketString({}, kStartTlsMessage); |
| xmpp_client_->Start(nullptr); |
| RunTasks(4); |
| } |
| |
| void StartWithState(XmppChannel::XmppState state) { |
| StartStream(); |
| xmpp_client_->set_state(state); |
| } |
| |
| void RunTasks(size_t count) { |
| while (count > 0) { |
| EXPECT_TRUE(fake_loop_.RunOnce(true /* may_block */)); |
| count--; |
| } |
| } |
| |
| void RunLoopUntilIdle() { |
| while (fake_loop_.RunOnce(false /* may_block */)) {} |
| } |
| |
| std::unique_ptr<FakeXmppChannel> xmpp_client_; |
| base::SimpleTestClock clock_; |
| chromeos::FakeMessageLoop fake_loop_{&clock_}; |
| }; |
| |
| TEST_F(XmppChannelTest, StartStream) { |
| EXPECT_EQ(XmppChannel::XmppState::kNotStarted, xmpp_client_->state()); |
| xmpp_client_->fake_stream_.ExpectWritePacketString({}, kStartStreamMessage); |
| xmpp_client_->Start(nullptr); |
| RunTasks(1); |
| EXPECT_EQ(XmppChannel::XmppState::kConnected, xmpp_client_->state()); |
| } |
| |
| TEST_F(XmppChannelTest, HandleStartedResponse) { |
| StartStream(); |
| EXPECT_EQ(XmppChannel::XmppState::kTlsStarted, xmpp_client_->state()); |
| } |
| |
| TEST_F(XmppChannelTest, HandleTLSCompleted) { |
| StartWithState(XmppChannel::XmppState::kTlsCompleted); |
| xmpp_client_->fake_stream_.AddReadPacketString({}, kTlsStreamResponse); |
| xmpp_client_->fake_stream_.ExpectWritePacketString({}, |
| kAuthenticationMessage); |
| RunTasks(4); |
| EXPECT_EQ(XmppChannel::XmppState::kAuthenticationStarted, |
| xmpp_client_->state()); |
| } |
| |
| TEST_F(XmppChannelTest, HandleAuthenticationSucceededResponse) { |
| StartWithState(XmppChannel::XmppState::kAuthenticationStarted); |
| xmpp_client_->fake_stream_.AddReadPacketString( |
| {}, kAuthenticationSucceededResponse); |
| xmpp_client_->fake_stream_.ExpectWritePacketString({}, kStartStreamMessage); |
| RunTasks(4); |
| EXPECT_EQ(XmppChannel::XmppState::kStreamRestartedPostAuthentication, |
| xmpp_client_->state()); |
| } |
| |
| TEST_F(XmppChannelTest, HandleAuthenticationFailedResponse) { |
| StartWithState(XmppChannel::XmppState::kAuthenticationStarted); |
| xmpp_client_->fake_stream_.AddReadPacketString({}, |
| kAuthenticationFailedResponse); |
| RunTasks(3); |
| EXPECT_EQ(XmppChannel::XmppState::kAuthenticationFailed, |
| xmpp_client_->state()); |
| } |
| |
| TEST_F(XmppChannelTest, HandleStreamRestartedResponse) { |
| StartWithState(XmppChannel::XmppState::kStreamRestartedPostAuthentication); |
| xmpp_client_->fake_stream_.AddReadPacketString({}, kRestartStreamResponse); |
| xmpp_client_->fake_stream_.ExpectWritePacketString({}, kBindMessage); |
| RunTasks(3); |
| EXPECT_EQ(XmppChannel::XmppState::kBindSent, xmpp_client_->state()); |
| EXPECT_TRUE(xmpp_client_->jid().empty()); |
| |
| xmpp_client_->fake_stream_.AddReadPacketString({}, kBindResponse); |
| xmpp_client_->fake_stream_.ExpectWritePacketString({}, kSessionMessage); |
| RunTasks(9); |
| EXPECT_EQ(XmppChannel::XmppState::kSessionStarted, xmpp_client_->state()); |
| EXPECT_EQ( |
| "110cc78f78d7032cc7bf2c6e14c1fa7d@clouddevices.gserviceaccount.com" |
| "/19853128", |
| xmpp_client_->jid()); |
| |
| xmpp_client_->fake_stream_.AddReadPacketString({}, kSessionResponse); |
| xmpp_client_->fake_stream_.ExpectWritePacketString({}, kSubscribeMessage); |
| RunTasks(4); |
| EXPECT_EQ(XmppChannel::XmppState::kSubscribeStarted, xmpp_client_->state()); |
| |
| xmpp_client_->fake_stream_.AddReadPacketString({}, kSubscribedResponse); |
| RunTasks(5); |
| EXPECT_EQ(XmppChannel::XmppState::kSubscribed, xmpp_client_->state()); |
| } |
| |
| } // namespace weave |