narayana/go-xmpp
2
0
Fork 0
Code Issues Pull requests Projects Releases Wiki Activity
go-xmpp/router_test.go

119 lines
3 KiB
Go
Raw Normal View History

Add router to make it easier to set up routing info - Using the router, the dispatch is not done anymore by receiving from receive channel, but by registering callback functions in routers, with matchers. - Make IQPayload a real interface to make it easier to match namespaces. - The StreamManager Run command is now blocking, waiting for StreamManager to terminate.
2019-06-13 15:22:39 +00:00
package xmpp_test
import (
"bytes"
"encoding/xml"
"testing"
"gosrc.io/xmpp"
)
Introduce Sender interface to abstract client sending in router handlers
2019-06-14 07:37:38 +00:00
// ============================================================================
// SenderMock
// ============================================================================
// Test route & matchers
Add router to make it easier to set up routing info - Using the router, the dispatch is not done anymore by receiving from receive channel, but by registering callback functions in routers, with matchers. - Make IQPayload a real interface to make it easier to match namespaces. - The StreamManager Run command is now blocking, waiting for StreamManager to terminate.
2019-06-13 15:22:39 +00:00
func TestNameMatcher(t *testing.T) {
router := xmpp.NewRouter()
Introduce Sender interface to abstract client sending in router handlers
2019-06-14 07:37:38 +00:00
router.HandleFunc("message", func(s xmpp.Sender, p xmpp.Packet) {
_ = s.SendRaw(successFlag)
Add router to make it easier to set up routing info - Using the router, the dispatch is not done anymore by receiving from receive channel, but by registering callback functions in routers, with matchers. - Make IQPayload a real interface to make it easier to match namespaces. - The StreamManager Run command is now blocking, waiting for StreamManager to terminate.
2019-06-13 15:22:39 +00:00
})
// Check that a message packet is properly matched
Introduce Sender interface to abstract client sending in router handlers
2019-06-14 07:37:38 +00:00
conn := NewSenderMock()
Add router to make it easier to set up routing info - Using the router, the dispatch is not done anymore by receiving from receive channel, but by registering callback functions in routers, with matchers. - Make IQPayload a real interface to make it easier to match namespaces. - The StreamManager Run command is now blocking, waiting for StreamManager to terminate.
2019-06-13 15:22:39 +00:00
// TODO: We want packet creation code to use struct to use default values
msg := xmpp.NewMessage("chat", "", "test@localhost", "1", "")
msg.Body = "Hello"
Introduce Sender interface to abstract client sending in router handlers
2019-06-14 07:37:38 +00:00
router.Route(conn, msg)
if conn.String() != successFlag {
Add router to make it easier to set up routing info - Using the router, the dispatch is not done anymore by receiving from receive channel, but by registering callback functions in routers, with matchers. - Make IQPayload a real interface to make it easier to match namespaces. - The StreamManager Run command is now blocking, waiting for StreamManager to terminate.
2019-06-13 15:22:39 +00:00
t.Error("Message was not matched and routed properly")
}
// Check that an IQ packet is not matched
Introduce Sender interface to abstract client sending in router handlers
2019-06-14 07:37:38 +00:00
conn = NewSenderMock()
Add router to make it easier to set up routing info - Using the router, the dispatch is not done anymore by receiving from receive channel, but by registering callback functions in routers, with matchers. - Make IQPayload a real interface to make it easier to match namespaces. - The StreamManager Run command is now blocking, waiting for StreamManager to terminate.
2019-06-13 15:22:39 +00:00
iq := xmpp.NewIQ("get", "", "localhost", "1", "")
An IQ can only have a single payload "An IQ stanza of type "get" or "set" MUST contain exactly one child element, which specifies the semantics of the particular request."
2019-06-19 08:21:57 +00:00
iq.Payload = &xmpp.DiscoInfo{}
Introduce Sender interface to abstract client sending in router handlers
2019-06-14 07:37:38 +00:00
router.Route(conn, iq)
if conn.String() == successFlag {
Add router to make it easier to set up routing info - Using the router, the dispatch is not done anymore by receiving from receive channel, but by registering callback functions in routers, with matchers. - Make IQPayload a real interface to make it easier to match namespaces. - The StreamManager Run command is now blocking, waiting for StreamManager to terminate.
2019-06-13 15:22:39 +00:00
t.Error("IQ should not have been matched and routed")
}
}
func TestIQNSMatcher(t *testing.T) {
router := xmpp.NewRouter()
router.NewRoute().
IQNamespaces(xmpp.NSDiscoInfo, xmpp.NSDiscoItems).
Introduce Sender interface to abstract client sending in router handlers
2019-06-14 07:37:38 +00:00
HandlerFunc(func(s xmpp.Sender, p xmpp.Packet) {
_ = s.SendRaw(successFlag)
Add router to make it easier to set up routing info - Using the router, the dispatch is not done anymore by receiving from receive channel, but by registering callback functions in routers, with matchers. - Make IQPayload a real interface to make it easier to match namespaces. - The StreamManager Run command is now blocking, waiting for StreamManager to terminate.
2019-06-13 15:22:39 +00:00
})
// Check that an IQ with proper namespace does match
Introduce Sender interface to abstract client sending in router handlers
2019-06-14 07:37:38 +00:00
conn := NewSenderMock()
Add router to make it easier to set up routing info - Using the router, the dispatch is not done anymore by receiving from receive channel, but by registering callback functions in routers, with matchers. - Make IQPayload a real interface to make it easier to match namespaces. - The StreamManager Run command is now blocking, waiting for StreamManager to terminate.
2019-06-13 15:22:39 +00:00
iqDisco := xmpp.NewIQ("get", "", "localhost", "1", "")
// TODO: Add a function to generate payload with proper namespace initialisation
An IQ can only have a single payload "An IQ stanza of type "get" or "set" MUST contain exactly one child element, which specifies the semantics of the particular request."
2019-06-19 08:21:57 +00:00
iqDisco.Payload = &xmpp.DiscoInfo{
Add router to make it easier to set up routing info - Using the router, the dispatch is not done anymore by receiving from receive channel, but by registering callback functions in routers, with matchers. - Make IQPayload a real interface to make it easier to match namespaces. - The StreamManager Run command is now blocking, waiting for StreamManager to terminate.
2019-06-13 15:22:39 +00:00
XMLName: xml.Name{
Space: xmpp.NSDiscoInfo,
Local: "query",
An IQ can only have a single payload "An IQ stanza of type "get" or "set" MUST contain exactly one child element, which specifies the semantics of the particular request."
2019-06-19 08:21:57 +00:00
}}
Introduce Sender interface to abstract client sending in router handlers
2019-06-14 07:37:38 +00:00
router.Route(conn, iqDisco)
if conn.String() != successFlag {
Add router to make it easier to set up routing info - Using the router, the dispatch is not done anymore by receiving from receive channel, but by registering callback functions in routers, with matchers. - Make IQPayload a real interface to make it easier to match namespaces. - The StreamManager Run command is now blocking, waiting for StreamManager to terminate.
2019-06-13 15:22:39 +00:00
t.Errorf("IQ should have been matched and routed: %v", iqDisco)
}
// Check that another namespace is not matched
Introduce Sender interface to abstract client sending in router handlers
2019-06-14 07:37:38 +00:00
conn = NewSenderMock()
Add router to make it easier to set up routing info - Using the router, the dispatch is not done anymore by receiving from receive channel, but by registering callback functions in routers, with matchers. - Make IQPayload a real interface to make it easier to match namespaces. - The StreamManager Run command is now blocking, waiting for StreamManager to terminate.
2019-06-13 15:22:39 +00:00
iqVersion := xmpp.NewIQ("get", "", "localhost", "1", "")
// TODO: Add a function to generate payload with proper namespace initialisation
An IQ can only have a single payload "An IQ stanza of type "get" or "set" MUST contain exactly one child element, which specifies the semantics of the particular request."
2019-06-19 08:21:57 +00:00
iqVersion.Payload = &xmpp.DiscoInfo{
Add router to make it easier to set up routing info - Using the router, the dispatch is not done anymore by receiving from receive channel, but by registering callback functions in routers, with matchers. - Make IQPayload a real interface to make it easier to match namespaces. - The StreamManager Run command is now blocking, waiting for StreamManager to terminate.
2019-06-13 15:22:39 +00:00
XMLName: xml.Name{
Space: "jabber:iq:version",
Local: "query",
An IQ can only have a single payload "An IQ stanza of type "get" or "set" MUST contain exactly one child element, which specifies the semantics of the particular request."
2019-06-19 08:21:57 +00:00
}}
Introduce Sender interface to abstract client sending in router handlers
2019-06-14 07:37:38 +00:00
router.Route(conn, iqVersion)
if conn.String() == successFlag {
Add router to make it easier to set up routing info - Using the router, the dispatch is not done anymore by receiving from receive channel, but by registering callback functions in routers, with matchers. - Make IQPayload a real interface to make it easier to match namespaces. - The StreamManager Run command is now blocking, waiting for StreamManager to terminate.
2019-06-13 15:22:39 +00:00
t.Errorf("IQ should not have been matched and routed: %v", iqVersion)
}
}
Introduce Sender interface to abstract client sending in router handlers
2019-06-14 07:37:38 +00:00
var successFlag = "matched"
type SenderMock struct {
buffer *bytes.Buffer
}
func NewSenderMock() SenderMock {
return SenderMock{buffer: new(bytes.Buffer)}
}
func (s SenderMock) Send(packet xmpp.Packet) error {
out, err := xml.Marshal(packet)
if err != nil {
return err
}
s.buffer.Write(out)
return nil
}
func (s SenderMock) SendRaw(str string) error {
s.buffer.WriteString(str)
return nil
}
func (s SenderMock) String() string {
return s.buffer.String()
}
func TestSenderMock(t *testing.T) {
conn := NewSenderMock()
msg := xmpp.NewMessage("", "", "test@localhost", "1", "")
msg.Body = "Hello"
if err := conn.Send(msg); err != nil {
t.Error("Could not send message")
}
if conn.String() != "<message id=\"1\" to=\"test@localhost\"><body>Hello</body></message>" {
t.Errorf("Incorrect packet sent: %s", conn.String())
}
}
Reference in a new issue Copy permalink