1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192 |
- #include "config.h"
- #include "rspamd.h"
- #include "libstat/stat_api.h"
- #include "lua/lua_common.h"
- #include "tests.h"
- #include "contrib/libev/ev.h"
-
- struct rspamd_main *rspamd_main = NULL;
- struct ev_loop *event_loop = NULL;
- worker_t *workers[] = { NULL };
-
- gchar *lua_test = NULL;
- gchar *lua_test_case = NULL;
- gboolean verbose = FALSE;
-
- static GOptionEntry entries[] =
- {
- { "test", 't', 0, G_OPTION_ARG_STRING, &lua_test,
- "Lua test to run (i.e. selectors.lua)", NULL },
- { "test-case", 'c', 0, G_OPTION_ARG_STRING, &lua_test_case,
- "Lua test to run, lua pattern i.e. \"case .* rcpts\"", NULL },
- { NULL, 0, 0, G_OPTION_ARG_NONE, NULL, NULL, NULL }
- };
-
- int
- main (int argc, char **argv)
- {
- struct rspamd_config *cfg;
- GOptionContext *context;
- GError *error = NULL;
-
- rspamd_main = (struct rspamd_main *)g_malloc (sizeof (struct rspamd_main));
- memset (rspamd_main, 0, sizeof (struct rspamd_main));
- rspamd_main->server_pool = rspamd_mempool_new (rspamd_mempool_suggest_size (), NULL, 0);
- cfg = rspamd_config_new (RSPAMD_CONFIG_INIT_DEFAULT);
- cfg->libs_ctx = rspamd_init_libs ();
- rspamd_main->cfg = cfg;
- cfg->cfg_pool = rspamd_mempool_new (rspamd_mempool_suggest_size (), NULL, 0);
- cfg->log_type = RSPAMD_LOG_CONSOLE;
- cfg->log_level = G_LOG_LEVEL_MESSAGE;
-
- rspamd_set_logger (cfg, g_quark_from_static_string("rspamd-test"),
- &rspamd_main->logger, rspamd_main->server_pool);
- (void)rspamd_log_open (rspamd_main->logger);
-
- g_test_init (&argc, &argv, NULL);
-
- context = g_option_context_new ("- run rspamd test");
- g_option_context_add_main_entries (context, entries, NULL);
-
- if (!g_option_context_parse (context, &argc, &argv, &error)) {
- fprintf (stderr, "option parsing failed: %s\n", error->message);
- g_option_context_free (context);
- exit (1);
- }
-
- rspamd_lua_set_path ((lua_State *)cfg->lua_state, NULL, NULL);
- event_loop = ev_default_loop (EVFLAG_SIGNALFD|EVBACKEND_ALL);
- rspamd_stat_init (cfg, event_loop);
- rspamd_url_init (NULL);
-
- if (g_test_verbose ()) {
- cfg->log_level = G_LOG_LEVEL_DEBUG;
- rspamd_set_logger (cfg, g_quark_from_static_string("rspamd-test"),
- &rspamd_main->logger, rspamd_main->server_pool);
- (void)rspamd_log_reopen (rspamd_main->logger);
- }
-
- g_log_set_default_handler (rspamd_glib_log_function, rspamd_main->logger);
-
- g_test_add_func ("/rspamd/mem_pool", rspamd_mem_pool_test_func);
- g_test_add_func ("/rspamd/radix", rspamd_radix_test_func);
- g_test_add_func ("/rspamd/dns", rspamd_dns_test_func);
- g_test_add_func ("/rspamd/dkim", rspamd_dkim_test_func);
- g_test_add_func ("/rspamd/rrd", rspamd_rrd_test_func);
- g_test_add_func ("/rspamd/upstream", rspamd_upstream_test_func);
- g_test_add_func ("/rspamd/shingles", rspamd_shingles_test_func);
- g_test_add_func ("/rspamd/lua", rspamd_lua_test_func);
- g_test_add_func ("/rspamd/cryptobox", rspamd_cryptobox_test_func);
- g_test_add_func ("/rspamd/heap", rspamd_heap_test_func);
- g_test_add_func ("/rspamd/lua_pcall", rspamd_lua_lua_pcall_vs_resume_test_func);
-
- #if 0
- g_test_add_func ("/rspamd/http", rspamd_http_test_func);
- g_test_add_func ("/rspamd/url", rspamd_url_test_func);
- g_test_add_func ("/rspamd/statfile", rspamd_statfile_test_func);
- g_test_add_func ("/rspamd/aio", rspamd_async_test_func);
- #endif
- g_test_run ();
-
- return 0;
- }
|