static void
test_Equals(TestBatchRunner *runner) {
ByteBuf *wanted = BB_new_bytes("foo", 4); // Include terminating NULL.
ByteBuf *got = BB_new_bytes("foo", 4);
TEST_TRUE(runner, BB_Equals(wanted, (Obj*)got), "Equals");
TEST_INT_EQ(runner, BB_Hash_Sum(got), BB_Hash_Sum(wanted), "Hash_Sum");
TEST_TRUE(runner, BB_Equals_Bytes(got, "foo", 4), "Equals_Bytes");
TEST_FALSE(runner, BB_Equals_Bytes(got, "foo", 3),
"Equals_Bytes spoiled by different size");
TEST_FALSE(runner, BB_Equals_Bytes(got, "bar", 4),
"Equals_Bytes spoiled by different content");
BB_Set_Size(got, 3);
TEST_FALSE(runner, BB_Equals(wanted, (Obj*)got),
"Different size spoils Equals");
TEST_FALSE(runner, BB_Hash_Sum(got) == BB_Hash_Sum(wanted),
"Different size spoils Hash_Sum (probably -- at least this one)");
BB_Mimic_Bytes(got, "bar", 4);
TEST_INT_EQ(runner, BB_Get_Size(wanted), BB_Get_Size(got),
"same length");
TEST_FALSE(runner, BB_Equals(wanted, (Obj*)got),
"Different content spoils Equals");
DECREF(got);
DECREF(wanted);
}
static void
test_Equals(TestBatchRunner *runner) {
ByteBuf *bb = BB_new_bytes("foo", 4); // Include terminating NULL.
{
ByteBuf *other = BB_new_bytes("foo", 4);
TEST_TRUE(runner, BB_Equals(bb, (Obj*)other), "Equals");
DECREF(other);
}
TEST_TRUE(runner, BB_Equals_Bytes(bb, "foo", 4), "Equals_Bytes");
TEST_FALSE(runner, BB_Equals_Bytes(bb, "foo", 3),
"Equals_Bytes spoiled by different size");
TEST_FALSE(runner, BB_Equals_Bytes(bb, "bar", 4),
"Equals_Bytes spoiled by different content");
{
ByteBuf *other = BB_new_bytes("foo", 3);
TEST_FALSE(runner, BB_Equals(bb, (Obj*)other),
"Different size spoils Equals");
DECREF(other);
}
{
ByteBuf *other = BB_new_bytes("bar", 4);
TEST_UINT_EQ(runner, BB_Get_Size(bb), BB_Get_Size(other),
"same length");
TEST_FALSE(runner, BB_Equals(bb, (Obj*)other),
"Different content spoils Equals");
DECREF(other);
}
DECREF(bb);
}
static void
test_Clone(TestBatchRunner *runner) {
ByteBuf *bb = BB_new_bytes("foo", 3);
ByteBuf *twin = BB_Clone(bb);
TEST_TRUE(runner, BB_Equals(bb, (Obj*)twin), "Clone");
DECREF(bb);
DECREF(twin);
}
static void
test_Cat(TestBatchRunner *runner) {
ByteBuf *wanted = BB_new_bytes("foobar", 6);
ByteBuf *got = BB_new_bytes("foo", 3);
ByteBuf *scratch = BB_new_bytes("bar", 3);
BB_Cat(got, scratch);
TEST_TRUE(runner, BB_Equals(wanted, (Obj*)got), "Cat");
BB_Mimic_Bytes(wanted, "foobarbaz", 9);
BB_Cat_Bytes(got, "baz", 3);
TEST_TRUE(runner, BB_Equals(wanted, (Obj*)got), "Cat_Bytes");
DECREF(scratch);
DECREF(got);
DECREF(wanted);
}
static void
test_compare(TestBatchRunner *runner) {
ByteBuf *a = BB_new_bytes("foo\0a", 5);
ByteBuf *b = BB_new_bytes("foo\0b", 5);
BB_Set_Size(a, 4);
BB_Set_Size(b, 4);
TEST_INT_EQ(runner, BB_compare(&a, &b), 0,
"BB_compare returns 0 for equal ByteBufs");
BB_Set_Size(a, 3);
TEST_TRUE(runner, BB_compare(&a, &b) < 0, "shorter ByteBuf sorts first");
BB_Set_Size(a, 5);
BB_Set_Size(b, 5);
TEST_TRUE(runner, BB_compare(&a, &b) < 0,
"NULL doesn't interfere with BB_compare");
DECREF(a);
DECREF(b);
}
static void
test_Cat(TestBatchRunner *runner) {
ByteBuf *bb = BB_new_bytes("foo", 3);
{
Blob *blob = Blob_new("bar", 3);
BB_Cat(bb, blob);
TEST_TRUE(runner, BB_Equals_Bytes(bb, "foobar", 6), "Cat");
DECREF(blob);
}
BB_Cat_Bytes(bb, "baz", 3);
TEST_TRUE(runner, BB_Equals_Bytes(bb, "foobarbaz", 9), "Cat_Bytes");
DECREF(bb);
}
static void
test_Mimic(TestBatchRunner *runner) {
ByteBuf *a = BB_new_bytes("foo", 3);
ByteBuf *b = BB_new(0);
BB_Mimic(b, (Obj*)a);
TEST_TRUE(runner, BB_Equals(a, (Obj*)b), "Mimic");
BB_Mimic_Bytes(a, "bar", 4);
TEST_TRUE(runner, strcmp(BB_Get_Buf(a), "bar") == 0,
"Mimic_Bytes content");
TEST_INT_EQ(runner, BB_Get_Size(a), 4, "Mimic_Bytes size");
BB_Mimic(b, (Obj*)a);
TEST_TRUE(runner, BB_Equals(a, (Obj*)b), "Mimic");
DECREF(a);
DECREF(b);
}
static void
test_Utf8_To_String(TestBatchRunner *runner) {
ByteBuf *bb = BB_new_bytes("foo", 3);
{
String *string = BB_Utf8_To_String(bb);
TEST_TRUE(runner, Str_Equals_Utf8(string, "foo", 3), "Utf8_To_String");
DECREF(string);
}
{
String *string = BB_Trusted_Utf8_To_String(bb);
TEST_TRUE(runner, Str_Equals_Utf8(string, "foo", 3),
"Trusted_Utf8_To_String");
DECREF(string);
}
DECREF(bb);
}
static Hash*
S_extract_tv_cache(ByteBuf *field_buf) {
Hash *tv_cache = Hash_new(0);
const char *tv_string = BB_Get_Buf(field_buf);
int32_t num_terms = NumUtil_decode_c32(&tv_string);
CharBuf *text_buf = CB_new(0);
// Read the number of highlightable terms in the field.
for (int32_t i = 0; i < num_terms; i++) {
size_t overlap = NumUtil_decode_c32(&tv_string);
size_t len = NumUtil_decode_c32(&tv_string);
// Decompress the term text.
CB_Set_Size(text_buf, overlap);
CB_Cat_Trusted_Utf8(text_buf, tv_string, len);
tv_string += len;
// Get positions & offsets string.
const char *bookmark_ptr = tv_string;
int32_t num_positions = NumUtil_decode_c32(&tv_string);
while (num_positions--) {
// Leave nums compressed to save a little mem.
NumUtil_skip_cint(&tv_string);
NumUtil_skip_cint(&tv_string);
NumUtil_skip_cint(&tv_string);
}
len = tv_string - bookmark_ptr;
// Store the $text => $posdata pair in the output hash.
String *text = CB_To_String(text_buf);
Hash_Store(tv_cache, (Obj*)text,
(Obj*)BB_new_bytes(bookmark_ptr, len));
DECREF(text);
}
DECREF(text_buf);
return tv_cache;
}
ByteBuf*
Str_To_ByteBuf_IMP(String *self) {
return BB_new_bytes(self->ptr, self->size);
}
