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

Fix tests

Browse source
This commit is contained in:
Bolke de Bruin 2022-08-08 12:05:43 +02:00
parent 9365fb03dc
commit b2f82c4ab9
1 changed files with 4 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

8
cmd/rdpgw/protocol/protocol_test.go
View file

@ -66,7 +66,7 @@ func TestHandshake(t *testing.T) {
t.Fatalf("handshakeRequest failed got ext auth %d, expected %d", extAuth, extAuth|HTTP_EXTENDED_AUTH_PAA) t.Fatalf("handshakeRequest failed got ext auth %d, expected %d", extAuth, extAuth|HTTP_EXTENDED_AUTH_PAA)
} }
data = h.handshakeResponse(0x0, 0x0) data = h.handshakeResponse(0x0, 0x0, ERROR_SUCCESS)
_, _, pkt, err = verifyPacketHeader(data, PKT_TYPE_HANDSHAKE_RESPONSE, HandshakeResponseLen) _, _, pkt, err = verifyPacketHeader(data, PKT_TYPE_HANDSHAKE_RESPONSE, HandshakeResponseLen)
if err != nil { if err != nil {
t.Fatalf("verifyHeader failed: %s", err) t.Fatalf("verifyHeader failed: %s", err)
@ -104,7 +104,7 @@ func TestTunnelCreation(t *testing.T) {
t.Fatalf("tunnelRequest failed got token %s, expected %s", token, client.PAAToken) t.Fatalf("tunnelRequest failed got token %s, expected %s", token, client.PAAToken)
} }
data = h.tunnelResponse() data = h.tunnelResponse(ERROR_SUCCESS)
_, _, pkt, err = verifyPacketHeader(data, PKT_TYPE_TUNNEL_RESPONSE, TunnelCreateResponseLen) _, _, pkt, err = verifyPacketHeader(data, PKT_TYPE_TUNNEL_RESPONSE, TunnelCreateResponseLen)
if err != nil { if err != nil {
t.Fatalf("verifyHeader failed: %s", err) t.Fatalf("verifyHeader failed: %s", err)
@ -148,7 +148,7 @@ func TestTunnelAuth(t *testing.T) {
t.Fatalf("tunnelAuthRequest failed got name %s, expected %s", n, name) t.Fatalf("tunnelAuthRequest failed got name %s, expected %s", n, name)
} }
data = h.tunnelAuthResponse() data = h.tunnelAuthResponse(ERROR_SUCCESS)
_, _, pkt, err = verifyPacketHeader(data, PKT_TYPE_TUNNEL_AUTH_RESPONSE, TunnelAuthResponseLen) _, _, pkt, err = verifyPacketHeader(data, PKT_TYPE_TUNNEL_AUTH_RESPONSE, TunnelAuthResponseLen)
if err != nil { if err != nil {
t.Fatalf("verifyHeader failed: %s", err) t.Fatalf("verifyHeader failed: %s", err)
@ -196,7 +196,7 @@ func TestChannelCreation(t *testing.T) {
t.Fatalf("channelRequest failed got port %d, expected %d", hPort, client.Port) t.Fatalf("channelRequest failed got port %d, expected %d", hPort, client.Port)
} }
data = h.channelResponse() data = h.channelResponse(ERROR_SUCCESS)
_, _, pkt, err = verifyPacketHeader(data, PKT_TYPE_CHANNEL_RESPONSE, uint32(ChannelResponseLen)) _, _, pkt, err = verifyPacketHeader(data, PKT_TYPE_CHANNEL_RESPONSE, uint32(ChannelResponseLen))
if err != nil { if err != nil {
t.Fatalf("verifyHeader failed: %s", err) t.Fatalf("verifyHeader failed: %s", err)