#include "test.h"
#include "store_mdb.h"
#include "assets.h"
static char *path = "/tmp/testdb";
/** @brief Test triple store.
*/
static int test_triple_store()
{
EXPECT_PASS (LSUP_mdbstore_setup (path, true));
LSUP_MDBStore *store = LSUP_mdbstore_new (path, NULL); // triple store.
ASSERT (store != NULL, "Error initializing store!");
LSUP_Triple *trp = create_triples();
LSUP_SerTriple ser_trp[NUM_TRP];
for (int i = 0; i < NUM_TRP; i++) {
LSUP_striple_init (ser_trp + i, BUF_DUMMY, BUF_DUMMY, BUF_DUMMY);
LSUP_triple_serialize (trp + i, ser_trp + i);
}
// Test adding.
size_t ins;
EXPECT_PASS (LSUP_mdbstore_add (store, NULL, ser_trp, NUM_TRP, &ins));
EXPECT_INT_EQ (ins, 8);
EXPECT_INT_EQ (LSUP_mdbstore_size (store), 8);
// Test lookups.
LSUP_SerTriple lut[12] = {
{NULL, NULL, NULL},
{ser_trp[0].s, NULL, NULL},
{ser_trp[2].s, NULL, NULL},
{NULL, ser_trp[0].p, NULL},
{NULL, ser_trp[0].s, NULL},
{NULL, NULL, ser_trp[6].o},
{ser_trp[4].s, ser_trp[4].p, NULL},
{NULL, ser_trp[7].p, ser_trp[7].o},
{ser_trp[5].s, NULL, ser_trp[5].o},
{ser_trp[5].s, NULL, ser_trp[5].o},
{ser_trp[4].s, ser_trp[4].p, ser_trp[4].o},
{ser_trp[4].s, ser_trp[4].p, ser_trp[6].o},
};
LSUP_Buffer *luc[12] = {
NULL,
NULL,
ser_trp[2].s,
NULL,
NULL,
NULL,
NULL,
NULL,
NULL,
ser_trp[2].s,
NULL,
NULL,
};
size_t results[12] = {
8,
5, 1, 1, 0, 1,
2, 1, 2, 2,
1, 0,
};
for (int i = 0; i < 12; i++) {
size_t ct;
TRACE ("Testing triple lookup #%d.\n", i);
LSUP_MDBIterator *it = LSUP_mdbstore_lookup(
store, lut + i, luc[i], &ct);
EXPECT_INT_EQ (ct, results[i]);
LSUP_mdbiter_free (it);
}
for (int i = 0; i < NUM_TRP; i++) {
LSUP_buffer_free (ser_trp[i].s);
LSUP_buffer_free (ser_trp[i].p);
LSUP_buffer_free (ser_trp[i].o);
}
LSUP_mdbstore_free (store);
free_triples (trp);
return 0;
}
/** @brief Test quad store.
*
* Insert the same triple set twice with different contexts.
*/
static int test_quad_store()
{
EXPECT_PASS (LSUP_mdbstore_setup (path, true));
LSUP_Term *ctx1 = LSUP_uri_new ("urn:c:1");
LSUP_Buffer *sc1 = LSUP_buffer_new_from_term (ctx1);
LSUP_MDBStore *store = LSUP_mdbstore_new (path, sc1); // quad store.
ASSERT (store != NULL, "Error initializing store!");
LSUP_Triple *trp = create_triples();
LSUP_SerTriple ser_trp[NUM_TRP];
for (int i = 0; i < NUM_TRP; i++) {
LSUP_striple_init (ser_trp + i, BUF_DUMMY, BUF_DUMMY, BUF_DUMMY);
LSUP_triple_serialize (trp + i, ser_trp + i);
}
// Only triples 0÷5 in default context.
size_t ct;
EXPECT_PASS (LSUP_mdbstore_add (store, NULL, ser_trp, 6, &ct));
EXPECT_INT_EQ (ct, 6);
LSUP_Term *ctx2 = LSUP_uri_new ("urn:c:2");
LSUP_Buffer *sc2 = LSUP_buffer_new_from_term (ctx2);
// Only triples 4÷9 in context 2 (effectively 4 non-duplicates).
EXPECT_PASS (LSUP_mdbstore_add (store, sc2, ser_trp + 4, 6, &ct));
EXPECT_INT_EQ (ct, 4);
// 6 triples in ctx1 + 6 in ctx2 - 2 duplicates
EXPECT_INT_EQ (LSUP_mdbstore_size (store), 10);
// This context has no triples.
LSUP_Term *ctx3 = LSUP_uri_new ("urn:c:3");
LSUP_Buffer *sc3 = LSUP_buffer_new_from_term (ctx3);
// Test lookups.
LSUP_SerTriple lut[41] = {
// Any context
{NULL, NULL, NULL}, // #0
{ser_trp[0].s, NULL, NULL}, // #1
{NULL, ser_trp[0].p, NULL}, // #2
{NULL, ser_trp[0].s, NULL}, // #3
{NULL, NULL, ser_trp[6].o}, // #4
{ser_trp[4].s, ser_trp[4].p, NULL}, // #5
{NULL, ser_trp[7].p, ser_trp[7].o}, // #6
{ser_trp[5].s, NULL, ser_trp[5].o}, // #7
{ser_trp[4].s, ser_trp[4].p, ser_trp[4].o}, // #8
{ser_trp[2].s, ser_trp[4].p, ser_trp[5].o}, // #9
// Context 1 (trp[0÷5])
{NULL, NULL, NULL}, // #10
{ser_trp[0].s, NULL, NULL}, // #11
{ser_trp[2].s, NULL, NULL}, // #12
{NULL, ser_trp[0].p, NULL}, // #13
{NULL, ser_trp[6].p, NULL}, // #14
{NULL, NULL, ser_trp[6].o}, // #15
{ser_trp[4].s, ser_trp[4].p, NULL}, // #16
{NULL, ser_trp[7].p, ser_trp[7].o}, // #17
{ser_trp[5].s, NULL, ser_trp[5].o}, // #18
{ser_trp[4].s, ser_trp[4].p, ser_trp[4].o}, // #19
{ser_trp[6].s, ser_trp[6].p, ser_trp[6].o}, // #20
// Context 2 (trp[4÷9])
{NULL, NULL, NULL}, // #21
{ser_trp[0].s, NULL, NULL}, // #22
{NULL, ser_trp[0].p, NULL}, // #23
{NULL, ser_trp[0].s, NULL}, // #24
{NULL, NULL, ser_trp[6].o}, // #25
{ser_trp[4].s, ser_trp[4].p, NULL}, // #26
{NULL, ser_trp[7].p, ser_trp[7].o}, // #27
{ser_trp[5].s, NULL, ser_trp[5].o}, // #28
{ser_trp[4].s, ser_trp[4].p, ser_trp[4].o}, // #29
{ser_trp[6].s, ser_trp[6].p, ser_trp[6].o}, // #30
// Non-existing context
{NULL, NULL, NULL}, // #31
{ser_trp[0].s, NULL, NULL}, // #32
{NULL, ser_trp[0].p, NULL}, // #33
{NULL, ser_trp[0].s, NULL}, // #34
{NULL, NULL, ser_trp[6].o}, // #35
{ser_trp[4].s, ser_trp[4].p, NULL}, // #36
{NULL, ser_trp[7].p, ser_trp[7].o}, // #37
{ser_trp[5].s, NULL, ser_trp[5].o}, // #38
{ser_trp[4].s, ser_trp[4].p, ser_trp[4].o}, // #39
{ser_trp[6].s, ser_trp[6].p, ser_trp[6].o}, // #40
};
LSUP_Buffer *luc[41] = {
// Any context
NULL, // #0
NULL, NULL, NULL, NULL, // #1-#4
NULL, NULL, NULL, // #5-#7
NULL, NULL, // #8-#9
// Context 1 (trp[0÷5])
sc1, // #10
sc1, sc1, sc1, sc1, sc1, // #11-#15
sc1, sc1, sc1, // #16-#18
sc1, sc1, // #19-#20
// Context 2 (trp[4÷9])
sc2, // #21
sc2, sc2, sc2, sc2, // #22-#25
sc2, sc2, sc2, // #26-#28
sc2, sc2, // #29-#30
// Non-existing context
sc3, // #31
sc3, sc3, sc3, sc3, // #32-#35
sc3, sc3, sc3, // #36-#38
sc3, sc3, // #39-#40
};
size_t results[41] = {
// NULL ctx
8, // #0
5, 1, 0, 1, // #1-#4
2, 1, 2, // #5-#7
1, 0, // #8-#9
// ctx1
6, // #10
4, 1, 1, 0, 0, // #11-#15
1, 0, 1, // #16-#18
1, 0, // #19-#20
// ctx2
4, // #21
3, 0, 0, 1, // #22-#25
2, 1, 2, // #26-#28
1, 1, // #29-#30
// ctx3
0, // #31-#32
0, 0, 0, 0, // #33-#36
0, 0, 0, // #37-#39
0, 0, // #40
};
for (int i = 0; i < 41; i++) {
size_t ct;
printf ("Testing triple lookup #%d: {", i);
if (lut[i].s) LSUP_buffer_print (lut[i].s);
else printf ("NULL");
printf (", ");
if (lut[i].p) LSUP_buffer_print (lut[i].p);
else printf ("NULL");
printf (", ");
if (lut[i].o) LSUP_buffer_print (lut[i].o);
else printf ("NULL");
printf (", ");
if (luc[i]) LSUP_buffer_print (luc[i]);
else printf ("NULL");
printf ("}\n");
LSUP_MDBIterator *it = LSUP_mdbstore_lookup(
store, lut + i, luc[i], &ct);
ASSERT (it != NULL, "Lookup error!");
EXPECT_INT_EQ (ct, results[i]);
LSUP_mdbiter_free (it);
}
for (int i = 0; i < NUM_TRP; i++) {
LSUP_buffer_free (ser_trp[i].s);
LSUP_buffer_free (ser_trp[i].p);
LSUP_buffer_free (ser_trp[i].o);
}
LSUP_term_free (ctx1);
LSUP_term_free (ctx2);
LSUP_term_free (ctx3);
LSUP_buffer_free (sc1);
LSUP_buffer_free (sc2);
LSUP_buffer_free (sc3);
free_triples (trp);
LSUP_mdbstore_free (store);
return 0;
}
int store_mdb_test()
{
RUN (test_triple_store);
RUN (test_quad_store);
return 0;
}