1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
|
/*
* Licensed to the Apache Software Foundation (ASF) under one or more
* contributor license agreements. See the NOTICE file distributed with
* this work for additional information regarding copyright ownership.
* The ASF licenses this file to you under the Apache License, Version 2.0
* (the "License"); you may not use this file except in compliance with
* the License. You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or
* implied. See the License for the specific language governing
* permissions and limitations under the License.
*/
#include <avro.h>
#include <stdio.h>
#include <stdlib.h>
#define check_exit(call) \
do { \
int __rc = call; \
if (__rc != 0) { \
fprintf(stderr, "Unexpected error:\n %s\n %s\n", \
avro_strerror(), #call); \
exit(EXIT_FAILURE); \
} \
} while (0)
#define expect_eof(call) \
do { \
int __rc = call; \
if (__rc != EOF) { \
fprintf(stderr, "Expected EOF:\n %s\n", #call); \
exit(EXIT_FAILURE); \
} \
} while (0)
#define expect_error(call) \
do { \
int __rc = call; \
if (__rc == 0) { \
fprintf(stderr, "Expected an error:\n %s\n", #call); \
exit(EXIT_FAILURE); \
} \
} while (0)
#define check_expected_value(actual, expected) \
do { \
if (!avro_value_equal_fast((actual), (expected))) { \
char *actual_json; \
char *expected_json; \
avro_value_to_json((actual), 1, &actual_json); \
avro_value_to_json((expected), 1, &expected_json); \
fprintf(stderr, "Expected %s\nGot %s\n", \
expected_json, actual_json); \
free(actual_json); \
free(expected_json); \
exit(EXIT_FAILURE); \
} \
} while (0)
int main(void)
{
avro_schema_t schema;
avro_file_reader_t reader;
avro_value_iface_t *iface;
avro_value_t actual;
avro_value_t expected;
avro_value_t branch;
schema = avro_schema_union();
avro_schema_union_append(schema, avro_schema_null());
avro_schema_union_append(schema, avro_schema_int());
iface = avro_generic_class_from_schema(schema);
avro_generic_value_new(iface, &actual);
avro_generic_value_new(iface, &expected);
/* First read the contents of the good file. */
check_exit(avro_file_reader("avro-1238-good.avro", &reader));
check_exit(avro_file_reader_read_value(reader, &actual));
check_exit(avro_value_set_branch(&expected, 0, &branch));
check_exit(avro_value_set_null(&branch));
check_expected_value(&actual, &expected);
check_exit(avro_file_reader_read_value(reader, &actual));
check_exit(avro_value_set_branch(&expected, 1, &branch));
check_exit(avro_value_set_int(&branch, 100));
check_expected_value(&actual, &expected);
expect_eof(avro_file_reader_read_value(reader, &actual));
check_exit(avro_file_reader_close(reader));
/* Then read from the truncated file. */
check_exit(avro_file_reader("avro-1238-truncated.avro", &reader));
check_exit(avro_file_reader_read_value(reader, &actual));
check_exit(avro_value_set_branch(&expected, 0, &branch));
check_exit(avro_value_set_null(&branch));
check_expected_value(&actual, &expected);
check_exit(avro_file_reader_read_value(reader, &actual));
check_exit(avro_value_set_branch(&expected, 1, &branch));
check_exit(avro_value_set_int(&branch, 100));
check_expected_value(&actual, &expected);
expect_error(avro_file_reader_read_value(reader, &actual));
check_exit(avro_file_reader_close(reader));
/* Clean up and exit */
avro_value_decref(&actual);
avro_value_decref(&expected);
avro_value_iface_decref(iface);
avro_schema_decref(schema);
exit(EXIT_SUCCESS);
}
|
