aboutsummaryrefslogtreecommitdiff
path: root/src/supplemental/websocket/websocket_test.c
diff options
context:
space:
mode:
Diffstat (limited to 'src/supplemental/websocket/websocket_test.c')
-rw-r--r--src/supplemental/websocket/websocket_test.c19
1 files changed, 19 insertions, 0 deletions
diff --git a/src/supplemental/websocket/websocket_test.c b/src/supplemental/websocket/websocket_test.c
index f648f740..781ca1d8 100644
--- a/src/supplemental/websocket/websocket_test.c
+++ b/src/supplemental/websocket/websocket_test.c
@@ -146,6 +146,12 @@ test_websocket_conn_props(void)
NUTS_PASS(nng_stream_listener_listen(l));
NUTS_PASS(nng_stream_dialer_alloc(&d, uri));
+ NUTS_PASS(nng_stream_dialer_set_string(
+ d, NNG_OPT_WS_REQUEST_HEADER "NNG-Req", "True"));
+
+ NUTS_PASS(nng_stream_listener_set_string(
+ l, NNG_OPT_WS_RESPONSE_HEADER "NNG-Rep", "True"));
+
nng_stream_dialer_dial(d, daio);
nng_stream_listener_accept(l, laio);
@@ -180,6 +186,19 @@ test_websocket_conn_props(void)
NUTS_FAIL(
nng_stream_get_size(c1, NNG_OPT_TCP_NODELAY, &sz), NNG_EBADTYPE);
+ NUTS_FAIL(nng_stream_get_string(
+ c1, NNG_OPT_WS_REQUEST_HEADER "No-Such-Header", &str),
+ NNG_ENOENT);
+ NUTS_PASS(nng_stream_get_string(
+ c1, NNG_OPT_WS_REQUEST_HEADER "NNG-Req", &str));
+ NUTS_MATCH(str, "True");
+ nng_strfree(str);
+
+ NUTS_PASS(nng_stream_get_string(
+ c2, NNG_OPT_WS_RESPONSE_HEADER "NNG-Rep", &str));
+ NUTS_MATCH(str, "True");
+ nng_strfree(str);
+
NUTS_PASS(nng_stream_get_string(
c1, NNG_OPT_WS_REQUEST_HEADER "Sec-WebSocket-Version", &str));
NUTS_TRUE(str != NULL);