1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148
|
package xmpp
import (
"encoding/xml"
"errors"
"github.com/twstrike/coyim/xmpp/data"
. "gopkg.in/check.v1"
)
type StreamsXmppSuite struct{}
var _ = Suite(&StreamsXmppSuite{})
func (s *StreamsXmppSuite) Test_sendInitialStreamHeader_returnsErrorIfSomethingGoesWrongWithFmtPrintf(c *C) {
conn := conn{
out: &mockConnIOReaderWriter{err: errors.New("Hello")},
originDomain: "foo.com",
}
err := conn.SendInitialStreamHeader()
c.Assert(err, Not(IsNil))
}
func (s *StreamsXmppSuite) Test_sendInitialStreamHeader_returnsErrorIfSomethingGoesWrongWithReadingAStream(c *C) {
mockIn := &mockConnIOReaderWriter{err: errors.New("Hello")}
conn := conn{
out: &mockConnIOReaderWriter{},
in: xml.NewDecoder(mockIn),
originDomain: "foo.com",
}
err := conn.SendInitialStreamHeader()
c.Assert(err, Not(IsNil))
}
func (s *StreamsXmppSuite) Test_sendInitialStreamHeader_sendsInitialStreamHeaderToOutput(c *C) {
mockOut := &mockConnIOReaderWriter{}
mockIn := &mockConnIOReaderWriter{err: errors.New("Hello")}
conn := conn{
out: mockOut,
in: xml.NewDecoder(mockIn),
originDomain: "somewhere.org",
}
conn.SendInitialStreamHeader()
c.Assert(string(mockOut.write), Equals, "<?xml version='1.0'?><stream:stream to='somewhere.org' xmlns='jabber:client' xmlns:stream='http://etherx.jabber.org/streams' version='1.0'>\n")
}
func (s *StreamsXmppSuite) Test_sendInitialStreamHeader_expectsResponseStreamHeaderInReturn(c *C) {
mockOut := &mockConnIOReaderWriter{}
mockIn := &mockConnIOReaderWriter{read: []byte("<?xml version='1.0'?><stream:stream xmlns:stream='http://etherx.jabber.org/streams' version='1.0'></stream:stream>")}
conn := conn{
out: mockOut,
in: xml.NewDecoder(mockIn),
originDomain: "somewhereElse.org",
}
err := conn.SendInitialStreamHeader()
c.Assert(err.Error(), Equals, "xmpp: error to unmarshal <features>: EOF")
}
func (s *StreamsXmppSuite) Test_sendInitialStreamHeader_failsIfReturnedStreamIsNotCorrectNamespace(c *C) {
mockOut := &mockConnIOReaderWriter{}
mockIn := &mockConnIOReaderWriter{read: []byte("<?xml version='1.0'?><str:stream xmlns:str='http://etherx.jabber.org/streams2' version='1.0'>")}
conn := conn{
out: mockOut,
in: xml.NewDecoder(mockIn),
originDomain: "somewhereElse.org",
}
err := conn.SendInitialStreamHeader()
c.Assert(err.Error(), Equals, "xmpp: expected <stream> but got <stream> in http://etherx.jabber.org/streams2")
}
func (s *StreamsXmppSuite) Test_sendInitialStreamHeader_failsIfReturnedElementIsNotStream(c *C) {
mockOut := &mockConnIOReaderWriter{}
mockIn := &mockConnIOReaderWriter{read: []byte("<?xml version='1.0'?><str:feature xmlns:str='http://etherx.jabber.org/streams' version='1.0'>")}
conn := conn{
out: mockOut,
in: xml.NewDecoder(mockIn),
originDomain: "somewhereElse.org",
}
err := conn.SendInitialStreamHeader()
c.Assert(err.Error(), Equals, "xmpp: expected <stream> but got <feature> in http://etherx.jabber.org/streams")
}
func (s *StreamsXmppSuite) Test_sendInitialStreamHeader_expectsFeaturesInReturn(c *C) {
mockOut := &mockConnIOReaderWriter{}
mockIn := &mockConnIOReaderWriter{read: []byte("<?xml version='1.0'?><str:stream xmlns:str='http://etherx.jabber.org/streams' version='1.0'><str:features></str:features>")}
conn := conn{
out: mockOut,
in: xml.NewDecoder(mockIn),
originDomain: "somewhereElse.org",
}
err := conn.SendInitialStreamHeader()
c.Assert(err, IsNil)
expected := data.StreamFeatures{}
expected.XMLName = xml.Name{Space: "http://etherx.jabber.org/streams", Local: "features"}
c.Assert(conn.features, DeepEquals, expected)
}
func (s *StreamsXmppSuite) Test_sendInitialStreamHeader_receiveResponseStreamHeaderInReturn(c *C) {
mockOut := &mockConnIOReaderWriter{}
mockIn := &mockConnIOReaderWriter{read: []byte(`
<?xml version='1.0'?>
<str:stream xmlns:str='http://etherx.jabber.org/streams' version='1.0'>
<str:features>
<bind xmlns='urn:ietf:params:xml:ns:xmpp-bind' />
<starttls xmlns='urn:ietf:params:xml:ns:xmpp-tls'>
<required/>
</starttls>
<mechanisms xmlns='urn:ietf:params:xml:ns:xmpp-sasl'>
<mechanism>PLAIN</mechanism>
<mechanism>X-OAUTH2</mechanism>
<mechanism>X-GOOGLE-TOKEN</mechanism>
</mechanisms>
<register xmlns='http://jabber.org/features/iq-register'/>
</str:features>
`)}
conn := conn{
out: mockOut,
in: xml.NewDecoder(mockIn),
originDomain: "somewhereElse.org",
}
err := conn.SendInitialStreamHeader()
c.Assert(err, IsNil)
expected := data.StreamFeatures{
XMLName: xml.Name{Space: "http://etherx.jabber.org/streams", Local: "features"},
Bind: data.BindBind{
XMLName: xml.Name{Space: "urn:ietf:params:xml:ns:xmpp-bind", Local: "bind"}, Resource: "", Jid: "",
},
StartTLS: data.StartTLS{
XMLName: xml.Name{Space: "urn:ietf:params:xml:ns:xmpp-tls", Local: "starttls"},
Required: xml.Name{Space: "urn:ietf:params:xml:ns:xmpp-tls", Local: "required"},
},
Mechanisms: data.SaslMechanisms{
XMLName: xml.Name{Space: "urn:ietf:params:xml:ns:xmpp-sasl", Local: "mechanisms"},
Mechanism: []string{"PLAIN", "X-OAUTH2", "X-GOOGLE-TOKEN"},
},
InBandRegistration: &data.InBandRegistration{xml.Name{Space: "http://jabber.org/features/iq-register", Local: "register"}},
}
c.Assert(conn.features, DeepEquals, expected)
}
|