Merge branch 'sk/reftable-clarify-tests'

The reftable unit tests are now ported to the "clar" unit testing
framework.

* sk/reftable-clarify-tests:
  t/unit-tests: finalize migration of reftable-related tests
  t/unit-tests: convert reftable stack test to use clar
  t/unit-tests: convert reftable record test to use clar
  t/unit-tests: convert reftable readwrite test to use clar
  t/unit-tests: convert reftable table test to use clar
  t/unit-tests: convert reftable pq test to use clar
  t/unit-tests: convert reftable merged test to use clar
  t/unit-tests: convert reftable block test to use clar
  t/unit-tests: convert reftable basics test to use clar test framework
  t/unit-tests: implement clar specific reftable test helper functions
This commit is contained in:
Junio C Hamano
2025-08-01 11:27:14 -07:00
14 changed files with 1117 additions and 1331 deletions

View File

@@ -1362,6 +1362,13 @@ CLAR_TEST_SUITES += u-oid-array
CLAR_TEST_SUITES += u-oidmap CLAR_TEST_SUITES += u-oidmap
CLAR_TEST_SUITES += u-oidtree CLAR_TEST_SUITES += u-oidtree
CLAR_TEST_SUITES += u-prio-queue CLAR_TEST_SUITES += u-prio-queue
CLAR_TEST_SUITES += u-reftable-basics
CLAR_TEST_SUITES += u-reftable-block
CLAR_TEST_SUITES += u-reftable-merged
CLAR_TEST_SUITES += u-reftable-pq
CLAR_TEST_SUITES += u-reftable-readwrite
CLAR_TEST_SUITES += u-reftable-stack
CLAR_TEST_SUITES += u-reftable-table
CLAR_TEST_SUITES += u-reftable-tree CLAR_TEST_SUITES += u-reftable-tree
CLAR_TEST_SUITES += u-strbuf CLAR_TEST_SUITES += u-strbuf
CLAR_TEST_SUITES += u-strcmp-offset CLAR_TEST_SUITES += u-strcmp-offset
@@ -1372,20 +1379,11 @@ CLAR_TEST_SUITES += u-urlmatch-normalization
CLAR_TEST_PROG = $(UNIT_TEST_BIN)/unit-tests$(X) CLAR_TEST_PROG = $(UNIT_TEST_BIN)/unit-tests$(X)
CLAR_TEST_OBJS = $(patsubst %,$(UNIT_TEST_DIR)/%.o,$(CLAR_TEST_SUITES)) CLAR_TEST_OBJS = $(patsubst %,$(UNIT_TEST_DIR)/%.o,$(CLAR_TEST_SUITES))
CLAR_TEST_OBJS += $(UNIT_TEST_DIR)/clar/clar.o CLAR_TEST_OBJS += $(UNIT_TEST_DIR)/clar/clar.o
CLAR_TEST_OBJS += $(UNIT_TEST_DIR)/unit-test.o
CLAR_TEST_OBJS += $(UNIT_TEST_DIR)/lib-oid.o CLAR_TEST_OBJS += $(UNIT_TEST_DIR)/lib-oid.o
CLAR_TEST_OBJS += $(UNIT_TEST_DIR)/lib-reftable.o
CLAR_TEST_OBJS += $(UNIT_TEST_DIR)/unit-test.o
UNIT_TEST_PROGRAMS += t-reftable-basics
UNIT_TEST_PROGRAMS += t-reftable-block
UNIT_TEST_PROGRAMS += t-reftable-merged
UNIT_TEST_PROGRAMS += t-reftable-pq
UNIT_TEST_PROGRAMS += t-reftable-readwrite
UNIT_TEST_PROGRAMS += t-reftable-record
UNIT_TEST_PROGRAMS += t-reftable-stack
UNIT_TEST_PROGRAMS += t-reftable-table
UNIT_TEST_PROGS = $(patsubst %,$(UNIT_TEST_BIN)/%$X,$(UNIT_TEST_PROGRAMS))
UNIT_TEST_OBJS += $(UNIT_TEST_DIR)/test-lib.o UNIT_TEST_OBJS += $(UNIT_TEST_DIR)/test-lib.o
UNIT_TEST_OBJS += $(UNIT_TEST_DIR)/lib-reftable.o
# xdiff and reftable libs may in turn depend on what is in libgit.a # xdiff and reftable libs may in turn depend on what is in libgit.a
GITLIBS = common-main.o $(LIB_FILE) $(XDIFF_LIB) $(REFTABLE_LIB) $(LIB_FILE) GITLIBS = common-main.o $(LIB_FILE) $(XDIFF_LIB) $(REFTABLE_LIB) $(LIB_FILE)

View File

@@ -125,7 +125,6 @@ check-meson:
@mkdir -p mesontmp && \ @mkdir -p mesontmp && \
printf "%s\n" \ printf "%s\n" \
"integration_tests t[0-9][0-9][0-9][0-9]-*.sh" \ "integration_tests t[0-9][0-9][0-9][0-9]-*.sh" \
"unit_test_programs unit-tests/t-*.c" \
"clar_test_suites unit-tests/u-*.c" | \ "clar_test_suites unit-tests/u-*.c" | \
while read -r variable pattern; do \ while read -r variable pattern; do \
awk "/^$$variable = \[\$$/ {flag=1 ; next } /^]$$/ { flag=0 } flag { gsub(/^ \047/, \"\"); gsub(/\047,\$$/, \"\"); print }" meson.build >mesontmp/meson.txt && \ awk "/^$$variable = \[\$$/ {flag=1 ; next } /^]$$/ { flag=0 } flag { gsub(/^ \047/, \"\"); gsub(/\047,\$$/, \"\"); print }" meson.build >mesontmp/meson.txt && \

View File

@@ -8,6 +8,14 @@ clar_test_suites = [
'unit-tests/u-oidmap.c', 'unit-tests/u-oidmap.c',
'unit-tests/u-oidtree.c', 'unit-tests/u-oidtree.c',
'unit-tests/u-prio-queue.c', 'unit-tests/u-prio-queue.c',
'unit-tests/u-reftable-basics.c',
'unit-tests/u-reftable-block.c',
'unit-tests/u-reftable-merged.c',
'unit-tests/u-reftable-pq.c',
'unit-tests/u-reftable-readwrite.c',
'unit-tests/u-reftable-record.c',
'unit-tests/u-reftable-stack.c',
'unit-tests/u-reftable-table.c',
'unit-tests/u-reftable-tree.c', 'unit-tests/u-reftable-tree.c',
'unit-tests/u-strbuf.c', 'unit-tests/u-strbuf.c',
'unit-tests/u-strcmp-offset.c', 'unit-tests/u-strcmp-offset.c',
@@ -20,7 +28,8 @@ clar_test_suites = [
clar_sources = [ clar_sources = [
'unit-tests/clar/clar.c', 'unit-tests/clar/clar.c',
'unit-tests/unit-test.c', 'unit-tests/unit-test.c',
'unit-tests/lib-oid.c' 'unit-tests/lib-oid.c',
'unit-tests/lib-reftable.c'
] ]
clar_decls_h = custom_target( clar_decls_h = custom_target(
@@ -54,33 +63,6 @@ clar_unit_tests = executable('unit-tests',
) )
test('unit-tests', clar_unit_tests, kwargs: test_kwargs) test('unit-tests', clar_unit_tests, kwargs: test_kwargs)
unit_test_programs = [
'unit-tests/t-reftable-basics.c',
'unit-tests/t-reftable-block.c',
'unit-tests/t-reftable-merged.c',
'unit-tests/t-reftable-pq.c',
'unit-tests/t-reftable-readwrite.c',
'unit-tests/t-reftable-record.c',
'unit-tests/t-reftable-stack.c',
'unit-tests/t-reftable-table.c',
]
foreach unit_test_program : unit_test_programs
unit_test_name = fs.stem(unit_test_program)
unit_test = executable(unit_test_name,
sources: [
'unit-tests/test-lib.c',
'unit-tests/lib-reftable.c',
unit_test_program,
],
dependencies: [libgit_commonmain],
)
test(unit_test_name, unit_test,
workdir: meson.current_source_dir(),
kwargs: test_kwargs,
)
endforeach
subdir('helper') subdir('helper')
integration_tests = [ integration_tests = [
@@ -1163,8 +1145,6 @@ benchmarks = [
# sufficient to catch missing test suites in our CI though. # sufficient to catch missing test suites in our CI though.
foreach glob, tests : { foreach glob, tests : {
't[0-9][0-9][0-9][0-9]-*.sh': integration_tests, 't[0-9][0-9][0-9][0-9]-*.sh': integration_tests,
'perf/p[0-9][0-9][0-9][0-9]-*.sh': benchmarks,
'unit-tests/t-*.c': unit_test_programs,
'unit-tests/u-*.c': clar_test_suites, 'unit-tests/u-*.c': clar_test_suites,
} }
actual_tests = run_command(shell, '-c', 'ls ' + glob, actual_tests = run_command(shell, '-c', 'ls ' + glob,
@@ -1231,4 +1211,4 @@ if perl.found() and time.found()
timeout: 0, timeout: 0,
) )
endforeach endforeach
endif endif

View File

@@ -1,12 +1,14 @@
#define DISABLE_SIGN_COMPARE_WARNINGS #include "unit-test.h"
#include "lib-reftable.h" #include "lib-reftable.h"
#include "test-lib.h" #include "hex.h"
#include "parse-options.h"
#include "reftable/constants.h" #include "reftable/constants.h"
#include "reftable/writer.h" #include "reftable/writer.h"
#include "strbuf.h" #include "strbuf.h"
#include "string-list.h"
#include "strvec.h"
void t_reftable_set_hash(uint8_t *p, int i, enum reftable_hash id) void cl_reftable_set_hash(uint8_t *p, int i, enum reftable_hash id)
{ {
memset(p, (uint8_t)i, hash_size(id)); memset(p, (uint8_t)i, hash_size(id));
} }
@@ -22,17 +24,17 @@ static int strbuf_writer_flush(void *arg UNUSED)
return 0; return 0;
} }
struct reftable_writer *t_reftable_strbuf_writer(struct reftable_buf *buf, struct reftable_writer *cl_reftable_strbuf_writer(struct reftable_buf *buf,
struct reftable_write_options *opts) struct reftable_write_options *opts)
{ {
struct reftable_writer *writer; struct reftable_writer *writer;
int ret = reftable_writer_new(&writer, &strbuf_writer_write, &strbuf_writer_flush, int ret = reftable_writer_new(&writer, &strbuf_writer_write, &strbuf_writer_flush,
buf, opts); buf, opts);
check(!ret); cl_assert(!ret);
return writer; return writer;
} }
void t_reftable_write_to_buf(struct reftable_buf *buf, void cl_reftable_write_to_buf(struct reftable_buf *buf,
struct reftable_ref_record *refs, struct reftable_ref_record *refs,
size_t nrefs, size_t nrefs,
struct reftable_log_record *logs, struct reftable_log_record *logs,
@@ -64,35 +66,36 @@ void t_reftable_write_to_buf(struct reftable_buf *buf,
min = ui; min = ui;
} }
writer = t_reftable_strbuf_writer(buf, &opts); writer = cl_reftable_strbuf_writer(buf, &opts);
reftable_writer_set_limits(writer, min, max); ret = reftable_writer_set_limits(writer, min, max);
cl_assert(!ret);
if (nrefs) { if (nrefs) {
ret = reftable_writer_add_refs(writer, refs, nrefs); ret = reftable_writer_add_refs(writer, refs, nrefs);
check_int(ret, ==, 0); cl_assert_equal_i(ret, 0);
} }
if (nlogs) { if (nlogs) {
ret = reftable_writer_add_logs(writer, logs, nlogs); ret = reftable_writer_add_logs(writer, logs, nlogs);
check_int(ret, ==, 0); cl_assert_equal_i(ret, 0);
} }
ret = reftable_writer_close(writer); ret = reftable_writer_close(writer);
check_int(ret, ==, 0); cl_assert_equal_i(ret, 0);
stats = reftable_writer_stats(writer); stats = reftable_writer_stats(writer);
for (size_t i = 0; i < stats->ref_stats.blocks; i++) { for (size_t i = 0; i < (size_t)stats->ref_stats.blocks; i++) {
size_t off = i * (opts.block_size ? opts.block_size size_t off = i * (opts.block_size ? opts.block_size
: DEFAULT_BLOCK_SIZE); : DEFAULT_BLOCK_SIZE);
if (!off) if (!off)
off = header_size(opts.hash_id == REFTABLE_HASH_SHA256 ? 2 : 1); off = header_size(opts.hash_id == REFTABLE_HASH_SHA256 ? 2 : 1);
check_char(buf->buf[off], ==, 'r'); cl_assert(buf->buf[off] == 'r');
} }
if (nrefs) if (nrefs)
check_int(stats->ref_stats.blocks, >, 0); cl_assert(stats->ref_stats.blocks > 0);
if (nlogs) if (nlogs)
check_int(stats->log_stats.blocks, >, 0); cl_assert(stats->log_stats.blocks > 0);
reftable_writer_free(writer); reftable_writer_free(writer);
} }

View File

@@ -1,21 +1,20 @@
#ifndef LIB_REFTABLE_H #include "git-compat-util.h"
#define LIB_REFTABLE_H #include "clar/clar.h"
#include "clar-decls.h"
#include "git-compat-util.h" #include "git-compat-util.h"
#include "reftable/reftable-writer.h" #include "reftable/reftable-writer.h"
#include "strbuf.h"
struct reftable_buf; struct reftable_buf;
void t_reftable_set_hash(uint8_t *p, int i, enum reftable_hash id); void cl_reftable_set_hash(uint8_t *p, int i, enum reftable_hash id);
struct reftable_writer *t_reftable_strbuf_writer(struct reftable_buf *buf, struct reftable_writer *cl_reftable_strbuf_writer(struct reftable_buf *buf,
struct reftable_write_options *opts); struct reftable_write_options *opts);
void t_reftable_write_to_buf(struct reftable_buf *buf, void cl_reftable_write_to_buf(struct reftable_buf *buf,
struct reftable_ref_record *refs, struct reftable_ref_record *refs,
size_t nrecords, size_t nrecords,
struct reftable_log_record *logs, struct reftable_log_record *logs,
size_t nlogs, size_t nlogs,
struct reftable_write_options *opts); struct reftable_write_options *opts);
#endif

View File

@@ -1,219 +0,0 @@
/*
Copyright 2020 Google LLC
Use of this source code is governed by a BSD-style
license that can be found in the LICENSE file or at
https://developers.google.com/open-source/licenses/bsd
*/
#include "test-lib.h"
#include "reftable/basics.h"
struct integer_needle_lesseq_args {
int needle;
int *haystack;
};
static int integer_needle_lesseq(size_t i, void *_args)
{
struct integer_needle_lesseq_args *args = _args;
return args->needle <= args->haystack[i];
}
static void *realloc_stub(void *p UNUSED, size_t size UNUSED)
{
return NULL;
}
int cmd_main(int argc UNUSED, const char *argv[] UNUSED)
{
if_test ("binary search with binsearch works") {
int haystack[] = { 2, 4, 6, 8, 10 };
struct {
int needle;
size_t expected_idx;
} testcases[] = {
{-9000, 0},
{-1, 0},
{0, 0},
{2, 0},
{3, 1},
{4, 1},
{7, 3},
{9, 4},
{10, 4},
{11, 5},
{9000, 5},
};
for (size_t i = 0; i < ARRAY_SIZE(testcases); i++) {
struct integer_needle_lesseq_args args = {
.haystack = haystack,
.needle = testcases[i].needle,
};
size_t idx;
idx = binsearch(ARRAY_SIZE(haystack),
&integer_needle_lesseq, &args);
check_int(idx, ==, testcases[i].expected_idx);
}
}
if_test ("names_length returns size of a NULL-terminated string array") {
const char *a[] = { "a", "b", NULL };
check_int(names_length(a), ==, 2);
}
if_test ("names_equal compares NULL-terminated string arrays") {
const char *a[] = { "a", "b", "c", NULL };
const char *b[] = { "a", "b", "d", NULL };
const char *c[] = { "a", "b", NULL };
check(names_equal(a, a));
check(!names_equal(a, b));
check(!names_equal(a, c));
}
if_test ("parse_names works for basic input") {
char in1[] = "line\n";
char in2[] = "a\nb\nc";
char **out = parse_names(in1, strlen(in1));
check(out != NULL);
check_str(out[0], "line");
check(!out[1]);
free_names(out);
out = parse_names(in2, strlen(in2));
check(out != NULL);
check_str(out[0], "a");
check_str(out[1], "b");
check_str(out[2], "c");
check(!out[3]);
free_names(out);
}
if_test ("parse_names drops empty string") {
char in[] = "a\n\nb\n";
char **out = parse_names(in, strlen(in));
check(out != NULL);
check_str(out[0], "a");
/* simply '\n' should be dropped as empty string */
check_str(out[1], "b");
check(!out[2]);
free_names(out);
}
if_test ("common_prefix_size works") {
struct reftable_buf a = REFTABLE_BUF_INIT;
struct reftable_buf b = REFTABLE_BUF_INIT;
struct {
const char *a, *b;
int want;
} cases[] = {
{"abcdef", "abc", 3},
{ "abc", "ab", 2 },
{ "", "abc", 0 },
{ "abc", "abd", 2 },
{ "abc", "pqr", 0 },
};
for (size_t i = 0; i < ARRAY_SIZE(cases); i++) {
check(!reftable_buf_addstr(&a, cases[i].a));
check(!reftable_buf_addstr(&b, cases[i].b));
check_uint(common_prefix_size(&a, &b), ==, cases[i].want);
reftable_buf_reset(&a);
reftable_buf_reset(&b);
}
reftable_buf_release(&a);
reftable_buf_release(&b);
}
if_test ("reftable_put_be64 and reftable_get_be64 work") {
uint64_t in = 0x1122334455667788;
uint8_t dest[8];
uint64_t out;
reftable_put_be64(dest, in);
out = reftable_get_be64(dest);
check_int(in, ==, out);
}
if_test ("reftable_put_be32 and reftable_get_be32 work") {
uint32_t in = 0x11223344;
uint8_t dest[4];
uint32_t out;
reftable_put_be32(dest, in);
out = reftable_get_be32(dest);
check_int(in, ==, out);
}
if_test ("reftable_put_be24 and reftable_get_be24 work") {
uint32_t in = 0x112233;
uint8_t dest[3];
uint32_t out;
reftable_put_be24(dest, in);
out = reftable_get_be24(dest);
check_int(in, ==, out);
}
if_test ("put_be16 and get_be16 work") {
uint32_t in = 0xfef1;
uint8_t dest[3];
uint32_t out;
reftable_put_be16(dest, in);
out = reftable_get_be16(dest);
check_int(in, ==, out);
}
if_test ("REFTABLE_ALLOC_GROW works") {
int *arr = NULL, *old_arr;
size_t alloc = 0, old_alloc;
check(!REFTABLE_ALLOC_GROW(arr, 1, alloc));
check(arr != NULL);
check_uint(alloc, >=, 1);
arr[0] = 42;
old_alloc = alloc;
old_arr = arr;
reftable_set_alloc(NULL, realloc_stub, NULL);
check(REFTABLE_ALLOC_GROW(arr, old_alloc + 1, alloc));
check(arr == old_arr);
check_uint(alloc, ==, old_alloc);
old_alloc = alloc;
reftable_set_alloc(NULL, NULL, NULL);
check(!REFTABLE_ALLOC_GROW(arr, old_alloc + 1, alloc));
check(arr != NULL);
check_uint(alloc, >, old_alloc);
arr[alloc - 1] = 42;
reftable_free(arr);
}
if_test ("REFTABLE_ALLOC_GROW_OR_NULL works") {
int *arr = NULL;
size_t alloc = 0, old_alloc;
REFTABLE_ALLOC_GROW_OR_NULL(arr, 1, alloc);
check(arr != NULL);
check_uint(alloc, >=, 1);
arr[0] = 42;
old_alloc = alloc;
REFTABLE_ALLOC_GROW_OR_NULL(arr, old_alloc + 1, alloc);
check(arr != NULL);
check_uint(alloc, >, old_alloc);
arr[alloc - 1] = 42;
old_alloc = alloc;
reftable_set_alloc(NULL, realloc_stub, NULL);
REFTABLE_ALLOC_GROW_OR_NULL(arr, old_alloc + 1, alloc);
check(arr == NULL);
check_uint(alloc, ==, 0);
reftable_set_alloc(NULL, NULL, NULL);
reftable_free(arr);
}
return test_done();
}

View File

@@ -0,0 +1,227 @@
/*
Copyright 2020 Google LLC
Use of this source code is governed by a BSD-style
license that can be found in the LICENSE file or at
https://developers.google.com/open-source/licenses/bsd
*/
#include "unit-test.h"
#include "lib-reftable.h"
#include "reftable/basics.h"
struct integer_needle_lesseq_args {
int needle;
int *haystack;
};
static int integer_needle_lesseq(size_t i, void *_args)
{
struct integer_needle_lesseq_args *args = _args;
return args->needle <= args->haystack[i];
}
static void *realloc_stub(void *p UNUSED, size_t size UNUSED)
{
return NULL;
}
void test_reftable_basics__binsearch(void)
{
int haystack[] = { 2, 4, 6, 8, 10 };
struct {
int needle;
size_t expected_idx;
} testcases[] = {
{-9000, 0},
{-1, 0},
{0, 0},
{2, 0},
{3, 1},
{4, 1},
{7, 3},
{9, 4},
{10, 4},
{11, 5},
{9000, 5},
};
for (size_t i = 0; i < ARRAY_SIZE(testcases); i++) {
struct integer_needle_lesseq_args args = {
.haystack = haystack,
.needle = testcases[i].needle,
};
size_t idx;
idx = binsearch(ARRAY_SIZE(haystack),
&integer_needle_lesseq, &args);
cl_assert_equal_i(idx, testcases[i].expected_idx);
}
}
void test_reftable_basics__names_length(void)
{
const char *a[] = { "a", "b", NULL };
cl_assert_equal_i(names_length(a), 2);
}
void test_reftable_basics__names_equal(void)
{
const char *a[] = { "a", "b", "c", NULL };
const char *b[] = { "a", "b", "d", NULL };
const char *c[] = { "a", "b", NULL };
cl_assert(names_equal(a, a));
cl_assert(!names_equal(a, b));
cl_assert(!names_equal(a, c));
}
void test_reftable_basics__parse_names(void)
{
char in1[] = "line\n";
char in2[] = "a\nb\nc";
char **out = parse_names(in1, strlen(in1));
cl_assert(out != NULL);
cl_assert_equal_s(out[0], "line");
cl_assert(!out[1]);
free_names(out);
out = parse_names(in2, strlen(in2));
cl_assert(out != NULL);
cl_assert_equal_s(out[0], "a");
cl_assert_equal_s(out[1], "b");
cl_assert_equal_s(out[2], "c");
cl_assert(!out[3]);
free_names(out);
}
void test_reftable_basics__parse_names_drop_empty_string(void)
{
char in[] = "a\n\nb\n";
char **out = parse_names(in, strlen(in));
cl_assert(out != NULL);
cl_assert_equal_s(out[0], "a");
/* simply '\n' should be dropped as empty string */
cl_assert_equal_s(out[1], "b");
cl_assert(out[2] == NULL);
free_names(out);
}
void test_reftable_basics__common_prefix_size(void)
{
struct reftable_buf a = REFTABLE_BUF_INIT;
struct reftable_buf b = REFTABLE_BUF_INIT;
struct {
const char *a, *b;
int want;
} cases[] = {
{"abcdef", "abc", 3},
{ "abc", "ab", 2 },
{ "", "abc", 0 },
{ "abc", "abd", 2 },
{ "abc", "pqr", 0 },
};
for (size_t i = 0; i < ARRAY_SIZE(cases); i++) {
cl_assert_equal_i(reftable_buf_addstr(&a, cases[i].a), 0);
cl_assert_equal_i(reftable_buf_addstr(&b, cases[i].b), 0);
cl_assert_equal_i(common_prefix_size(&a, &b), cases[i].want);
reftable_buf_reset(&a);
reftable_buf_reset(&b);
}
reftable_buf_release(&a);
reftable_buf_release(&b);
}
void test_reftable_basics__put_get_be64(void)
{
uint64_t in = 0x1122334455667788;
uint8_t dest[8];
uint64_t out;
reftable_put_be64(dest, in);
out = reftable_get_be64(dest);
cl_assert(in == out);
}
void test_reftable_basics__put_get_be32(void)
{
uint32_t in = 0x11223344;
uint8_t dest[4];
uint32_t out;
reftable_put_be32(dest, in);
out = reftable_get_be32(dest);
cl_assert_equal_i(in, out);
}
void test_reftable_basics__put_get_be24(void)
{
uint32_t in = 0x112233;
uint8_t dest[3];
uint32_t out;
reftable_put_be24(dest, in);
out = reftable_get_be24(dest);
cl_assert_equal_i(in, out);
}
void test_reftable_basics__put_get_be16(void)
{
uint32_t in = 0xfef1;
uint8_t dest[3];
uint32_t out;
reftable_put_be16(dest, in);
out = reftable_get_be16(dest);
cl_assert_equal_i(in, out);
}
void test_reftable_basics__alloc_grow(void)
{
int *arr = NULL, *old_arr;
size_t alloc = 0, old_alloc;
cl_assert_equal_i(REFTABLE_ALLOC_GROW(arr, 1, alloc), 0);
cl_assert(arr != NULL);
cl_assert(alloc >= 1);
arr[0] = 42;
old_alloc = alloc;
old_arr = arr;
reftable_set_alloc(NULL, realloc_stub, NULL);
cl_assert(REFTABLE_ALLOC_GROW(arr, old_alloc + 1, alloc));
cl_assert(arr == old_arr);
cl_assert_equal_i(alloc, old_alloc);
old_alloc = alloc;
reftable_set_alloc(NULL, NULL, NULL);
cl_assert_equal_i(REFTABLE_ALLOC_GROW(arr, old_alloc + 1, alloc), 0);
cl_assert(arr != NULL);
cl_assert(alloc > old_alloc);
arr[alloc - 1] = 42;
reftable_free(arr);
}
void test_reftable_basics__alloc_grow_or_null(void)
{
int *arr = NULL;
size_t alloc = 0, old_alloc;
REFTABLE_ALLOC_GROW_OR_NULL(arr, 1, alloc);
cl_assert(arr != NULL);
cl_assert(alloc >= 1);
arr[0] = 42;
old_alloc = alloc;
REFTABLE_ALLOC_GROW_OR_NULL(arr, old_alloc + 1, alloc);
cl_assert(arr != NULL);
cl_assert(alloc > old_alloc);
arr[alloc - 1] = 42;
old_alloc = alloc;
reftable_set_alloc(NULL, realloc_stub, NULL);
REFTABLE_ALLOC_GROW_OR_NULL(arr, old_alloc + 1, alloc);
cl_assert(arr == NULL);
cl_assert_equal_i(alloc, 0);
reftable_set_alloc(NULL, NULL, NULL);
reftable_free(arr);
}

View File

@@ -6,14 +6,15 @@ license that can be found in the LICENSE file or at
https://developers.google.com/open-source/licenses/bsd https://developers.google.com/open-source/licenses/bsd
*/ */
#include "test-lib.h" #include "unit-test.h"
#include "lib-reftable.h"
#include "reftable/block.h" #include "reftable/block.h"
#include "reftable/blocksource.h" #include "reftable/blocksource.h"
#include "reftable/constants.h" #include "reftable/constants.h"
#include "reftable/reftable-error.h" #include "reftable/reftable-error.h"
#include "strbuf.h" #include "strbuf.h"
static void t_ref_block_read_write(void) void test_reftable_block__read_write(void)
{ {
const int header_off = 21; /* random */ const int header_off = 21; /* random */
struct reftable_record recs[30]; struct reftable_record recs[30];
@@ -34,17 +35,18 @@ static void t_ref_block_read_write(void)
struct reftable_buf block_data = REFTABLE_BUF_INIT; struct reftable_buf block_data = REFTABLE_BUF_INIT;
REFTABLE_CALLOC_ARRAY(block_data.buf, block_size); REFTABLE_CALLOC_ARRAY(block_data.buf, block_size);
check(block_data.buf != NULL); cl_assert(block_data.buf != NULL);
block_data.len = block_size; block_data.len = block_size;
ret = block_writer_init(&bw, REFTABLE_BLOCK_TYPE_REF, (uint8_t *) block_data.buf, block_size, ret = block_writer_init(&bw, REFTABLE_BLOCK_TYPE_REF,
(uint8_t *) block_data.buf, block_size,
header_off, hash_size(REFTABLE_HASH_SHA1)); header_off, hash_size(REFTABLE_HASH_SHA1));
check(!ret); cl_assert(!ret);
rec.u.ref.refname = (char *) ""; rec.u.ref.refname = (char *) "";
rec.u.ref.value_type = REFTABLE_REF_DELETION; rec.u.ref.value_type = REFTABLE_REF_DELETION;
ret = block_writer_add(&bw, &rec); ret = block_writer_add(&bw, &rec);
check_int(ret, ==, REFTABLE_API_ERROR); cl_assert_equal_i(ret, REFTABLE_API_ERROR);
for (i = 0; i < N; i++) { for (i = 0; i < N; i++) {
rec.u.ref.refname = xstrfmt("branch%02"PRIuMAX, (uintmax_t)i); rec.u.ref.refname = xstrfmt("branch%02"PRIuMAX, (uintmax_t)i);
@@ -55,11 +57,11 @@ static void t_ref_block_read_write(void)
ret = block_writer_add(&bw, &rec); ret = block_writer_add(&bw, &rec);
rec.u.ref.refname = NULL; rec.u.ref.refname = NULL;
rec.u.ref.value_type = REFTABLE_REF_DELETION; rec.u.ref.value_type = REFTABLE_REF_DELETION;
check_int(ret, ==, 0); cl_assert_equal_i(ret, 0);
} }
ret = block_writer_finish(&bw); ret = block_writer_finish(&bw);
check_int(ret, >, 0); cl_assert(ret > 0);
block_writer_release(&bw); block_writer_release(&bw);
@@ -71,32 +73,32 @@ static void t_ref_block_read_write(void)
for (i = 0; ; i++) { for (i = 0; ; i++) {
ret = block_iter_next(&it, &rec); ret = block_iter_next(&it, &rec);
check_int(ret, >=, 0); cl_assert(ret >= 0);
if (ret > 0) { if (ret > 0) {
check_int(i, ==, N); cl_assert_equal_i(i, N);
break; break;
} }
check(reftable_record_equal(&recs[i], &rec, REFTABLE_HASH_SIZE_SHA1)); cl_assert_equal_i(reftable_record_equal(&recs[i], &rec, REFTABLE_HASH_SIZE_SHA1), 1);
} }
for (i = 0; i < N; i++) { for (i = 0; i < N; i++) {
reftable_record_key(&recs[i], &want); reftable_record_key(&recs[i], &want);
ret = block_iter_seek_key(&it, &want); ret = block_iter_seek_key(&it, &want);
check_int(ret, ==, 0); cl_assert_equal_i(ret, 0);
ret = block_iter_next(&it, &rec); ret = block_iter_next(&it, &rec);
check_int(ret, ==, 0); cl_assert_equal_i(ret, 0);
check(reftable_record_equal(&recs[i], &rec, REFTABLE_HASH_SIZE_SHA1)); cl_assert_equal_i(reftable_record_equal(&recs[i], &rec, REFTABLE_HASH_SIZE_SHA1), 1);
want.len--; want.len--;
ret = block_iter_seek_key(&it, &want); ret = block_iter_seek_key(&it, &want);
check_int(ret, ==, 0); cl_assert_equal_i(ret, 0);
ret = block_iter_next(&it, &rec); ret = block_iter_next(&it, &rec);
check_int(ret, ==, 0); cl_assert_equal_i(ret, 0);
check(reftable_record_equal(&recs[10 * (i / 10)], &rec, REFTABLE_HASH_SIZE_SHA1)); cl_assert_equal_i(reftable_record_equal(&recs[10 * (i / 10)], &rec, REFTABLE_HASH_SIZE_SHA1), 1);
} }
reftable_block_release(&block); reftable_block_release(&block);
@@ -108,7 +110,7 @@ static void t_ref_block_read_write(void)
reftable_record_release(&recs[i]); reftable_record_release(&recs[i]);
} }
static void t_log_block_read_write(void) void test_reftable_block__log_read_write(void)
{ {
const int header_off = 21; const int header_off = 21;
struct reftable_record recs[30]; struct reftable_record recs[30];
@@ -129,12 +131,12 @@ static void t_log_block_read_write(void)
struct reftable_buf block_data = REFTABLE_BUF_INIT; struct reftable_buf block_data = REFTABLE_BUF_INIT;
REFTABLE_CALLOC_ARRAY(block_data.buf, block_size); REFTABLE_CALLOC_ARRAY(block_data.buf, block_size);
check(block_data.buf != NULL); cl_assert(block_data.buf != NULL);
block_data.len = block_size; block_data.len = block_size;
ret = block_writer_init(&bw, REFTABLE_BLOCK_TYPE_LOG, (uint8_t *) block_data.buf, block_size, ret = block_writer_init(&bw, REFTABLE_BLOCK_TYPE_LOG, (uint8_t *) block_data.buf, block_size,
header_off, hash_size(REFTABLE_HASH_SHA1)); header_off, hash_size(REFTABLE_HASH_SHA1));
check(!ret); cl_assert(!ret);
for (i = 0; i < N; i++) { for (i = 0; i < N; i++) {
rec.u.log.refname = xstrfmt("branch%02"PRIuMAX , (uintmax_t)i); rec.u.log.refname = xstrfmt("branch%02"PRIuMAX , (uintmax_t)i);
@@ -145,11 +147,11 @@ static void t_log_block_read_write(void)
ret = block_writer_add(&bw, &rec); ret = block_writer_add(&bw, &rec);
rec.u.log.refname = NULL; rec.u.log.refname = NULL;
rec.u.log.value_type = REFTABLE_LOG_DELETION; rec.u.log.value_type = REFTABLE_LOG_DELETION;
check_int(ret, ==, 0); cl_assert_equal_i(ret, 0);
} }
ret = block_writer_finish(&bw); ret = block_writer_finish(&bw);
check_int(ret, >, 0); cl_assert(ret > 0);
block_writer_release(&bw); block_writer_release(&bw);
@@ -161,33 +163,33 @@ static void t_log_block_read_write(void)
for (i = 0; ; i++) { for (i = 0; ; i++) {
ret = block_iter_next(&it, &rec); ret = block_iter_next(&it, &rec);
check_int(ret, >=, 0); cl_assert(ret >= 0);
if (ret > 0) { if (ret > 0) {
check_int(i, ==, N); cl_assert_equal_i(i, N);
break; break;
} }
check(reftable_record_equal(&recs[i], &rec, REFTABLE_HASH_SIZE_SHA1)); cl_assert_equal_i(reftable_record_equal(&recs[i], &rec, REFTABLE_HASH_SIZE_SHA1), 1);
} }
for (i = 0; i < N; i++) { for (i = 0; i < N; i++) {
reftable_buf_reset(&want); reftable_buf_reset(&want);
check(!reftable_buf_addstr(&want, recs[i].u.log.refname)); cl_assert(reftable_buf_addstr(&want, recs[i].u.log.refname) == 0);
ret = block_iter_seek_key(&it, &want); ret = block_iter_seek_key(&it, &want);
check_int(ret, ==, 0); cl_assert_equal_i(ret, 0);
ret = block_iter_next(&it, &rec); ret = block_iter_next(&it, &rec);
check_int(ret, ==, 0); cl_assert_equal_i(ret, 0);
check(reftable_record_equal(&recs[i], &rec, REFTABLE_HASH_SIZE_SHA1)); cl_assert_equal_i(reftable_record_equal(&recs[i], &rec, REFTABLE_HASH_SIZE_SHA1), 1);
want.len--; want.len--;
ret = block_iter_seek_key(&it, &want); ret = block_iter_seek_key(&it, &want);
check_int(ret, ==, 0); cl_assert_equal_i(ret, 0);
ret = block_iter_next(&it, &rec); ret = block_iter_next(&it, &rec);
check_int(ret, ==, 0); cl_assert_equal_i(ret, 0);
check(reftable_record_equal(&recs[10 * (i / 10)], &rec, REFTABLE_HASH_SIZE_SHA1)); cl_assert_equal_i(reftable_record_equal(&recs[10 * (i / 10)], &rec, REFTABLE_HASH_SIZE_SHA1), 1);
} }
reftable_block_release(&block); reftable_block_release(&block);
@@ -199,7 +201,7 @@ static void t_log_block_read_write(void)
reftable_record_release(&recs[i]); reftable_record_release(&recs[i]);
} }
static void t_obj_block_read_write(void) void test_reftable_block__obj_read_write(void)
{ {
const int header_off = 21; const int header_off = 21;
struct reftable_record recs[30]; struct reftable_record recs[30];
@@ -220,12 +222,12 @@ static void t_obj_block_read_write(void)
struct reftable_buf block_data = REFTABLE_BUF_INIT; struct reftable_buf block_data = REFTABLE_BUF_INIT;
REFTABLE_CALLOC_ARRAY(block_data.buf, block_size); REFTABLE_CALLOC_ARRAY(block_data.buf, block_size);
check(block_data.buf != NULL); cl_assert(block_data.buf != NULL);
block_data.len = block_size; block_data.len = block_size;
ret = block_writer_init(&bw, REFTABLE_BLOCK_TYPE_OBJ, (uint8_t *) block_data.buf, block_size, ret = block_writer_init(&bw, REFTABLE_BLOCK_TYPE_OBJ, (uint8_t *) block_data.buf, block_size,
header_off, hash_size(REFTABLE_HASH_SHA1)); header_off, hash_size(REFTABLE_HASH_SHA1));
check(!ret); cl_assert(!ret);
for (i = 0; i < N; i++) { for (i = 0; i < N; i++) {
uint8_t bytes[] = { i, i + 1, i + 2, i + 3, i + 5 }, *allocated; uint8_t bytes[] = { i, i + 1, i + 2, i + 3, i + 5 }, *allocated;
@@ -238,11 +240,11 @@ static void t_obj_block_read_write(void)
ret = block_writer_add(&bw, &rec); ret = block_writer_add(&bw, &rec);
rec.u.obj.hash_prefix = NULL; rec.u.obj.hash_prefix = NULL;
rec.u.obj.hash_prefix_len = 0; rec.u.obj.hash_prefix_len = 0;
check_int(ret, ==, 0); cl_assert_equal_i(ret, 0);
} }
ret = block_writer_finish(&bw); ret = block_writer_finish(&bw);
check_int(ret, >, 0); cl_assert(ret > 0);
block_writer_release(&bw); block_writer_release(&bw);
@@ -254,24 +256,24 @@ static void t_obj_block_read_write(void)
for (i = 0; ; i++) { for (i = 0; ; i++) {
ret = block_iter_next(&it, &rec); ret = block_iter_next(&it, &rec);
check_int(ret, >=, 0); cl_assert(ret >= 0);
if (ret > 0) { if (ret > 0) {
check_int(i, ==, N); cl_assert_equal_i(i, N);
break; break;
} }
check(reftable_record_equal(&recs[i], &rec, REFTABLE_HASH_SIZE_SHA1)); cl_assert_equal_i(reftable_record_equal(&recs[i], &rec, REFTABLE_HASH_SIZE_SHA1), 1);
} }
for (i = 0; i < N; i++) { for (i = 0; i < N; i++) {
reftable_record_key(&recs[i], &want); reftable_record_key(&recs[i], &want);
ret = block_iter_seek_key(&it, &want); ret = block_iter_seek_key(&it, &want);
check_int(ret, ==, 0); cl_assert_equal_i(ret, 0);
ret = block_iter_next(&it, &rec); ret = block_iter_next(&it, &rec);
check_int(ret, ==, 0); cl_assert_equal_i(ret, 0);
check(reftable_record_equal(&recs[i], &rec, REFTABLE_HASH_SIZE_SHA1)); cl_assert_equal_i(reftable_record_equal(&recs[i], &rec, REFTABLE_HASH_SIZE_SHA1), 1);
} }
reftable_block_release(&block); reftable_block_release(&block);
@@ -283,7 +285,7 @@ static void t_obj_block_read_write(void)
reftable_record_release(&recs[i]); reftable_record_release(&recs[i]);
} }
static void t_index_block_read_write(void) void test_reftable_block__ref_read_write(void)
{ {
const int header_off = 21; const int header_off = 21;
struct reftable_record recs[30]; struct reftable_record recs[30];
@@ -305,12 +307,12 @@ static void t_index_block_read_write(void)
struct reftable_buf block_data = REFTABLE_BUF_INIT; struct reftable_buf block_data = REFTABLE_BUF_INIT;
REFTABLE_CALLOC_ARRAY(block_data.buf, block_size); REFTABLE_CALLOC_ARRAY(block_data.buf, block_size);
check(block_data.buf != NULL); cl_assert(block_data.buf != NULL);
block_data.len = block_size; block_data.len = block_size;
ret = block_writer_init(&bw, REFTABLE_BLOCK_TYPE_INDEX, (uint8_t *) block_data.buf, block_size, ret = block_writer_init(&bw, REFTABLE_BLOCK_TYPE_INDEX, (uint8_t *) block_data.buf, block_size,
header_off, hash_size(REFTABLE_HASH_SHA1)); header_off, hash_size(REFTABLE_HASH_SHA1));
check(!ret); cl_assert(!ret);
for (i = 0; i < N; i++) { for (i = 0; i < N; i++) {
char buf[128]; char buf[128];
@@ -319,15 +321,15 @@ static void t_index_block_read_write(void)
reftable_buf_init(&recs[i].u.idx.last_key); reftable_buf_init(&recs[i].u.idx.last_key);
recs[i].type = REFTABLE_BLOCK_TYPE_INDEX; recs[i].type = REFTABLE_BLOCK_TYPE_INDEX;
check(!reftable_buf_addstr(&recs[i].u.idx.last_key, buf)); cl_assert(!reftable_buf_addstr(&recs[i].u.idx.last_key, buf));
recs[i].u.idx.offset = i; recs[i].u.idx.offset = i;
ret = block_writer_add(&bw, &recs[i]); ret = block_writer_add(&bw, &recs[i]);
check_int(ret, ==, 0); cl_assert_equal_i(ret, 0);
} }
ret = block_writer_finish(&bw); ret = block_writer_finish(&bw);
check_int(ret, >, 0); cl_assert(ret > 0);
block_writer_release(&bw); block_writer_release(&bw);
@@ -339,32 +341,32 @@ static void t_index_block_read_write(void)
for (i = 0; ; i++) { for (i = 0; ; i++) {
ret = block_iter_next(&it, &rec); ret = block_iter_next(&it, &rec);
check_int(ret, >=, 0); cl_assert(ret >= 0);
if (ret > 0) { if (ret > 0) {
check_int(i, ==, N); cl_assert_equal_i(i, N);
break; break;
} }
check(reftable_record_equal(&recs[i], &rec, REFTABLE_HASH_SIZE_SHA1)); cl_assert_equal_i(reftable_record_equal(&recs[i], &rec, REFTABLE_HASH_SIZE_SHA1), 1);
} }
for (i = 0; i < N; i++) { for (i = 0; i < N; i++) {
reftable_record_key(&recs[i], &want); reftable_record_key(&recs[i], &want);
ret = block_iter_seek_key(&it, &want); ret = block_iter_seek_key(&it, &want);
check_int(ret, ==, 0); cl_assert_equal_i(ret, 0);
ret = block_iter_next(&it, &rec); ret = block_iter_next(&it, &rec);
check_int(ret, ==, 0); cl_assert_equal_i(ret, 0);
check(reftable_record_equal(&recs[i], &rec, REFTABLE_HASH_SIZE_SHA1)); cl_assert_equal_i(reftable_record_equal(&recs[i], &rec, REFTABLE_HASH_SIZE_SHA1), 1);
want.len--; want.len--;
ret = block_iter_seek_key(&it, &want); ret = block_iter_seek_key(&it, &want);
check_int(ret, ==, 0); cl_assert_equal_i(ret, 0);
ret = block_iter_next(&it, &rec); ret = block_iter_next(&it, &rec);
check_int(ret, ==, 0); cl_assert_equal_i(ret, 0);
check(reftable_record_equal(&recs[10 * (i / 10)], &rec, REFTABLE_HASH_SIZE_SHA1)); cl_assert_equal_i(reftable_record_equal(&recs[10 * (i / 10)], &rec, REFTABLE_HASH_SIZE_SHA1), 1);
} }
reftable_block_release(&block); reftable_block_release(&block);
@@ -376,7 +378,7 @@ static void t_index_block_read_write(void)
reftable_record_release(&recs[i]); reftable_record_release(&recs[i]);
} }
static void t_block_iterator(void) void test_reftable_block__iterator(void)
{ {
struct reftable_block_source source = { 0 }; struct reftable_block_source source = { 0 };
struct block_writer writer = { struct block_writer writer = {
@@ -391,11 +393,12 @@ static void t_block_iterator(void)
data.len = 1024; data.len = 1024;
REFTABLE_CALLOC_ARRAY(data.buf, data.len); REFTABLE_CALLOC_ARRAY(data.buf, data.len);
check(data.buf != NULL); cl_assert(data.buf != NULL);
err = block_writer_init(&writer, REFTABLE_BLOCK_TYPE_REF, (uint8_t *) data.buf, data.len, err = block_writer_init(&writer, REFTABLE_BLOCK_TYPE_REF,
(uint8_t *) data.buf, data.len,
0, hash_size(REFTABLE_HASH_SHA1)); 0, hash_size(REFTABLE_HASH_SHA1));
check(!err); cl_assert(!err);
for (size_t i = 0; i < ARRAY_SIZE(expected_refs); i++) { for (size_t i = 0; i < ARRAY_SIZE(expected_refs); i++) {
expected_refs[i] = (struct reftable_record) { expected_refs[i] = (struct reftable_record) {
@@ -408,42 +411,42 @@ static void t_block_iterator(void)
memset(expected_refs[i].u.ref.value.val1, i, REFTABLE_HASH_SIZE_SHA1); memset(expected_refs[i].u.ref.value.val1, i, REFTABLE_HASH_SIZE_SHA1);
err = block_writer_add(&writer, &expected_refs[i]); err = block_writer_add(&writer, &expected_refs[i]);
check_int(err, ==, 0); cl_assert_equal_i(err, 0);
} }
err = block_writer_finish(&writer); err = block_writer_finish(&writer);
check_int(err, >, 0); cl_assert(err > 0);
block_source_from_buf(&source, &data); block_source_from_buf(&source, &data);
reftable_block_init(&block, &source, 0, 0, data.len, reftable_block_init(&block, &source, 0, 0, data.len,
REFTABLE_HASH_SIZE_SHA1, REFTABLE_BLOCK_TYPE_REF); REFTABLE_HASH_SIZE_SHA1, REFTABLE_BLOCK_TYPE_REF);
err = reftable_block_init_iterator(&block, &it); err = reftable_block_init_iterator(&block, &it);
check_int(err, ==, 0); cl_assert_equal_i(err, 0);
for (size_t i = 0; ; i++) { for (size_t i = 0; ; i++) {
err = reftable_iterator_next_ref(&it, &ref); err = reftable_iterator_next_ref(&it, &ref);
if (err > 0) { if (err > 0) {
check_int(i, ==, ARRAY_SIZE(expected_refs)); cl_assert_equal_i(i, ARRAY_SIZE(expected_refs));
break; break;
} }
check_int(err, ==, 0); cl_assert_equal_i(err, 0);
check(reftable_ref_record_equal(&ref, &expected_refs[i].u.ref, cl_assert(reftable_ref_record_equal(&ref,
REFTABLE_HASH_SIZE_SHA1)); &expected_refs[i].u.ref, REFTABLE_HASH_SIZE_SHA1));
} }
err = reftable_iterator_seek_ref(&it, "refs/heads/does-not-exist"); err = reftable_iterator_seek_ref(&it, "refs/heads/does-not-exist");
check_int(err, ==, 0); cl_assert_equal_i(err, 0);
err = reftable_iterator_next_ref(&it, &ref); err = reftable_iterator_next_ref(&it, &ref);
check_int(err, ==, 1); cl_assert_equal_i(err, 1);
err = reftable_iterator_seek_ref(&it, "refs/heads/branch-13"); err = reftable_iterator_seek_ref(&it, "refs/heads/branch-13");
check_int(err, ==, 0); cl_assert_equal_i(err, 0);
err = reftable_iterator_next_ref(&it, &ref); err = reftable_iterator_next_ref(&it, &ref);
check_int(err, ==, 0); cl_assert_equal_i(err, 0);
check(reftable_ref_record_equal(&ref, &expected_refs[13].u.ref, cl_assert(reftable_ref_record_equal(&ref,
REFTABLE_HASH_SIZE_SHA1)); &expected_refs[13].u.ref,REFTABLE_HASH_SIZE_SHA1));
for (size_t i = 0; i < ARRAY_SIZE(expected_refs); i++) for (size_t i = 0; i < ARRAY_SIZE(expected_refs); i++)
reftable_free(expected_refs[i].u.ref.refname); reftable_free(expected_refs[i].u.ref.refname);
@@ -453,14 +456,3 @@ static void t_block_iterator(void)
block_writer_release(&writer); block_writer_release(&writer);
reftable_buf_release(&data); reftable_buf_release(&data);
} }
int cmd_main(int argc UNUSED, const char *argv[] UNUSED)
{
TEST(t_index_block_read_write(), "read-write operations on index blocks work");
TEST(t_log_block_read_write(), "read-write operations on log blocks work");
TEST(t_obj_block_read_write(), "read-write operations on obj blocks work");
TEST(t_ref_block_read_write(), "read-write operations on ref blocks work");
TEST(t_block_iterator(), "block iterator works");
return test_done();
}

View File

@@ -6,7 +6,7 @@ license that can be found in the LICENSE file or at
https://developers.google.com/open-source/licenses/bsd https://developers.google.com/open-source/licenses/bsd
*/ */
#include "test-lib.h" #include "unit-test.h"
#include "lib-reftable.h" #include "lib-reftable.h"
#include "reftable/blocksource.h" #include "reftable/blocksource.h"
#include "reftable/constants.h" #include "reftable/constants.h"
@@ -29,21 +29,21 @@ merged_table_from_records(struct reftable_ref_record **refs,
int err; int err;
REFTABLE_CALLOC_ARRAY(*tables, n); REFTABLE_CALLOC_ARRAY(*tables, n);
check(*tables != NULL); cl_assert(*tables != NULL);
REFTABLE_CALLOC_ARRAY(*source, n); REFTABLE_CALLOC_ARRAY(*source, n);
check(*source != NULL); cl_assert(*source != NULL);
for (size_t i = 0; i < n; i++) { for (size_t i = 0; i < n; i++) {
t_reftable_write_to_buf(&buf[i], refs[i], sizes[i], NULL, 0, &opts); cl_reftable_write_to_buf(&buf[i], refs[i], sizes[i], NULL, 0, &opts);
block_source_from_buf(&(*source)[i], &buf[i]); block_source_from_buf(&(*source)[i], &buf[i]);
err = reftable_table_new(&(*tables)[i], &(*source)[i], err = reftable_table_new(&(*tables)[i], &(*source)[i],
"name"); "name");
check(!err); cl_assert(!err);
} }
err = reftable_merged_table_new(&mt, *tables, n, REFTABLE_HASH_SHA1); err = reftable_merged_table_new(&mt, *tables, n, REFTABLE_HASH_SHA1);
check(!err); cl_assert(!err);
return mt; return mt;
} }
@@ -54,7 +54,7 @@ static void tables_destroy(struct reftable_table **tables, const size_t n)
reftable_free(tables); reftable_free(tables);
} }
static void t_merged_single_record(void) void test_reftable_merged__single_record(void)
{ {
struct reftable_ref_record r1[] = { { struct reftable_ref_record r1[] = { {
.refname = (char *) "b", .refname = (char *) "b",
@@ -85,13 +85,14 @@ static void t_merged_single_record(void)
int err; int err;
err = merged_table_init_iter(mt, &it, REFTABLE_BLOCK_TYPE_REF); err = merged_table_init_iter(mt, &it, REFTABLE_BLOCK_TYPE_REF);
check(!err); cl_assert(!err);
err = reftable_iterator_seek_ref(&it, "a"); err = reftable_iterator_seek_ref(&it, "a");
check(!err); cl_assert(!err);
err = reftable_iterator_next_ref(&it, &ref); err = reftable_iterator_next_ref(&it, &ref);
check(!err); cl_assert(!err);
check(reftable_ref_record_equal(&r2[0], &ref, REFTABLE_HASH_SIZE_SHA1)); cl_assert(reftable_ref_record_equal(&r2[0], &ref,
REFTABLE_HASH_SIZE_SHA1) != 0);
reftable_ref_record_release(&ref); reftable_ref_record_release(&ref);
reftable_iterator_destroy(&it); reftable_iterator_destroy(&it);
tables_destroy(tables, 3); tables_destroy(tables, 3);
@@ -101,7 +102,7 @@ static void t_merged_single_record(void)
reftable_free(bs); reftable_free(bs);
} }
static void t_merged_refs(void) void test_reftable_merged__refs(void)
{ {
struct reftable_ref_record r1[] = { struct reftable_ref_record r1[] = {
{ {
@@ -165,12 +166,12 @@ static void t_merged_refs(void)
size_t i; size_t i;
err = merged_table_init_iter(mt, &it, REFTABLE_BLOCK_TYPE_REF); err = merged_table_init_iter(mt, &it, REFTABLE_BLOCK_TYPE_REF);
check(!err); cl_assert(!err);
err = reftable_iterator_seek_ref(&it, "a"); err = reftable_iterator_seek_ref(&it, "a");
check(!err); cl_assert(err == 0);
check_int(reftable_merged_table_hash_id(mt), ==, REFTABLE_HASH_SHA1); cl_assert_equal_i(reftable_merged_table_hash_id(mt), REFTABLE_HASH_SHA1);
check_int(reftable_merged_table_min_update_index(mt), ==, 1); cl_assert_equal_i(reftable_merged_table_min_update_index(mt), 1);
check_int(reftable_merged_table_max_update_index(mt), ==, 3); cl_assert_equal_i(reftable_merged_table_max_update_index(mt), 3);
while (len < 100) { /* cap loops/recursion. */ while (len < 100) { /* cap loops/recursion. */
struct reftable_ref_record ref = { 0 }; struct reftable_ref_record ref = { 0 };
@@ -178,15 +179,15 @@ static void t_merged_refs(void)
if (err > 0) if (err > 0)
break; break;
check(!REFTABLE_ALLOC_GROW(out, len + 1, cap)); cl_assert(REFTABLE_ALLOC_GROW(out, len + 1, cap) == 0);
out[len++] = ref; out[len++] = ref;
} }
reftable_iterator_destroy(&it); reftable_iterator_destroy(&it);
check_int(ARRAY_SIZE(want), ==, len); cl_assert_equal_i(ARRAY_SIZE(want), len);
for (i = 0; i < len; i++) for (i = 0; i < len; i++)
check(reftable_ref_record_equal(want[i], &out[i], cl_assert(reftable_ref_record_equal(want[i], &out[i],
REFTABLE_HASH_SIZE_SHA1)); REFTABLE_HASH_SIZE_SHA1) != 0);
for (i = 0; i < len; i++) for (i = 0; i < len; i++)
reftable_ref_record_release(&out[i]); reftable_ref_record_release(&out[i]);
reftable_free(out); reftable_free(out);
@@ -198,7 +199,7 @@ static void t_merged_refs(void)
reftable_free(bs); reftable_free(bs);
} }
static void t_merged_seek_multiple_times(void) void test_reftable_merged__seek_multiple_times(void)
{ {
struct reftable_ref_record r1[] = { struct reftable_ref_record r1[] = {
{ {
@@ -248,20 +249,17 @@ static void t_merged_seek_multiple_times(void)
for (size_t i = 0; i < 5; i++) { for (size_t i = 0; i < 5; i++) {
int err = reftable_iterator_seek_ref(&it, "c"); int err = reftable_iterator_seek_ref(&it, "c");
check(!err); cl_assert(!err);
err = reftable_iterator_next_ref(&it, &rec); cl_assert(reftable_iterator_next_ref(&it, &rec) == 0);
check(!err); cl_assert_equal_i(reftable_ref_record_equal(&rec, &r1[1],
err = reftable_ref_record_equal(&rec, &r1[1], REFTABLE_HASH_SIZE_SHA1); REFTABLE_HASH_SIZE_SHA1), 1);
check(err == 1);
err = reftable_iterator_next_ref(&it, &rec); cl_assert(reftable_iterator_next_ref(&it, &rec) == 0);
check(!err); cl_assert_equal_i(reftable_ref_record_equal(&rec, &r2[1],
err = reftable_ref_record_equal(&rec, &r2[1], REFTABLE_HASH_SIZE_SHA1); REFTABLE_HASH_SIZE_SHA1), 1);
check(err == 1);
err = reftable_iterator_next_ref(&it, &rec); cl_assert(reftable_iterator_next_ref(&it, &rec) > 0);
check(err > 0);
} }
for (size_t i = 0; i < ARRAY_SIZE(bufs); i++) for (size_t i = 0; i < ARRAY_SIZE(bufs); i++)
@@ -273,7 +271,7 @@ static void t_merged_seek_multiple_times(void)
reftable_free(sources); reftable_free(sources);
} }
static void t_merged_seek_multiple_times_without_draining(void) void test_reftable_merged__seek_multiple_times_no_drain(void)
{ {
struct reftable_ref_record r1[] = { struct reftable_ref_record r1[] = {
{ {
@@ -317,24 +315,19 @@ static void t_merged_seek_multiple_times_without_draining(void)
struct reftable_ref_record rec = { 0 }; struct reftable_ref_record rec = { 0 };
struct reftable_iterator it = { 0 }; struct reftable_iterator it = { 0 };
struct reftable_merged_table *mt; struct reftable_merged_table *mt;
int err;
mt = merged_table_from_records(refs, &sources, &tables, sizes, bufs, 2); mt = merged_table_from_records(refs, &sources, &tables, sizes, bufs, 2);
merged_table_init_iter(mt, &it, REFTABLE_BLOCK_TYPE_REF); merged_table_init_iter(mt, &it, REFTABLE_BLOCK_TYPE_REF);
err = reftable_iterator_seek_ref(&it, "b"); cl_assert(reftable_iterator_seek_ref(&it, "b") == 0);
check(!err); cl_assert(reftable_iterator_next_ref(&it, &rec) == 0);
err = reftable_iterator_next_ref(&it, &rec); cl_assert_equal_i(reftable_ref_record_equal(&rec, &r2[0],
check(!err); REFTABLE_HASH_SIZE_SHA1), 1);
err = reftable_ref_record_equal(&rec, &r2[0], REFTABLE_HASH_SIZE_SHA1);
check(err == 1);
err = reftable_iterator_seek_ref(&it, "a"); cl_assert(reftable_iterator_seek_ref(&it, "a") == 0);
check(!err); cl_assert(reftable_iterator_next_ref(&it, &rec) == 0);
err = reftable_iterator_next_ref(&it, &rec); cl_assert_equal_i(reftable_ref_record_equal(&rec, &r1[0],
check(!err); REFTABLE_HASH_SIZE_SHA1), 1);
err = reftable_ref_record_equal(&rec, &r1[0], REFTABLE_HASH_SIZE_SHA1);
check(err == 1);
for (size_t i = 0; i < ARRAY_SIZE(bufs); i++) for (size_t i = 0; i < ARRAY_SIZE(bufs); i++)
reftable_buf_release(&bufs[i]); reftable_buf_release(&bufs[i]);
@@ -359,25 +352,25 @@ merged_table_from_log_records(struct reftable_log_record **logs,
int err; int err;
REFTABLE_CALLOC_ARRAY(*tables, n); REFTABLE_CALLOC_ARRAY(*tables, n);
check(*tables != NULL); cl_assert(*tables != NULL);
REFTABLE_CALLOC_ARRAY(*source, n); REFTABLE_CALLOC_ARRAY(*source, n);
check(*source != NULL); cl_assert(*source != NULL);
for (size_t i = 0; i < n; i++) { for (size_t i = 0; i < n; i++) {
t_reftable_write_to_buf(&buf[i], NULL, 0, logs[i], sizes[i], &opts); cl_reftable_write_to_buf(&buf[i], NULL, 0, logs[i], sizes[i], &opts);
block_source_from_buf(&(*source)[i], &buf[i]); block_source_from_buf(&(*source)[i], &buf[i]);
err = reftable_table_new(&(*tables)[i], &(*source)[i], err = reftable_table_new(&(*tables)[i], &(*source)[i],
"name"); "name");
check(!err); cl_assert(!err);
} }
err = reftable_merged_table_new(&mt, *tables, n, REFTABLE_HASH_SHA1); err = reftable_merged_table_new(&mt, *tables, n, REFTABLE_HASH_SHA1);
check(!err); cl_assert(!err);
return mt; return mt;
} }
static void t_merged_logs(void) void test_reftable_merged__logs(void)
{ {
struct reftable_log_record r1[] = { struct reftable_log_record r1[] = {
{ {
@@ -439,19 +432,19 @@ static void t_merged_logs(void)
struct reftable_merged_table *mt = merged_table_from_log_records( struct reftable_merged_table *mt = merged_table_from_log_records(
logs, &bs, &tables, sizes, bufs, 3); logs, &bs, &tables, sizes, bufs, 3);
struct reftable_iterator it = { 0 }; struct reftable_iterator it = { 0 };
int err;
struct reftable_log_record *out = NULL; struct reftable_log_record *out = NULL;
size_t len = 0; size_t len = 0;
size_t cap = 0; size_t cap = 0;
size_t i; size_t i;
int err;
err = merged_table_init_iter(mt, &it, REFTABLE_BLOCK_TYPE_LOG); err = merged_table_init_iter(mt, &it, REFTABLE_BLOCK_TYPE_LOG);
check(!err); cl_assert(!err);
err = reftable_iterator_seek_log(&it, "a"); err = reftable_iterator_seek_log(&it, "a");
check(!err); cl_assert(!err);
check_int(reftable_merged_table_hash_id(mt), ==, REFTABLE_HASH_SHA1); cl_assert_equal_i(reftable_merged_table_hash_id(mt), REFTABLE_HASH_SHA1);
check_int(reftable_merged_table_min_update_index(mt), ==, 1); cl_assert_equal_i(reftable_merged_table_min_update_index(mt), 1);
check_int(reftable_merged_table_max_update_index(mt), ==, 3); cl_assert_equal_i(reftable_merged_table_max_update_index(mt), 3);
while (len < 100) { /* cap loops/recursion. */ while (len < 100) { /* cap loops/recursion. */
struct reftable_log_record log = { 0 }; struct reftable_log_record log = { 0 };
@@ -459,24 +452,24 @@ static void t_merged_logs(void)
if (err > 0) if (err > 0)
break; break;
check(!REFTABLE_ALLOC_GROW(out, len + 1, cap)); cl_assert(REFTABLE_ALLOC_GROW(out, len + 1, cap) == 0);
out[len++] = log; out[len++] = log;
} }
reftable_iterator_destroy(&it); reftable_iterator_destroy(&it);
check_int(ARRAY_SIZE(want), ==, len); cl_assert_equal_i(ARRAY_SIZE(want), len);
for (i = 0; i < len; i++) for (i = 0; i < len; i++)
check(reftable_log_record_equal(want[i], &out[i], cl_assert(reftable_log_record_equal(want[i], &out[i],
REFTABLE_HASH_SIZE_SHA1)); REFTABLE_HASH_SIZE_SHA1) != 0);
err = merged_table_init_iter(mt, &it, REFTABLE_BLOCK_TYPE_LOG); err = merged_table_init_iter(mt, &it, REFTABLE_BLOCK_TYPE_LOG);
check(!err); cl_assert(!err);
err = reftable_iterator_seek_log_at(&it, "a", 2); err = reftable_iterator_seek_log_at(&it, "a", 2);
check(!err); cl_assert(!err);
reftable_log_record_release(&out[0]); reftable_log_record_release(&out[0]);
err = reftable_iterator_next_log(&it, &out[0]); cl_assert(reftable_iterator_next_log(&it, &out[0]) == 0);
check(!err); cl_assert(reftable_log_record_equal(&out[0], &r3[0],
check(reftable_log_record_equal(&out[0], &r3[0], REFTABLE_HASH_SIZE_SHA1)); REFTABLE_HASH_SIZE_SHA1) != 0);
reftable_iterator_destroy(&it); reftable_iterator_destroy(&it);
for (i = 0; i < len; i++) for (i = 0; i < len; i++)
@@ -490,11 +483,11 @@ static void t_merged_logs(void)
reftable_free(bs); reftable_free(bs);
} }
static void t_default_write_opts(void) void test_reftable_merged__default_write_opts(void)
{ {
struct reftable_write_options opts = { 0 }; struct reftable_write_options opts = { 0 };
struct reftable_buf buf = REFTABLE_BUF_INIT; struct reftable_buf buf = REFTABLE_BUF_INIT;
struct reftable_writer *w = t_reftable_strbuf_writer(&buf, &opts); struct reftable_writer *w = cl_reftable_strbuf_writer(&buf, &opts);
struct reftable_ref_record rec = { struct reftable_ref_record rec = {
.refname = (char *) "master", .refname = (char *) "master",
.update_index = 1, .update_index = 1,
@@ -507,40 +500,25 @@ static void t_default_write_opts(void)
reftable_writer_set_limits(w, 1, 1); reftable_writer_set_limits(w, 1, 1);
err = reftable_writer_add_ref(w, &rec); cl_assert_equal_i(reftable_writer_add_ref(w, &rec), 0);
check(!err);
err = reftable_writer_close(w); cl_assert_equal_i(reftable_writer_close(w), 0);
check(!err);
reftable_writer_free(w); reftable_writer_free(w);
block_source_from_buf(&source, &buf); block_source_from_buf(&source, &buf);
err = reftable_table_new(&table, &source, "filename"); err = reftable_table_new(&table, &source, "filename");
check(!err); cl_assert(!err);
hash_id = reftable_table_hash_id(table); hash_id = reftable_table_hash_id(table);
check_int(hash_id, ==, REFTABLE_HASH_SHA1); cl_assert_equal_i(hash_id, REFTABLE_HASH_SHA1);
err = reftable_merged_table_new(&merged, &table, 1, REFTABLE_HASH_SHA256); err = reftable_merged_table_new(&merged, &table, 1, REFTABLE_HASH_SHA256);
check_int(err, ==, REFTABLE_FORMAT_ERROR); cl_assert_equal_i(err, REFTABLE_FORMAT_ERROR);
err = reftable_merged_table_new(&merged, &table, 1, REFTABLE_HASH_SHA1); err = reftable_merged_table_new(&merged, &table, 1, REFTABLE_HASH_SHA1);
check(!err); cl_assert(!err);
reftable_table_decref(table); reftable_table_decref(table);
reftable_merged_table_free(merged); reftable_merged_table_free(merged);
reftable_buf_release(&buf); reftable_buf_release(&buf);
} }
int cmd_main(int argc UNUSED, const char *argv[] UNUSED)
{
TEST(t_default_write_opts(), "merged table with default write opts");
TEST(t_merged_logs(), "merged table with multiple log updates for same ref");
TEST(t_merged_refs(), "merged table with multiple updates to same ref");
TEST(t_merged_seek_multiple_times(), "merged table can seek multiple times");
TEST(t_merged_seek_multiple_times_without_draining(), "merged table can seek multiple times without draining");
TEST(t_merged_single_record(), "ref occurring in only one record can be fetched");
return test_done();
}

View File

@@ -6,7 +6,8 @@ license that can be found in the LICENSE file or at
https://developers.google.com/open-source/licenses/bsd https://developers.google.com/open-source/licenses/bsd
*/ */
#include "test-lib.h" #include "unit-test.h"
#include "lib-reftable.h"
#include "reftable/constants.h" #include "reftable/constants.h"
#include "reftable/pq.h" #include "reftable/pq.h"
#include "strbuf.h" #include "strbuf.h"
@@ -15,18 +16,18 @@ static void merged_iter_pqueue_check(const struct merged_iter_pqueue *pq)
{ {
for (size_t i = 1; i < pq->len; i++) { for (size_t i = 1; i < pq->len; i++) {
size_t parent = (i - 1) / 2; size_t parent = (i - 1) / 2;
check(pq_less(&pq->heap[parent], &pq->heap[i])); cl_assert(pq_less(&pq->heap[parent], &pq->heap[i]) != 0);
} }
} }
static int pq_entry_equal(struct pq_entry *a, struct pq_entry *b) static int pq_entry_equal(struct pq_entry *a, struct pq_entry *b)
{ {
int cmp; int cmp;
check(!reftable_record_cmp(a->rec, b->rec, &cmp)); cl_assert_equal_i(reftable_record_cmp(a->rec, b->rec, &cmp), 0);
return !cmp && (a->index == b->index); return !cmp && (a->index == b->index);
} }
static void t_pq_record(void) void test_reftable_pq__record(void)
{ {
struct merged_iter_pqueue pq = { 0 }; struct merged_iter_pqueue pq = { 0 };
struct reftable_record recs[54]; struct reftable_record recs[54];
@@ -34,7 +35,8 @@ static void t_pq_record(void)
char *last = NULL; char *last = NULL;
for (i = 0; i < N; i++) { for (i = 0; i < N; i++) {
check(!reftable_record_init(&recs[i], REFTABLE_BLOCK_TYPE_REF)); cl_assert(!reftable_record_init(&recs[i],
REFTABLE_BLOCK_TYPE_REF));
recs[i].u.ref.refname = xstrfmt("%02"PRIuMAX, (uintmax_t)i); recs[i].u.ref.refname = xstrfmt("%02"PRIuMAX, (uintmax_t)i);
} }
@@ -53,13 +55,13 @@ static void t_pq_record(void)
struct pq_entry top = merged_iter_pqueue_top(pq); struct pq_entry top = merged_iter_pqueue_top(pq);
struct pq_entry e; struct pq_entry e;
check(!merged_iter_pqueue_remove(&pq, &e)); cl_assert_equal_i(merged_iter_pqueue_remove(&pq, &e), 0);
merged_iter_pqueue_check(&pq); merged_iter_pqueue_check(&pq);
check(pq_entry_equal(&top, &e)); cl_assert(pq_entry_equal(&top, &e));
check(reftable_record_type(e.rec) == REFTABLE_BLOCK_TYPE_REF); cl_assert(reftable_record_type(e.rec) == REFTABLE_BLOCK_TYPE_REF);
if (last) if (last)
check_int(strcmp(last, e.rec->u.ref.refname), <, 0); cl_assert(strcmp(last, e.rec->u.ref.refname) < 0);
last = e.rec->u.ref.refname; last = e.rec->u.ref.refname;
} }
@@ -68,7 +70,7 @@ static void t_pq_record(void)
merged_iter_pqueue_release(&pq); merged_iter_pqueue_release(&pq);
} }
static void t_pq_index(void) void test_reftable_pq__index(void)
{ {
struct merged_iter_pqueue pq = { 0 }; struct merged_iter_pqueue pq = { 0 };
struct reftable_record recs[13]; struct reftable_record recs[13];
@@ -76,7 +78,8 @@ static void t_pq_index(void)
size_t N = ARRAY_SIZE(recs), i; size_t N = ARRAY_SIZE(recs), i;
for (i = 0; i < N; i++) { for (i = 0; i < N; i++) {
check(!reftable_record_init(&recs[i], REFTABLE_BLOCK_TYPE_REF)); cl_assert(!reftable_record_init(&recs[i],
REFTABLE_BLOCK_TYPE_REF));
recs[i].u.ref.refname = (char *) "refs/heads/master"; recs[i].u.ref.refname = (char *) "refs/heads/master";
} }
@@ -96,28 +99,29 @@ static void t_pq_index(void)
struct pq_entry top = merged_iter_pqueue_top(pq); struct pq_entry top = merged_iter_pqueue_top(pq);
struct pq_entry e; struct pq_entry e;
check(!merged_iter_pqueue_remove(&pq, &e)); cl_assert_equal_i(merged_iter_pqueue_remove(&pq, &e), 0);
merged_iter_pqueue_check(&pq); merged_iter_pqueue_check(&pq);
check(pq_entry_equal(&top, &e)); cl_assert(pq_entry_equal(&top, &e));
check(reftable_record_type(e.rec) == REFTABLE_BLOCK_TYPE_REF); cl_assert(reftable_record_type(e.rec) == REFTABLE_BLOCK_TYPE_REF);
check_int(e.index, ==, i); cl_assert_equal_i(e.index, i);
if (last) if (last)
check_str(last, e.rec->u.ref.refname); cl_assert_equal_s(last, e.rec->u.ref.refname);
last = e.rec->u.ref.refname; last = e.rec->u.ref.refname;
} }
merged_iter_pqueue_release(&pq); merged_iter_pqueue_release(&pq);
} }
static void t_merged_iter_pqueue_top(void) void test_reftable_pq__merged_iter_pqueue_top(void)
{ {
struct merged_iter_pqueue pq = { 0 }; struct merged_iter_pqueue pq = { 0 };
struct reftable_record recs[13]; struct reftable_record recs[13];
size_t N = ARRAY_SIZE(recs), i; size_t N = ARRAY_SIZE(recs), i;
for (i = 0; i < N; i++) { for (i = 0; i < N; i++) {
check(!reftable_record_init(&recs[i], REFTABLE_BLOCK_TYPE_REF)); cl_assert(!reftable_record_init(&recs[i],
REFTABLE_BLOCK_TYPE_REF));
recs[i].u.ref.refname = (char *) "refs/heads/master"; recs[i].u.ref.refname = (char *) "refs/heads/master";
} }
@@ -137,25 +141,16 @@ static void t_merged_iter_pqueue_top(void)
struct pq_entry top = merged_iter_pqueue_top(pq); struct pq_entry top = merged_iter_pqueue_top(pq);
struct pq_entry e; struct pq_entry e;
check(!merged_iter_pqueue_remove(&pq, &e)); cl_assert_equal_i(merged_iter_pqueue_remove(&pq, &e), 0);
merged_iter_pqueue_check(&pq); merged_iter_pqueue_check(&pq);
check(pq_entry_equal(&top, &e)); cl_assert(pq_entry_equal(&top, &e) != 0);
check(reftable_record_equal(top.rec, &recs[i], REFTABLE_HASH_SIZE_SHA1)); cl_assert(reftable_record_equal(top.rec, &recs[i], REFTABLE_HASH_SIZE_SHA1) != 0);
for (size_t j = 0; i < pq.len; j++) { for (size_t j = 0; i < pq.len; j++) {
check(pq_less(&top, &pq.heap[j])); cl_assert(pq_less(&top, &pq.heap[j]) != 0);
check_int(top.index, >, j); cl_assert(top.index > j);
} }
} }
merged_iter_pqueue_release(&pq); merged_iter_pqueue_release(&pq);
} }
int cmd_main(int argc UNUSED, const char *argv[] UNUSED)
{
TEST(t_pq_record(), "pq works with record-based comparison");
TEST(t_pq_index(), "pq works with index-based comparison");
TEST(t_merged_iter_pqueue_top(), "merged_iter_pqueue_top works");
return test_done();
}

View File

@@ -8,7 +8,7 @@ https://developers.google.com/open-source/licenses/bsd
#define DISABLE_SIGN_COMPARE_WARNINGS #define DISABLE_SIGN_COMPARE_WARNINGS
#include "test-lib.h" #include "unit-test.h"
#include "lib-reftable.h" #include "lib-reftable.h"
#include "reftable/basics.h" #include "reftable/basics.h"
#include "reftable/blocksource.h" #include "reftable/blocksource.h"
@@ -19,24 +19,24 @@ https://developers.google.com/open-source/licenses/bsd
static const int update_index = 5; static const int update_index = 5;
static void t_buffer(void) void test_reftable_readwrite__buffer(void)
{ {
struct reftable_buf buf = REFTABLE_BUF_INIT; struct reftable_buf buf = REFTABLE_BUF_INIT;
struct reftable_block_source source = { 0 }; struct reftable_block_source source = { 0 };
struct reftable_block_data out = { 0 }; struct reftable_block_data out = { 0 };
int n; int n;
uint8_t in[] = "hello"; uint8_t in[] = "hello";
check(!reftable_buf_add(&buf, in, sizeof(in))); cl_assert_equal_i(reftable_buf_add(&buf, in, sizeof(in)), 0);
block_source_from_buf(&source, &buf); block_source_from_buf(&source, &buf);
check_int(block_source_size(&source), ==, 6); cl_assert_equal_i(block_source_size(&source), 6);
n = block_source_read_data(&source, &out, 0, sizeof(in)); n = block_source_read_data(&source, &out, 0, sizeof(in));
check_int(n, ==, sizeof(in)); cl_assert_equal_i(n, sizeof(in));
check(!memcmp(in, out.data, n)); cl_assert(!memcmp(in, out.data, n));
block_source_release_data(&out); block_source_release_data(&out);
n = block_source_read_data(&source, &out, 1, 2); n = block_source_read_data(&source, &out, 1, 2);
check_int(n, ==, 2); cl_assert_equal_i(n, 2);
check(!memcmp(out.data, "el", 2)); cl_assert(!memcmp(out.data, "el", 2));
block_source_release_data(&out); block_source_release_data(&out);
block_source_close(&source); block_source_close(&source);
@@ -55,41 +55,41 @@ static void write_table(char ***names, struct reftable_buf *buf, int N,
int i; int i;
REFTABLE_CALLOC_ARRAY(*names, N + 1); REFTABLE_CALLOC_ARRAY(*names, N + 1);
check(*names != NULL); cl_assert(*names != NULL);
REFTABLE_CALLOC_ARRAY(refs, N); REFTABLE_CALLOC_ARRAY(refs, N);
check(refs != NULL); cl_assert(refs != NULL);
REFTABLE_CALLOC_ARRAY(logs, N); REFTABLE_CALLOC_ARRAY(logs, N);
check(logs != NULL); cl_assert(logs != NULL);
for (i = 0; i < N; i++) { for (i = 0; i < N; i++) {
refs[i].refname = (*names)[i] = xstrfmt("refs/heads/branch%02d", i); refs[i].refname = (*names)[i] = xstrfmt("refs/heads/branch%02d", i);
refs[i].update_index = update_index; refs[i].update_index = update_index;
refs[i].value_type = REFTABLE_REF_VAL1; refs[i].value_type = REFTABLE_REF_VAL1;
t_reftable_set_hash(refs[i].value.val1, i, REFTABLE_HASH_SHA1); cl_reftable_set_hash(refs[i].value.val1, i,
REFTABLE_HASH_SHA1);
} }
for (i = 0; i < N; i++) { for (i = 0; i < N; i++) {
logs[i].refname = (*names)[i]; logs[i].refname = (*names)[i];
logs[i].update_index = update_index; logs[i].update_index = update_index;
logs[i].value_type = REFTABLE_LOG_UPDATE; logs[i].value_type = REFTABLE_LOG_UPDATE;
t_reftable_set_hash(logs[i].value.update.new_hash, i, cl_reftable_set_hash(logs[i].value.update.new_hash, i,
REFTABLE_HASH_SHA1); REFTABLE_HASH_SHA1);
logs[i].value.update.message = (char *) "message"; logs[i].value.update.message = (char *) "message";
} }
t_reftable_write_to_buf(buf, refs, N, logs, N, &opts); cl_reftable_write_to_buf(buf, refs, N, logs, N, &opts);
reftable_free(refs); reftable_free(refs);
reftable_free(logs); reftable_free(logs);
} }
static void t_log_buffer_size(void) void test_reftable_readwrite__log_buffer_size(void)
{ {
struct reftable_buf buf = REFTABLE_BUF_INIT; struct reftable_buf buf = REFTABLE_BUF_INIT;
struct reftable_write_options opts = { struct reftable_write_options opts = {
.block_size = 4096, .block_size = 4096,
}; };
int err;
int i; int i;
struct reftable_log_record struct reftable_log_record
log = { .refname = (char *) "refs/heads/master", log = { .refname = (char *) "refs/heads/master",
@@ -102,7 +102,8 @@ static void t_log_buffer_size(void)
.time = 0x5e430672, .time = 0x5e430672,
.message = (char *) "commit: 9\n", .message = (char *) "commit: 9\n",
} } }; } } };
struct reftable_writer *w = t_reftable_strbuf_writer(&buf, &opts); struct reftable_writer *w = cl_reftable_strbuf_writer(&buf,
&opts);
/* This tests buffer extension for log compression. Must use a random /* This tests buffer extension for log compression. Must use a random
hash, to ensure that the compressed part is larger than the original. hash, to ensure that the compressed part is larger than the original.
@@ -112,22 +113,19 @@ static void t_log_buffer_size(void)
log.value.update.new_hash[i] = (uint8_t)(git_rand(0) % 256); log.value.update.new_hash[i] = (uint8_t)(git_rand(0) % 256);
} }
reftable_writer_set_limits(w, update_index, update_index); reftable_writer_set_limits(w, update_index, update_index);
err = reftable_writer_add_log(w, &log); cl_assert_equal_i(reftable_writer_add_log(w, &log), 0);
check(!err); cl_assert_equal_i(reftable_writer_close(w), 0);
err = reftable_writer_close(w);
check(!err);
reftable_writer_free(w); reftable_writer_free(w);
reftable_buf_release(&buf); reftable_buf_release(&buf);
} }
static void t_log_overflow(void) void test_reftable_readwrite__log_overflow(void)
{ {
struct reftable_buf buf = REFTABLE_BUF_INIT; struct reftable_buf buf = REFTABLE_BUF_INIT;
char msg[256] = { 0 }; char msg[256] = { 0 };
struct reftable_write_options opts = { struct reftable_write_options opts = {
.block_size = ARRAY_SIZE(msg), .block_size = ARRAY_SIZE(msg),
}; };
int err;
struct reftable_log_record log = { struct reftable_log_record log = {
.refname = (char *) "refs/heads/master", .refname = (char *) "refs/heads/master",
.update_index = update_index, .update_index = update_index,
@@ -144,21 +142,22 @@ static void t_log_overflow(void)
}, },
}, },
}; };
struct reftable_writer *w = t_reftable_strbuf_writer(&buf, &opts); struct reftable_writer *w = cl_reftable_strbuf_writer(&buf,
&opts);
memset(msg, 'x', sizeof(msg) - 1); memset(msg, 'x', sizeof(msg) - 1);
reftable_writer_set_limits(w, update_index, update_index); reftable_writer_set_limits(w, update_index, update_index);
err = reftable_writer_add_log(w, &log); cl_assert_equal_i(reftable_writer_add_log(w, &log), REFTABLE_ENTRY_TOO_BIG_ERROR);
check_int(err, ==, REFTABLE_ENTRY_TOO_BIG_ERROR);
reftable_writer_free(w); reftable_writer_free(w);
reftable_buf_release(&buf); reftable_buf_release(&buf);
} }
static void t_log_write_limits(void) void test_reftable_readwrite__log_write_limits(void)
{ {
struct reftable_write_options opts = { 0 }; struct reftable_write_options opts = { 0 };
struct reftable_buf buf = REFTABLE_BUF_INIT; struct reftable_buf buf = REFTABLE_BUF_INIT;
struct reftable_writer *w = t_reftable_strbuf_writer(&buf, &opts); struct reftable_writer *w = cl_reftable_strbuf_writer(&buf,
&opts);
struct reftable_log_record log = { struct reftable_log_record log = {
.refname = (char *)"refs/head/master", .refname = (char *)"refs/head/master",
.update_index = 0, .update_index = 0,
@@ -174,29 +173,25 @@ static void t_log_write_limits(void)
}, },
}, },
}; };
int err;
reftable_writer_set_limits(w, 1, 1); reftable_writer_set_limits(w, 1, 1);
/* write with update_index (0) below set limits (1, 1) */ /* write with update_index (0) below set limits (1, 1) */
err = reftable_writer_add_log(w, &log); cl_assert_equal_i(reftable_writer_add_log(w, &log), 0);
check_int(err, ==, 0);
/* write with update_index (1) in the set limits (1, 1) */ /* write with update_index (1) in the set limits (1, 1) */
log.update_index = 1; log.update_index = 1;
err = reftable_writer_add_log(w, &log); cl_assert_equal_i(reftable_writer_add_log(w, &log), 0);
check_int(err, ==, 0);
/* write with update_index (3) above set limits (1, 1) */ /* write with update_index (3) above set limits (1, 1) */
log.update_index = 3; log.update_index = 3;
err = reftable_writer_add_log(w, &log); cl_assert_equal_i(reftable_writer_add_log(w, &log), REFTABLE_API_ERROR);
check_int(err, ==, REFTABLE_API_ERROR);
reftable_writer_free(w); reftable_writer_free(w);
reftable_buf_release(&buf); reftable_buf_release(&buf);
} }
static void t_log_write_read(void) void test_reftable_readwrite__log_write_read(void)
{ {
struct reftable_write_options opts = { struct reftable_write_options opts = {
.block_size = 256, .block_size = 256,
@@ -207,13 +202,14 @@ static void t_log_write_read(void)
struct reftable_table *table; struct reftable_table *table;
struct reftable_block_source source = { 0 }; struct reftable_block_source source = { 0 };
struct reftable_buf buf = REFTABLE_BUF_INIT; struct reftable_buf buf = REFTABLE_BUF_INIT;
struct reftable_writer *w = t_reftable_strbuf_writer(&buf, &opts); struct reftable_writer *w = cl_reftable_strbuf_writer(&buf, &opts);
const struct reftable_stats *stats = NULL; const struct reftable_stats *stats = NULL;
int N = 2, err, i, n; int N = 2, i;
char **names; char **names;
int err;
names = reftable_calloc(N + 1, sizeof(*names)); names = reftable_calloc(N + 1, sizeof(*names));
check(names != NULL); cl_assert(names != NULL);
reftable_writer_set_limits(w, 0, N); reftable_writer_set_limits(w, 0, N);
@@ -225,8 +221,7 @@ static void t_log_write_read(void)
ref.refname = name; ref.refname = name;
ref.update_index = i; ref.update_index = i;
err = reftable_writer_add_ref(w, &ref); cl_assert_equal_i(reftable_writer_add_ref(w, &ref), 0);
check(!err);
} }
for (i = 0; i < N; i++) { for (i = 0; i < N; i++) {
@@ -235,60 +230,57 @@ static void t_log_write_read(void)
log.refname = names[i]; log.refname = names[i];
log.update_index = i; log.update_index = i;
log.value_type = REFTABLE_LOG_UPDATE; log.value_type = REFTABLE_LOG_UPDATE;
t_reftable_set_hash(log.value.update.old_hash, i, cl_reftable_set_hash(log.value.update.old_hash, i,
REFTABLE_HASH_SHA1); REFTABLE_HASH_SHA1);
t_reftable_set_hash(log.value.update.new_hash, i + 1, cl_reftable_set_hash(log.value.update.new_hash, i + 1,
REFTABLE_HASH_SHA1); REFTABLE_HASH_SHA1);
err = reftable_writer_add_log(w, &log); cl_assert_equal_i(reftable_writer_add_log(w, &log), 0);
check(!err);
} }
n = reftable_writer_close(w); cl_assert_equal_i(reftable_writer_close(w), 0);
check_int(n, ==, 0);
stats = reftable_writer_stats(w); stats = reftable_writer_stats(w);
check_int(stats->log_stats.blocks, >, 0); cl_assert(stats->log_stats.blocks > 0);
reftable_writer_free(w); reftable_writer_free(w);
w = NULL; w = NULL;
block_source_from_buf(&source, &buf); block_source_from_buf(&source, &buf);
err = reftable_table_new(&table, &source, "file.log"); err = reftable_table_new(&table, &source, "file.log");
check(!err); cl_assert(!err);
err = reftable_table_init_ref_iterator(table, &it); err = reftable_table_init_ref_iterator(table, &it);
check(!err); cl_assert(!err);
err = reftable_iterator_seek_ref(&it, names[N - 1]); err = reftable_iterator_seek_ref(&it, names[N - 1]);
check(!err); cl_assert(!err);
err = reftable_iterator_next_ref(&it, &ref); err = reftable_iterator_next_ref(&it, &ref);
check(!err); cl_assert(!err);
/* end of iteration. */ /* end of iteration. */
err = reftable_iterator_next_ref(&it, &ref); cl_assert(reftable_iterator_next_ref(&it, &ref) > 0);
check_int(err, >, 0);
reftable_iterator_destroy(&it); reftable_iterator_destroy(&it);
reftable_ref_record_release(&ref); reftable_ref_record_release(&ref);
err = reftable_table_init_log_iterator(table, &it); err = reftable_table_init_log_iterator(table, &it);
check(!err); cl_assert(!err);
err = reftable_iterator_seek_log(&it, ""); err = reftable_iterator_seek_log(&it, "");
check(!err); cl_assert(!err);
for (i = 0; ; i++) { for (i = 0; ; i++) {
int err = reftable_iterator_next_log(&it, &log); int err = reftable_iterator_next_log(&it, &log);
if (err > 0) if (err > 0)
break; break;
check(!err); cl_assert(!err);
check_str(names[i], log.refname); cl_assert_equal_s(names[i], log.refname);
check_int(i, ==, log.update_index); cl_assert_equal_i(i, log.update_index);
reftable_log_record_release(&log); reftable_log_record_release(&log);
} }
check_int(i, ==, N); cl_assert_equal_i(i, N);
reftable_iterator_destroy(&it); reftable_iterator_destroy(&it);
/* cleanup. */ /* cleanup. */
@@ -297,7 +289,7 @@ static void t_log_write_read(void)
reftable_table_decref(table); reftable_table_decref(table);
} }
static void t_log_zlib_corruption(void) void test_reftable_readwrite__log_zlib_corruption(void)
{ {
struct reftable_write_options opts = { struct reftable_write_options opts = {
.block_size = 256, .block_size = 256,
@@ -306,10 +298,12 @@ static void t_log_zlib_corruption(void)
struct reftable_table *table; struct reftable_table *table;
struct reftable_block_source source = { 0 }; struct reftable_block_source source = { 0 };
struct reftable_buf buf = REFTABLE_BUF_INIT; struct reftable_buf buf = REFTABLE_BUF_INIT;
struct reftable_writer *w = t_reftable_strbuf_writer(&buf, &opts); struct reftable_writer *w = cl_reftable_strbuf_writer(&buf,
&opts);
const struct reftable_stats *stats = NULL; const struct reftable_stats *stats = NULL;
char message[100] = { 0 }; char message[100] = { 0 };
int err, i, n; int i;
int err;
struct reftable_log_record log = { struct reftable_log_record log = {
.refname = (char *) "refname", .refname = (char *) "refname",
.value_type = REFTABLE_LOG_UPDATE, .value_type = REFTABLE_LOG_UPDATE,
@@ -329,14 +323,11 @@ static void t_log_zlib_corruption(void)
reftable_writer_set_limits(w, 1, 1); reftable_writer_set_limits(w, 1, 1);
err = reftable_writer_add_log(w, &log); cl_assert_equal_i(reftable_writer_add_log(w, &log), 0);
check(!err); cl_assert_equal_i(reftable_writer_close(w), 0);
n = reftable_writer_close(w);
check_int(n, ==, 0);
stats = reftable_writer_stats(w); stats = reftable_writer_stats(w);
check_int(stats->log_stats.blocks, >, 0); cl_assert(stats->log_stats.blocks > 0);
reftable_writer_free(w); reftable_writer_free(w);
w = NULL; w = NULL;
@@ -346,12 +337,12 @@ static void t_log_zlib_corruption(void)
block_source_from_buf(&source, &buf); block_source_from_buf(&source, &buf);
err = reftable_table_new(&table, &source, "file.log"); err = reftable_table_new(&table, &source, "file.log");
check(!err); cl_assert(!err);
err = reftable_table_init_log_iterator(table, &it); err = reftable_table_init_log_iterator(table, &it);
check(!err); cl_assert(!err);
err = reftable_iterator_seek_log(&it, "refname"); err = reftable_iterator_seek_log(&it, "refname");
check_int(err, ==, REFTABLE_ZLIB_ERROR); cl_assert_equal_i(err, REFTABLE_ZLIB_ERROR);
reftable_iterator_destroy(&it); reftable_iterator_destroy(&it);
@@ -360,7 +351,7 @@ static void t_log_zlib_corruption(void)
reftable_buf_release(&buf); reftable_buf_release(&buf);
} }
static void t_table_read_write_sequential(void) void test_reftable_readwrite__table_read_write_sequential(void)
{ {
char **names; char **names;
struct reftable_buf buf = REFTABLE_BUF_INIT; struct reftable_buf buf = REFTABLE_BUF_INIT;
@@ -376,24 +367,24 @@ static void t_table_read_write_sequential(void)
block_source_from_buf(&source, &buf); block_source_from_buf(&source, &buf);
err = reftable_table_new(&table, &source, "file.ref"); err = reftable_table_new(&table, &source, "file.ref");
check(!err); cl_assert(!err);
err = reftable_table_init_ref_iterator(table, &it); err = reftable_table_init_ref_iterator(table, &it);
check(!err); cl_assert(!err);
err = reftable_iterator_seek_ref(&it, ""); err = reftable_iterator_seek_ref(&it, "");
check(!err); cl_assert(!err);
for (j = 0; ; j++) { for (j = 0; ; j++) {
struct reftable_ref_record ref = { 0 }; struct reftable_ref_record ref = { 0 };
int r = reftable_iterator_next_ref(&it, &ref); int r = reftable_iterator_next_ref(&it, &ref);
check_int(r, >=, 0); cl_assert(r >= 0);
if (r > 0) if (r > 0)
break; break;
check_str(names[j], ref.refname); cl_assert_equal_s(names[j], ref.refname);
check_int(update_index, ==, ref.update_index); cl_assert_equal_i(update_index, ref.update_index);
reftable_ref_record_release(&ref); reftable_ref_record_release(&ref);
} }
check_int(j, ==, N); cl_assert_equal_i(j, N);
reftable_iterator_destroy(&it); reftable_iterator_destroy(&it);
reftable_table_decref(table); reftable_table_decref(table);
@@ -401,42 +392,42 @@ static void t_table_read_write_sequential(void)
free_names(names); free_names(names);
} }
static void t_table_write_small_table(void) void test_reftable_readwrite__table_write_small_table(void)
{ {
char **names; char **names;
struct reftable_buf buf = REFTABLE_BUF_INIT; struct reftable_buf buf = REFTABLE_BUF_INIT;
int N = 1; int N = 1;
write_table(&names, &buf, N, 4096, REFTABLE_HASH_SHA1); write_table(&names, &buf, N, 4096, REFTABLE_HASH_SHA1);
check_int(buf.len, <, 200); cl_assert(buf.len < 200);
reftable_buf_release(&buf); reftable_buf_release(&buf);
free_names(names); free_names(names);
} }
static void t_table_read_api(void) void test_reftable_readwrite__table_read_api(void)
{ {
char **names; char **names;
struct reftable_buf buf = REFTABLE_BUF_INIT; struct reftable_buf buf = REFTABLE_BUF_INIT;
int N = 50; int N = 50;
struct reftable_table *table; struct reftable_table *table;
struct reftable_block_source source = { 0 }; struct reftable_block_source source = { 0 };
int err;
struct reftable_log_record log = { 0 }; struct reftable_log_record log = { 0 };
struct reftable_iterator it = { 0 }; struct reftable_iterator it = { 0 };
int err;
write_table(&names, &buf, N, 256, REFTABLE_HASH_SHA1); write_table(&names, &buf, N, 256, REFTABLE_HASH_SHA1);
block_source_from_buf(&source, &buf); block_source_from_buf(&source, &buf);
err = reftable_table_new(&table, &source, "file.ref"); err = reftable_table_new(&table, &source, "file.ref");
check(!err); cl_assert(!err);
err = reftable_table_init_ref_iterator(table, &it); err = reftable_table_init_ref_iterator(table, &it);
check(!err); cl_assert(!err);
err = reftable_iterator_seek_ref(&it, names[0]); err = reftable_iterator_seek_ref(&it, names[0]);
check(!err); cl_assert(!err);
err = reftable_iterator_next_log(&it, &log); err = reftable_iterator_next_log(&it, &log);
check_int(err, ==, REFTABLE_API_ERROR); cl_assert_equal_i(err, REFTABLE_API_ERROR);
reftable_buf_release(&buf); reftable_buf_release(&buf);
free_names(names); free_names(names);
@@ -464,42 +455,43 @@ static void t_table_read_write_seek(int index, enum reftable_hash hash_id)
block_source_from_buf(&source, &buf); block_source_from_buf(&source, &buf);
err = reftable_table_new(&table, &source, "file.ref"); err = reftable_table_new(&table, &source, "file.ref");
check(!err); cl_assert(!err);
check_int(hash_id, ==, reftable_table_hash_id(table)); cl_assert_equal_i(hash_id, reftable_table_hash_id(table));
if (!index) { if (!index) {
table->ref_offsets.index_offset = 0; table->ref_offsets.index_offset = 0;
} else { } else {
check_int(table->ref_offsets.index_offset, >, 0); cl_assert(table->ref_offsets.index_offset > 0);
} }
for (i = 1; i < N; i++) { for (i = 1; i < N; i++) {
err = reftable_table_init_ref_iterator(table, &it); err = reftable_table_init_ref_iterator(table, &it);
check(!err); cl_assert(!err);
err = reftable_iterator_seek_ref(&it, names[i]); err = reftable_iterator_seek_ref(&it, names[i]);
check(!err); cl_assert(!err);
err = reftable_iterator_next_ref(&it, &ref); err = reftable_iterator_next_ref(&it, &ref);
check(!err); cl_assert(!err);
check_str(names[i], ref.refname); cl_assert_equal_s(names[i], ref.refname);
check_int(REFTABLE_REF_VAL1, ==, ref.value_type); cl_assert_equal_i(REFTABLE_REF_VAL1, ref.value_type);
check_int(i, ==, ref.value.val1[0]); cl_assert_equal_i(i, ref.value.val1[0]);
reftable_ref_record_release(&ref); reftable_ref_record_release(&ref);
reftable_iterator_destroy(&it); reftable_iterator_destroy(&it);
} }
check(!reftable_buf_addstr(&pastLast, names[N - 1])); cl_assert_equal_i(reftable_buf_addstr(&pastLast, names[N - 1]),
check(!reftable_buf_addstr(&pastLast, "/")); 0);
cl_assert_equal_i(reftable_buf_addstr(&pastLast, "/"), 0);
err = reftable_table_init_ref_iterator(table, &it); err = reftable_table_init_ref_iterator(table, &it);
check(!err); cl_assert(!err);
err = reftable_iterator_seek_ref(&it, pastLast.buf); err = reftable_iterator_seek_ref(&it, pastLast.buf);
if (err == 0) { if (err == 0) {
struct reftable_ref_record ref = { 0 }; struct reftable_ref_record ref = { 0 };
int err = reftable_iterator_next_ref(&it, &ref); int err = reftable_iterator_next_ref(&it, &ref);
check_int(err, >, 0); cl_assert(err > 0);
} else { } else {
check_int(err, >, 0); cl_assert(err > 0);
} }
reftable_buf_release(&pastLast); reftable_buf_release(&pastLast);
@@ -510,17 +502,17 @@ static void t_table_read_write_seek(int index, enum reftable_hash hash_id)
reftable_table_decref(table); reftable_table_decref(table);
} }
static void t_table_read_write_seek_linear(void) void test_reftable_readwrite__table_read_write_seek_linear(void)
{ {
t_table_read_write_seek(0, REFTABLE_HASH_SHA1); t_table_read_write_seek(0, REFTABLE_HASH_SHA1);
} }
static void t_table_read_write_seek_linear_sha256(void) void test_reftable_readwrite__table_read_write_seek_linear_sha256(void)
{ {
t_table_read_write_seek(0, REFTABLE_HASH_SHA256); t_table_read_write_seek(0, REFTABLE_HASH_SHA256);
} }
static void t_table_read_write_seek_index(void) void test_reftable_readwrite__table_read_write_seek_index(void)
{ {
t_table_read_write_seek(1, REFTABLE_HASH_SHA1); t_table_read_write_seek(1, REFTABLE_HASH_SHA1);
} }
@@ -538,14 +530,16 @@ static void t_table_refs_for(int indexed)
struct reftable_table *table; struct reftable_table *table;
struct reftable_block_source source = { 0 }; struct reftable_block_source source = { 0 };
struct reftable_buf buf = REFTABLE_BUF_INIT; struct reftable_buf buf = REFTABLE_BUF_INIT;
struct reftable_writer *w = t_reftable_strbuf_writer(&buf, &opts); struct reftable_writer *w = cl_reftable_strbuf_writer(&buf,
&opts);
struct reftable_iterator it = { 0 }; struct reftable_iterator it = { 0 };
int N = 50, n, j, err, i; int N = 50, j, i;
int err;
want_names = reftable_calloc(N + 1, sizeof(*want_names)); want_names = reftable_calloc(N + 1, sizeof(*want_names));
check(want_names != NULL); cl_assert(want_names != NULL);
t_reftable_set_hash(want_hash, 4, REFTABLE_HASH_SHA1); cl_reftable_set_hash(want_hash, 4, REFTABLE_HASH_SHA1);
for (i = 0; i < N; i++) { for (i = 0; i < N; i++) {
uint8_t hash[REFTABLE_HASH_SIZE_SHA1]; uint8_t hash[REFTABLE_HASH_SIZE_SHA1];
@@ -561,24 +555,22 @@ static void t_table_refs_for(int indexed)
ref.refname = name; ref.refname = name;
ref.value_type = REFTABLE_REF_VAL2; ref.value_type = REFTABLE_REF_VAL2;
t_reftable_set_hash(ref.value.val2.value, i / 4, cl_reftable_set_hash(ref.value.val2.value, i / 4,
REFTABLE_HASH_SHA1); REFTABLE_HASH_SHA1);
t_reftable_set_hash(ref.value.val2.target_value, 3 + i / 4, cl_reftable_set_hash(ref.value.val2.target_value,
REFTABLE_HASH_SHA1); 3 + i / 4, REFTABLE_HASH_SHA1);
/* 80 bytes / entry, so 3 entries per block. Yields 17 /* 80 bytes / entry, so 3 entries per block. Yields 17
*/ */
/* blocks. */ /* blocks. */
n = reftable_writer_add_ref(w, &ref); cl_assert_equal_i(reftable_writer_add_ref(w, &ref), 0);
check_int(n, ==, 0);
if (!memcmp(ref.value.val2.value, want_hash, REFTABLE_HASH_SIZE_SHA1) || if (!memcmp(ref.value.val2.value, want_hash, REFTABLE_HASH_SIZE_SHA1) ||
!memcmp(ref.value.val2.target_value, want_hash, REFTABLE_HASH_SIZE_SHA1)) !memcmp(ref.value.val2.target_value, want_hash, REFTABLE_HASH_SIZE_SHA1))
want_names[want_names_len++] = xstrdup(name); want_names[want_names_len++] = xstrdup(name);
} }
n = reftable_writer_close(w); cl_assert_equal_i(reftable_writer_close(w), 0);
check_int(n, ==, 0);
reftable_writer_free(w); reftable_writer_free(w);
w = NULL; w = NULL;
@@ -586,29 +578,29 @@ static void t_table_refs_for(int indexed)
block_source_from_buf(&source, &buf); block_source_from_buf(&source, &buf);
err = reftable_table_new(&table, &source, "file.ref"); err = reftable_table_new(&table, &source, "file.ref");
check(!err); cl_assert(!err);
if (!indexed) if (!indexed)
table->obj_offsets.is_present = 0; table->obj_offsets.is_present = 0;
err = reftable_table_init_ref_iterator(table, &it); err = reftable_table_init_ref_iterator(table, &it);
check(!err); cl_assert(!err);
err = reftable_iterator_seek_ref(&it, ""); err = reftable_iterator_seek_ref(&it, "");
check(!err); cl_assert(!err);
reftable_iterator_destroy(&it); reftable_iterator_destroy(&it);
err = reftable_table_refs_for(table, &it, want_hash); err = reftable_table_refs_for(table, &it, want_hash);
check(!err); cl_assert(!err);
for (j = 0; ; j++) { for (j = 0; ; j++) {
int err = reftable_iterator_next_ref(&it, &ref); int err = reftable_iterator_next_ref(&it, &ref);
check_int(err, >=, 0); cl_assert(err >= 0);
if (err > 0) if (err > 0)
break; break;
check_int(j, <, want_names_len); cl_assert(j < want_names_len);
check_str(ref.refname, want_names[j]); cl_assert_equal_s(ref.refname, want_names[j]);
reftable_ref_record_release(&ref); reftable_ref_record_release(&ref);
} }
check_int(j, ==, want_names_len); cl_assert_equal_i(j, want_names_len);
reftable_buf_release(&buf); reftable_buf_release(&buf);
free_names(want_names); free_names(want_names);
@@ -616,21 +608,21 @@ static void t_table_refs_for(int indexed)
reftable_table_decref(table); reftable_table_decref(table);
} }
static void t_table_refs_for_no_index(void) void test_reftable_readwrite__table_refs_for_no_index(void)
{ {
t_table_refs_for(0); t_table_refs_for(0);
} }
static void t_table_refs_for_obj_index(void) void test_reftable_readwrite__table_refs_for_obj_index(void)
{ {
t_table_refs_for(1); t_table_refs_for(1);
} }
static void t_write_empty_table(void) void test_reftable_readwrite__write_empty_table(void)
{ {
struct reftable_write_options opts = { 0 }; struct reftable_write_options opts = { 0 };
struct reftable_buf buf = REFTABLE_BUF_INIT; struct reftable_buf buf = REFTABLE_BUF_INIT;
struct reftable_writer *w = t_reftable_strbuf_writer(&buf, &opts); struct reftable_writer *w = cl_reftable_strbuf_writer(&buf, &opts);
struct reftable_block_source source = { 0 }; struct reftable_block_source source = { 0 };
struct reftable_table *table = NULL; struct reftable_table *table = NULL;
struct reftable_ref_record rec = { 0 }; struct reftable_ref_record rec = { 0 };
@@ -639,43 +631,41 @@ static void t_write_empty_table(void)
reftable_writer_set_limits(w, 1, 1); reftable_writer_set_limits(w, 1, 1);
err = reftable_writer_close(w); cl_assert_equal_i(reftable_writer_close(w), REFTABLE_EMPTY_TABLE_ERROR);
check_int(err, ==, REFTABLE_EMPTY_TABLE_ERROR);
reftable_writer_free(w); reftable_writer_free(w);
check_uint(buf.len, ==, header_size(1) + footer_size(1)); cl_assert_equal_i(buf.len, header_size(1) + footer_size(1));
block_source_from_buf(&source, &buf); block_source_from_buf(&source, &buf);
err = reftable_table_new(&table, &source, "filename"); err = reftable_table_new(&table, &source, "filename");
check(!err); cl_assert(!err);
err = reftable_table_init_ref_iterator(table, &it); err = reftable_table_init_ref_iterator(table, &it);
check(!err); cl_assert(!err);
err = reftable_iterator_seek_ref(&it, ""); err = reftable_iterator_seek_ref(&it, "");
check(!err); cl_assert(!err);
err = reftable_iterator_next_ref(&it, &rec); err = reftable_iterator_next_ref(&it, &rec);
check_int(err, >, 0); cl_assert(err > 0);
reftable_iterator_destroy(&it); reftable_iterator_destroy(&it);
reftable_table_decref(table); reftable_table_decref(table);
reftable_buf_release(&buf); reftable_buf_release(&buf);
} }
static void t_write_object_id_min_length(void) void test_reftable_readwrite__write_object_id_min_length(void)
{ {
struct reftable_write_options opts = { struct reftable_write_options opts = {
.block_size = 75, .block_size = 75,
}; };
struct reftable_buf buf = REFTABLE_BUF_INIT; struct reftable_buf buf = REFTABLE_BUF_INIT;
struct reftable_writer *w = t_reftable_strbuf_writer(&buf, &opts); struct reftable_writer *w = cl_reftable_strbuf_writer(&buf, &opts);
struct reftable_ref_record ref = { struct reftable_ref_record ref = {
.update_index = 1, .update_index = 1,
.value_type = REFTABLE_REF_VAL1, .value_type = REFTABLE_REF_VAL1,
.value.val1 = {42}, .value.val1 = {42},
}; };
int err;
int i; int i;
reftable_writer_set_limits(w, 1, 1); reftable_writer_set_limits(w, 1, 1);
@@ -686,30 +676,27 @@ static void t_write_object_id_min_length(void)
char name[256]; char name[256];
snprintf(name, sizeof(name), "ref%05d", i); snprintf(name, sizeof(name), "ref%05d", i);
ref.refname = name; ref.refname = name;
err = reftable_writer_add_ref(w, &ref); cl_assert_equal_i(reftable_writer_add_ref(w, &ref), 0);
check(!err);
} }
err = reftable_writer_close(w); cl_assert_equal_i(reftable_writer_close(w), 0);
check(!err); cl_assert_equal_i(reftable_writer_stats(w)->object_id_len, 2);
check_int(reftable_writer_stats(w)->object_id_len, ==, 2);
reftable_writer_free(w); reftable_writer_free(w);
reftable_buf_release(&buf); reftable_buf_release(&buf);
} }
static void t_write_object_id_length(void) void test_reftable_readwrite__write_object_id_length(void)
{ {
struct reftable_write_options opts = { struct reftable_write_options opts = {
.block_size = 75, .block_size = 75,
}; };
struct reftable_buf buf = REFTABLE_BUF_INIT; struct reftable_buf buf = REFTABLE_BUF_INIT;
struct reftable_writer *w = t_reftable_strbuf_writer(&buf, &opts); struct reftable_writer *w = cl_reftable_strbuf_writer(&buf, &opts);
struct reftable_ref_record ref = { struct reftable_ref_record ref = {
.update_index = 1, .update_index = 1,
.value_type = REFTABLE_REF_VAL1, .value_type = REFTABLE_REF_VAL1,
.value.val1 = {42}, .value.val1 = {42},
}; };
int err;
int i; int i;
reftable_writer_set_limits(w, 1, 1); reftable_writer_set_limits(w, 1, 1);
@@ -721,44 +708,39 @@ static void t_write_object_id_length(void)
snprintf(name, sizeof(name), "ref%05d", i); snprintf(name, sizeof(name), "ref%05d", i);
ref.refname = name; ref.refname = name;
ref.value.val1[15] = i; ref.value.val1[15] = i;
err = reftable_writer_add_ref(w, &ref); cl_assert(reftable_writer_add_ref(w, &ref) == 0);
check(!err);
} }
err = reftable_writer_close(w); cl_assert_equal_i(reftable_writer_close(w), 0);
check(!err); cl_assert_equal_i(reftable_writer_stats(w)->object_id_len, 16);
check_int(reftable_writer_stats(w)->object_id_len, ==, 16);
reftable_writer_free(w); reftable_writer_free(w);
reftable_buf_release(&buf); reftable_buf_release(&buf);
} }
static void t_write_empty_key(void) void test_reftable_readwrite__write_empty_key(void)
{ {
struct reftable_write_options opts = { 0 }; struct reftable_write_options opts = { 0 };
struct reftable_buf buf = REFTABLE_BUF_INIT; struct reftable_buf buf = REFTABLE_BUF_INIT;
struct reftable_writer *w = t_reftable_strbuf_writer(&buf, &opts); struct reftable_writer *w = cl_reftable_strbuf_writer(&buf, &opts);
struct reftable_ref_record ref = { struct reftable_ref_record ref = {
.refname = (char *) "", .refname = (char *) "",
.update_index = 1, .update_index = 1,
.value_type = REFTABLE_REF_DELETION, .value_type = REFTABLE_REF_DELETION,
}; };
int err;
reftable_writer_set_limits(w, 1, 1); reftable_writer_set_limits(w, 1, 1);
err = reftable_writer_add_ref(w, &ref); cl_assert_equal_i(reftable_writer_add_ref(w, &ref), REFTABLE_API_ERROR);
check_int(err, ==, REFTABLE_API_ERROR); cl_assert_equal_i(reftable_writer_close(w),
REFTABLE_EMPTY_TABLE_ERROR);
err = reftable_writer_close(w);
check_int(err, ==, REFTABLE_EMPTY_TABLE_ERROR);
reftable_writer_free(w); reftable_writer_free(w);
reftable_buf_release(&buf); reftable_buf_release(&buf);
} }
static void t_write_key_order(void) void test_reftable_readwrite__write_key_order(void)
{ {
struct reftable_write_options opts = { 0 }; struct reftable_write_options opts = { 0 };
struct reftable_buf buf = REFTABLE_BUF_INIT; struct reftable_buf buf = REFTABLE_BUF_INIT;
struct reftable_writer *w = t_reftable_strbuf_writer(&buf, &opts); struct reftable_writer *w = cl_reftable_strbuf_writer(&buf, &opts);
struct reftable_ref_record refs[2] = { struct reftable_ref_record refs[2] = {
{ {
.refname = (char *) "b", .refname = (char *) "b",
@@ -776,24 +758,21 @@ static void t_write_key_order(void)
}, },
} }
}; };
int err;
reftable_writer_set_limits(w, 1, 1); reftable_writer_set_limits(w, 1, 1);
err = reftable_writer_add_ref(w, &refs[0]); cl_assert_equal_i(reftable_writer_add_ref(w, &refs[0]), 0);
check(!err); cl_assert_equal_i(reftable_writer_add_ref(w, &refs[1]),
err = reftable_writer_add_ref(w, &refs[1]); REFTABLE_API_ERROR);
check_int(err, ==, REFTABLE_API_ERROR);
refs[0].update_index = 2; refs[0].update_index = 2;
err = reftable_writer_add_ref(w, &refs[0]); cl_assert_equal_i(reftable_writer_add_ref(w, &refs[0]), REFTABLE_API_ERROR);
check_int(err, ==, REFTABLE_API_ERROR);
reftable_writer_close(w); reftable_writer_close(w);
reftable_writer_free(w); reftable_writer_free(w);
reftable_buf_release(&buf); reftable_buf_release(&buf);
} }
static void t_write_multiple_indices(void) void test_reftable_readwrite__write_multiple_indices(void)
{ {
struct reftable_write_options opts = { struct reftable_write_options opts = {
.block_size = 100, .block_size = 100,
@@ -805,9 +784,10 @@ static void t_write_multiple_indices(void)
struct reftable_writer *writer; struct reftable_writer *writer;
struct reftable_table *table; struct reftable_table *table;
char buf[128]; char buf[128];
int err, i; int i;
int err;
writer = t_reftable_strbuf_writer(&writer_buf, &opts); writer = cl_reftable_strbuf_writer(&writer_buf, &opts);
reftable_writer_set_limits(writer, 1, 1); reftable_writer_set_limits(writer, 1, 1);
for (i = 0; i < 100; i++) { for (i = 0; i < 100; i++) {
struct reftable_ref_record ref = { struct reftable_ref_record ref = {
@@ -819,8 +799,7 @@ static void t_write_multiple_indices(void)
snprintf(buf, sizeof(buf), "refs/heads/%04d", i); snprintf(buf, sizeof(buf), "refs/heads/%04d", i);
ref.refname = buf; ref.refname = buf;
err = reftable_writer_add_ref(writer, &ref); cl_assert_equal_i(reftable_writer_add_ref(writer, &ref), 0);
check(!err);
} }
for (i = 0; i < 100; i++) { for (i = 0; i < 100; i++) {
@@ -836,8 +815,7 @@ static void t_write_multiple_indices(void)
snprintf(buf, sizeof(buf), "refs/heads/%04d", i); snprintf(buf, sizeof(buf), "refs/heads/%04d", i);
log.refname = buf; log.refname = buf;
err = reftable_writer_add_log(writer, &log); cl_assert_equal_i(reftable_writer_add_log(writer, &log), 0);
check(!err);
} }
reftable_writer_close(writer); reftable_writer_close(writer);
@@ -847,22 +825,22 @@ static void t_write_multiple_indices(void)
* for each of the block types. * for each of the block types.
*/ */
stats = reftable_writer_stats(writer); stats = reftable_writer_stats(writer);
check_int(stats->ref_stats.index_offset, >, 0); cl_assert(stats->ref_stats.index_offset > 0);
check_int(stats->obj_stats.index_offset, >, 0); cl_assert(stats->obj_stats.index_offset > 0);
check_int(stats->log_stats.index_offset, >, 0); cl_assert(stats->log_stats.index_offset > 0);
block_source_from_buf(&source, &writer_buf); block_source_from_buf(&source, &writer_buf);
err = reftable_table_new(&table, &source, "filename"); err = reftable_table_new(&table, &source, "filename");
check(!err); cl_assert(!err);
/* /*
* Seeking the log uses the log index now. In case there is any * Seeking the log uses the log index now. In case there is any
* confusion regarding indices we would notice here. * confusion regarding indices we would notice here.
*/ */
err = reftable_table_init_log_iterator(table, &it); err = reftable_table_init_log_iterator(table, &it);
check(!err); cl_assert(!err);
err = reftable_iterator_seek_log(&it, ""); err = reftable_iterator_seek_log(&it, "");
check(!err); cl_assert(!err);
reftable_iterator_destroy(&it); reftable_iterator_destroy(&it);
reftable_writer_free(writer); reftable_writer_free(writer);
@@ -870,7 +848,7 @@ static void t_write_multiple_indices(void)
reftable_buf_release(&writer_buf); reftable_buf_release(&writer_buf);
} }
static void t_write_multi_level_index(void) void test_reftable_readwrite__write_multi_level_index(void)
{ {
struct reftable_write_options opts = { struct reftable_write_options opts = {
.block_size = 100, .block_size = 100,
@@ -883,7 +861,7 @@ static void t_write_multi_level_index(void)
struct reftable_table *table; struct reftable_table *table;
int err; int err;
writer = t_reftable_strbuf_writer(&writer_buf, &opts); writer = cl_reftable_strbuf_writer(&writer_buf, &opts);
reftable_writer_set_limits(writer, 1, 1); reftable_writer_set_limits(writer, 1, 1);
for (size_t i = 0; i < 200; i++) { for (size_t i = 0; i < 200; i++) {
struct reftable_ref_record ref = { struct reftable_ref_record ref = {
@@ -896,8 +874,7 @@ static void t_write_multi_level_index(void)
snprintf(buf, sizeof(buf), "refs/heads/%03" PRIuMAX, (uintmax_t)i); snprintf(buf, sizeof(buf), "refs/heads/%03" PRIuMAX, (uintmax_t)i);
ref.refname = buf; ref.refname = buf;
err = reftable_writer_add_ref(writer, &ref); cl_assert_equal_i(reftable_writer_add_ref(writer, &ref), 0);
check(!err);
} }
reftable_writer_close(writer); reftable_writer_close(writer);
@@ -906,19 +883,19 @@ static void t_write_multi_level_index(void)
* multi-level index. * multi-level index.
*/ */
stats = reftable_writer_stats(writer); stats = reftable_writer_stats(writer);
check_int(stats->ref_stats.max_index_level, ==, 2); cl_assert_equal_i(stats->ref_stats.max_index_level, 2);
block_source_from_buf(&source, &writer_buf); block_source_from_buf(&source, &writer_buf);
err = reftable_table_new(&table, &source, "filename"); err = reftable_table_new(&table, &source, "filename");
check(!err); cl_assert(!err);
/* /*
* Seeking the last ref should work as expected. * Seeking the last ref should work as expected.
*/ */
err = reftable_table_init_ref_iterator(table, &it); err = reftable_table_init_ref_iterator(table, &it);
check(!err); cl_assert(!err);
err = reftable_iterator_seek_ref(&it, "refs/heads/199"); err = reftable_iterator_seek_ref(&it, "refs/heads/199");
check(!err); cl_assert(!err);
reftable_iterator_destroy(&it); reftable_iterator_destroy(&it);
reftable_writer_free(writer); reftable_writer_free(writer);
@@ -927,7 +904,7 @@ static void t_write_multi_level_index(void)
reftable_buf_release(&buf); reftable_buf_release(&buf);
} }
static void t_corrupt_table_empty(void) void test_reftable_readwrite__corrupt_table_empty(void)
{ {
struct reftable_buf buf = REFTABLE_BUF_INIT; struct reftable_buf buf = REFTABLE_BUF_INIT;
struct reftable_block_source source = { 0 }; struct reftable_block_source source = { 0 };
@@ -936,50 +913,22 @@ static void t_corrupt_table_empty(void)
block_source_from_buf(&source, &buf); block_source_from_buf(&source, &buf);
err = reftable_table_new(&table, &source, "file.log"); err = reftable_table_new(&table, &source, "file.log");
check_int(err, ==, REFTABLE_FORMAT_ERROR); cl_assert_equal_i(err, REFTABLE_FORMAT_ERROR);
} }
static void t_corrupt_table(void) void test_reftable_readwrite__corrupt_table(void)
{ {
uint8_t zeros[1024] = { 0 }; uint8_t zeros[1024] = { 0 };
struct reftable_buf buf = REFTABLE_BUF_INIT; struct reftable_buf buf = REFTABLE_BUF_INIT;
struct reftable_block_source source = { 0 }; struct reftable_block_source source = { 0 };
struct reftable_table *table; struct reftable_table *table;
int err; int err;
check(!reftable_buf_add(&buf, zeros, sizeof(zeros)));
cl_assert(!reftable_buf_add(&buf, zeros, sizeof(zeros)));
block_source_from_buf(&source, &buf); block_source_from_buf(&source, &buf);
err = reftable_table_new(&table, &source, "file.log"); err = reftable_table_new(&table, &source, "file.log");
check_int(err, ==, REFTABLE_FORMAT_ERROR); cl_assert_equal_i(err, REFTABLE_FORMAT_ERROR);
reftable_buf_release(&buf); reftable_buf_release(&buf);
} }
int cmd_main(int argc UNUSED, const char *argv[] UNUSED)
{
TEST(t_buffer(), "strbuf works as blocksource");
TEST(t_corrupt_table(), "read-write on corrupted table");
TEST(t_corrupt_table_empty(), "read-write on an empty table");
TEST(t_log_buffer_size(), "buffer extension for log compression");
TEST(t_log_overflow(), "log overflow returns expected error");
TEST(t_log_write_limits(), "writer limits for writing log records");
TEST(t_log_write_read(), "read-write on log records");
TEST(t_log_zlib_corruption(), "reading corrupted log record returns expected error");
TEST(t_table_read_api(), "read on a table");
TEST(t_table_read_write_seek_index(), "read-write on a table with index");
TEST(t_table_read_write_seek_linear(), "read-write on a table without index (SHA1)");
TEST(t_table_read_write_seek_linear_sha256(), "read-write on a table without index (SHA256)");
TEST(t_table_read_write_sequential(), "sequential read-write on a table");
TEST(t_table_refs_for_no_index(), "refs-only table with no index");
TEST(t_table_refs_for_obj_index(), "refs-only table with index");
TEST(t_table_write_small_table(), "write_table works");
TEST(t_write_empty_key(), "write on refs with empty keys");
TEST(t_write_empty_table(), "read-write on empty tables");
TEST(t_write_key_order(), "refs must be written in increasing order");
TEST(t_write_multi_level_index(), "table with multi-level index");
TEST(t_write_multiple_indices(), "table with indices for multiple block types");
TEST(t_write_object_id_length(), "prefix compression on writing refs");
TEST(t_write_object_id_min_length(), "prefix compression on writing refs");
return test_done();
}

View File

@@ -6,7 +6,8 @@
https://developers.google.com/open-source/licenses/bsd https://developers.google.com/open-source/licenses/bsd
*/ */
#include "test-lib.h" #include "unit-test.h"
#include "lib-reftable.h"
#include "reftable/basics.h" #include "reftable/basics.h"
#include "reftable/constants.h" #include "reftable/constants.h"
#include "reftable/record.h" #include "reftable/record.h"
@@ -17,16 +18,17 @@ static void t_copy(struct reftable_record *rec)
uint8_t typ; uint8_t typ;
typ = reftable_record_type(rec); typ = reftable_record_type(rec);
check(!reftable_record_init(&copy, typ)); cl_assert_equal_i(reftable_record_init(&copy, typ), 0);
reftable_record_copy_from(&copy, rec, REFTABLE_HASH_SIZE_SHA1); reftable_record_copy_from(&copy, rec, REFTABLE_HASH_SIZE_SHA1);
/* do it twice to catch memory leaks */ /* do it twice to catch memory leaks */
reftable_record_copy_from(&copy, rec, REFTABLE_HASH_SIZE_SHA1); reftable_record_copy_from(&copy, rec, REFTABLE_HASH_SIZE_SHA1);
check(reftable_record_equal(rec, &copy, REFTABLE_HASH_SIZE_SHA1)); cl_assert(reftable_record_equal(rec, &copy,
REFTABLE_HASH_SIZE_SHA1) != 0);
reftable_record_release(&copy); reftable_record_release(&copy);
} }
static void t_varint_roundtrip(void) void test_reftable_record__varint_roundtrip(void)
{ {
uint64_t inputs[] = { 0, uint64_t inputs[] = { 0,
1, 1,
@@ -49,16 +51,16 @@ static void t_varint_roundtrip(void)
int n = put_var_int(&out, in); int n = put_var_int(&out, in);
uint64_t got = 0; uint64_t got = 0;
check_int(n, >, 0); cl_assert(n > 0);
out.len = n; out.len = n;
n = get_var_int(&got, &out); n = get_var_int(&got, &out);
check_int(n, >, 0); cl_assert(n > 0);
check_int(got, ==, in); cl_assert_equal_i(got, in);
} }
} }
static void t_varint_overflow(void) void test_reftable_record__varint_overflow(void)
{ {
unsigned char buf[] = { unsigned char buf[] = {
0xFF, 0xFF, 0xFF, 0xFF, 0xFF, 0xFF, 0xFF, 0xFF,
@@ -70,8 +72,7 @@ static void t_varint_overflow(void)
.len = sizeof(buf), .len = sizeof(buf),
}; };
uint64_t value; uint64_t value;
int err = get_var_int(&value, &view); cl_assert_equal_i(get_var_int(&value, &view), -1);
check_int(err, ==, -1);
} }
static void set_hash(uint8_t *h, int j) static void set_hash(uint8_t *h, int j)
@@ -80,7 +81,7 @@ static void set_hash(uint8_t *h, int j)
h[i] = (j >> i) & 0xff; h[i] = (j >> i) & 0xff;
} }
static void t_reftable_ref_record_comparison(void) void test_reftable_record__ref_record_comparison(void)
{ {
struct reftable_record in[3] = { struct reftable_record in[3] = {
{ {
@@ -102,21 +103,23 @@ static void t_reftable_ref_record_comparison(void)
}; };
int cmp; int cmp;
check(!reftable_record_equal(&in[0], &in[1], REFTABLE_HASH_SIZE_SHA1)); cl_assert(reftable_record_equal(&in[0], &in[1], REFTABLE_HASH_SIZE_SHA1) == 0);
check(!reftable_record_cmp(&in[0], &in[1], &cmp)); cl_assert_equal_i(reftable_record_cmp(&in[0], &in[1], &cmp), 0);
check(!cmp); cl_assert(!cmp);
check(!reftable_record_equal(&in[1], &in[2], REFTABLE_HASH_SIZE_SHA1)); cl_assert(reftable_record_equal(&in[1], &in[2],
check(!reftable_record_cmp(&in[1], &in[2], &cmp)); REFTABLE_HASH_SIZE_SHA1) == 0);
check_int(cmp, >, 0); cl_assert_equal_i(reftable_record_cmp(&in[1], &in[2], &cmp), 0);
cl_assert(cmp > 0);
in[1].u.ref.value_type = in[0].u.ref.value_type; in[1].u.ref.value_type = in[0].u.ref.value_type;
check(reftable_record_equal(&in[0], &in[1], REFTABLE_HASH_SIZE_SHA1)); cl_assert(reftable_record_equal(&in[0], &in[1],
check(!reftable_record_cmp(&in[0], &in[1], &cmp)); REFTABLE_HASH_SIZE_SHA1) != 0);
check(!cmp); cl_assert_equal_i(reftable_record_cmp(&in[0], &in[1], &cmp), 0);
cl_assert(!cmp);
} }
static void t_reftable_ref_record_compare_name(void) void test_reftable_record__ref_record_compare_name(void)
{ {
struct reftable_ref_record recs[3] = { struct reftable_ref_record recs[3] = {
{ {
@@ -130,12 +133,15 @@ static void t_reftable_ref_record_compare_name(void)
}, },
}; };
check_int(reftable_ref_record_compare_name(&recs[0], &recs[1]), <, 0); cl_assert(reftable_ref_record_compare_name(&recs[0],
check_int(reftable_ref_record_compare_name(&recs[1], &recs[0]), >, 0); &recs[1]) < 0);
check_int(reftable_ref_record_compare_name(&recs[0], &recs[2]), ==, 0); cl_assert(reftable_ref_record_compare_name(&recs[1],
&recs[0]) > 0);
cl_assert_equal_i(reftable_ref_record_compare_name(&recs[0],
&recs[2]), 0);
} }
static void t_reftable_ref_record_roundtrip(void) void test_reftable_record__ref_record_roundtrip(void)
{ {
struct reftable_buf scratch = REFTABLE_BUF_INIT; struct reftable_buf scratch = REFTABLE_BUF_INIT;
@@ -172,19 +178,21 @@ static void t_reftable_ref_record_roundtrip(void)
t_copy(&in); t_copy(&in);
check_int(reftable_record_val_type(&in), ==, i); cl_assert_equal_i(reftable_record_val_type(&in), i);
check_int(reftable_record_is_deletion(&in), ==, i == REFTABLE_REF_DELETION); cl_assert_equal_i(reftable_record_is_deletion(&in),
i == REFTABLE_REF_DELETION);
reftable_record_key(&in, &key); reftable_record_key(&in, &key);
n = reftable_record_encode(&in, dest, REFTABLE_HASH_SIZE_SHA1); n = reftable_record_encode(&in, dest, REFTABLE_HASH_SIZE_SHA1);
check_int(n, >, 0); cl_assert(n > 0);
/* decode into a non-zero reftable_record to test for leaks. */ /* decode into a non-zero reftable_record to test for leaks. */
m = reftable_record_decode(&out, key, i, dest, REFTABLE_HASH_SIZE_SHA1, &scratch); m = reftable_record_decode(&out, key, i, dest, REFTABLE_HASH_SIZE_SHA1, &scratch);
check_int(n, ==, m); cl_assert_equal_i(n, m);
check(reftable_ref_record_equal(&in.u.ref, &out.u.ref, cl_assert(reftable_ref_record_equal(&in.u.ref,
REFTABLE_HASH_SIZE_SHA1)); &out.u.ref,
REFTABLE_HASH_SIZE_SHA1) != 0);
reftable_record_release(&in); reftable_record_release(&in);
reftable_buf_release(&key); reftable_buf_release(&key);
@@ -194,7 +202,7 @@ static void t_reftable_ref_record_roundtrip(void)
reftable_buf_release(&scratch); reftable_buf_release(&scratch);
} }
static void t_reftable_log_record_comparison(void) void test_reftable_record__log_record_comparison(void)
{ {
struct reftable_record in[3] = { struct reftable_record in[3] = {
{ {
@@ -215,21 +223,24 @@ static void t_reftable_log_record_comparison(void)
}; };
int cmp; int cmp;
check(!reftable_record_equal(&in[0], &in[1], REFTABLE_HASH_SIZE_SHA1)); cl_assert_equal_i(reftable_record_equal(&in[0], &in[1],
check(!reftable_record_equal(&in[1], &in[2], REFTABLE_HASH_SIZE_SHA1)); REFTABLE_HASH_SIZE_SHA1), 0);
check(!reftable_record_cmp(&in[1], &in[2], &cmp)); cl_assert_equal_i(reftable_record_equal(&in[1], &in[2],
check_int(cmp, >, 0); REFTABLE_HASH_SIZE_SHA1), 0);
cl_assert_equal_i(reftable_record_cmp(&in[1], &in[2], &cmp), 0);
cl_assert(cmp > 0);
/* comparison should be reversed for equal keys, because /* comparison should be reversed for equal keys, because
* comparison is now performed on the basis of update indices */ * comparison is now performed on the basis of update indices */
check(!reftable_record_cmp(&in[0], &in[1], &cmp)); cl_assert_equal_i(reftable_record_cmp(&in[0], &in[1], &cmp), 0);
check_int(cmp, <, 0); cl_assert(cmp < 0);
in[1].u.log.update_index = in[0].u.log.update_index; in[1].u.log.update_index = in[0].u.log.update_index;
check(reftable_record_equal(&in[0], &in[1], REFTABLE_HASH_SIZE_SHA1)); cl_assert(reftable_record_equal(&in[0], &in[1],
check(!reftable_record_cmp(&in[0], &in[1], &cmp)); REFTABLE_HASH_SIZE_SHA1) != 0);
cl_assert_equal_i(reftable_record_cmp(&in[0], &in[1], &cmp), 0);
} }
static void t_reftable_log_record_compare_key(void) void test_reftable_record__log_record_compare_key(void)
{ {
struct reftable_log_record logs[3] = { struct reftable_log_record logs[3] = {
{ {
@@ -246,19 +257,24 @@ static void t_reftable_log_record_compare_key(void)
}, },
}; };
check_int(reftable_log_record_compare_key(&logs[0], &logs[1]), <, 0); cl_assert(reftable_log_record_compare_key(&logs[0],
check_int(reftable_log_record_compare_key(&logs[1], &logs[0]), >, 0); &logs[1]) < 0);
cl_assert(reftable_log_record_compare_key(&logs[1],
&logs[0]) > 0);
logs[1].update_index = logs[0].update_index; logs[1].update_index = logs[0].update_index;
check_int(reftable_log_record_compare_key(&logs[0], &logs[1]), <, 0); cl_assert(reftable_log_record_compare_key(&logs[0],
&logs[1]) < 0);
check_int(reftable_log_record_compare_key(&logs[0], &logs[2]), >, 0); cl_assert(reftable_log_record_compare_key(&logs[0],
check_int(reftable_log_record_compare_key(&logs[2], &logs[0]), <, 0); &logs[2]) > 0);
cl_assert(reftable_log_record_compare_key(&logs[2],
&logs[0]) < 0);
logs[2].update_index = logs[0].update_index; logs[2].update_index = logs[0].update_index;
check_int(reftable_log_record_compare_key(&logs[0], &logs[2]), ==, 0); cl_assert_equal_i(reftable_log_record_compare_key(&logs[0], &logs[2]), 0);
} }
static void t_reftable_log_record_roundtrip(void) void test_reftable_record__log_record_roundtrip(void)
{ {
struct reftable_log_record in[] = { struct reftable_log_record in[] = {
{ {
@@ -292,9 +308,9 @@ static void t_reftable_log_record_roundtrip(void)
set_hash(in[2].value.update.new_hash, 3); set_hash(in[2].value.update.new_hash, 3);
set_hash(in[2].value.update.old_hash, 4); set_hash(in[2].value.update.old_hash, 4);
check(!reftable_log_record_is_deletion(&in[0])); cl_assert_equal_i(reftable_log_record_is_deletion(&in[0]), 0);
check(reftable_log_record_is_deletion(&in[1])); cl_assert(reftable_log_record_is_deletion(&in[1]) != 0);
check(!reftable_log_record_is_deletion(&in[2])); cl_assert_equal_i(reftable_log_record_is_deletion(&in[2]), 0);
for (size_t i = 0; i < ARRAY_SIZE(in); i++) { for (size_t i = 0; i < ARRAY_SIZE(in); i++) {
struct reftable_record rec = { .type = REFTABLE_BLOCK_TYPE_LOG }; struct reftable_record rec = { .type = REFTABLE_BLOCK_TYPE_LOG };
@@ -328,14 +344,14 @@ static void t_reftable_log_record_roundtrip(void)
reftable_record_key(&rec, &key); reftable_record_key(&rec, &key);
n = reftable_record_encode(&rec, dest, REFTABLE_HASH_SIZE_SHA1); n = reftable_record_encode(&rec, dest, REFTABLE_HASH_SIZE_SHA1);
check_int(n, >=, 0); cl_assert(n >= 0);
valtype = reftable_record_val_type(&rec); valtype = reftable_record_val_type(&rec);
m = reftable_record_decode(&out, key, valtype, dest, m = reftable_record_decode(&out, key, valtype, dest,
REFTABLE_HASH_SIZE_SHA1, &scratch); REFTABLE_HASH_SIZE_SHA1, &scratch);
check_int(n, ==, m); cl_assert_equal_i(n, m);
check(reftable_log_record_equal(&in[i], &out.u.log, cl_assert(reftable_log_record_equal(&in[i], &out.u.log,
REFTABLE_HASH_SIZE_SHA1)); REFTABLE_HASH_SIZE_SHA1) != 0);
reftable_log_record_release(&in[i]); reftable_log_record_release(&in[i]);
reftable_buf_release(&key); reftable_buf_release(&key);
reftable_record_release(&out); reftable_record_release(&out);
@@ -344,7 +360,7 @@ static void t_reftable_log_record_roundtrip(void)
reftable_buf_release(&scratch); reftable_buf_release(&scratch);
} }
static void t_key_roundtrip(void) void test_reftable_record__key_roundtrip(void)
{ {
uint8_t buffer[1024] = { 0 }; uint8_t buffer[1024] = { 0 };
struct string_view dest = { struct string_view dest = {
@@ -359,25 +375,28 @@ static void t_key_roundtrip(void)
int n, m; int n, m;
uint8_t rt_extra; uint8_t rt_extra;
check(!reftable_buf_addstr(&last_key, "refs/heads/master")); cl_assert_equal_i(reftable_buf_addstr(&last_key,
check(!reftable_buf_addstr(&key, "refs/tags/bla")); "refs/heads/master"), 0);
cl_assert_equal_i(reftable_buf_addstr(&key,
"refs/tags/bla"), 0);
extra = 6; extra = 6;
n = reftable_encode_key(&restart, dest, last_key, key, extra); n = reftable_encode_key(&restart, dest, last_key, key, extra);
check(!restart); cl_assert(!restart);
check_int(n, >, 0); cl_assert(n > 0);
check(!reftable_buf_addstr(&roundtrip, "refs/heads/master")); cl_assert_equal_i(reftable_buf_addstr(&roundtrip,
"refs/heads/master"), 0);
m = reftable_decode_key(&roundtrip, &rt_extra, dest); m = reftable_decode_key(&roundtrip, &rt_extra, dest);
check_int(n, ==, m); cl_assert_equal_i(n, m);
check(!reftable_buf_cmp(&key, &roundtrip)); cl_assert_equal_i(reftable_buf_cmp(&key, &roundtrip), 0);
check_int(rt_extra, ==, extra); cl_assert_equal_i(rt_extra, extra);
reftable_buf_release(&last_key); reftable_buf_release(&last_key);
reftable_buf_release(&key); reftable_buf_release(&key);
reftable_buf_release(&roundtrip); reftable_buf_release(&roundtrip);
} }
static void t_reftable_obj_record_comparison(void) void test_reftable_record__obj_record_comparison(void)
{ {
uint8_t id_bytes[] = { 0, 1, 2, 3, 4, 5, 6 }; uint8_t id_bytes[] = { 0, 1, 2, 3, 4, 5, 6 };
@@ -405,21 +424,23 @@ static void t_reftable_obj_record_comparison(void)
}; };
int cmp; int cmp;
check(!reftable_record_equal(&in[0], &in[1], REFTABLE_HASH_SIZE_SHA1)); cl_assert_equal_i(reftable_record_equal(&in[0], &in[1],
check(!reftable_record_cmp(&in[0], &in[1], &cmp)); REFTABLE_HASH_SIZE_SHA1), 0);
check(!cmp); cl_assert_equal_i(reftable_record_cmp(&in[0], &in[1], &cmp), 0);
cl_assert(!cmp);
check(!reftable_record_equal(&in[1], &in[2], REFTABLE_HASH_SIZE_SHA1)); cl_assert_equal_i(reftable_record_equal(&in[1], &in[2],
check(!reftable_record_cmp(&in[1], &in[2], &cmp)); REFTABLE_HASH_SIZE_SHA1), 0);
check_int(cmp, >, 0); cl_assert_equal_i(reftable_record_cmp(&in[1], &in[2], &cmp), 0);
cl_assert(cmp > 0);
in[1].u.obj.offset_len = in[0].u.obj.offset_len; in[1].u.obj.offset_len = in[0].u.obj.offset_len;
check(reftable_record_equal(&in[0], &in[1], REFTABLE_HASH_SIZE_SHA1)); cl_assert(reftable_record_equal(&in[0], &in[1], REFTABLE_HASH_SIZE_SHA1) != 0);
check(!reftable_record_cmp(&in[0], &in[1], &cmp)); cl_assert_equal_i(reftable_record_cmp(&in[0], &in[1], &cmp), 0);
check(!cmp); cl_assert(!cmp);
} }
static void t_reftable_obj_record_roundtrip(void) void test_reftable_record__obj_record_roundtrip(void)
{ {
uint8_t testHash1[REFTABLE_HASH_SIZE_SHA1] = { 1, 2, 3, 4, 0 }; uint8_t testHash1[REFTABLE_HASH_SIZE_SHA1] = { 1, 2, 3, 4, 0 };
uint64_t till9[] = { 1, 2, 3, 4, 500, 600, 700, 800, 9000 }; uint64_t till9[] = { 1, 2, 3, 4, 500, 600, 700, 800, 9000 };
@@ -460,17 +481,18 @@ static void t_reftable_obj_record_roundtrip(void)
int n, m; int n, m;
uint8_t extra; uint8_t extra;
check(!reftable_record_is_deletion(&in)); cl_assert_equal_i(reftable_record_is_deletion(&in), 0);
t_copy(&in); t_copy(&in);
reftable_record_key(&in, &key); reftable_record_key(&in, &key);
n = reftable_record_encode(&in, dest, REFTABLE_HASH_SIZE_SHA1); n = reftable_record_encode(&in, dest, REFTABLE_HASH_SIZE_SHA1);
check_int(n, >, 0); cl_assert(n > 0);
extra = reftable_record_val_type(&in); extra = reftable_record_val_type(&in);
m = reftable_record_decode(&out, key, extra, dest, m = reftable_record_decode(&out, key, extra, dest,
REFTABLE_HASH_SIZE_SHA1, &scratch); REFTABLE_HASH_SIZE_SHA1, &scratch);
check_int(n, ==, m); cl_assert_equal_i(n, m);
check(reftable_record_equal(&in, &out, REFTABLE_HASH_SIZE_SHA1)); cl_assert(reftable_record_equal(&in, &out,
REFTABLE_HASH_SIZE_SHA1) != 0);
reftable_buf_release(&key); reftable_buf_release(&key);
reftable_record_release(&out); reftable_record_release(&out);
} }
@@ -478,7 +500,7 @@ static void t_reftable_obj_record_roundtrip(void)
reftable_buf_release(&scratch); reftable_buf_release(&scratch);
} }
static void t_reftable_index_record_comparison(void) void test_reftable_record__index_record_comparison(void)
{ {
struct reftable_record in[3] = { struct reftable_record in[3] = {
{ {
@@ -499,28 +521,33 @@ static void t_reftable_index_record_comparison(void)
}; };
int cmp; int cmp;
check(!reftable_buf_addstr(&in[0].u.idx.last_key, "refs/heads/master")); cl_assert_equal_i(reftable_buf_addstr(&in[0].u.idx.last_key,
check(!reftable_buf_addstr(&in[1].u.idx.last_key, "refs/heads/master")); "refs/heads/master"), 0);
check(!reftable_buf_addstr(&in[2].u.idx.last_key, "refs/heads/branch")); cl_assert_equal_i(reftable_buf_addstr(&in[1].u.idx.last_key, "refs/heads/master"), 0);
cl_assert(reftable_buf_addstr(&in[2].u.idx.last_key,
"refs/heads/branch") == 0);
check(!reftable_record_equal(&in[0], &in[1], REFTABLE_HASH_SIZE_SHA1)); cl_assert_equal_i(reftable_record_equal(&in[0], &in[1],
check(!reftable_record_cmp(&in[0], &in[1], &cmp)); REFTABLE_HASH_SIZE_SHA1), 0);
check(!cmp); cl_assert_equal_i(reftable_record_cmp(&in[0], &in[1], &cmp), 0);
cl_assert(!cmp);
check(!reftable_record_equal(&in[1], &in[2], REFTABLE_HASH_SIZE_SHA1)); cl_assert_equal_i(reftable_record_equal(&in[1], &in[2],
check(!reftable_record_cmp(&in[1], &in[2], &cmp)); REFTABLE_HASH_SIZE_SHA1), 0);
check_int(cmp, >, 0); cl_assert_equal_i(reftable_record_cmp(&in[1], &in[2], &cmp), 0);
cl_assert(cmp > 0);
in[1].u.idx.offset = in[0].u.idx.offset; in[1].u.idx.offset = in[0].u.idx.offset;
check(reftable_record_equal(&in[0], &in[1], REFTABLE_HASH_SIZE_SHA1)); cl_assert(reftable_record_equal(&in[0], &in[1],
check(!reftable_record_cmp(&in[0], &in[1], &cmp)); REFTABLE_HASH_SIZE_SHA1) != 0);
check(!cmp); cl_assert_equal_i(reftable_record_cmp(&in[0], &in[1], &cmp), 0);
cl_assert(!cmp);
for (size_t i = 0; i < ARRAY_SIZE(in); i++) for (size_t i = 0; i < ARRAY_SIZE(in); i++)
reftable_record_release(&in[i]); reftable_record_release(&in[i]);
} }
static void t_reftable_index_record_roundtrip(void) void test_reftable_record__index_record_roundtrip(void)
{ {
struct reftable_record in = { struct reftable_record in = {
.type = REFTABLE_BLOCK_TYPE_INDEX, .type = REFTABLE_BLOCK_TYPE_INDEX,
@@ -543,43 +570,26 @@ static void t_reftable_index_record_roundtrip(void)
int n, m; int n, m;
uint8_t extra; uint8_t extra;
check(!reftable_buf_addstr(&in.u.idx.last_key, "refs/heads/master")); cl_assert_equal_i(reftable_buf_addstr(&in.u.idx.last_key,
"refs/heads/master"), 0);
reftable_record_key(&in, &key); reftable_record_key(&in, &key);
t_copy(&in); t_copy(&in);
check(!reftable_record_is_deletion(&in)); cl_assert_equal_i(reftable_record_is_deletion(&in), 0);
check(!reftable_buf_cmp(&key, &in.u.idx.last_key)); cl_assert_equal_i(reftable_buf_cmp(&key, &in.u.idx.last_key), 0);
n = reftable_record_encode(&in, dest, REFTABLE_HASH_SIZE_SHA1); n = reftable_record_encode(&in, dest, REFTABLE_HASH_SIZE_SHA1);
check_int(n, >, 0); cl_assert(n > 0);
extra = reftable_record_val_type(&in); extra = reftable_record_val_type(&in);
m = reftable_record_decode(&out, key, extra, dest, REFTABLE_HASH_SIZE_SHA1, m = reftable_record_decode(&out, key, extra, dest,
&scratch); REFTABLE_HASH_SIZE_SHA1, &scratch);
check_int(m, ==, n); cl_assert_equal_i(m, n);
check(reftable_record_equal(&in, &out, REFTABLE_HASH_SIZE_SHA1)); cl_assert(reftable_record_equal(&in, &out,
REFTABLE_HASH_SIZE_SHA1) != 0);
reftable_record_release(&out); reftable_record_release(&out);
reftable_buf_release(&key); reftable_buf_release(&key);
reftable_buf_release(&scratch); reftable_buf_release(&scratch);
reftable_buf_release(&in.u.idx.last_key); reftable_buf_release(&in.u.idx.last_key);
} }
int cmd_main(int argc UNUSED, const char *argv[] UNUSED)
{
TEST(t_reftable_ref_record_comparison(), "comparison operations work on ref record");
TEST(t_reftable_log_record_comparison(), "comparison operations work on log record");
TEST(t_reftable_index_record_comparison(), "comparison operations work on index record");
TEST(t_reftable_obj_record_comparison(), "comparison operations work on obj record");
TEST(t_reftable_ref_record_compare_name(), "reftable_ref_record_compare_name works");
TEST(t_reftable_log_record_compare_key(), "reftable_log_record_compare_key works");
TEST(t_reftable_log_record_roundtrip(), "record operations work on log record");
TEST(t_reftable_ref_record_roundtrip(), "record operations work on ref record");
TEST(t_varint_roundtrip(), "put_var_int and get_var_int work");
TEST(t_varint_overflow(), "get_var_int notices an integer overflow");
TEST(t_key_roundtrip(), "reftable_encode_key and reftable_decode_key work");
TEST(t_reftable_obj_record_roundtrip(), "record operations work on obj record");
TEST(t_reftable_index_record_roundtrip(), "record operations work on index record");
return test_done();
}

View File

@@ -1,4 +1,4 @@
#include "test-lib.h" #include "unit-test.h"
#include "lib-reftable.h" #include "lib-reftable.h"
#include "reftable/blocksource.h" #include "reftable/blocksource.h"
#include "reftable/constants.h" #include "reftable/constants.h"
@@ -6,7 +6,7 @@
#include "reftable/table.h" #include "reftable/table.h"
#include "strbuf.h" #include "strbuf.h"
static int t_table_seek_once(void) void test_reftable_table__seek_once(void)
{ {
struct reftable_ref_record records[] = { struct reftable_ref_record records[] = {
{ {
@@ -22,32 +22,32 @@ static int t_table_seek_once(void)
struct reftable_buf buf = REFTABLE_BUF_INIT; struct reftable_buf buf = REFTABLE_BUF_INIT;
int ret; int ret;
t_reftable_write_to_buf(&buf, records, ARRAY_SIZE(records), NULL, 0, NULL); cl_reftable_write_to_buf(&buf, records, ARRAY_SIZE(records), NULL, 0, NULL);
block_source_from_buf(&source, &buf); block_source_from_buf(&source, &buf);
ret = reftable_table_new(&table, &source, "name"); ret = reftable_table_new(&table, &source, "name");
check(!ret); cl_assert(!ret);
reftable_table_init_ref_iterator(table, &it); reftable_table_init_ref_iterator(table, &it);
ret = reftable_iterator_seek_ref(&it, ""); ret = reftable_iterator_seek_ref(&it, "");
check(!ret); cl_assert(!ret);
ret = reftable_iterator_next_ref(&it, &ref); ret = reftable_iterator_next_ref(&it, &ref);
check(!ret); cl_assert(!ret);
ret = reftable_ref_record_equal(&ref, &records[0], REFTABLE_HASH_SIZE_SHA1); ret = reftable_ref_record_equal(&ref, &records[0],
check_int(ret, ==, 1); REFTABLE_HASH_SIZE_SHA1);
cl_assert_equal_i(ret, 1);
ret = reftable_iterator_next_ref(&it, &ref); ret = reftable_iterator_next_ref(&it, &ref);
check_int(ret, ==, 1); cl_assert_equal_i(ret, 1);
reftable_ref_record_release(&ref); reftable_ref_record_release(&ref);
reftable_iterator_destroy(&it); reftable_iterator_destroy(&it);
reftable_table_decref(table); reftable_table_decref(table);
reftable_buf_release(&buf); reftable_buf_release(&buf);
return 0;
} }
static int t_table_reseek(void) void test_reftable_table__reseek(void)
{ {
struct reftable_ref_record records[] = { struct reftable_ref_record records[] = {
{ {
@@ -63,35 +63,35 @@ static int t_table_reseek(void)
struct reftable_buf buf = REFTABLE_BUF_INIT; struct reftable_buf buf = REFTABLE_BUF_INIT;
int ret; int ret;
t_reftable_write_to_buf(&buf, records, ARRAY_SIZE(records), NULL, 0, NULL); cl_reftable_write_to_buf(&buf, records, ARRAY_SIZE(records),
NULL, 0, NULL);
block_source_from_buf(&source, &buf); block_source_from_buf(&source, &buf);
ret = reftable_table_new(&table, &source, "name"); ret = reftable_table_new(&table, &source, "name");
check(!ret); cl_assert(!ret);
reftable_table_init_ref_iterator(table, &it); reftable_table_init_ref_iterator(table, &it);
for (size_t i = 0; i < 5; i++) { for (size_t i = 0; i < 5; i++) {
ret = reftable_iterator_seek_ref(&it, ""); ret = reftable_iterator_seek_ref(&it, "");
check(!ret); cl_assert(!ret);
ret = reftable_iterator_next_ref(&it, &ref); ret = reftable_iterator_next_ref(&it, &ref);
check(!ret); cl_assert(!ret);
ret = reftable_ref_record_equal(&ref, &records[0], REFTABLE_HASH_SIZE_SHA1); ret = reftable_ref_record_equal(&ref, &records[0], REFTABLE_HASH_SIZE_SHA1);
check_int(ret, ==, 1); cl_assert_equal_i(ret, 1);
ret = reftable_iterator_next_ref(&it, &ref); ret = reftable_iterator_next_ref(&it, &ref);
check_int(ret, ==, 1); cl_assert_equal_i(ret, 1);
} }
reftable_ref_record_release(&ref); reftable_ref_record_release(&ref);
reftable_iterator_destroy(&it); reftable_iterator_destroy(&it);
reftable_table_decref(table); reftable_table_decref(table);
reftable_buf_release(&buf); reftable_buf_release(&buf);
return 0;
} }
static int t_table_block_iterator(void) void test_reftable_table__block_iterator(void)
{ {
struct reftable_block_source source = { 0 }; struct reftable_block_source source = { 0 };
struct reftable_table_iterator it = { 0 }; struct reftable_table_iterator it = { 0 };
@@ -147,14 +147,14 @@ static int t_table_block_iterator(void)
(uintmax_t) i); (uintmax_t) i);
} }
t_reftable_write_to_buf(&buf, records, nrecords, NULL, 0, NULL); cl_reftable_write_to_buf(&buf, records, nrecords, NULL, 0, NULL);
block_source_from_buf(&source, &buf); block_source_from_buf(&source, &buf);
ret = reftable_table_new(&table, &source, "name"); ret = reftable_table_new(&table, &source, "name");
check(!ret); cl_assert(!ret);
ret = reftable_table_iterator_init(&it, table); ret = reftable_table_iterator_init(&it, table);
check(!ret); cl_assert(!ret);
for (size_t i = 0; i < ARRAY_SIZE(expected_blocks); i++) { for (size_t i = 0; i < ARRAY_SIZE(expected_blocks); i++) {
struct reftable_iterator record_it = { 0 }; struct reftable_iterator record_it = { 0 };
@@ -163,22 +163,26 @@ static int t_table_block_iterator(void)
}; };
ret = reftable_table_iterator_next(&it, &block); ret = reftable_table_iterator_next(&it, &block);
check(!ret); cl_assert(!ret);
check_int(block->block_type, ==, expected_blocks[i].block_type); cl_assert_equal_i(block->block_type,
check_int(block->header_off, ==, expected_blocks[i].header_off); expected_blocks[i].block_type);
check_int(block->restart_count, ==, expected_blocks[i].restart_count); cl_assert_equal_i(block->header_off,
expected_blocks[i].header_off);
cl_assert_equal_i(block->restart_count,
expected_blocks[i].restart_count);
ret = reftable_block_init_iterator(block, &record_it); ret = reftable_block_init_iterator(block, &record_it);
check(!ret); cl_assert(!ret);
for (size_t j = 0; ; j++) { for (size_t j = 0; ; j++) {
ret = iterator_next(&record_it, &record); ret = iterator_next(&record_it, &record);
if (ret > 0) { if (ret > 0) {
check_int(j, ==, expected_blocks[i].record_count); cl_assert_equal_i(j,
expected_blocks[i].record_count);
break; break;
} }
check(!ret); cl_assert(!ret);
} }
reftable_iterator_destroy(&record_it); reftable_iterator_destroy(&record_it);
@@ -186,7 +190,7 @@ static int t_table_block_iterator(void)
} }
ret = reftable_table_iterator_next(&it, &block); ret = reftable_table_iterator_next(&it, &block);
check_int(ret, ==, 1); cl_assert_equal_i(ret, 1);
for (size_t i = 0; i < nrecords; i++) for (size_t i = 0; i < nrecords; i++)
reftable_free(records[i].refname); reftable_free(records[i].refname);
@@ -194,13 +198,4 @@ static int t_table_block_iterator(void)
reftable_table_decref(table); reftable_table_decref(table);
reftable_buf_release(&buf); reftable_buf_release(&buf);
reftable_free(records); reftable_free(records);
return 0;
}
int cmd_main(int argc UNUSED, const char *argv[] UNUSED)
{
TEST(t_table_seek_once(), "table can seek once");
TEST(t_table_reseek(), "table can reseek multiple times");
TEST(t_table_block_iterator(), "table can iterate through blocks");
return test_done();
} }