Added unit tests for testing the server blacklist.

diff --git a/source/tests/xestiaab_common.h b/source/tests/xestiaab_common.h
index 8d60863..1d1e0c3 100644 (file)
--- a/source/tests/xestiaab_common.h
+++ b/source/tests/xestiaab_common.h
@@ -20,6 +20,7 @@
 #include "../common/dirs.h"
 #include "../common/filename.h"
 #include "../common/mime.h"
+#include "../common/svrblist.h"
 
 TEST(CommonFunctions, GetUserDir){
        
@@ -229,7 +230,18 @@ TEST(CommonFunctions, CreateFilenamePath){
 
 TEST(CommonFunctions, GetMIME){
        
+       // Get the MIME type based on the files given.
+       
        ASSERT_EQ("text/plain", GetMIME("TextMIMEExample.txt"));
        ASSERT_EQ("image/png", GetMIME("PictureMIMEExample.png"));
        
+}
+
+TEST(CommonFunctions, CheckBlacklist){
+       
+       // Test that the blacklist is actually working.
+       
+       ASSERT_EQ(false, CheckBlacklist("example.com"));
+       ASSERT_EQ(true, CheckBlacklist("google.com"));
+       
 }
\ No newline at end of file