105 lines
2.7 KiB
C++
105 lines
2.7 KiB
C++
#include <cassert>
|
|
#include <iostream>
|
|
|
|
#include "Arena.h"
|
|
#include "Phonebook.h"
|
|
#include "test_fixtures.h"
|
|
|
|
|
|
constexpr char TEST_PBSTR1[] = "foo";
|
|
constexpr uint8_t TEST_PBSTRLEN1 = 3;
|
|
constexpr char TEST_PBSTR2[] = "baz";
|
|
constexpr uint8_t TEST_PBSTRLEN2 = 3;
|
|
constexpr char TEST_PBSTR3[] = "quux";
|
|
constexpr uint8_t TEST_PBSTRLEN3 = 4;
|
|
constexpr char TEST_PBSTR4[] = "spam";
|
|
constexpr uint8_t TEST_PBSTRLEN4 = 4;
|
|
constexpr char TEST_PBSTR5[] = "xyzzx";
|
|
constexpr uint8_t TEST_PBSTRLEN5 = 5;
|
|
constexpr char TEST_PBSTR6[] = "corvid";
|
|
constexpr uint8_t TEST_PBSTRLEN6 = 6;
|
|
|
|
|
|
static bool
|
|
test_setpb(Phonebook &pb, const char *k, uint8_t kl, const char *v,
|
|
uint8_t vl)
|
|
{
|
|
bool ok;
|
|
std::cout << "test Set " << k << "->" << v << std::endl;
|
|
ok = pb.Set(k, kl, v, vl) == 0;
|
|
std::cout << "\tSet complete\n";
|
|
return ok;
|
|
}
|
|
|
|
|
|
int
|
|
main(int argc, const char *argv[])
|
|
{
|
|
Arena arena;
|
|
TLV::Record value;
|
|
TLV::Record expect;
|
|
|
|
std::cout << "TESTPROG: " << argv[0] << std::endl;
|
|
InitializeArena(arena);
|
|
|
|
#if defined(__linux__)
|
|
if (CreateArena(arena, ARENA_FILE, ARENA_SIZE, 0644) == -1) {
|
|
abort();
|
|
}
|
|
#else
|
|
if (AllocNewArena(arena, ARENA_SIZE) == -1) {
|
|
abort();
|
|
}
|
|
#endif
|
|
DisplayArena(arena);
|
|
|
|
TLV::SetRecord(expect, PHONEBOOK_VAL_TAG, TEST_PBSTRLEN3, TEST_PBSTR3);
|
|
|
|
Phonebook pb(arena);
|
|
assert(!pb.Has(TEST_PBSTR2, TEST_PBSTRLEN2));
|
|
|
|
assert(test_setpb(pb, TEST_PBSTR1, TEST_PBSTRLEN1, TEST_PBSTR3,
|
|
TEST_PBSTRLEN3));
|
|
pb.DumpKVPairs();
|
|
assert(test_setpb(pb, TEST_PBSTR2, TEST_PBSTRLEN2, TEST_PBSTR3,
|
|
TEST_PBSTRLEN3));
|
|
pb.DumpKVPairs();
|
|
assert(pb.Has(TEST_PBSTR2, TEST_PBSTRLEN2));
|
|
assert(test_setpb(pb, TEST_PBSTR4, TEST_PBSTRLEN4, TEST_PBSTR5,
|
|
TEST_PBSTRLEN5));
|
|
pb.DumpKVPairs();
|
|
assert(pb.Lookup(TEST_PBSTR2, TEST_PBSTRLEN2, value));
|
|
|
|
assert(cmp_record(value, expect));
|
|
|
|
std::cout << "test overwriting key" << std::endl;
|
|
assert(test_setpb(pb, TEST_PBSTR2, TEST_PBSTRLEN2, TEST_PBSTR6,
|
|
TEST_PBSTRLEN6));
|
|
pb.DumpKVPairs();
|
|
TLV::SetRecord(expect, PHONEBOOK_VAL_TAG, TEST_PBSTRLEN6, TEST_PBSTR6);
|
|
std::cout << "\tlookup" << std::endl;
|
|
assert(pb.Lookup(TEST_PBSTR2, TEST_PBSTRLEN2, value));
|
|
std::cout << "\tcompare records" << std::endl;
|
|
assert(cmp_record(value, expect));
|
|
|
|
std::cout << "\tadd new key to phonebook" << std::endl;
|
|
assert(test_setpb(pb, TEST_PBSTR3, TEST_PBSTRLEN3, TEST_PBSTR5,
|
|
TEST_PBSTRLEN5));
|
|
pb.DumpKVPairs();
|
|
|
|
TLV::SetRecord(expect, PHONEBOOK_VAL_TAG, TEST_PBSTRLEN5, TEST_PBSTR5);
|
|
assert(pb.Lookup(TEST_PBSTR4, TEST_PBSTRLEN4, value));
|
|
assert(cmp_record(value, expect));
|
|
|
|
std::cout << "OK" <<std::endl;
|
|
|
|
// Dump the generated arena for inspection later.
|
|
#if defined(__linux__)
|
|
#else
|
|
pb.DumpToFile(ARENA_FILE);
|
|
#endif
|
|
|
|
ClearArena(arena);
|
|
pb.DumpKVPairs();
|
|
}
|