denozord/strongswan-ext
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

unit-tests: Add a test case for HTTP response codes

Browse Source
This commit is contained in:
Tobias Brunner
2014-05-19 14:24:12 +02:00
parent 9a18593752
commit deb8975bd2
1 changed files with 57 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/libstrongswan/tests/suites/test_fetch_http.c
+57 -8
View File
@@ -18,6 +18,8 @@
#include <unistd.h> #include <unistd.h>
#include <time.h> #include <time.h>
#define HTTP_SUCCESS(status) ((status) >= 200 && (status) < 300)
/** /**
* HTTP test definition * HTTP test definition
*/ */
@@ -42,6 +44,8 @@ typedef struct {
void *res; void *res;
/* length of response data */ /* length of response data */
int res_len; int res_len;
/* status code, defaults to 200 */
u_int code;
} test_service_t; } test_service_t;
static char large[] = { static char large[] = {
@@ -147,8 +151,13 @@ static bool servicing(void *data, stream_t *stream)
ck_assert(memeq(body, test->req, test->req_len)); ck_assert(memeq(body, test->req, test->req_len));
} }
if (!test->code)
{
test->code = 200;
}
/* response headers */ /* response headers */
snprintf(buf, sizeof(buf), "HTTP/1.%u 200 OK\r\n", test->minor); snprintf(buf, sizeof(buf), "HTTP/1.%u %u OK\r\n", test->minor, test->code);
ck_assert(stream->write_all(stream, buf, strlen(buf))); ck_assert(stream->write_all(stream, buf, strlen(buf)));
t = time(NULL); t = time(NULL);
gmtime_r(&t, &tm); gmtime_r(&t, &tm);
@@ -174,13 +183,13 @@ static bool servicing(void *data, stream_t *stream)
static test_service_t gtests[] = { static test_service_t gtests[] = {
{ "GET", 1, "127.0.0.1", 6543, "/a/test/?b=c", NULL, { "GET", 1, "127.0.0.1", 6543, "/a/test/?b=c", NULL,
NULL, 0, "\x12\x34", 2 }, NULL, 0, "\x12\x34", 2, 0 },
{ "GET", 0, "localhost", 6543, "/", NULL, { "GET", 0, "localhost", 6543, "/", NULL,
NULL, 0, NULL, 0 }, NULL, 0, NULL, 0, 0 },
{ "GET", 0, "127.0.0.1", 6543, "/largefile", NULL, { "GET", 0, "127.0.0.1", 6543, "/largefile", NULL,
NULL, 0, large, sizeof(large) }, NULL, 0, large, sizeof(large), 0 },
{ "GET", 1, "[::1]", 6543, "/ipv6-url", NULL, { "GET", 1, "[::1]", 6543, "/ipv6-url", NULL,
NULL, 0, "\x00\r\n\r\x00testdatablabla", 20 }, NULL, 0, "\x00\r\n\r\x00testdatablabla", 20, 0 },
}; };
START_TEST(test_get) START_TEST(test_get)
@@ -215,11 +224,11 @@ END_TEST
static test_service_t ptests[] = { static test_service_t ptests[] = {
{ "POST", 1, "127.0.0.1", 6543, "/a/test/?b=c", "application/binary", { "POST", 1, "127.0.0.1", 6543, "/a/test/?b=c", "application/binary",
"\x23\x45", 2, "\x12\x34", 2 }, "\x23\x45", 2, "\x12\x34", 2, 0 },
{ "POST", 0, "localhost", 6543, "/largefile", "application/x-large", { "POST", 0, "localhost", 6543, "/largefile", "application/x-large",
large, sizeof(large), large, sizeof(large) }, large, sizeof(large), large, sizeof(large), 0 },
{ "POST", 1, "[::1]", 6543, "/ipv6-url", "text/plain", { "POST", 1, "[::1]", 6543, "/ipv6-url", "text/plain",
"\x00\r\n\r\x00testdatablabla", 20, "\x00\r\n\r\x00testdatablabla", 20 }, "\x00\r\n\r\x00testdatablabla", 20, "\x00\r\n\r\x00testdatablabla", 20, 0 },
}; };
START_TEST(test_post) START_TEST(test_post)
@@ -254,6 +263,42 @@ START_TEST(test_post)
} }
END_TEST END_TEST
static test_service_t rtests[] = {
{ "GET", 1, "localhost", 6544, "/", NULL, NULL, 0, NULL, 0, 200 },
{ "GET", 1, "localhost", 6544, "/", NULL, NULL, 0, NULL, 0, 204 },
{ "GET", 1, "localhost", 6544, "/", NULL, NULL, 0, NULL, 0, 400 },
{ "GET", 1, "localhost", 6544, "/", NULL, NULL, 0, NULL, 0, 404 },
{ "GET", 1, "localhost", 6544, "/", NULL, NULL, 0, NULL, 0, 500 },
};
START_TEST(test_response_code)
{
stream_service_t *service;
status_t status;
chunk_t data;
char uri[256];
u_int code;
lib->processor->set_threads(lib->processor, 8);
snprintf(uri, sizeof(uri), "tcp://%s:%u", rtests[_i].host, rtests[_i].port);
service = lib->streams->create_service(lib->streams, uri, 1);
ck_assert(service != NULL);
service->on_accept(service, servicing, &rtests[_i], JOB_PRIO_HIGH, 0);
snprintf(uri, sizeof(uri), "http://%s:%u%s",
rtests[_i].host, rtests[_i].port, rtests[_i].path);
status = lib->fetcher->fetch(lib->fetcher, uri, &data,
FETCH_RESPONSE_CODE, &code, FETCH_END);
ck_assert_int_eq(status, HTTP_SUCCESS(rtests[_i].code) ? SUCCESS : FAILED);
ck_assert_int_eq(code, rtests[_i].code);
free(data.ptr);
service->destroy(service);
}
END_TEST
Suite *fetch_http_suite_create() Suite *fetch_http_suite_create()
{ {
Suite *s; Suite *s;
@@ -269,5 +314,9 @@ Suite *fetch_http_suite_create()
tcase_add_loop_test(tc, test_post, 0, countof(ptests)); tcase_add_loop_test(tc, test_post, 0, countof(ptests));
suite_add_tcase(s, tc); suite_add_tcase(s, tc);
tc = tcase_create("response code");
tcase_add_loop_test(tc, test_response_code, 0, countof(rtests));
suite_add_tcase(s, tc);
return s; return s;
} }