97 lines
1.9 KiB
Go
97 lines
1.9 KiB
Go
|
package client
|
||
|
|
|
||
|
|
import (
|
||
|
|
"encoding/xml"
|
||
|
|
"net"
|
||
|
|
"sync"
|
||
|
|
"testing"
|
||
|
|
|
||
|
|
"github.com/stretchr/testify/assert"
|
||
|
|
|
||
|
|
log "github.com/sirupsen/logrus"
|
||
|
|
|
||
|
|
"dev.sum7.eu/genofire/yaja/xmpp"
|
||
|
|
"dev.sum7.eu/genofire/yaja/xmpp/base"
|
||
|
|
)
|
||
|
|
|
||
|
|
func TestRead(t *testing.T) {
|
||
|
|
assert := assert.New(t)
|
||
|
|
|
||
|
|
server, clientConn := net.Pipe()
|
||
|
|
client := &Client{}
|
||
|
|
client.setConnection(clientConn)
|
||
|
|
|
||
|
|
go server.Write([]byte(`<message>`))
|
||
|
|
|
||
|
|
element, err := client.Read()
|
||
|
|
assert.NoError(err)
|
||
|
|
assert.Equal("message", element.Name.Local)
|
||
|
|
|
||
|
|
go server.Write([]byte(`<>`))
|
||
|
|
element, err = client.Read()
|
||
|
|
assert.Error(err)
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestSend(t *testing.T) {
|
||
|
|
assert := assert.New(t)
|
||
|
|
|
||
|
|
server, clientConn := net.Pipe()
|
||
|
|
client := &Client{
|
||
|
|
Logging: log.WithField("test", "send"),
|
||
|
|
}
|
||
|
|
client.setConnection(clientConn)
|
||
|
|
serverDecoder := xml.NewDecoder(server)
|
||
|
|
wgWait := &sync.WaitGroup{}
|
||
|
|
|
||
|
|
wgWait.Add(1)
|
||
|
|
go func() {
|
||
|
|
err := client.Send(3)
|
||
|
|
assert.NoError(err)
|
||
|
|
wgWait.Done()
|
||
|
|
}()
|
||
|
|
|
||
|
|
element, err := read(serverDecoder)
|
||
|
|
wgWait.Wait()
|
||
|
|
assert.NoError(err)
|
||
|
|
assert.Equal("int", element.Name.Local)
|
||
|
|
|
||
|
|
wgWait.Add(1)
|
||
|
|
go func() {
|
||
|
|
err := client.Send(&xmpp.MessageClient{To: xmppbase.NewJID("a@a.de")})
|
||
|
|
assert.NoError(err)
|
||
|
|
wgWait.Done()
|
||
|
|
}()
|
||
|
|
|
||
|
|
element, err = read(serverDecoder)
|
||
|
|
wgWait.Wait()
|
||
|
|
assert.NoError(err)
|
||
|
|
assert.Equal("message", element.Name.Local)
|
||
|
|
assert.Equal("a@a.de", element.Attr[1].Value)
|
||
|
|
|
||
|
|
wgWait.Add(1)
|
||
|
|
go func() {
|
||
|
|
err := client.Send(&xmpp.IQClient{Type: xmpp.IQTypeGet})
|
||
|
|
assert.NoError(err)
|
||
|
|
wgWait.Done()
|
||
|
|
}()
|
||
|
|
|
||
|
|
element, err = read(serverDecoder)
|
||
|
|
wgWait.Wait()
|
||
|
|
assert.NoError(err)
|
||
|
|
assert.Equal("iq", element.Name.Local)
|
||
|
|
assert.Equal("get", element.Attr[2].Value)
|
||
|
|
|
||
|
|
wgWait.Add(1)
|
||
|
|
go func() {
|
||
|
|
err := client.Send(&xmpp.PresenceClient{Type: xmpp.PresenceTypeSubscribe})
|
||
|
|
assert.NoError(err)
|
||
|
|
wgWait.Done()
|
||
|
|
}()
|
||
|
|
|
||
|
|
element, err = read(serverDecoder)
|
||
|
|
wgWait.Wait()
|
||
|
|
assert.NoError(err)
|
||
|
|
assert.Equal("presence", element.Name.Local)
|
||
|
|
assert.Equal("subscribe", element.Attr[1].Value)
|
||
|
|
}
|