Home | History | Annotate | Line # | Download | only in tests
kodDatabase.c revision 1.1.1.3.2.2 
      1 /*	$NetBSD: kodDatabase.c,v 1.1.1.3.2.2 2015/11/07 22:26:45 snj Exp $	*/
      2 
      3 #include "config.h"
      4 
      5 #include "ntp_types.h"
      6 #include "sntptest.h"
      7 #include "ntp_stdlib.h"
      8 #include "sntp-opts.h"
      9 #include "kod_management.h"
     10 #include "ntp_io.h"
     11 
     12 #include "unity.h"
     13 
     14 void setUp(void);
     15 void test_SingleEntryHandling(void);
     16 void test_MultipleEntryHandling(void);
     17 void test_NoMatchInSearch(void);
     18 void test_AddDuplicate(void);
     19 void test_DeleteEntry(void);
     20 
     21 
     22 void
     23 setUp(void) {
     24 	kod_init_kod_db("/dev/null", TRUE);
     25 }
     26 
     27 
     28 void
     29 test_SingleEntryHandling(void) {
     30 	const char HOST[] = "192.0.2.5";
     31 	const char REASON[] = "DENY";
     32 
     33 	add_entry(HOST, REASON);
     34 
     35 	struct kod_entry* result;
     36 
     37 	TEST_ASSERT_EQUAL(1, search_entry(HOST, &result));
     38 	TEST_ASSERT_EQUAL_STRING(HOST, result->hostname);
     39 	TEST_ASSERT_EQUAL_STRING(REASON, result->type);
     40 }
     41 
     42 
     43 void
     44 test_MultipleEntryHandling(void) {
     45 	const char HOST1[] = "192.0.2.3";
     46 	const char REASON1[] = "DENY";
     47 
     48 	const char HOST2[] = "192.0.5.5";
     49 	const char REASON2[] = "RATE";
     50 
     51 	const char HOST3[] = "192.0.10.1";
     52 	const char REASON3[] = "DENY";
     53 
     54 	add_entry(HOST1, REASON1);
     55 	add_entry(HOST2, REASON2);
     56 	add_entry(HOST3, REASON3);
     57 
     58 	struct kod_entry* result;
     59 
     60 	TEST_ASSERT_EQUAL(1, search_entry(HOST1, &result));
     61 	TEST_ASSERT_EQUAL_STRING(HOST1, result->hostname);
     62 	TEST_ASSERT_EQUAL_STRING(REASON1, result->type);
     63 
     64 	TEST_ASSERT_EQUAL(1, search_entry(HOST2, &result));
     65 	TEST_ASSERT_EQUAL_STRING(HOST2, result->hostname);
     66 	TEST_ASSERT_EQUAL_STRING(REASON2, result->type);
     67 
     68 	TEST_ASSERT_EQUAL(1, search_entry(HOST3, &result));
     69 	TEST_ASSERT_EQUAL_STRING(HOST3, result->hostname);
     70 	TEST_ASSERT_EQUAL_STRING(REASON3, result->type);
     71 
     72 	free(result);
     73 }
     74 
     75 
     76 void
     77 test_NoMatchInSearch(void) {
     78 	const char HOST_ADD[] = "192.0.2.6";
     79 	const char HOST_NOTADD[] = "192.0.6.1";
     80 	const char REASON[] = "DENY";
     81 
     82 	add_entry(HOST_ADD, REASON);
     83 
     84 	struct kod_entry* result;
     85 
     86 	TEST_ASSERT_EQUAL(0, search_entry(HOST_NOTADD, &result));
     87 	TEST_ASSERT_TRUE(result == NULL);
     88 }
     89 
     90 
     91 void
     92 test_AddDuplicate(void) {
     93 	const char HOST[] = "192.0.2.3";
     94 	const char REASON1[] = "RATE";
     95 	const char REASON2[] = "DENY";
     96 
     97 	add_entry(HOST, REASON1);
     98 	struct kod_entry* result1;
     99 	TEST_ASSERT_EQUAL(1, search_entry(HOST, &result1));
    100 
    101 	/*
    102 	 * Sleeps for two seconds since we want to ensure that
    103 	 * the timestamp is updated to a new value.
    104 	 */
    105 	sleep(2);
    106 
    107 	add_entry(HOST, REASON2);
    108 	struct kod_entry* result2;
    109 	TEST_ASSERT_EQUAL(1, search_entry(HOST, &result2));
    110 
    111 	TEST_ASSERT_FALSE(result1->timestamp == result2->timestamp);
    112 
    113 	free(result1);
    114 	free(result2);
    115 }
    116 
    117 
    118 void
    119 test_DeleteEntry(void) {
    120 	const char HOST1[] = "192.0.2.1";
    121 	const char HOST2[] = "192.0.2.2";
    122 	const char HOST3[] = "192.0.2.3";
    123 	const char REASON[] = "DENY";
    124 
    125 	add_entry(HOST1, REASON);
    126 	add_entry(HOST2, REASON);
    127 	add_entry(HOST3, REASON);
    128 
    129 	struct kod_entry* result;
    130 
    131 	TEST_ASSERT_EQUAL(1, search_entry(HOST2, &result));
    132 	free(result);
    133 
    134 	delete_entry(HOST2, REASON);
    135 
    136 	TEST_ASSERT_EQUAL(0, search_entry(HOST2, &result));
    137 
    138 	// Ensure that the other entry is still there.
    139 	TEST_ASSERT_EQUAL(1, search_entry(HOST1, &result));
    140 	free(result);
    141 }
    142