Gitiles
Code Review Sign In
gerrit.openfyde.cn / webrtc.googlesource.com / src / 43cb716e55eab1cf1ae8afd2ac79b51a604d0fa5 / . / webrtc / libjingle / xmpp / xmppstanzaparser_unittest.cc
blob: 65fcac9d639e7682e23e65247d0b6652aefcc964 [file] [log] [blame]
henrike@webrtc.org269fb4b2014-10-28 22:20:11 +0000[diff] [blame]1/*
2 * Copyright 2004 The WebRTC Project Authors. All rights reserved.
3 *
4 * Use of this source code is governed by a BSD-style license
5 * that can be found in the LICENSE file in the root of the source
6 * tree. An additional intellectual property rights grant can be found
7 * in the file PATENTS. All contributing project authors may
8 * be found in the AUTHORS file in the root of the source tree.
9 */
10
11#include <iostream>
12#include <sstream>
13#include <string>
14#include "webrtc/libjingle/xmllite/xmlelement.h"
15#include "webrtc/libjingle/xmpp/xmppstanzaparser.h"
16#include "webrtc/base/common.h"
17#include "webrtc/base/gunit.h"
18
19using buzz::QName;
20using buzz::XmlElement;
21using buzz::XmppStanzaParser;
22using buzz::XmppStanzaParseHandler;
23
24class XmppStanzaParserTestHandler : public XmppStanzaParseHandler {
25 public:
26 virtual void StartStream(const XmlElement * element) {
27 ss_ << "START" << element->Str();
28 }
29 virtual void Stanza(const XmlElement * element) {
30 ss_ << "STANZA" << element->Str();
31 }
32 virtual void EndStream() {
33 ss_ << "END";
34 }
35 virtual void XmlError() {
36 ss_ << "ERROR";
37 }
38
39 std::string Str() {
40 return ss_.str();
41 }
42
43 std::string StrClear() {
44 std::string result = ss_.str();
45 ss_.str("");
46 return result;
47 }
48
49 private:
50 std::stringstream ss_;
51};
52
53
54TEST(XmppStanzaParserTest, TestTrivial) {
55 XmppStanzaParserTestHandler handler;
56 XmppStanzaParser parser(&handler);
57 std::string fragment;
58
59 fragment = "<trivial/>";
60 parser.Parse(fragment.c_str(), fragment.length(), false);
61 EXPECT_EQ("START<trivial/>END", handler.StrClear());
62}
63
64TEST(XmppStanzaParserTest, TestStanzaAtATime) {
65 XmppStanzaParserTestHandler handler;
66 XmppStanzaParser parser(&handler);
67 std::string fragment;
68
69 fragment = "<stream:stream id='abc' xmlns='j:c' xmlns:stream='str'>";
70 parser.Parse(fragment.c_str(), fragment.length(), false);
71 EXPECT_EQ("START<stream:stream id=\"abc\" xmlns=\"j:c\" "
72 "xmlns:stream=\"str\"/>", handler.StrClear());
73
74 fragment = "<message type='foo'><body>hello</body></message>";
75 parser.Parse(fragment.c_str(), fragment.length(), false);
76 EXPECT_EQ("STANZA<c:message type=\"foo\" xmlns:c=\"j:c\">"
77 "<c:body>hello</c:body></c:message>", handler.StrClear());
78
79 fragment = " SOME TEXT TO IGNORE ";
80 parser.Parse(fragment.c_str(), fragment.length(), false);
81 EXPECT_EQ("", handler.StrClear());
82
83 fragment = "<iq type='set' id='123'><abc xmlns='def'/></iq>";
84 parser.Parse(fragment.c_str(), fragment.length(), false);
85 EXPECT_EQ("STANZA<c:iq type=\"set\" id=\"123\" xmlns:c=\"j:c\">"
86 "<abc xmlns=\"def\"/></c:iq>", handler.StrClear());
87
88 fragment = "</stream:stream>";
89 parser.Parse(fragment.c_str(), fragment.length(), false);
90 EXPECT_EQ("END", handler.StrClear());
91}
92
93TEST(XmppStanzaParserTest, TestFragmentedStanzas) {
94 XmppStanzaParserTestHandler handler;
95 XmppStanzaParser parser(&handler);
96 std::string fragment;
97
98 fragment = "<stream:stream id='abc' xmlns='j:c' xml";
99 parser.Parse(fragment.c_str(), fragment.length(), false);
100 EXPECT_EQ("", handler.StrClear());
101
102 fragment = "ns:stream='str'><message type='foo'><body>hel";
103 parser.Parse(fragment.c_str(), fragment.length(), false);
104 EXPECT_EQ("START<stream:stream id=\"abc\" xmlns=\"j:c\" "
105 "xmlns:stream=\"str\"/>", handler.StrClear());
106
107 fragment = "lo</body></message> IGNORE ME <iq type='set' id='123'>"
108 "<abc xmlns='def'/></iq></st";
109 parser.Parse(fragment.c_str(), fragment.length(), false);
110 EXPECT_EQ("STANZA<c:message type=\"foo\" xmlns:c=\"j:c\">"
111 "<c:body>hello</c:body></c:message>STANZA<c:iq type=\"set\" id=\"123\" "
112 "xmlns:c=\"j:c\"><abc xmlns=\"def\"/></c:iq>", handler.StrClear());
113
114 fragment = "ream:stream>";
115 parser.Parse(fragment.c_str(), fragment.length(), false);
116 EXPECT_EQ("END", handler.StrClear());
117}
118
119TEST(XmppStanzaParserTest, TestReset) {
120 XmppStanzaParserTestHandler handler;
121 XmppStanzaParser parser(&handler);
122 std::string fragment;
123
124 fragment = "<stream:stream id='abc' xmlns='j:c' xml";
125 parser.Parse(fragment.c_str(), fragment.length(), false);
126 EXPECT_EQ("", handler.StrClear());
127
128 parser.Reset();
129 fragment = "<stream:stream id='abc' xmlns='j:c' xml";
130 parser.Parse(fragment.c_str(), fragment.length(), false);
131 EXPECT_EQ("", handler.StrClear());
132
133 fragment = "ns:stream='str'><message type='foo'><body>hel";
134 parser.Parse(fragment.c_str(), fragment.length(), false);
135 EXPECT_EQ("START<stream:stream id=\"abc\" xmlns=\"j:c\" "
136 "xmlns:stream=\"str\"/>", handler.StrClear());
137 parser.Reset();
138
139 fragment = "<stream:stream id='abc' xmlns='j:c' xmlns:stream='str'>";
140 parser.Parse(fragment.c_str(), fragment.length(), false);
141 EXPECT_EQ("START<stream:stream id=\"abc\" xmlns=\"j:c\" "
142 "xmlns:stream=\"str\"/>", handler.StrClear());
143
144 fragment = "<message type='foo'><body>hello</body></message>";
145 parser.Parse(fragment.c_str(), fragment.length(), false);
146 EXPECT_EQ("STANZA<c:message type=\"foo\" xmlns:c=\"j:c\">"
147 "<c:body>hello</c:body></c:message>", handler.StrClear());
148}
149
150TEST(XmppStanzaParserTest, TestError) {
151 XmppStanzaParserTestHandler handler;
152 XmppStanzaParser parser(&handler);
153 std::string fragment;
154
155 fragment = "<-foobar/>";
156 parser.Parse(fragment.c_str(), fragment.length(), false);
157 EXPECT_EQ("ERROR", handler.StrClear());
158
159 parser.Reset();
160 fragment = "<stream:stream/>";
161 parser.Parse(fragment.c_str(), fragment.length(), false);
162 EXPECT_EQ("ERROR", handler.StrClear());
163 parser.Reset();
164
165 fragment = "ns:stream='str'><message type='foo'><body>hel";
166 parser.Parse(fragment.c_str(), fragment.length(), false);
167 EXPECT_EQ("ERROR", handler.StrClear());
168 parser.Reset();
169
170 fragment = "<stream:stream xmlns:stream='st' xmlns='jc'>"
171 "<foo/><bar><st:foobar/></bar>";
172 parser.Parse(fragment.c_str(), fragment.length(), false);
173 EXPECT_EQ("START<stream:stream xmlns:stream=\"st\" xmlns=\"jc\"/>STANZA"
174 "<jc:foo xmlns:jc=\"jc\"/>ERROR", handler.StrClear());
175}