diff --git a/Makefile b/Makefile index 0f4c34c3..ad1295ea 100644 --- a/Makefile +++ b/Makefile @@ -82,7 +82,7 @@ test: unit_test coverage: test gcov -l -n *.gcno | sed '/^$$/d' | sed 'N;s/\n/ /' - gcov mongoose.gcno >/dev/null + gcov -t mongoose.c > mongoose.gcov upload-coverage: coverage curl -s https://codecov.io/bash | /bin/bash diff --git a/test/unit_test.c b/test/unit_test.c index ef436426..e3345e82 100644 --- a/test/unit_test.c +++ b/test/unit_test.c @@ -1709,7 +1709,12 @@ static void test_check_ip_acl(void) { static void w3(struct mg_connection *c, int ev, void *ev_data, void *fn_data) { // MG_INFO(("ev %d", ev)); if (ev == MG_EV_WS_OPEN) { + char buf[8192]; + memset(buf, 'A', sizeof(buf)); mg_ws_send(c, "hi there!", 9, WEBSOCKET_OP_TEXT); + mg_printf(c, "%s", "boo"); + mg_ws_wrap(c, 3, WEBSOCKET_OP_TEXT), + mg_ws_send(c, buf, sizeof(buf), WEBSOCKET_OP_TEXT); } else if (ev == MG_EV_WS_MSG) { struct mg_ws_message *wm = (struct mg_ws_message *) ev_data; ASSERT(mg_strcmp(wm->data, mg_str("lebowski")) == 0); @@ -1742,7 +1747,13 @@ static void w2(struct mg_connection *c, int ev, void *ev_data, void *fn_data) { } } else if (ev == MG_EV_WS_MSG) { struct mg_ws_message *wm = (struct mg_ws_message *) ev_data; - ASSERT(mg_strcmp(wm->data, mg_str("hi there!")) == 0); + if (wm->data.len == 9) { + ASSERT(mg_strcmp(wm->data, mg_str("hi there!")) == 0); + } else if (wm->data.len == 3) { + ASSERT(mg_strcmp(wm->data, mg_str("boo")) == 0); + } else { + ASSERT(wm->data.len == 8192); + } } }