Commonize unit test main()

CL: none

PUBLISHED_FROM=ee76407d89f3fcbe3da23810117a374d06af0a25
This commit is contained in:
Deomid Ryabkov 2018-02-07 14:53:15 +00:00 committed by Cesanta Bot
parent cd89db5c2f
commit 50ecc30b4d
3 changed files with 26 additions and 72 deletions

View File

@ -27,7 +27,7 @@
extern "C" { extern "C" {
#endif #endif
extern int num_tests; extern int g_num_tests;
#ifdef MG_TEST_ABORT_ON_FAIL #ifdef MG_TEST_ABORT_ON_FAIL
#define MG_TEST_ABORT abort() #define MG_TEST_ABORT abort()
@ -48,7 +48,7 @@ void _strfail(const char *a, const char *e, int len);
#define ASSERT(expr) \ #define ASSERT(expr) \
do { \ do { \
num_tests++; \ g_num_tests++; \
if (!(expr)) FAIL(#expr, __LINE__); \ if (!(expr)) FAIL(#expr, __LINE__); \
} while (0) } while (0)
#define ASSERT_TRUE(expr) ASSERT(expr) #define ASSERT_TRUE(expr) ASSERT(expr)
@ -71,7 +71,6 @@ void _strfail(const char *a, const char *e, int len);
elapsed = cs_time() - elapsed; \ elapsed = cs_time() - elapsed; \
printf(" [%.3f] %s\n", elapsed, test_name); \ printf(" [%.3f] %s\n", elapsed, test_name); \
fflush(stdout); \ fflush(stdout); \
*total_elapsed += elapsed; \
} \ } \
if (msg) return msg; \ if (msg) return msg; \
} while (0) } while (0)
@ -89,7 +88,7 @@ void _strfail(const char *a, const char *e, int len);
*/ */
#define ASSERT_EQ(actual, expected) \ #define ASSERT_EQ(actual, expected) \
do { \ do { \
num_tests++; \ g_num_tests++; \
if (!((actual) == (expected))) { \ if (!((actual) == (expected))) { \
printf("%f != %f\n", AS_DOUBLE(actual), AS_DOUBLE(expected)); \ printf("%f != %f\n", AS_DOUBLE(actual), AS_DOUBLE(expected)); \
FAIL(#actual " == " #expected, __LINE__); \ FAIL(#actual " == " #expected, __LINE__); \
@ -99,7 +98,7 @@ void _strfail(const char *a, const char *e, int len);
/* "Less than" assertion. */ /* "Less than" assertion. */
#define ASSERT_LT(a, b) \ #define ASSERT_LT(a, b) \
do { \ do { \
num_tests++; \ g_num_tests++; \
if (!((a) < (b))) { \ if (!((a) < (b))) { \
printf("%f >= %f\n", AS_DOUBLE(a), AS_DOUBLE(b)); \ printf("%f >= %f\n", AS_DOUBLE(a), AS_DOUBLE(b)); \
FAIL(#a " < " #b, __LINE__); \ FAIL(#a " < " #b, __LINE__); \
@ -109,7 +108,7 @@ void _strfail(const char *a, const char *e, int len);
/* "Greater than" assertion. */ /* "Greater than" assertion. */
#define ASSERT_GT(a, b) \ #define ASSERT_GT(a, b) \
do { \ do { \
num_tests++; \ g_num_tests++; \
if (!((a) > (b))) { \ if (!((a) > (b))) { \
printf("%f <= %f\n", AS_DOUBLE(a), AS_DOUBLE(b)); \ printf("%f <= %f\n", AS_DOUBLE(a), AS_DOUBLE(b)); \
FAIL(#a " > " #b, __LINE__); \ FAIL(#a " > " #b, __LINE__); \
@ -119,7 +118,7 @@ void _strfail(const char *a, const char *e, int len);
/* Assert that actual == expected, where both are NUL-terminated. */ /* Assert that actual == expected, where both are NUL-terminated. */
#define ASSERT_STREQ(actual, expected) \ #define ASSERT_STREQ(actual, expected) \
do { \ do { \
num_tests++; \ g_num_tests++; \
if (!_assert_streq(actual, expected)) { \ if (!_assert_streq(actual, expected)) { \
FAIL("ASSERT_STREQ(" #actual ", " #expected ")", __LINE__); \ FAIL("ASSERT_STREQ(" #actual ", " #expected ")", __LINE__); \
} \ } \
@ -128,7 +127,7 @@ void _strfail(const char *a, const char *e, int len);
/* Assert that actual == expected, where both are pointers */ /* Assert that actual == expected, where both are pointers */
#define ASSERT_PTREQ(actual, expected) \ #define ASSERT_PTREQ(actual, expected) \
do { \ do { \
num_tests++; \ g_num_tests++; \
if (actual != expected) { \ if (actual != expected) { \
printf("%p != %p\n", actual, expected); \ printf("%p != %p\n", actual, expected); \
FAIL("ASSERT_PTREQ(" #actual ", " #expected ")", __LINE__); \ FAIL("ASSERT_PTREQ(" #actual ", " #expected ")", __LINE__); \
@ -138,7 +137,7 @@ void _strfail(const char *a, const char *e, int len);
/* Assert that actual != expected, where both are pointers */ /* Assert that actual != expected, where both are pointers */
#define ASSERT_PTRNEQ(actual, expected) \ #define ASSERT_PTRNEQ(actual, expected) \
do { \ do { \
num_tests++; \ g_num_tests++; \
if (actual == expected) { \ if (actual == expected) { \
printf("%p == %p\n", actual, expected); \ printf("%p == %p\n", actual, expected); \
FAIL("ASSERT_PTRNEQ(" #actual ", " #expected ")", __LINE__); \ FAIL("ASSERT_PTRNEQ(" #actual ", " #expected ")", __LINE__); \
@ -148,7 +147,7 @@ void _strfail(const char *a, const char *e, int len);
/* Same as STREQ, but only expected is NUL-terminated. */ /* Same as STREQ, but only expected is NUL-terminated. */
#define ASSERT_STREQ_NZ(actual, expected) \ #define ASSERT_STREQ_NZ(actual, expected) \
do { \ do { \
num_tests++; \ g_num_tests++; \
if (!_assert_streq_nz(actual, expected)) { \ if (!_assert_streq_nz(actual, expected)) { \
FAIL("ASSERT_STREQ_NZ(" #actual ", " #expected ")", __LINE__); \ FAIL("ASSERT_STREQ_NZ(" #actual ", " #expected ")", __LINE__); \
} \ } \
@ -156,7 +155,7 @@ void _strfail(const char *a, const char *e, int len);
#define ASSERT_MG_STREQ(actual, expected) \ #define ASSERT_MG_STREQ(actual, expected) \
do { \ do { \
num_tests++; \ g_num_tests++; \
if ((actual).len != strlen(expected) || \ if ((actual).len != strlen(expected) || \
memcmp((actual).p, expected, (actual).len) != 0) { \ memcmp((actual).p, expected, (actual).len) != 0) { \
printf("'%.*s' (%d) != '%s'\n", (int)(actual).len, (actual).p, \ printf("'%.*s' (%d) != '%s'\n", (int)(actual).len, (actual).p, \

View File

@ -15,7 +15,7 @@ else
COMMON_PARENT = $(SRC_DIR) COMMON_PARENT = $(SRC_DIR)
endif endif
TEST_SOURCES = unit_test.c test_util.c TEST_SOURCES = unit_test.c test_util.c test_main.c
AMALGAMATED_SOURCES = ../mongoose.c AMALGAMATED_SOURCES = ../mongoose.c
KRYPTON_PATH = $(REPO_ROOT)/krypton KRYPTON_PATH = $(REPO_ROOT)/krypton

View File

@ -18,13 +18,10 @@
#include "mongoose.h" #include "mongoose.h"
#include "src/mg_internal.h" #include "src/mg_internal.h"
#include "unit_test.h" #include "unit_test.h"
#include "common/test_main.h"
#include "common/test_util.h" #include "common/test_util.h"
#include "common/cs_md5.h" #include "common/cs_md5.h"
#if defined(_MSC_VER) && _MSC_VER >= 1900
#include <crtdbg.h>
#endif
#if defined(__STDC_VERSION__) && __STDC_VERSION__ < 199901L && !defined(WIN32) #if defined(__STDC_VERSION__) && __STDC_VERSION__ < 199901L && !defined(WIN32)
#define __func__ "" #define __func__ ""
#endif #endif
@ -35,7 +32,6 @@
#define LOOPBACK_IP "127.0.0.1" #define LOOPBACK_IP "127.0.0.1"
#define LISTENING_ADDR LOOPBACK_IP ":" HTTP_PORT #define LISTENING_ADDR LOOPBACK_IP ":" HTTP_PORT
static const char *s_argv_0 = NULL;
static struct mg_serve_http_opts s_http_server_opts; static struct mg_serve_http_opts s_http_server_opts;
static int s_listening_port = 23456; static int s_listening_port = 23456;
@ -1648,8 +1644,8 @@ static void cb7(struct mg_connection *nc, int ev, void *ev_data) {
if (ev == MG_EV_HTTP_REPLY) { if (ev == MG_EV_HTTP_REPLY) {
/* Make sure that we've downloaded this executable, byte-to-byte */ /* Make sure that we've downloaded this executable, byte-to-byte */
data = read_file(s_argv_0, &size); data = read_file(g_argv_0, &size);
DBG(("file %s, size %d; got %d", s_argv_0, (int) size, (int) hm->body.len)); DBG(("file %s, size %d; got %d", g_argv_0, (int) size, (int) hm->body.len));
if (data != NULL && size == hm->body.len && if (data != NULL && size == hm->body.len &&
memcmp(hm->body.p, data, size) == 0) { memcmp(hm->body.p, data, size) == 0) {
strcpy(user_data, "success"); strcpy(user_data, "success");
@ -2025,12 +2021,12 @@ static const char *test_http(void) {
nc->user_data = status; nc->user_data = status;
/* Wine and GDB set argv0 to full path: strip the dir component */ /* Wine and GDB set argv0 to full path: strip the dir component */
if ((this_binary = strrchr(s_argv_0, '\\')) != NULL) { if ((this_binary = strrchr(g_argv_0, '\\')) != NULL) {
this_binary++; this_binary++;
} else if ((this_binary = strrchr(s_argv_0, '/')) != NULL) { } else if ((this_binary = strrchr(g_argv_0, '/')) != NULL) {
this_binary++; this_binary++;
} else { } else {
this_binary = s_argv_0; this_binary = g_argv_0;
} }
mg_printf(nc, "GET /%s HTTP/1.0\n\n", this_binary); mg_printf(nc, "GET /%s HTTP/1.0\n\n", this_binary);
/* Test mime type for static file */ /* Test mime type for static file */
@ -5394,12 +5390,12 @@ static const char *test_socks(void) {
mg_set_protocol_http_websocket(c); mg_set_protocol_http_websocket(c);
c->user_data = status; c->user_data = status;
/* Wine and GDB set argv0 to full path: strip the dir component */ /* Wine and GDB set argv0 to full path: strip the dir component */
if ((this_binary = strrchr(s_argv_0, '\\')) != NULL) { if ((this_binary = strrchr(g_argv_0, '\\')) != NULL) {
this_binary++; this_binary++;
} else if ((this_binary = strrchr(s_argv_0, '/')) != NULL) { } else if ((this_binary = strrchr(g_argv_0, '/')) != NULL) {
this_binary++; this_binary++;
} else { } else {
this_binary = s_argv_0; this_binary = g_argv_0;
} }
mg_printf(c, "GET /%s HTTP/1.0\n\n", this_binary); mg_printf(c, "GET /%s HTTP/1.0\n\n", this_binary);
@ -5412,7 +5408,11 @@ static const char *test_socks(void) {
} }
#endif #endif
static const char *run_tests(const char *filter, double *total_elapsed) { void tests_setup(void) {
test_iface = mg_if_create_iface(mg_ifaces[MG_MAIN_IFACE], NULL);
}
const char *tests_run(const char *filter) {
RUN_TEST(test_mbuf); RUN_TEST(test_mbuf);
RUN_TEST(test_parse_uri); RUN_TEST(test_parse_uri);
RUN_TEST(test_assemble_uri); RUN_TEST(test_assemble_uri);
@ -5525,51 +5525,6 @@ static const char *run_tests(const char *filter, double *total_elapsed) {
return NULL; return NULL;
} }
#if defined(_MSC_VER) && _MSC_VER >= 1900 void tests_teardown(void) {
int __cdecl CrtDbgHook(int nReportType, char *szMsg, int *pnRet) {
(void) nReportType;
(void) szMsg;
(void) pnRet;
fprintf(stderr, "CRT debug hook: type: %d, msg: %s\n", nReportType, szMsg);
/* Return true - Abort,Retry,Ignore dialog will *not* be displayed */
return 1;
}
#endif
void setup() {
test_iface = mg_if_create_iface(mg_ifaces[MG_MAIN_IFACE], NULL);
}
void teardown() {
free(test_iface); free(test_iface);
} }
int __cdecl main(int argc, char *argv[]) {
const char *fail_msg;
const char *filter = argc > 1 ? argv[1] : "";
double total_elapsed = 0.0;
setvbuf(stdout, NULL, _IONBF, 0);
setvbuf(stderr, NULL, _IONBF, 0);
#if defined(_MSC_VER) && _MSC_VER >= 1900
/* NOTE: not available on wine/vc6 */
_CrtSetReportHook2(_CRT_RPTHOOK_INSTALL, CrtDbgHook);
#endif
setup();
s_argv_0 = argv[0];
fail_msg = run_tests(filter, &total_elapsed);
printf("%s, run %d in %.3lfs\n", fail_msg ? "FAIL" : "PASS", num_tests,
total_elapsed);
teardown();
if (fail_msg != NULL) {
/* Prevent leak analyzer from running: there will be "leaks" because of
* premature return from the test, and in this case we don't care. */
_exit(EXIT_FAILURE);
}
return EXIT_SUCCESS;
}