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
126
127
128
129
130
|
#include "testing.h"
extern void tpl_reset();
extern struct tpl_item *tpl_pool[];
extern unsigned tpl_pool_used;
extern unsigned tpl_pool_func_used;
static int adder(struct tplfunc_frame *frame, void *result) {
int a = (int) strtol(frame->argv[0].t_char_ptr, NULL, 10);
int b = (int) strtol(frame->argv[1].t_char_ptr, NULL, 10);
char **ptr = (char **) result;
const size_t sz = 100;
*ptr = calloc(sz, sizeof(*ptr));
snprintf(*ptr, sz, "%d", a + b);
return 0;
}
static int subtractor(struct tplfunc_frame *frame, void *result) {
int a = (int) strtol(frame->argv[0].t_char_ptr, NULL, 10);
int b = (int) strtol(frame->argv[1].t_char_ptr, NULL, 10);
char **ptr = (char **) result;
const size_t sz = 100;
*ptr = calloc(sz, sizeof(*ptr));
snprintf(*ptr, sz, "%d", a - b);
return 0;
}
static int multiplier(struct tplfunc_frame *frame, void *result) {
int a = (int) strtol(frame->argv[0].t_char_ptr, NULL, 10);
int b = (int) strtol(frame->argv[1].t_char_ptr, NULL, 10);
char **ptr = (char **) result;
const size_t sz = 100;
*ptr = calloc(sz, sizeof(*ptr));
snprintf(*ptr, sz, "%d", a * b);
return 0;
}
static int divider(struct tplfunc_frame *frame, void *result) {
int a = (int) strtol(frame->argv[0].t_char_ptr, NULL, 10);
int b = (int) strtol(frame->argv[1].t_char_ptr, NULL, 10);
char **ptr = (char **) result;
size_t sz = 100;
*ptr = calloc(sz, sizeof(*ptr));
snprintf(*ptr, sz, "%d", a / b);
return 0;
}
void test_tpl_workflow() {
char *data = strdup("Hello world!");
tpl_reset();
tpl_register("hello_message", &data);
char *result = NULL;
result = tpl_render("I said, \"{{ hello_message }}\"");
STASIS_ASSERT(strcmp(result, "I said, \"Hello world!\"") == 0, "stored value in key is incorrect");
guard_free(result);
setenv("HELLO", "Hello environment!", 1);
result = tpl_render("{{ env:HELLO }}");
STASIS_ASSERT(strcmp(result, "Hello environment!") == 0, "environment variable content mismatch");
guard_free(result);
unsetenv("HELLO");
guard_free(data);
}
void test_tpl_register() {
char *data = strdup("Hello world!");
tpl_reset();
unsigned used_before_register = tpl_pool_used;
tpl_register("hello_message", &data);
STASIS_ASSERT(tpl_pool_used == (used_before_register + 1), "tpl_register did not increment allocation counter");
STASIS_ASSERT(tpl_pool[used_before_register] != NULL, "register did not allocate a tpl_item record in the pool");
free(data);
}
void test_tpl_register_func() {
tpl_reset();
struct testcase {
const char *key;
int argc;
void *func;
};
struct testcase tc[] = {
{.key = "add", .argc = 2, .func = &adder},
{.key = "sub", .argc = 2, .func = &subtractor},
{.key = "mul", .argc = 2, .func = &multiplier},
{.key = "div", .argc = 2, .func = ÷r},
};
tpl_register_func("add", tc[0].func, tc[0].argc, NULL);
tpl_register_func("sub", tc[1].func, tc[1].argc, NULL);
tpl_register_func("mul", tc[2].func, tc[2].argc, NULL);
tpl_register_func("div", tc[3].func, tc[3].argc, NULL);
STASIS_ASSERT(tpl_pool_func_used == sizeof(tc) / sizeof(*tc), "unexpected function pool used");
char *result = NULL;
result = tpl_render("{{ func:add(0,3) }}");
STASIS_ASSERT(result != NULL && strcmp(result, "3") == 0, "Answer was not 3");
guard_free(result);
result = tpl_render("{{ func:add(1,2) }}");
STASIS_ASSERT(result != NULL && strcmp(result, "3") == 0, "Answer was not 3");
guard_free(result);
result = tpl_render("{{ func:sub(6,3) }}");
STASIS_ASSERT(result != NULL && strcmp(result, "3") == 0, "Answer was not 3");
guard_free(result);
result = tpl_render("{{ func:sub(4,1) }}");
STASIS_ASSERT(result != NULL && strcmp(result, "3") == 0, "Answer was not 3");
guard_free(result);
result = tpl_render("{{ func:mul(1, 3) }}");
STASIS_ASSERT(result != NULL && strcmp(result, "3") == 0, "Answer was not 3");
guard_free(result);
result = tpl_render("{{ func:div(6,2) }}");
STASIS_ASSERT(result != NULL && strcmp(result, "3") == 0, "Answer was not 3");
guard_free(result);
result = tpl_render("{{ func:div(3,1) }}");
STASIS_ASSERT(result != NULL && strcmp(result, "3") == 0, "Answer was not 3");
guard_free(result);
}
int main(int argc, char *argv[]) {
STASIS_TEST_BEGIN_MAIN();
STASIS_TEST_FUNC *tests[] = {
test_tpl_workflow,
test_tpl_register_func,
test_tpl_register,
};
STASIS_TEST_RUN(tests);
STASIS_TEST_END_MAIN();
}
|
