[thirdparty/git.git] / reftable / merged_test.c

/*
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 "merged.h"

#include "system.h"

#include "basics.h"
#include "blocksource.h"
#include "reader.h"
#include "record.h"
#include "test_framework.h"
#include "reftable-merged.h"
#include "reftable-tests.h"
#include "reftable-generic.h"
#include "reftable-writer.h"

static void write_test_table(struct strbuf *buf,
			     struct reftable_ref_record refs[], int n)
{
	uint64_t min = 0xffffffff;
	uint64_t max = 0;
	int i = 0;
	int err;

	struct reftable_write_options opts = {
		.block_size = 256,
	};
	struct reftable_writer *w = NULL;
	for (i = 0; i < n; i++) {
		uint64_t ui = refs[i].update_index;
		if (ui > max) {
			max = ui;
		}
		if (ui < min) {
			min = ui;
		}
	}

	w = reftable_new_writer(&strbuf_add_void, &noop_flush, buf, &opts);
	reftable_writer_set_limits(w, min, max);

	for (i = 0; i < n; i++) {
		uint64_t before = refs[i].update_index;
		int n = reftable_writer_add_ref(w, &refs[i]);
		EXPECT(n == 0);
		EXPECT(before == refs[i].update_index);
	}

	err = reftable_writer_close(w);
	EXPECT_ERR(err);

	reftable_writer_free(w);
}

static void write_test_log_table(struct strbuf *buf,
				 struct reftable_log_record logs[], int n,
				 uint64_t update_index)
{
	int i = 0;
	int err;

	struct reftable_write_options opts = {
		.block_size = 256,
		.exact_log_message = 1,
	};
	struct reftable_writer *w = NULL;
	w = reftable_new_writer(&strbuf_add_void, &noop_flush, buf, &opts);
	reftable_writer_set_limits(w, update_index, update_index);

	for (i = 0; i < n; i++) {
		int err = reftable_writer_add_log(w, &logs[i]);
		EXPECT_ERR(err);
	}

	err = reftable_writer_close(w);
	EXPECT_ERR(err);

	reftable_writer_free(w);
}

static struct reftable_merged_table *
merged_table_from_records(struct reftable_ref_record **refs,
			  struct reftable_block_source **source,
			  struct reftable_reader ***readers, int *sizes,
			  struct strbuf *buf, size_t n)
{
	struct reftable_merged_table *mt = NULL;
	struct reftable_table *tabs;
	int err;

	REFTABLE_CALLOC_ARRAY(tabs, n);
	REFTABLE_CALLOC_ARRAY(*readers, n);
	REFTABLE_CALLOC_ARRAY(*source, n);

	for (size_t i = 0; i < n; i++) {
		write_test_table(&buf[i], refs[i], sizes[i]);
		block_source_from_strbuf(&(*source)[i], &buf[i]);

		err = reftable_new_reader(&(*readers)[i], &(*source)[i],
					  "name");
		EXPECT_ERR(err);
		reftable_table_from_reader(&tabs[i], (*readers)[i]);
	}

	err = reftable_new_merged_table(&mt, tabs, n, GIT_SHA1_FORMAT_ID);
	EXPECT_ERR(err);
	return mt;
}

static void readers_destroy(struct reftable_reader **readers, size_t n)
{
	int i = 0;
	for (; i < n; i++)
		reftable_reader_free(readers[i]);
	reftable_free(readers);
}

static void test_merged_between(void)
{
	struct reftable_ref_record r1[] = { {
		.refname = "b",
		.update_index = 1,
		.value_type = REFTABLE_REF_VAL1,
		.value.val1 = { 1, 2, 3, 0 },
	} };
	struct reftable_ref_record r2[] = { {
		.refname = "a",
		.update_index = 2,
		.value_type = REFTABLE_REF_DELETION,
	} };

	struct reftable_ref_record *refs[] = { r1, r2 };
	int sizes[] = { 1, 1 };
	struct strbuf bufs[2] = { STRBUF_INIT, STRBUF_INIT };
	struct reftable_block_source *bs = NULL;
	struct reftable_reader **readers = NULL;
	struct reftable_merged_table *mt =
		merged_table_from_records(refs, &bs, &readers, sizes, bufs, 2);
	int i;
	struct reftable_ref_record ref = { NULL };
	struct reftable_iterator it = { NULL };
	int err = reftable_merged_table_seek_ref(mt, &it, "a");
	EXPECT_ERR(err);

	err = reftable_iterator_next_ref(&it, &ref);
	EXPECT_ERR(err);
	EXPECT(ref.update_index == 2);
	reftable_ref_record_release(&ref);
	reftable_iterator_destroy(&it);
	readers_destroy(readers, 2);
	reftable_merged_table_free(mt);
	for (i = 0; i < ARRAY_SIZE(bufs); i++) {
		strbuf_release(&bufs[i]);
	}
	reftable_free(bs);
}

static void test_merged(void)
{
	struct reftable_ref_record r1[] = {
		{
			.refname = "a",
			.update_index = 1,
			.value_type = REFTABLE_REF_VAL1,
			.value.val1 = { 1 },
		},
		{
			.refname = "b",
			.update_index = 1,
			.value_type = REFTABLE_REF_VAL1,
			.value.val1 = { 1 },
		},
		{
			.refname = "c",
			.update_index = 1,
			.value_type = REFTABLE_REF_VAL1,
			.value.val1 = { 1 },
		}
	};
	struct reftable_ref_record r2[] = { {
		.refname = "a",
		.update_index = 2,
		.value_type = REFTABLE_REF_DELETION,
	} };
	struct reftable_ref_record r3[] = {
		{
			.refname = "c",
			.update_index = 3,
			.value_type = REFTABLE_REF_VAL1,
			.value.val1 = { 2 },
		},
		{
			.refname = "d",
			.update_index = 3,
			.value_type = REFTABLE_REF_VAL1,
			.value.val1 = { 1 },
		},
	};

	struct reftable_ref_record *want[] = {
		&r2[0],
		&r1[1],
		&r3[0],
		&r3[1],
	};

	struct reftable_ref_record *refs[] = { r1, r2, r3 };
	int sizes[3] = { 3, 1, 2 };
	struct strbuf bufs[3] = { STRBUF_INIT, STRBUF_INIT, STRBUF_INIT };
	struct reftable_block_source *bs = NULL;
	struct reftable_reader **readers = NULL;
	struct reftable_merged_table *mt =
		merged_table_from_records(refs, &bs, &readers, sizes, bufs, 3);

	struct reftable_iterator it = { NULL };
	int err = reftable_merged_table_seek_ref(mt, &it, "a");
	struct reftable_ref_record *out = NULL;
	size_t len = 0;
	size_t cap = 0;
	int i = 0;

	EXPECT_ERR(err);
	EXPECT(reftable_merged_table_hash_id(mt) == GIT_SHA1_FORMAT_ID);
	EXPECT(reftable_merged_table_min_update_index(mt) == 1);

	while (len < 100) { /* cap loops/recursion. */
		struct reftable_ref_record ref = { NULL };
		int err = reftable_iterator_next_ref(&it, &ref);
		if (err > 0)
			break;

		REFTABLE_ALLOC_GROW(out, len + 1, cap);
		out[len++] = ref;
	}
	reftable_iterator_destroy(&it);

	EXPECT(ARRAY_SIZE(want) == len);
	for (i = 0; i < len; i++) {
		EXPECT(reftable_ref_record_equal(want[i], &out[i],
						 GIT_SHA1_RAWSZ));
	}
	for (i = 0; i < len; i++) {
		reftable_ref_record_release(&out[i]);
	}
	reftable_free(out);

	for (i = 0; i < 3; i++) {
		strbuf_release(&bufs[i]);
	}
	readers_destroy(readers, 3);
	reftable_merged_table_free(mt);
	reftable_free(bs);
}

static struct reftable_merged_table *
merged_table_from_log_records(struct reftable_log_record **logs,
			      struct reftable_block_source **source,
			      struct reftable_reader ***readers, int *sizes,
			      struct strbuf *buf, size_t n)
{
	struct reftable_merged_table *mt = NULL;
	struct reftable_table *tabs;
	int err;

	REFTABLE_CALLOC_ARRAY(tabs, n);
	REFTABLE_CALLOC_ARRAY(*readers, n);
	REFTABLE_CALLOC_ARRAY(*source, n);

	for (size_t i = 0; i < n; i++) {
		write_test_log_table(&buf[i], logs[i], sizes[i], i + 1);
		block_source_from_strbuf(&(*source)[i], &buf[i]);

		err = reftable_new_reader(&(*readers)[i], &(*source)[i],
					  "name");
		EXPECT_ERR(err);
		reftable_table_from_reader(&tabs[i], (*readers)[i]);
	}

	err = reftable_new_merged_table(&mt, tabs, n, GIT_SHA1_FORMAT_ID);
	EXPECT_ERR(err);
	return mt;
}

static void test_merged_logs(void)
{
	struct reftable_log_record r1[] = {
		{
			.refname = "a",
			.update_index = 2,
			.value_type = REFTABLE_LOG_UPDATE,
			.value.update = {
				.old_hash = { 2 },
				/* deletion */
				.name = "jane doe",
				.email = "jane@invalid",
				.message = "message2",
			}
		},
		{
			.refname = "a",
			.update_index = 1,
			.value_type = REFTABLE_LOG_UPDATE,
			.value.update = {
				.old_hash = { 1 },
				.new_hash = { 2 },
				.name = "jane doe",
				.email = "jane@invalid",
				.message = "message1",
			}
		},
	};
	struct reftable_log_record r2[] = {
		{
			.refname = "a",
			.update_index = 3,
			.value_type = REFTABLE_LOG_UPDATE,
			.value.update = {
				.new_hash = { 3 },
				.name = "jane doe",
				.email = "jane@invalid",
				.message = "message3",
			}
		},
	};
	struct reftable_log_record r3[] = {
		{
			.refname = "a",
			.update_index = 2,
			.value_type = REFTABLE_LOG_DELETION,
		},
	};
	struct reftable_log_record *want[] = {
		&r2[0],
		&r3[0],
		&r1[1],
	};

	struct reftable_log_record *logs[] = { r1, r2, r3 };
	int sizes[3] = { 2, 1, 1 };
	struct strbuf bufs[3] = { STRBUF_INIT, STRBUF_INIT, STRBUF_INIT };
	struct reftable_block_source *bs = NULL;
	struct reftable_reader **readers = NULL;
	struct reftable_merged_table *mt = merged_table_from_log_records(
		logs, &bs, &readers, sizes, bufs, 3);

	struct reftable_iterator it = { NULL };
	int err = reftable_merged_table_seek_log(mt, &it, "a");
	struct reftable_log_record *out = NULL;
	size_t len = 0;
	size_t cap = 0;
	int i = 0;

	EXPECT_ERR(err);
	EXPECT(reftable_merged_table_hash_id(mt) == GIT_SHA1_FORMAT_ID);
	EXPECT(reftable_merged_table_min_update_index(mt) == 1);

	while (len < 100) { /* cap loops/recursion. */
		struct reftable_log_record log = { NULL };
		int err = reftable_iterator_next_log(&it, &log);
		if (err > 0)
			break;

		REFTABLE_ALLOC_GROW(out, len + 1, cap);
		out[len++] = log;
	}
	reftable_iterator_destroy(&it);

	EXPECT(ARRAY_SIZE(want) == len);
	for (i = 0; i < len; i++) {
		EXPECT(reftable_log_record_equal(want[i], &out[i],
						 GIT_SHA1_RAWSZ));
	}

	err = reftable_merged_table_seek_log_at(mt, &it, "a", 2);
	EXPECT_ERR(err);
	reftable_log_record_release(&out[0]);
	err = reftable_iterator_next_log(&it, &out[0]);
	EXPECT_ERR(err);
	EXPECT(reftable_log_record_equal(&out[0], &r3[0], GIT_SHA1_RAWSZ));
	reftable_iterator_destroy(&it);

	for (i = 0; i < len; i++) {
		reftable_log_record_release(&out[i]);
	}
	reftable_free(out);

	for (i = 0; i < 3; i++) {
		strbuf_release(&bufs[i]);
	}
	readers_destroy(readers, 3);
	reftable_merged_table_free(mt);
	reftable_free(bs);
}

static void test_default_write_opts(void)
{
	struct reftable_write_options opts = { 0 };
	struct strbuf buf = STRBUF_INIT;
	struct reftable_writer *w =
		reftable_new_writer(&strbuf_add_void, &noop_flush, &buf, &opts);

	struct reftable_ref_record rec = {
		.refname = "master",
		.update_index = 1,
	};
	int err;
	struct reftable_block_source source = { NULL };
	struct reftable_table *tab = reftable_calloc(1, sizeof(*tab));
	uint32_t hash_id;
	struct reftable_reader *rd = NULL;
	struct reftable_merged_table *merged = NULL;

	reftable_writer_set_limits(w, 1, 1);

	err = reftable_writer_add_ref(w, &rec);
	EXPECT_ERR(err);

	err = reftable_writer_close(w);
	EXPECT_ERR(err);
	reftable_writer_free(w);

	block_source_from_strbuf(&source, &buf);

	err = reftable_new_reader(&rd, &source, "filename");
	EXPECT_ERR(err);

	hash_id = reftable_reader_hash_id(rd);
	EXPECT(hash_id == GIT_SHA1_FORMAT_ID);

	reftable_table_from_reader(&tab[0], rd);
	err = reftable_new_merged_table(&merged, tab, 1, GIT_SHA1_FORMAT_ID);
	EXPECT_ERR(err);

	reftable_reader_free(rd);
	reftable_merged_table_free(merged);
	strbuf_release(&buf);
}

/* XXX test refs_for(oid) */

int merged_test_main(int argc, const char *argv[])
{
	RUN_TEST(test_merged_logs);
	RUN_TEST(test_merged_between);
	RUN_TEST(test_merged);
	RUN_TEST(test_default_write_opts);
	return 0;
}
Commit	Line	Data
1ae2b8cd HWN	1	/*
	2	Copyright 2020 Google LLC
	3
	4	Use of this source code is governed by a BSD-style
	5	license that can be found in the LICENSE file or at
	6	https://developers.google.com/open-source/licenses/bsd
	7	*/
	8
	9	#include "merged.h"
	10
	11	#include "system.h"
	12
	13	#include "basics.h"
	14	#include "blocksource.h"
1ae2b8cd HWN	15	#include "reader.h"
	16	#include "record.h"
	17	#include "test_framework.h"
	18	#include "reftable-merged.h"
	19	#include "reftable-tests.h"
	20	#include "reftable-generic.h"
	21	#include "reftable-writer.h"
	22
	23	static void write_test_table(struct strbuf *buf,
	24	struct reftable_ref_record refs[], int n)
	25	{
22d2f70e ÆAB	26	uint64_t min = 0xffffffff;
22d2f70e ÆAB	27	uint64_t max = 0;
1ae2b8cd HWN	28	int i = 0;
	29	int err;
	30
	31	struct reftable_write_options opts = {
	32	.block_size = 256,
	33	};
	34	struct reftable_writer *w = NULL;
	35	for (i = 0; i < n; i++) {
	36	uint64_t ui = refs[i].update_index;
	37	if (ui > max) {
	38	max = ui;
	39	}
	40	if (ui < min) {
	41	min = ui;
	42	}
	43	}
	44
1df18a1c	45	w = reftable_new_writer(&strbuf_add_void, &noop_flush, buf, &opts);
1ae2b8cd HWN	46	reftable_writer_set_limits(w, min, max);
	47
	48	for (i = 0; i < n; i++) {
	49	uint64_t before = refs[i].update_index;
	50	int n = reftable_writer_add_ref(w, &refs[i]);
	51	EXPECT(n == 0);
	52	EXPECT(before == refs[i].update_index);
	53	}
	54
	55	err = reftable_writer_close(w);
	56	EXPECT_ERR(err);
	57
	58	reftable_writer_free(w);
	59	}
	60
	61	static void write_test_log_table(struct strbuf *buf,
	62	struct reftable_log_record logs[], int n,
	63	uint64_t update_index)
	64	{
	65	int i = 0;
	66	int err;
	67
	68	struct reftable_write_options opts = {
	69	.block_size = 256,
	70	.exact_log_message = 1,
	71	};
	72	struct reftable_writer *w = NULL;
1df18a1c	73	w = reftable_new_writer(&strbuf_add_void, &noop_flush, buf, &opts);
1ae2b8cd HWN	74	reftable_writer_set_limits(w, update_index, update_index);
	75
	76	for (i = 0; i < n; i++) {
	77	int err = reftable_writer_add_log(w, &logs[i]);
	78	EXPECT_ERR(err);
	79	}
	80
	81	err = reftable_writer_close(w);
	82	EXPECT_ERR(err);
	83
	84	reftable_writer_free(w);
	85	}
	86
	87	static struct reftable_merged_table *
	88	merged_table_from_records(struct reftable_ref_record **refs,
	89	struct reftable_block_source **source,
	90	struct reftable_reader **readers, int sizes,
81879123	91	struct strbuf *buf, size_t n)
1ae2b8cd	92	{
1ae2b8cd	93	struct reftable_merged_table *mt = NULL;
b4ff12c8	94	struct reftable_table *tabs;
1ae2b8cd	95	int err;
b4ff12c8 PS	96
	97	REFTABLE_CALLOC_ARRAY(tabs, n);
	98	REFTABLE_CALLOC_ARRAY(*readers, n);
	99	REFTABLE_CALLOC_ARRAY(*source, n);
	100
81879123	101	for (size_t i = 0; i < n; i++) {
1ae2b8cd HWN	102	write_test_table(&buf[i], refs[i], sizes[i]);
	103	block_source_from_strbuf(&(*source)[i], &buf[i]);
	104
	105	err = reftable_new_reader(&(readers)[i], &(source)[i],
	106	"name");
	107	EXPECT_ERR(err);
	108	reftable_table_from_reader(&tabs[i], (*readers)[i]);
	109	}
	110
	111	err = reftable_new_merged_table(&mt, tabs, n, GIT_SHA1_FORMAT_ID);
	112	EXPECT_ERR(err);
	113	return mt;
	114	}
	115
	116	static void readers_destroy(struct reftable_reader **readers, size_t n)
	117	{
	118	int i = 0;
	119	for (; i < n; i++)
	120	reftable_reader_free(readers[i]);
	121	reftable_free(readers);
	122	}
	123
	124	static void test_merged_between(void)
	125	{
1ae2b8cd HWN	126	struct reftable_ref_record r1[] = { {
	127	.refname = "b",
	128	.update_index = 1,
	129	.value_type = REFTABLE_REF_VAL1,
7af607c5	130	.value.val1 = { 1, 2, 3, 0 },
1ae2b8cd HWN	131	} };
	132	struct reftable_ref_record r2[] = { {
	133	.refname = "a",
	134	.update_index = 2,
	135	.value_type = REFTABLE_REF_DELETION,
	136	} };
	137
	138	struct reftable_ref_record *refs[] = { r1, r2 };
	139	int sizes[] = { 1, 1 };
	140	struct strbuf bufs[2] = { STRBUF_INIT, STRBUF_INIT };
	141	struct reftable_block_source *bs = NULL;
	142	struct reftable_reader **readers = NULL;
	143	struct reftable_merged_table *mt =
	144	merged_table_from_records(refs, &bs, &readers, sizes, bufs, 2);
	145	int i;
	146	struct reftable_ref_record ref = { NULL };
	147	struct reftable_iterator it = { NULL };
	148	int err = reftable_merged_table_seek_ref(mt, &it, "a");
	149	EXPECT_ERR(err);
	150
	151	err = reftable_iterator_next_ref(&it, &ref);
	152	EXPECT_ERR(err);
	153	EXPECT(ref.update_index == 2);
	154	reftable_ref_record_release(&ref);
	155	reftable_iterator_destroy(&it);
	156	readers_destroy(readers, 2);
	157	reftable_merged_table_free(mt);
	158	for (i = 0; i < ARRAY_SIZE(bufs); i++) {
	159	strbuf_release(&bufs[i]);
	160	}
	161	reftable_free(bs);
	162	}
	163
	164	static void test_merged(void)
	165	{
1ae2b8cd HWN	166	struct reftable_ref_record r1[] = {
	167	{
	168	.refname = "a",
	169	.update_index = 1,
	170	.value_type = REFTABLE_REF_VAL1,
7af607c5	171	.value.val1 = { 1 },
1ae2b8cd HWN	172	},
	173	{
	174	.refname = "b",
	175	.update_index = 1,
	176	.value_type = REFTABLE_REF_VAL1,
7af607c5	177	.value.val1 = { 1 },
1ae2b8cd HWN	178	},
	179	{
	180	.refname = "c",
	181	.update_index = 1,
	182	.value_type = REFTABLE_REF_VAL1,
7af607c5	183	.value.val1 = { 1 },
1ae2b8cd HWN	184	}
	185	};
	186	struct reftable_ref_record r2[] = { {
	187	.refname = "a",
	188	.update_index = 2,
	189	.value_type = REFTABLE_REF_DELETION,
	190	} };
	191	struct reftable_ref_record r3[] = {
	192	{
	193	.refname = "c",
	194	.update_index = 3,
	195	.value_type = REFTABLE_REF_VAL1,
7af607c5	196	.value.val1 = { 2 },
1ae2b8cd HWN	197	},
	198	{
	199	.refname = "d",
	200	.update_index = 3,
	201	.value_type = REFTABLE_REF_VAL1,
7af607c5	202	.value.val1 = { 1 },
1ae2b8cd HWN	203	},
	204	};
	205
f2b25514 HWN	206	struct reftable_ref_record *want[] = {
	207	&r2[0],
	208	&r1[1],
	209	&r3[0],
	210	&r3[1],
1ae2b8cd HWN	211	};
	212
	213	struct reftable_ref_record *refs[] = { r1, r2, r3 };
	214	int sizes[3] = { 3, 1, 2 };
	215	struct strbuf bufs[3] = { STRBUF_INIT, STRBUF_INIT, STRBUF_INIT };
	216	struct reftable_block_source *bs = NULL;
	217	struct reftable_reader **readers = NULL;
	218	struct reftable_merged_table *mt =
	219	merged_table_from_records(refs, &bs, &readers, sizes, bufs, 3);
	220
	221	struct reftable_iterator it = { NULL };
	222	int err = reftable_merged_table_seek_ref(mt, &it, "a");
	223	struct reftable_ref_record *out = NULL;
	224	size_t len = 0;
	225	size_t cap = 0;
	226	int i = 0;
	227
	228	EXPECT_ERR(err);
	229	EXPECT(reftable_merged_table_hash_id(mt) == GIT_SHA1_FORMAT_ID);
	230	EXPECT(reftable_merged_table_min_update_index(mt) == 1);
	231
	232	while (len < 100) { /* cap loops/recursion. */
	233	struct reftable_ref_record ref = { NULL };
	234	int err = reftable_iterator_next_ref(&it, &ref);
f6b58c1b	235	if (err > 0)
1ae2b8cd	236	break;
f6b58c1b PS	237
f6b58c1b PS	238	REFTABLE_ALLOC_GROW(out, len + 1, cap);
1ae2b8cd HWN	239	out[len++] = ref;
	240	}
	241	reftable_iterator_destroy(&it);
	242
	243	EXPECT(ARRAY_SIZE(want) == len);
	244	for (i = 0; i < len; i++) {
f2b25514	245	EXPECT(reftable_ref_record_equal(want[i], &out[i],
1ae2b8cd HWN	246	GIT_SHA1_RAWSZ));
	247	}
	248	for (i = 0; i < len; i++) {
	249	reftable_ref_record_release(&out[i]);
	250	}
	251	reftable_free(out);
	252
	253	for (i = 0; i < 3; i++) {
	254	strbuf_release(&bufs[i]);
	255	}
	256	readers_destroy(readers, 3);
	257	reftable_merged_table_free(mt);
	258	reftable_free(bs);
	259	}
	260
	261	static struct reftable_merged_table *
	262	merged_table_from_log_records(struct reftable_log_record **logs,
	263	struct reftable_block_source **source,
	264	struct reftable_reader **readers, int sizes,
81879123	265	struct strbuf *buf, size_t n)
1ae2b8cd	266	{
1ae2b8cd	267	struct reftable_merged_table *mt = NULL;
b4ff12c8	268	struct reftable_table *tabs;
1ae2b8cd	269	int err;
b4ff12c8 PS	270
	271	REFTABLE_CALLOC_ARRAY(tabs, n);
	272	REFTABLE_CALLOC_ARRAY(*readers, n);
	273	REFTABLE_CALLOC_ARRAY(*source, n);
	274
81879123	275	for (size_t i = 0; i < n; i++) {
1ae2b8cd HWN	276	write_test_log_table(&buf[i], logs[i], sizes[i], i + 1);
	277	block_source_from_strbuf(&(*source)[i], &buf[i]);
	278
	279	err = reftable_new_reader(&(readers)[i], &(source)[i],
	280	"name");
	281	EXPECT_ERR(err);
	282	reftable_table_from_reader(&tabs[i], (*readers)[i]);
	283	}
	284
	285	err = reftable_new_merged_table(&mt, tabs, n, GIT_SHA1_FORMAT_ID);
	286	EXPECT_ERR(err);
	287	return mt;
	288	}
	289
	290	static void test_merged_logs(void)
	291	{
1ae2b8cd HWN	292	struct reftable_log_record r1[] = {
	293	{
	294	.refname = "a",
	295	.update_index = 2,
	296	.value_type = REFTABLE_LOG_UPDATE,
	297	.value.update = {
87ff7230	298	.old_hash = { 2 },
1ae2b8cd HWN	299	/* deletion */
	300	.name = "jane doe",
	301	.email = "jane@invalid",
	302	.message = "message2",
	303	}
	304	},
	305	{
	306	.refname = "a",
	307	.update_index = 1,
	308	.value_type = REFTABLE_LOG_UPDATE,
	309	.value.update = {
87ff7230 PS	310	.old_hash = { 1 },
87ff7230 PS	311	.new_hash = { 2 },
1ae2b8cd HWN	312	.name = "jane doe",
	313	.email = "jane@invalid",
	314	.message = "message1",
	315	}
	316	},
	317	};
	318	struct reftable_log_record r2[] = {
	319	{
	320	.refname = "a",
	321	.update_index = 3,
	322	.value_type = REFTABLE_LOG_UPDATE,
	323	.value.update = {
87ff7230	324	.new_hash = { 3 },
1ae2b8cd HWN	325	.name = "jane doe",
	326	.email = "jane@invalid",
	327	.message = "message3",
	328	}
	329	},
	330	};
	331	struct reftable_log_record r3[] = {
	332	{
	333	.refname = "a",
	334	.update_index = 2,
	335	.value_type = REFTABLE_LOG_DELETION,
	336	},
	337	};
f2b25514 HWN	338	struct reftable_log_record *want[] = {
	339	&r2[0],
	340	&r3[0],
	341	&r1[1],
1ae2b8cd HWN	342	};
	343
	344	struct reftable_log_record *logs[] = { r1, r2, r3 };
	345	int sizes[3] = { 2, 1, 1 };
	346	struct strbuf bufs[3] = { STRBUF_INIT, STRBUF_INIT, STRBUF_INIT };
	347	struct reftable_block_source *bs = NULL;
	348	struct reftable_reader **readers = NULL;
	349	struct reftable_merged_table *mt = merged_table_from_log_records(
	350	logs, &bs, &readers, sizes, bufs, 3);
	351
	352	struct reftable_iterator it = { NULL };
	353	int err = reftable_merged_table_seek_log(mt, &it, "a");
	354	struct reftable_log_record *out = NULL;
	355	size_t len = 0;
	356	size_t cap = 0;
	357	int i = 0;
	358
	359	EXPECT_ERR(err);
	360	EXPECT(reftable_merged_table_hash_id(mt) == GIT_SHA1_FORMAT_ID);
	361	EXPECT(reftable_merged_table_min_update_index(mt) == 1);
	362
	363	while (len < 100) { /* cap loops/recursion. */
	364	struct reftable_log_record log = { NULL };
	365	int err = reftable_iterator_next_log(&it, &log);
f6b58c1b	366	if (err > 0)
1ae2b8cd	367	break;
f6b58c1b PS	368
f6b58c1b PS	369	REFTABLE_ALLOC_GROW(out, len + 1, cap);
1ae2b8cd HWN	370	out[len++] = log;
	371	}
	372	reftable_iterator_destroy(&it);
	373
	374	EXPECT(ARRAY_SIZE(want) == len);
	375	for (i = 0; i < len; i++) {
f2b25514	376	EXPECT(reftable_log_record_equal(want[i], &out[i],
1ae2b8cd HWN	377	GIT_SHA1_RAWSZ));
	378	}
	379
	380	err = reftable_merged_table_seek_log_at(mt, &it, "a", 2);
	381	EXPECT_ERR(err);
	382	reftable_log_record_release(&out[0]);
	383	err = reftable_iterator_next_log(&it, &out[0]);
	384	EXPECT_ERR(err);
	385	EXPECT(reftable_log_record_equal(&out[0], &r3[0], GIT_SHA1_RAWSZ));
	386	reftable_iterator_destroy(&it);
	387
	388	for (i = 0; i < len; i++) {
	389	reftable_log_record_release(&out[i]);
	390	}
	391	reftable_free(out);
	392
	393	for (i = 0; i < 3; i++) {
	394	strbuf_release(&bufs[i]);
	395	}
	396	readers_destroy(readers, 3);
	397	reftable_merged_table_free(mt);
	398	reftable_free(bs);
	399	}
	400
	401	static void test_default_write_opts(void)
	402	{
	403	struct reftable_write_options opts = { 0 };
	404	struct strbuf buf = STRBUF_INIT;
	405	struct reftable_writer *w =
1df18a1c	406	reftable_new_writer(&strbuf_add_void, &noop_flush, &buf, &opts);
1ae2b8cd HWN	407
	408	struct reftable_ref_record rec = {
	409	.refname = "master",
	410	.update_index = 1,
	411	};
	412	int err;
	413	struct reftable_block_source source = { NULL };
b4ff12c8	414	struct reftable_table tab = reftable_calloc(1, sizeof(tab));
1ae2b8cd HWN	415	uint32_t hash_id;
	416	struct reftable_reader *rd = NULL;
	417	struct reftable_merged_table *merged = NULL;
	418
	419	reftable_writer_set_limits(w, 1, 1);
	420
	421	err = reftable_writer_add_ref(w, &rec);
	422	EXPECT_ERR(err);
	423
	424	err = reftable_writer_close(w);
	425	EXPECT_ERR(err);
	426	reftable_writer_free(w);
	427
	428	block_source_from_strbuf(&source, &buf);
	429
	430	err = reftable_new_reader(&rd, &source, "filename");
	431	EXPECT_ERR(err);
	432
	433	hash_id = reftable_reader_hash_id(rd);
	434	EXPECT(hash_id == GIT_SHA1_FORMAT_ID);
	435
	436	reftable_table_from_reader(&tab[0], rd);
	437	err = reftable_new_merged_table(&merged, tab, 1, GIT_SHA1_FORMAT_ID);
	438	EXPECT_ERR(err);
	439
	440	reftable_reader_free(rd);
	441	reftable_merged_table_free(merged);
	442	strbuf_release(&buf);
	443	}
	444
	445	/* XXX test refs_for(oid) */
	446
	447	int merged_test_main(int argc, const char *argv[])
	448	{
	449	RUN_TEST(test_merged_logs);
	450	RUN_TEST(test_merged_between);
	451	RUN_TEST(test_merged);
	452	RUN_TEST(test_default_write_opts);
	453	return 0;
	454	}