#include <CUnit/CUnit.h>
|
#include <CUnit/Basic.h>
|
|
#include <arpa/inet.h>
|
|
#include "packet_mosq.h"
|
|
/* ========================================================================
|
* BYTE TESTS
|
* ======================================================================== */
|
|
/* This tests writing a Byte to an incoming packet. */
|
static void TEST_byte_write(void)
|
{
|
uint8_t payload[260];
|
struct mosquitto__packet packet;
|
int i;
|
|
memset(&packet, 0, sizeof(struct mosquitto__packet));
|
memset(payload, 0, sizeof(payload));
|
packet.payload = payload;
|
packet.packet_length = 256;
|
|
for(i=0; i<256; i++){
|
packet__write_byte(&packet, 255-i);
|
}
|
|
CU_ASSERT_EQUAL(packet.pos, 256);
|
for(i=0; i<256; i++){
|
CU_ASSERT_EQUAL(payload[i], 255-i);
|
}
|
}
|
|
|
/* ========================================================================
|
* TWO BYTE INTEGER TESTS
|
* ======================================================================== */
|
|
/* This tests writing a Two Byte Integer to an incoming packet. */
|
static void TEST_uint16_write(void)
|
{
|
uint8_t payload[650];
|
uint16_t *payload16;
|
struct mosquitto__packet packet;
|
int i;
|
|
memset(&packet, 0, sizeof(struct mosquitto__packet));
|
memset(payload, 0, sizeof(payload));
|
packet.payload = payload;
|
packet.packet_length = 650;
|
|
for(i=0; i<325; i++){
|
packet__write_uint16(&packet, 100*i);
|
}
|
|
CU_ASSERT_EQUAL(packet.pos, 650);
|
payload16 = (uint16_t *)payload;
|
for(i=0; i<325; i++){
|
CU_ASSERT_EQUAL(payload16[i], htons(100*i));
|
}
|
}
|
|
|
/* ========================================================================
|
* FOUR BYTE INTEGER TESTS
|
* ======================================================================== */
|
|
/* This tests writing a Four Byte Integer to an incoming packet. */
|
static void TEST_uint32_write(void)
|
{
|
uint8_t *payload;
|
uint32_t *payload32;
|
struct mosquitto__packet packet;
|
int i;
|
|
payload = calloc(42000, sizeof(uint32_t));
|
if(!payload){
|
CU_FAIL_FATAL("Out of memory");
|
}
|
|
memset(&packet, 0, sizeof(struct mosquitto__packet));
|
packet.payload = payload;
|
packet.packet_length = 42000;
|
|
for(i=0; i<10500; i++){
|
packet__write_uint32(&packet, 1000*i);
|
}
|
|
CU_ASSERT_EQUAL(packet.pos, 42000);
|
payload32 = (uint32_t *)payload;
|
for(i=0; i<10500; i++){
|
CU_ASSERT_EQUAL(payload32[i], htonl(1000*i));
|
}
|
free(payload);
|
}
|
|
|
/* ========================================================================
|
* UTF-8 STRING TESTS
|
* ======================================================================== */
|
|
/* This tests writing a UTF-8 String to an incoming packet. */
|
static void TEST_string_write(void)
|
{
|
uint8_t payload[100];
|
struct mosquitto__packet packet;
|
|
memset(&packet, 0, sizeof(struct mosquitto__packet));
|
memset(payload, 0, 100);
|
|
packet.payload = payload;
|
packet.packet_length = 100;
|
|
packet__write_string(&packet, "first test", strlen("first test"));
|
packet__write_string(&packet, "second test", strlen("second test"));
|
|
CU_ASSERT_EQUAL(packet.pos, 2+10+2+11);
|
CU_ASSERT_EQUAL(payload[0], 0);
|
CU_ASSERT_EQUAL(payload[1], 10);
|
CU_ASSERT_NSTRING_EQUAL(payload+2, "first test", 10);
|
CU_ASSERT_EQUAL(payload[2+10+0], 0);
|
CU_ASSERT_EQUAL(payload[2+10+1], 11);
|
CU_ASSERT_NSTRING_EQUAL(payload+2+10+2, "second test", 11);
|
}
|
|
|
/* ========================================================================
|
* TEST SUITE SETUP
|
* ======================================================================== */
|
|
int init_datatype_write_tests(void)
|
{
|
CU_pSuite test_suite = NULL;
|
|
test_suite = CU_add_suite("Datatype write", NULL, NULL);
|
if(!test_suite){
|
printf("Error adding CUnit test suite.\n");
|
return 1;
|
}
|
|
if(0
|
|| !CU_add_test(test_suite, "Byte write", TEST_byte_write)
|
|| !CU_add_test(test_suite, "Two Byte Integer write", TEST_uint16_write)
|
|| !CU_add_test(test_suite, "Four Byte Integer write", TEST_uint32_write)
|
|| !CU_add_test(test_suite, "UTF-8 String write", TEST_string_write)
|
){
|
|
printf("Error adding Datatype write CUnit tests.\n");
|
return 1;
|
}
|
|
return 0;
|
}
|
