webrtc/talk/xmpp/xmppstanzaparser_unittest.cc
henrike@webrtc.org 28e2075280 Adds trunk/talk folder of revision 359 from libjingles google code to
trunk/talk


git-svn-id: http://webrtc.googlecode.com/svn/trunk@4318 4adac7df-926f-26a2-2b94-8c16560cd09d
2013-07-10 00:45:36 +00:00

169 lines
5.5 KiB
C++

// Copyright 2004 Google Inc. All Rights Reserved
#include <string>
#include <sstream>
#include <iostream>
#include "talk/base/common.h"
#include "talk/base/gunit.h"
#include "talk/xmllite/xmlelement.h"
#include "talk/xmpp/xmppstanzaparser.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());
}