added unit tests for mydomains

unit_tests/Makefile.in

@@ -24,7 +24,7 @@ RUNNING_GROUP = `@id_bin@ -gn $(RUNNING_USER)`
 
 INSTALL = @INSTALL@
 
-all:	check_parser_utils check_rules check_digest smtp
+all:	check_parser_utils check_rules check_digest check_mydomains smtp
 
 check_parser_utils: check_parser_utils.c ../src/libpiler.a
 	$(CC) $(CFLAGS) $(INCDIR) $(DEFS) -o $@ $< -lpiler $(LIBS) $(LIBDIR)
@@ -35,6 +35,9 @@ check_rules: check_rules.c ../src/libpiler.a
 check_digest: check_digest.c ../src/libpiler.a
 	$(CC) $(CFLAGS) $(INCDIR) $(DEFS) -o $@ $< -lpiler $(LIBS) $(LIBDIR)
 
+check_mydomains: check_mydomains.c ../src/libpiler.a
+	$(CC) $(CFLAGS) $(INCDIR) $(DEFS) -o $@ $< -lpiler $(LIBS) $(LIBDIR)
+
 smtp:	smtp.c ../src/libpiler.a
 	$(CC) $(CFLAGS) $(INCDIR) $(DEFS) -o $@ $< -lpiler $(LIBS) $(LIBDIR)
 
@@ -42,7 +45,7 @@ install:
 	@echo
 
 clean:
-	rm -f check_parser_utils check_rules check_digest smtp
+	rm -f check_parser_utils check_rules check_digest check_mydomains smtp
 
 distclean: clean
 	rm -f Makefile

unit_tests/check_mydomains.c

@@ -0,0 +1,136 @@
+/*
+ * check_mydomains.c, SJ
+ */
+
+#include <stdio.h>
+#include <sys/stat.h>
+#include <unistd.h>
+#include <locale.h>
+#include <stdbool.h>
+#include <assert.h>
+#include "../src/piler.h"
+
+
+char *domains[] = {
+   "aaaa.fu",
+   "hajaja.kom",
+   "auajajau.zzz",
+   "akauejujedhj.com",
+   "738383.com",
+   "27281919.com",
+   "uazazahhja.ddd"
+};
+
+
+struct emails {
+   char email[SMALLBUFSIZE];
+   int match;
+};
+
+
+static void fill_domain_table(struct __config *cfg){
+   int i;
+   char buf[SMALLBUFSIZE];
+   struct session_data sdata;
+
+   if(open_database(&sdata, cfg) == ERR){
+      printf("cannot open database\n");
+      return;
+   }
+
+   printf("adding testing domains...\n");
+
+   for(i=0; i<sizeof(domains)/sizeof(char*); i++){
+      snprintf(buf, sizeof(buf)-1, "insert into domain (domain, mapped) values('%s', '%s')", domains[i], domains[i]);
+      p_query(&sdata, buf, cfg);
+   }
+
+   close_database(&sdata);
+}
+
+
+static void restore_domain_table(struct __config *cfg){
+   int i;
+   char buf[SMALLBUFSIZE];
+   struct session_data sdata;
+
+   if(open_database(&sdata, cfg) == ERR){
+      printf("cannot open database\n");
+      return;
+   }
+
+   printf("removing testing domains...\n");
+
+   for(i=0; i<sizeof(domains)/sizeof(char*); i++){
+      snprintf(buf, sizeof(buf)-1, "delete from domain where domain='%s'", domains[i]);
+      p_query(&sdata, buf, cfg);
+   }
+
+   close_database(&sdata);
+}
+
+
+static void test_mydomains(struct __config *cfg){
+   int i;
+   struct session_data sdata;
+   struct __data data;
+   struct emails emails[] = {
+      {"ajaja@aaaa.fu ", 1},
+      {"ajahahah@aaa.fu ", 0},
+      {"kajajaj@auajajau.zzz", 1},
+      {"djdkdjkdjkd@gmail.com ", 0},
+      {"akajajajw@738383.com ", 1},
+      {"eziezuiru@hotmail.fu ", 0},
+      {"mysjdjdjf@uazazahhja.ddd ", 1},
+      {"akdieckeeod@auajajau.zzz ", 1},
+      {"aukjwdkldkld@acts.hu ", 0},
+      {"aujwhcpajdl@fbi.gov.us ", 0},
+      {"Uahjajahsjsh@aaaa.fu ", 1},
+      {"aujedhkakd@aaaa.biz ", 0},
+      {"aujdkdkfi@aaaa.fu ", 1}
+   };
+
+   if(open_database(&sdata, cfg) == ERR){
+      printf("cannot open database\n");
+      return;
+   }
+
+   inithash(data.mydomains);
+
+   load_mydomains(&sdata, &data, cfg);
+
+   for(i=0; i<sizeof(emails)/sizeof(struct emails); i++){
+      //printf("%s / %d\n", emails[i].email, emails[i].match);
+      assert(is_email_address_on_my_domains(emails[i].email, &data) == emails[i].match && "test_mydomains()");
+   }
+
+   clearhash(data.mydomains);
+
+   close_database(&sdata);
+
+   printf("test_mydomains() OK\n");
+}
+
+
+int main(){
+   struct __config cfg;
+
+   if(!can_i_write_current_directory()) __fatal("cannot write current directory!");
+
+   (void) openlog("mydomains_test", LOG_PID, LOG_MAIL);
+
+   cfg = read_config("test.conf");
+
+   setlocale(LC_MESSAGES, cfg.locale);
+   setlocale(LC_CTYPE, cfg.locale);
+
+   fill_domain_table(&cfg);
+
+   test_mydomains(&cfg);
+
+   restore_domain_table(&cfg);
+
+
+   return 0;
+}
+