From b2f82c4ab9e877e97b44b9ca13c4a0cc0dbf5162 Mon Sep 17 00:00:00 2001
From: Bolke de Bruin <bolke@xs4all.nl>
Date: Mon, 8 Aug 2022 12:05:43 +0200
Subject: [PATCH] Fix tests
---
cmd/rdpgw/protocol/protocol_test.go | 8 ++++----
1 file changed, 4 insertions(+), 4 deletions(-)
diff --git a/cmd/rdpgw/protocol/protocol_test.go b/cmd/rdpgw/protocol/protocol_test.go
index b8a5974..4ae4194 100644
--- a/cmd/rdpgw/protocol/protocol_test.go
+++ b/cmd/rdpgw/protocol/protocol_test.go
@@ -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)
}
- data = h.handshakeResponse(0x0, 0x0)
+ data = h.handshakeResponse(0x0, 0x0, ERROR_SUCCESS)
_, _, pkt, err = verifyPacketHeader(data, PKT_TYPE_HANDSHAKE_RESPONSE, HandshakeResponseLen)
if err != nil {
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)
}
- data = h.tunnelResponse()
+ data = h.tunnelResponse(ERROR_SUCCESS)
_, _, pkt, err = verifyPacketHeader(data, PKT_TYPE_TUNNEL_RESPONSE, TunnelCreateResponseLen)
if err != nil {
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)
}
- data = h.tunnelAuthResponse()
+ data = h.tunnelAuthResponse(ERROR_SUCCESS)
_, _, pkt, err = verifyPacketHeader(data, PKT_TYPE_TUNNEL_AUTH_RESPONSE, TunnelAuthResponseLen)
if err != nil {
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)
}
- data = h.channelResponse()
+ data = h.channelResponse(ERROR_SUCCESS)
_, _, pkt, err = verifyPacketHeader(data, PKT_TYPE_CHANNEL_RESPONSE, uint32(ChannelResponseLen))
if err != nil {
t.Fatalf("verifyHeader failed: %s", err)
--
GitLab