diff options
-rw-r--r-- | test/CMakeLists.txt | 1 | ||||
-rw-r--r-- | test/rspamd_test_suite.c | 1 | ||||
-rw-r--r-- | test/rspamd_upstream_test.c | 73 | ||||
-rw-r--r-- | test/tests.h | 2 |
4 files changed, 77 insertions, 0 deletions
diff --git a/test/CMakeLists.txt b/test/CMakeLists.txt index c0c4b5d9c..021eb0212 100644 --- a/test/CMakeLists.txt +++ b/test/CMakeLists.txt @@ -8,6 +8,7 @@ SET(TESTSRC rspamd_expression_test.c rspamd_dkim_test.c rspamd_rrd_test.c rspamd_radix_test.c + rspamd_upstream_test.c rspamd_test_suite.c) ADD_EXECUTABLE(rspamd-test EXCLUDE_FROM_ALL ${TESTSRC}) diff --git a/test/rspamd_test_suite.c b/test/rspamd_test_suite.c index b3a154b63..1d78fac49 100644 --- a/test/rspamd_test_suite.c +++ b/test/rspamd_test_suite.c @@ -53,6 +53,7 @@ main (int argc, char **argv) g_test_add_func ("/rspamd/aio", rspamd_async_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_run (); diff --git a/test/rspamd_upstream_test.c b/test/rspamd_upstream_test.c new file mode 100644 index 000000000..531f0bd27 --- /dev/null +++ b/test/rspamd_upstream_test.c @@ -0,0 +1,73 @@ +/* + * Copyright (c) 2014, Vsevolod Stakhov + * + * All rights reserved. + * + * Redistribution and use in source and binary forms, with or without + * modification, are permitted provided that the following conditions are met: + * * Redistributions of source code must retain the above copyright + * notice, this list of conditions and the following disclaimer. + * * Redistributions in binary form must reproduce the above copyright + * notice, this list of conditions and the following disclaimer in the + * documentation and/or other materials provided with the distribution. + * + * THIS SOFTWARE IS PROVIDED BY AUTHOR ''AS IS'' AND ANY + * EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED + * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE + * DISCLAIMED. IN NO EVENT SHALL AUTHOR BE LIABLE FOR ANY + * DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES + * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; + * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND + * ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT + * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS + * SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. + */ + +#include "config.h" +#include "main.h" +#include "upstream.h" + +const char *test_upstream_list = "microsoft.com:443:1,google.com:2,kernel.org:443:3"; + + +static void +rspamd_upstream_test_method (struct upstream_list *ls, + enum rspamd_upstream_rotation rot, const gchar *expected) +{ + struct upstream *up; + + up = rspamd_upstream_get (ls, rot); + g_assert (up != NULL); + g_assert (strcmp (rspamd_upstream_name (up), expected) == 0); +} + +void +rspamd_upstream_test_func (void) +{ + struct upstream_list *ls; + struct event_base *ev_base = event_init (); + struct rspamd_dns_resolver *resolver; + struct rspamd_config *cfg; + + cfg = (struct rspamd_config *)g_malloc (sizeof (struct rspamd_config)); + bzero (cfg, sizeof (struct rspamd_config)); + cfg->cfg_pool = rspamd_mempool_new (rspamd_mempool_suggest_size ()); + cfg->dns_retransmits = 2; + cfg->dns_timeout = 0.5; + + resolver = dns_resolver_init (NULL, ev_base, cfg); + + rspamd_upstreams_library_init (resolver->r, ev_base); + + ls = rspamd_upstreams_create (); + g_assert (rspamd_upstreams_parse_line (ls, test_upstream_list, 443, NULL)); + + /* Test master-slave rotation */ + rspamd_upstream_test_method (ls, RSPAMD_UPSTREAM_MASTER_SLAVE, "kernel.org"); + rspamd_upstream_test_method (ls, RSPAMD_UPSTREAM_MASTER_SLAVE, "kernel.org"); + + /* Test round-robin rotation */ + rspamd_upstream_test_method (ls, RSPAMD_UPSTREAM_ROUND_ROBIN, "kernel.org"); + rspamd_upstream_test_method (ls, RSPAMD_UPSTREAM_ROUND_ROBIN, "google.com"); + rspamd_upstream_test_method (ls, RSPAMD_UPSTREAM_ROUND_ROBIN, "microsoft.com"); +} diff --git a/test/tests.h b/test/tests.h index 129068920..f5472bd71 100644 --- a/test/tests.h +++ b/test/tests.h @@ -35,4 +35,6 @@ void rspamd_dkim_test_func (void); /* RRD test */ void rspamd_rrd_test_func (void); +void rspamd_upstream_test_func (void); + #endif |