zydronium/rdpgw
1
0
Fork 0
mirror of https://github.com/bolkedebruin/rdpgw.git synced 2025-08-18 06:23:49 +02:00
Code Issues Projects Releases Packages Wiki Activity

Fix tests

Browse source
This commit is contained in:
Bolke de Bruin 2020-08-02 21:38:41 +02:00
parent 9c19a1b40a
commit fdffd962f0
1 changed files with 13 additions and 9 deletions
Download patch file Download diff file Expand all files Collapse all files

22
protocol/protocol_test.go
View file

@ -123,7 +123,10 @@ func TestTunnelCreation(t *testing.T) {
}
func TestTunnelAuth(t *testing.T) {
client := ClientConfig{}
name := "test_name"
client := ClientConfig{
Name: name,
}
s := &SessionInfo{}
hc := &ServerConf{
TokenAuth: true,
@ -133,9 +136,8 @@ func TestTunnelAuth(t *testing.T) {
},
}
h := NewServer(s, hc)
name := "test_name"
data := client.tunnelAuthRequest(name)
data := client.tunnelAuthRequest()
_, _, pkt, err := verifyPacketHeader(data, PKT_TYPE_TUNNEL_AUTH, uint32(TunnelAuthLen+len(name)*2))
if err != nil {
t.Fatalf("verifyHeader failed: %s", err)
@ -166,7 +168,11 @@ func TestTunnelAuth(t *testing.T) {
}
func TestChannelCreation(t *testing.T) {
client := ClientConfig{}
server := "test_server"
client := ClientConfig{
Server: server,
Port: 3389,
}
s := &SessionInfo{}
hc := &ServerConf{
TokenAuth: true,
@ -176,10 +182,8 @@ func TestChannelCreation(t *testing.T) {
},
}
h := NewServer(s, hc)
server := "test_server"
port := uint16(3389)
data := client.channelRequest(server, port)
data := client.channelRequest()
_, _, pkt, err := verifyPacketHeader(data, PKT_TYPE_CHANNEL_CREATE, uint32(ChannelCreateLen+len(server)*2))
if err != nil {
t.Fatalf("verifyHeader failed: %s", err)
@ -188,8 +192,8 @@ func TestChannelCreation(t *testing.T) {
if hServer != server {
t.Fatalf("channelRequest failed got server %s, expected %s", hServer, server)
}
if hPort != port {
t.Fatalf("channelRequest failed got port %d, expected %d", hPort, port)
if int(hPort) != client.Port {
t.Fatalf("channelRequest failed got port %d, expected %d", hPort, client.Port)
}
data = h.channelResponse()