176 lines
5.8 KiB
C++
176 lines
5.8 KiB
C++
|
/*
|
||
|
|
* Copyright 2004 The WebRTC Project Authors. All rights reserved.
|
||
|
|
*
|
||
|
|
* Use of this source code is governed by a BSD-style license
|
||
|
|
* that can be found in the LICENSE file in the root of the source
|
||
|
|
* tree. An additional intellectual property rights grant can be found
|
||
|
|
* in the file PATENTS. All contributing project authors may
|
||
|
|
* be found in the AUTHORS file in the root of the source tree.
|
||
|
|
*/
|
||
|
|
|
||
|
|
#include <iostream>
|
||
|
|
#include <sstream>
|
||
|
|
#include <string>
|
||
|
|
#include "webrtc/libjingle/xmllite/xmlelement.h"
|
||
|
|
#include "webrtc/libjingle/xmpp/xmppstanzaparser.h"
|
||
|
|
#include "webrtc/base/common.h"
|
||
|
|
#include "webrtc/base/gunit.h"
|
||
|
|
|
||
|
|
using buzz::QName;
|
||
|
|
using buzz::XmlElement;
|
||
|
|
using buzz::XmppStanzaParser;
|
||
|
|
using buzz::XmppStanzaParseHandler;
|
||
|
|
|
||
|
|
class XmppStanzaParserTestHandler : public XmppStanzaParseHandler {
|
||
|
|
public:
|
||
|
|
virtual void StartStream(const XmlElement * element) {
|
||
|
|
ss_ << "START" << element->Str();
|
||
|
|
}
|
||
|
|
virtual void Stanza(const XmlElement * element) {
|
||
|
|
ss_ << "STANZA" << element->Str();
|
||
|
|
}
|
||
|
|
virtual void EndStream() {
|
||
|
|
ss_ << "END";
|
||
|
|
}
|
||
|
|
virtual void XmlError() {
|
||
|
|
ss_ << "ERROR";
|
||
|
|
}
|
||
|
|
|
||
|
|
std::string Str() {
|
||
|
|
return ss_.str();
|
||
|
|
}
|
||
|
|
|
||
|
|
std::string StrClear() {
|
||
|
|
std::string result = ss_.str();
|
||
|
|
ss_.str("");
|
||
|
|
return result;
|
||
|
|
}
|
||
|
|
|
||
|
|
private:
|
||
|
|
std::stringstream ss_;
|
||
|
|
};
|
||
|
|
|
||
|
|
|
||
|
|
TEST(XmppStanzaParserTest, TestTrivial) {
|
||
|
|
XmppStanzaParserTestHandler handler;
|
||
|
|
XmppStanzaParser parser(&handler);
|
||
|
|
std::string fragment;
|
||
|
|
|
||
|
|
fragment = "<trivial/>";
|
||
|
|
parser.Parse(fragment.c_str(), fragment.length(), false);
|
||
|
|
EXPECT_EQ("START<trivial/>END", handler.StrClear());
|
||
|
|
}
|
||
|
|
|
||
|
|
TEST(XmppStanzaParserTest, TestStanzaAtATime) {
|
||
|
|
XmppStanzaParserTestHandler handler;
|
||
|
|
XmppStanzaParser parser(&handler);
|
||
|
|
std::string fragment;
|
||
|
|
|
||
|
|
fragment = "<stream:stream id='abc' xmlns='j:c' xmlns:stream='str'>";
|
||
|
|
parser.Parse(fragment.c_str(), fragment.length(), false);
|
||
|
|
EXPECT_EQ("START<stream:stream id=\"abc\" xmlns=\"j:c\" "
|
||
|
|
"xmlns:stream=\"str\"/>", handler.StrClear());
|
||
|
|
|
||
|
|
fragment = "<message type='foo'><body>hello</body></message>";
|
||
|
|
parser.Parse(fragment.c_str(), fragment.length(), false);
|
||
|
|
EXPECT_EQ("STANZA<c:message type=\"foo\" xmlns:c=\"j:c\">"
|
||
|
|
"<c:body>hello</c:body></c:message>", handler.StrClear());
|
||
|
|
|
||
|
|
fragment = " SOME TEXT TO IGNORE ";
|
||
|
|
parser.Parse(fragment.c_str(), fragment.length(), false);
|
||
|
|
EXPECT_EQ("", handler.StrClear());
|
||
|
|
|
||
|
|
fragment = "<iq type='set' id='123'><abc xmlns='def'/></iq>";
|
||
|
|
parser.Parse(fragment.c_str(), fragment.length(), false);
|
||
|
|
EXPECT_EQ("STANZA<c:iq type=\"set\" id=\"123\" xmlns:c=\"j:c\">"
|
||
|
|
"<abc xmlns=\"def\"/></c:iq>", handler.StrClear());
|
||
|
|
|
||
|
|
fragment = "</stream:stream>";
|
||
|
|
parser.Parse(fragment.c_str(), fragment.length(), false);
|
||
|
|
EXPECT_EQ("END", handler.StrClear());
|
||
|
|
}
|
||
|
|
|
||
|
|
TEST(XmppStanzaParserTest, TestFragmentedStanzas) {
|
||
|
|
XmppStanzaParserTestHandler handler;
|
||
|
|
XmppStanzaParser parser(&handler);
|
||
|
|
std::string fragment;
|
||
|
|
|
||
|
|
fragment = "<stream:stream id='abc' xmlns='j:c' xml";
|
||
|
|
parser.Parse(fragment.c_str(), fragment.length(), false);
|
||
|
|
EXPECT_EQ("", handler.StrClear());
|
||
|
|
|
||
|
|
fragment = "ns:stream='str'><message type='foo'><body>hel";
|
||
|
|
parser.Parse(fragment.c_str(), fragment.length(), false);
|
||
|
|
EXPECT_EQ("START<stream:stream id=\"abc\" xmlns=\"j:c\" "
|
||
|
|
"xmlns:stream=\"str\"/>", handler.StrClear());
|
||
|
|
|
||
|
|
fragment = "lo</body></message> IGNORE ME <iq type='set' id='123'>"
|
||
|
|
"<abc xmlns='def'/></iq></st";
|
||
|
|
parser.Parse(fragment.c_str(), fragment.length(), false);
|
||
|
|
EXPECT_EQ("STANZA<c:message type=\"foo\" xmlns:c=\"j:c\">"
|
||
|
|
"<c:body>hello</c:body></c:message>STANZA<c:iq type=\"set\" id=\"123\" "
|
||
|
|
"xmlns:c=\"j:c\"><abc xmlns=\"def\"/></c:iq>", handler.StrClear());
|
||
|
|
|
||
|
|
fragment = "ream:stream>";
|
||
|
|
parser.Parse(fragment.c_str(), fragment.length(), false);
|
||
|
|
EXPECT_EQ("END", handler.StrClear());
|
||
|
|
}
|
||
|
|
|
||
|
|
TEST(XmppStanzaParserTest, TestReset) {
|
||
|
|
XmppStanzaParserTestHandler handler;
|
||
|
|
XmppStanzaParser parser(&handler);
|
||
|
|
std::string fragment;
|
||
|
|
|
||
|
|
fragment = "<stream:stream id='abc' xmlns='j:c' xml";
|
||
|
|
parser.Parse(fragment.c_str(), fragment.length(), false);
|
||
|
|
EXPECT_EQ("", handler.StrClear());
|
||
|
|
|
||
|
|
parser.Reset();
|
||
|
|
fragment = "<stream:stream id='abc' xmlns='j:c' xml";
|
||
|
|
parser.Parse(fragment.c_str(), fragment.length(), false);
|
||
|
|
EXPECT_EQ("", handler.StrClear());
|
||
|
|
|
||
|
|
fragment = "ns:stream='str'><message type='foo'><body>hel";
|
||
|
|
parser.Parse(fragment.c_str(), fragment.length(), false);
|
||
|
|
EXPECT_EQ("START<stream:stream id=\"abc\" xmlns=\"j:c\" "
|
||
|
|
"xmlns:stream=\"str\"/>", handler.StrClear());
|
||
|
|
parser.Reset();
|
||
|
|
|
||
|
|
fragment = "<stream:stream id='abc' xmlns='j:c' xmlns:stream='str'>";
|
||
|
|
parser.Parse(fragment.c_str(), fragment.length(), false);
|
||
|
|
EXPECT_EQ("START<stream:stream id=\"abc\" xmlns=\"j:c\" "
|
||
|
|
"xmlns:stream=\"str\"/>", handler.StrClear());
|
||
|
|
|
||
|
|
fragment = "<message type='foo'><body>hello</body></message>";
|
||
|
|
parser.Parse(fragment.c_str(), fragment.length(), false);
|
||
|
|
EXPECT_EQ("STANZA<c:message type=\"foo\" xmlns:c=\"j:c\">"
|
||
|
|
"<c:body>hello</c:body></c:message>", handler.StrClear());
|
||
|
|
}
|
||
|
|
|
||
|
|
TEST(XmppStanzaParserTest, TestError) {
|
||
|
|
XmppStanzaParserTestHandler handler;
|
||
|
|
XmppStanzaParser parser(&handler);
|
||
|
|
std::string fragment;
|
||
|
|
|
||
|
|
fragment = "<-foobar/>";
|
||
|
|
parser.Parse(fragment.c_str(), fragment.length(), false);
|
||
|
|
EXPECT_EQ("ERROR", handler.StrClear());
|
||
|
|
|
||
|
|
parser.Reset();
|
||
|
|
fragment = "<stream:stream/>";
|
||
|
|
parser.Parse(fragment.c_str(), fragment.length(), false);
|
||
|
|
EXPECT_EQ("ERROR", handler.StrClear());
|
||
|
|
parser.Reset();
|
||
|
|
|
||
|
|
fragment = "ns:stream='str'><message type='foo'><body>hel";
|
||
|
|
parser.Parse(fragment.c_str(), fragment.length(), false);
|
||
|
|
EXPECT_EQ("ERROR", handler.StrClear());
|
||
|
|
parser.Reset();
|
||
|
|
|
||
|
|
fragment = "<stream:stream xmlns:stream='st' xmlns='jc'>"
|
||
|
|
"<foo/><bar><st:foobar/></bar>";
|
||
|
|
parser.Parse(fragment.c_str(), fragment.length(), false);
|
||
|
|
EXPECT_EQ("START<stream:stream xmlns:stream=\"st\" xmlns=\"jc\"/>STANZA"
|
||
|
|
"<jc:foo xmlns:jc=\"jc\"/>ERROR", handler.StrClear());
|
||
|
|
}
|