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
|
/* Copyright (c) 2021 Dovecot authors, see the included COPYING file */
#include "lib.h"
#include "ioloop.h"
#include "dlua-script-private.h"
#include "dict-private.h"
#include "dict-lua.h"
#include "test-common.h"
#include <fcntl.h>
#include <sys/stat.h>
static void test_dict_register(void)
{
dict_driver_register(&dict_driver_file);
}
static void test_dict_finished(lua_State *L, struct ioloop *ioloop, int res)
{
if (res < 0)
i_error("%s", lua_tostring(L, -1));
io_loop_stop(ioloop);
}
static void test_dict_lua(void)
{
static const char *luascript =
"function test_dict(dict)\n"
" local trans = dict:transaction_begin()\n"
" trans:set('shared/testkey', 'testvalue')\n"
" trans:set('shared/testkey2', 'testvalue2')\n"
" trans:commit()\n"
"\n"
" assert(dict:lookup('shared/testkey')[1] == 'testvalue')\n"
" assert(dict:lookup('shared/testkey2')[1] == 'testvalue2')\n"
"\n"
" local key, values\n"
" local table = {}\n"
" for key, values in dict:iterate('shared/', 0) do\n"
" assert(#values == 1)\n"
" table[key] = values[1]\n"
" end\n"
" assert(table['shared/testkey'] == 'testvalue')\n"
" assert(table['shared/testkey2'] == 'testvalue2')\n"
"\n"
" trans = dict:transaction_begin()\n"
" trans:set_timestamp({['tv_sec'] = 1631278269, ['tv_nsec'] = 999999999})\n"
" trans:set('shared/testkey', 'updated')\n"
" trans:unset('shared/testkey2')\n"
" trans:commit()\n"
"\n"
" assert(dict:lookup('shared/testkey')[1] == 'updated')\n"
" assert(dict:lookup('shared/testkey2') == nil)\n"
"end\n";
struct dict_settings set = {
.base_dir = NULL,
};
struct dict *dict;
const char *error;
test_begin("dict lua");
struct ioloop *ioloop = io_loop_create();
i_unlink_if_exists(".test.dict");
if (dict_init("file:.test.dict", &set, &dict, &error) < 0)
i_fatal("dict_init(.test.dict) failed: %s", error);
struct dlua_script *script;
if (dlua_script_create_string(luascript, &script, NULL, &error) < 0)
i_fatal("dlua_script_create_string() failed: %s", error);
dlua_dovecot_register(script);
if (dlua_script_init(script, &error) < 0)
i_fatal("dlua_script_init() failed: %s", error);
lua_State *thread = dlua_script_new_thread(script);
dlua_push_dict(thread, dict);
if (dlua_pcall_yieldable(thread, "test_dict", 1, test_dict_finished,
ioloop, &error) < 0)
i_fatal("dlua_pcall() failed: %s", error);
io_loop_run(ioloop);
i_assert(lua_gettop(thread) == 0);
dlua_script_close_thread(script, &thread);
dlua_script_unref(&script);
dict_deinit(&dict);
io_loop_destroy(&ioloop);
i_unlink(".test.dict");
test_end();
}
int main(void)
{
static void (*const test_functions[])(void) = {
test_dict_register,
test_dict_lua,
NULL
};
return test_run(test_functions);
}
|
