125 lines
2.5 KiB
C++
125 lines
2.5 KiB
C++
#include <cassert>
|
|
#include <cstring>
|
|
#include <iostream>
|
|
|
|
#include "Arena.h"
|
|
#include "TLV.h"
|
|
|
|
#include "test_fixtures.h"
|
|
|
|
|
|
static uint8_t arena_buffer[ARENA_SIZE];
|
|
|
|
|
|
static bool
|
|
cmp_record(TLV::Record &a, TLV::Record &b)
|
|
{
|
|
if (a.Tag != b .Tag) {
|
|
return false;
|
|
}
|
|
|
|
if (a.Len != b.Len) {
|
|
return false;
|
|
}
|
|
|
|
if (memcmp(a.Val, b.Val, a.Len) != 0) {
|
|
return false;
|
|
}
|
|
|
|
return true;
|
|
}
|
|
|
|
|
|
bool
|
|
tlv_test_suite(Arena &backend)
|
|
{
|
|
TLV::Record rec1, rec2, rec3, rec4;
|
|
uint8_t *cursor = NULL;
|
|
|
|
TLV::set_record(rec1, 1, TEST_STRLEN1, TEST_STR1);
|
|
TLV::set_record(rec2, 2, TEST_STRLEN2, TEST_STR2);
|
|
TLV::set_record(rec3, 1, TEST_STRLEN4, TEST_STR4);
|
|
rec4.Tag = 1;
|
|
|
|
assert(TLV::write_to_memory(backend, cursor, rec1) != NULL);
|
|
assert((cursor = TLV::write_to_memory(backend, cursor, rec2)) != NULL);
|
|
assert(TLV::write_to_memory(backend, cursor, rec3) != NULL);
|
|
cursor = NULL;
|
|
|
|
// the cursor should point at the next record,
|
|
// and rec4 should contain the same data as rec1.
|
|
cursor = TLV::find_tag(backend, cursor, rec4);
|
|
assert(cursor != NULL);
|
|
assert(cursor != backend.store);
|
|
assert(cmp_record(rec1, rec4));
|
|
|
|
cursor = TLV::find_tag(backend, cursor, rec4);
|
|
assert(cursor != NULL);
|
|
assert(cmp_record(rec3, rec4));
|
|
|
|
TLV::set_record(rec4, 3, TEST_STRLEN3, TEST_STR3);
|
|
assert(TLV::write_to_memory(backend, NULL, rec4));
|
|
|
|
rec4.Tag = 2;
|
|
cursor = TLV::find_tag(backend, NULL, rec4);
|
|
assert(cursor != NULL);
|
|
|
|
TLV::delete_record(backend, cursor);
|
|
assert(cursor[0] == 3);
|
|
|
|
return true;
|
|
}
|
|
|
|
bool
|
|
run_suite(Arena &backend, const char *label)
|
|
{
|
|
std::cout << "running test suite " << label << ": ";
|
|
if (!tlv_test_suite(backend)) {
|
|
std::cout << "FAILED" << std::endl;
|
|
return false;
|
|
}
|
|
std::cout << "OK" << std::endl;
|
|
|
|
std::cout << "\tdestroying arena: ";
|
|
if (arena_destroy(backend) != 0) {
|
|
std::cout << "FAILED" << std::endl;
|
|
return false;
|
|
}
|
|
|
|
std::cout << "OK" << std::endl;
|
|
return true;
|
|
}
|
|
|
|
|
|
int
|
|
main(int argc, const char *argv[])
|
|
{
|
|
Arena arena_static;
|
|
Arena arena_mem;
|
|
Arena arena_file;
|
|
|
|
std::cout << "TESTPROG: " << argv[0] << std::endl;
|
|
|
|
if (-1 == new_arena(arena_static, arena_buffer, ARENA_SIZE)) {
|
|
abort();
|
|
} else if (!run_suite(arena_static, "arena_static")) {
|
|
abort();
|
|
}
|
|
|
|
if (-1 == create_arena(arena_file, ARENA_FILE, ARENA_SIZE, 0644)) {
|
|
abort();
|
|
} else if (!run_suite(arena_file, "arena_file")) {
|
|
abort();
|
|
}
|
|
|
|
if (-1 == alloc_new_arena(arena_mem, ARENA_SIZE)) {
|
|
abort();
|
|
} else if (!run_suite(arena_mem, "arena_mem")) {
|
|
abort();
|
|
}
|
|
|
|
|
|
std::cout << "OK" << std::endl;
|
|
return 0;
|
|
}
|