static void test_newentry(void **context)
{
setup();
UpdateLastSawHost("SHA-12345", "127.0.0.64", true, 666);
DBHandle *db;
OpenDB(&db, dbid_lastseen);
KeyHostSeen q;
assert_int_equal(ReadDB(db, "qiSHA-12345", &q, sizeof(q)), true);
assert_int_equal(q.lastseen, 666);
assert_double_close(q.Q.q, 0.0);
assert_double_close(q.Q.dq, 0.0);
assert_double_close(q.Q.expect, 0.0);
assert_double_close(q.Q.var, 0.0);
assert_int_equal(ReadDB(db, "qoSHA-12345", &q, sizeof(q)), false);
char address[CF_BUFSIZE];
assert_int_equal(ReadDB(db, "kSHA-12345", address, sizeof(address)), true);
assert_string_equal(address, "127.0.0.64");
char hostkey[CF_BUFSIZE];
assert_int_equal(ReadDB(db, "a127.0.0.64", hostkey, sizeof(hostkey)), true);
assert_string_equal(hostkey, "SHA-12345");
CloseDB(db);
}
void test_migrate_single(const char *expected_old_key,
const char *expected_quality_key)
{
/* Test migration of single entry */
DBHandle *db = setup(true);
KeyHostSeen0 khs0 = {
.q = 666777.0,
.expect = 12345.0,
.var = 6543210.0,
};
strcpy(khs0.address, "1.2.3.4");
assert_int_equal(WriteDB(db, expected_old_key, &khs0, sizeof(khs0)), true);
CloseDB(db);
assert_int_equal(OpenDB(&db, dbid_lastseen), true);
/* Old entry migrated */
assert_int_equal(HasKeyDB(db, expected_old_key,
strlen(expected_old_key) + 1), false);
/* Version marker */
assert_int_equal(HasKeyDB(db, "version", strlen("version") + 1), true);
/* Incoming connection quality */
KeyHostSeen khs;
assert_int_equal(ReadDB(db, expected_quality_key, &khs, sizeof(khs)), true);
assert_int_equal(khs.lastseen, 666777);
assert_double_close(khs.Q.q, 12345.0);
assert_double_close(khs.Q.expect, 12345.0);
assert_double_close(khs.Q.var, 6543210.0);
/* Address mapping */
char address[CF_BUFSIZE];
assert_int_equal(ReadDB(db, KEYHASH_KEY, address, sizeof(address)), true);
assert_string_equal(address, "1.2.3.4");
/* Reverse mapping */
char keyhash[CF_BUFSIZE];
assert_int_equal(ReadDB(db, "a1.2.3.4", keyhash, sizeof(keyhash)), true);
assert_string_equal(keyhash, KEYHASH);
CloseDB(db);
}
void test_migrate_incoming(void)
{
test_migrate_single(KEYHASH_IN, QUALITY_IN);
}
static void test_update(void **context)
{
setup();
UpdateLastSawHost("SHA-12345", "127.0.0.64", true, 555);
UpdateLastSawHost("SHA-12345", "127.0.0.64", true, 1110);
DBHandle *db;
OpenDB(&db, dbid_lastseen);
KeyHostSeen q;
assert_int_equal(ReadDB(db, "qiSHA-12345", &q, sizeof(q)), true);
assert_int_equal(q.lastseen, 1110);
assert_double_close(q.Q.q, 555.0);
assert_double_close(q.Q.dq, 555.0);
assert_double_close(q.Q.expect, 222.0);
assert_double_close(q.Q.var, 123210.0);
CloseDB(db);
}
static void test_double_from_string(void)
{
{
double val;
assert_true(DoubleFromString("1.2k", &val));
assert_double_close(1200.0, val);
}
{
double val;
assert_false(DoubleFromString("abc", &val));
}
}
static void test_parse_primitives(void)
{
JsonElement *pri = NULL;
const char *data = "\"foo\"";
assert_int_equal(JSON_PARSE_OK, JsonParse(&data, &pri));
assert_string_equal("foo", JsonPrimitiveGetAsString(pri));
JsonDestroy(pri);
data = "-123";
assert_int_equal(JSON_PARSE_OK, JsonParse(&data, &pri));
assert_true(-123 == JsonPrimitiveGetAsInteger(pri));
JsonDestroy(pri);
data = "1.23";
assert_int_equal(JSON_PARSE_OK, JsonParse(&data, &pri));
assert_double_close(1.23, JsonPrimitiveGetAsReal(pri));
JsonDestroy(pri);
data = "true";
assert_int_equal(JSON_PARSE_OK, JsonParse(&data, &pri));
assert_true(JsonPrimitiveGetAsBool(pri));
JsonDestroy(pri);
}
