]>
Commit | Line | Data |
---|---|---|
e581fd72 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 "block.h" | |
10 | ||
11 | #include "system.h" | |
12 | #include "blocksource.h" | |
13 | #include "basics.h" | |
14 | #include "constants.h" | |
15 | #include "record.h" | |
16 | #include "test_framework.h" | |
17 | #include "reftable-tests.h" | |
18 | ||
19 | static void test_block_read_write(void) | |
20 | { | |
21 | const int header_off = 21; /* random */ | |
22 | char *names[30]; | |
23 | const int N = ARRAY_SIZE(names); | |
24 | const int block_size = 1024; | |
25 | struct reftable_block block = { NULL }; | |
26 | struct block_writer bw = { | |
27 | .last_key = STRBUF_INIT, | |
28 | }; | |
66c0daba HWN |
29 | struct reftable_record rec = { |
30 | .type = BLOCK_TYPE_REF, | |
31 | }; | |
e581fd72 HWN |
32 | int i = 0; |
33 | int n; | |
34 | struct block_reader br = { 0 }; | |
35 | struct block_iter it = { .last_key = STRBUF_INIT }; | |
36 | int j = 0; | |
37 | struct strbuf want = STRBUF_INIT; | |
38 | ||
39 | block.data = reftable_calloc(block_size); | |
40 | block.len = block_size; | |
41 | block.source = malloc_block_source(); | |
42 | block_writer_init(&bw, BLOCK_TYPE_REF, block.data, block_size, | |
43 | header_off, hash_size(GIT_SHA1_FORMAT_ID)); | |
e581fd72 HWN |
44 | |
45 | for (i = 0; i < N; i++) { | |
46 | char name[100]; | |
47 | uint8_t hash[GIT_SHA1_RAWSZ]; | |
48 | snprintf(name, sizeof(name), "branch%02d", i); | |
49 | memset(hash, i, sizeof(hash)); | |
50 | ||
66c0daba HWN |
51 | rec.u.ref.refname = name; |
52 | rec.u.ref.value_type = REFTABLE_REF_VAL1; | |
53 | rec.u.ref.value.val1 = hash; | |
e581fd72 HWN |
54 | |
55 | names[i] = xstrdup(name); | |
56 | n = block_writer_add(&bw, &rec); | |
66c0daba HWN |
57 | rec.u.ref.refname = NULL; |
58 | rec.u.ref.value_type = REFTABLE_REF_DELETION; | |
e581fd72 HWN |
59 | EXPECT(n == 0); |
60 | } | |
61 | ||
62 | n = block_writer_finish(&bw); | |
63 | EXPECT(n > 0); | |
64 | ||
65 | block_writer_release(&bw); | |
66 | ||
67 | block_reader_init(&br, &block, header_off, block_size, GIT_SHA1_RAWSZ); | |
68 | ||
69 | block_reader_start(&br, &it); | |
70 | ||
71 | while (1) { | |
72 | int r = block_iter_next(&it, &rec); | |
73 | EXPECT(r >= 0); | |
74 | if (r > 0) { | |
75 | break; | |
76 | } | |
66c0daba | 77 | EXPECT_STREQ(names[j], rec.u.ref.refname); |
e581fd72 HWN |
78 | j++; |
79 | } | |
80 | ||
81 | reftable_record_release(&rec); | |
82 | block_iter_close(&it); | |
83 | ||
84 | for (i = 0; i < N; i++) { | |
85 | struct block_iter it = { .last_key = STRBUF_INIT }; | |
86 | strbuf_reset(&want); | |
87 | strbuf_addstr(&want, names[i]); | |
88 | ||
89 | n = block_reader_seek(&br, &it, &want); | |
90 | EXPECT(n == 0); | |
91 | ||
92 | n = block_iter_next(&it, &rec); | |
93 | EXPECT(n == 0); | |
94 | ||
66c0daba | 95 | EXPECT_STREQ(names[i], rec.u.ref.refname); |
e581fd72 HWN |
96 | |
97 | want.len--; | |
98 | n = block_reader_seek(&br, &it, &want); | |
99 | EXPECT(n == 0); | |
100 | ||
101 | n = block_iter_next(&it, &rec); | |
102 | EXPECT(n == 0); | |
66c0daba | 103 | EXPECT_STREQ(names[10 * (i / 10)], rec.u.ref.refname); |
e581fd72 HWN |
104 | |
105 | block_iter_close(&it); | |
106 | } | |
107 | ||
108 | reftable_record_release(&rec); | |
109 | reftable_block_done(&br.block); | |
110 | strbuf_release(&want); | |
111 | for (i = 0; i < N; i++) { | |
112 | reftable_free(names[i]); | |
113 | } | |
114 | } | |
115 | ||
116 | int block_test_main(int argc, const char *argv[]) | |
117 | { | |
118 | RUN_TEST(test_block_read_write); | |
119 | return 0; | |
120 | } |