From 2f218889f8b09ecdabfd25c601e98d0e594f7042 Mon Sep 17 00:00:00 2001 From: Steve Brokenshire Date: Sun, 20 Mar 2016 23:07:14 +0000 Subject: [PATCH] Added unit tests for testing the server blacklist. --- source/tests/xestiaab_common.h | 12 ++++++++++++ 1 file changed, 12 insertions(+) diff --git a/source/tests/xestiaab_common.h b/source/tests/xestiaab_common.h index 8d60863..1d1e0c3 100644 --- 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 -- 2.39.5