zydronium/rdpgw
1
0
Fork 0
mirror of https://github.com/bolkedebruin/rdpgw.git synced 2025-08-20 07:23:48 +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) { func TestTunnelAuth(t *testing.T) {
client := ClientConfig{} name := "test_name"
client := ClientConfig{
Name: name,
}
s := &SessionInfo{} s := &SessionInfo{}
hc := &ServerConf{ hc := &ServerConf{
TokenAuth: true, TokenAuth: true,
@ -133,9 +136,8 @@ func TestTunnelAuth(t *testing.T) {
}, },
} }
h := NewServer(s, hc) 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)) _, _, pkt, err := verifyPacketHeader(data, PKT_TYPE_TUNNEL_AUTH, uint32(TunnelAuthLen+len(name)*2))
if err != nil { if err != nil {
t.Fatalf("verifyHeader failed: %s", err) t.Fatalf("verifyHeader failed: %s", err)
@ -166,7 +168,11 @@ func TestTunnelAuth(t *testing.T) {
} }
func TestChannelCreation(t *testing.T) { func TestChannelCreation(t *testing.T) {
client := ClientConfig{} server := "test_server"
client := ClientConfig{
Server: server,
Port: 3389,
}
s := &SessionInfo{} s := &SessionInfo{}
hc := &ServerConf{ hc := &ServerConf{
TokenAuth: true, TokenAuth: true,
@ -176,10 +182,8 @@ func TestChannelCreation(t *testing.T) {
}, },
} }
h := NewServer(s, hc) 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)) _, _, pkt, err := verifyPacketHeader(data, PKT_TYPE_CHANNEL_CREATE, uint32(ChannelCreateLen+len(server)*2))
if err != nil { if err != nil {
t.Fatalf("verifyHeader failed: %s", err) t.Fatalf("verifyHeader failed: %s", err)
@ -188,8 +192,8 @@ func TestChannelCreation(t *testing.T) {
if hServer != server { if hServer != server {
t.Fatalf("channelRequest failed got server %s, expected %s", hServer, server) t.Fatalf("channelRequest failed got server %s, expected %s", hServer, server)
} }
if hPort != port { if int(hPort) != client.Port {
t.Fatalf("channelRequest failed got port %d, expected %d", hPort, port) t.Fatalf("channelRequest failed got port %d, expected %d", hPort, client.Port)
} }
data = h.channelResponse() data = h.channelResponse()