Function test_reftable_block__iterator

t/unit-tests/u-reftable-block.c:381–458

}

void test_reftable_block__iterator(void)
{
	struct reftable_block_source source = { 0 };
	struct block_writer writer = {
		.last_key = REFTABLE_BUF_INIT,
	};
	struct reftable_record expected_refs[20];
	struct reftable_ref_record ref = { 0 };
	struct reftable_iterator it = { 0 };
	struct reftable_block block = { 0 };
	struct reftable_buf data;
	int err;

	data.len = 1024;
	REFTABLE_CALLOC_ARRAY(data.buf, data.len);
	cl_assert(data.buf != NULL);

	err = block_writer_init(&writer, REFTABLE_BLOCK_TYPE_REF,
				(uint8_t *) data.buf, data.len,
				0, hash_size(REFTABLE_HASH_SHA1));
	cl_assert(!err);

	for (size_t i = 0; i < ARRAY_SIZE(expected_refs); i++) {
		expected_refs[i] = (struct reftable_record) {
			.type = REFTABLE_BLOCK_TYPE_REF,
			.u.ref = {
				.value_type = REFTABLE_REF_VAL1,
				.refname = xstrfmt("refs/heads/branch-%02"PRIuMAX, (uintmax_t)i),
			},
		};
		memset(expected_refs[i].u.ref.value.val1, i, REFTABLE_HASH_SIZE_SHA1);

		err = block_writer_add(&writer, &expected_refs[i]);
		cl_assert_equal_i(err, 0);
	}

	err = block_writer_finish(&writer);
	cl_assert(err > 0);

	block_source_from_buf(&source, &data);
	reftable_block_init(&block, &source, 0, 0, data.len,
			    REFTABLE_HASH_SIZE_SHA1, REFTABLE_BLOCK_TYPE_REF);

	err = reftable_block_init_iterator(&block, &it);
	cl_assert_equal_i(err, 0);

	for (size_t i = 0; ; i++) {
		err = reftable_iterator_next_ref(&it, &ref);
		if (err > 0) {
			cl_assert_equal_i(i, ARRAY_SIZE(expected_refs));
			break;
		}
		cl_assert_equal_i(err, 0);

		cl_assert(reftable_ref_record_equal(&ref,
						    &expected_refs[i].u.ref, REFTABLE_HASH_SIZE_SHA1));
	}