]>
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 }; | |
a8305bc6 | 35 | struct block_iter it = BLOCK_ITER_INIT; |
e581fd72 HWN |
36 | int j = 0; |
37 | struct strbuf want = STRBUF_INIT; | |
38 | ||
b4ff12c8 | 39 | REFTABLE_CALLOC_ARRAY(block.data, block_size); |
e581fd72 HWN |
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 | 44 | |
45c2fcc2 HWN |
45 | rec.u.ref.refname = ""; |
46 | rec.u.ref.value_type = REFTABLE_REF_DELETION; | |
47 | n = block_writer_add(&bw, &rec); | |
48 | EXPECT(n == REFTABLE_API_ERROR); | |
49 | ||
e581fd72 HWN |
50 | for (i = 0; i < N; i++) { |
51 | char name[100]; | |
e581fd72 | 52 | snprintf(name, sizeof(name), "branch%02d", i); |
e581fd72 | 53 | |
66c0daba HWN |
54 | rec.u.ref.refname = name; |
55 | rec.u.ref.value_type = REFTABLE_REF_VAL1; | |
7af607c5 | 56 | memset(rec.u.ref.value.val1, i, GIT_SHA1_RAWSZ); |
e581fd72 HWN |
57 | |
58 | names[i] = xstrdup(name); | |
59 | n = block_writer_add(&bw, &rec); | |
66c0daba HWN |
60 | rec.u.ref.refname = NULL; |
61 | rec.u.ref.value_type = REFTABLE_REF_DELETION; | |
e581fd72 HWN |
62 | EXPECT(n == 0); |
63 | } | |
64 | ||
65 | n = block_writer_finish(&bw); | |
66 | EXPECT(n > 0); | |
67 | ||
68 | block_writer_release(&bw); | |
69 | ||
70 | block_reader_init(&br, &block, header_off, block_size, GIT_SHA1_RAWSZ); | |
71 | ||
72 | block_reader_start(&br, &it); | |
73 | ||
74 | while (1) { | |
75 | int r = block_iter_next(&it, &rec); | |
76 | EXPECT(r >= 0); | |
77 | if (r > 0) { | |
78 | break; | |
79 | } | |
66c0daba | 80 | EXPECT_STREQ(names[j], rec.u.ref.refname); |
e581fd72 HWN |
81 | j++; |
82 | } | |
83 | ||
84 | reftable_record_release(&rec); | |
85 | block_iter_close(&it); | |
86 | ||
87 | for (i = 0; i < N; i++) { | |
a8305bc6 | 88 | struct block_iter it = BLOCK_ITER_INIT; |
e581fd72 HWN |
89 | strbuf_reset(&want); |
90 | strbuf_addstr(&want, names[i]); | |
91 | ||
92 | n = block_reader_seek(&br, &it, &want); | |
93 | EXPECT(n == 0); | |
94 | ||
95 | n = block_iter_next(&it, &rec); | |
96 | EXPECT(n == 0); | |
97 | ||
66c0daba | 98 | EXPECT_STREQ(names[i], rec.u.ref.refname); |
e581fd72 HWN |
99 | |
100 | want.len--; | |
101 | n = block_reader_seek(&br, &it, &want); | |
102 | EXPECT(n == 0); | |
103 | ||
104 | n = block_iter_next(&it, &rec); | |
105 | EXPECT(n == 0); | |
66c0daba | 106 | EXPECT_STREQ(names[10 * (i / 10)], rec.u.ref.refname); |
e581fd72 HWN |
107 | |
108 | block_iter_close(&it); | |
109 | } | |
110 | ||
111 | reftable_record_release(&rec); | |
112 | reftable_block_done(&br.block); | |
113 | strbuf_release(&want); | |
114 | for (i = 0; i < N; i++) { | |
115 | reftable_free(names[i]); | |
116 | } | |
117 | } | |
118 | ||
119 | int block_test_main(int argc, const char *argv[]) | |
120 | { | |
121 | RUN_TEST(test_block_read_write); | |
122 | return 0; | |
123 | } |