From b8e54d28343fa3464a99676aed05d11733185cd1 Mon Sep 17 00:00:00 2001 From: Denis Kenzior Date: Tue, 16 Aug 2016 15:28:48 -0500 Subject: [PATCH] unit: Add M1 builder unit test --- unit/test-wsc.c | 20 ++++++++++++++++++++ 1 file changed, 20 insertions(+) diff --git a/unit/test-wsc.c b/unit/test-wsc.c index 6da5fac3..2e9d0497 100644 --- a/unit/test-wsc.c +++ b/unit/test-wsc.c @@ -623,6 +623,25 @@ static void wsc_test_parse_m1(const void *data) assert(!memcmp(test->public_key, m1.public_key, 192)); } +static void wsc_test_build_m1(const void *data) +{ + const struct m1_data *test = data; + struct wsc_m1 m1; + uint8_t *out; + size_t out_len; + + memcpy(&m1, &test->expected, sizeof(m1)); + memcpy(m1.public_key, test->public_key, 192); + + out = wsc_build_m1(&m1, &out_len); + assert(out); + + assert(out_len == test->len); + assert(!memcmp(test->pdu, out, test->len)); + + l_free(out); +} + int main(int argc, char *argv[]) { l_test_init(&argc, &argv); @@ -649,6 +668,7 @@ int main(int argc, char *argv[]) &dh_generate_pubkey_test_data_1); l_test_add("/wsc/parse/m1 1", wsc_test_parse_m1, &m1_data_1); + l_test_add("/wsc/build/m1 1", wsc_test_build_m1, &m1_data_1); return l_test_run(); }