| OLD | NEW |
| (Empty) | |
| 1 // Copyright 2015 The Chromium Authors. All rights reserved. |
| 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. |
| 4 |
| 5 #include "remoting/signaling/xmpp_stream_parser.h" |
| 6 |
| 7 #include "base/bind.h" |
| 8 #include "base/memory/scoped_vector.h" |
| 9 #include "base/message_loop/message_loop.h" |
| 10 #include "base/run_loop.h" |
| 11 #include "testing/gtest/include/gtest/gtest.h" |
| 12 #include "third_party/webrtc/libjingle/xmllite/xmlelement.h" |
| 13 |
| 14 namespace remoting { |
| 15 |
| 16 class XmppStreamParserTest : public testing::Test { |
| 17 public: |
| 18 XmppStreamParserTest() : error_(false) {} |
| 19 |
| 20 void SetUp() override { |
| 21 parser_.reset(new remoting::XmppStreamParser()); |
| 22 parser_->SetCallbacks( |
| 23 base::Bind(&XmppStreamParserTest::OnStanza, base::Unretained(this)), |
| 24 base::Bind(&XmppStreamParserTest::OnError, base::Unretained(this))); |
| 25 } |
| 26 |
| 27 void TearDown() override { |
| 28 parser_.reset(); |
| 29 base::RunLoop().RunUntilIdle(); |
| 30 } |
| 31 |
| 32 void OnStanza(scoped_ptr<buzz::XmlElement> stanza) { |
| 33 received_stanzas_.push_back(stanza.Pass()); |
| 34 } |
| 35 |
| 36 void OnError() { |
| 37 error_ = true; |
| 38 } |
| 39 |
| 40 protected: |
| 41 base::MessageLoop message_loop_; |
| 42 |
| 43 scoped_ptr<XmppStreamParser> parser_; |
| 44 ScopedVector<buzz::XmlElement> received_stanzas_; |
| 45 bool error_; |
| 46 }; |
| 47 |
| 48 TEST_F(XmppStreamParserTest, ParseXmppStream) { |
| 49 parser_->AppendData("<stream><iq>text</iq>"); |
| 50 EXPECT_EQ(received_stanzas_[0]->Str(), "<iq>text</iq>"); |
| 51 }; |
| 52 |
| 53 TEST_F(XmppStreamParserTest, HandleMultipleIncomingStanzas) { |
| 54 parser_->AppendData("<stream><iq>text</iq><iq>more text</iq>"); |
| 55 EXPECT_EQ(received_stanzas_[0]->Str(), "<iq>text</iq>"); |
| 56 EXPECT_EQ(received_stanzas_[1]->Str(), "<iq>more text</iq>"); |
| 57 }; |
| 58 |
| 59 TEST_F(XmppStreamParserTest, IgnoreWhitespaceBetweenStanzas) { |
| 60 parser_->AppendData("<stream> <iq>text</iq>"); |
| 61 EXPECT_EQ(received_stanzas_[0]->Str(), "<iq>text</iq>"); |
| 62 }; |
| 63 |
| 64 TEST_F(XmppStreamParserTest, AssembleMessagesFromChunks) { |
| 65 parser_->AppendData("<stream><i"); |
| 66 parser_->AppendData("q>"); |
| 67 |
| 68 // Split one UTF-8 sequence into two chunks |
| 69 std::string data = "😃"; |
| 70 parser_->AppendData(data.substr(0, 2)); |
| 71 parser_->AppendData(data.substr(2)); |
| 72 |
| 73 parser_->AppendData("</iq>"); |
| 74 |
| 75 EXPECT_EQ(received_stanzas_[0]->Str(), "<iq>😃</iq>"); |
| 76 }; |
| 77 |
| 78 TEST_F(XmppStreamParserTest, StopParsingOnErrors) { |
| 79 parser_->AppendData("<stream><invalidtag p!='a'></invalidtag><iq>text</iq>"); |
| 80 EXPECT_TRUE(error_); |
| 81 EXPECT_TRUE(received_stanzas_.empty()); |
| 82 }; |
| 83 |
| 84 TEST_F(XmppStreamParserTest, FailOnInvalidStreamHeader) { |
| 85 parser_->AppendData("<stream p!='a'>"); |
| 86 EXPECT_TRUE(error_); |
| 87 }; |
| 88 |
| 89 TEST_F(XmppStreamParserTest, FailOnLooseText) { |
| 90 parser_->AppendData("stream<"); |
| 91 EXPECT_TRUE(error_); |
| 92 }; |
| 93 |
| 94 } // namespace remoting |
| OLD | NEW |