Added unit tests for testing the server blacklist.

[xestiaab/.git] / source / tests / xestiaab_common.h

diff --git a/source/tests/xestiaab_common.h b/source/tests/xestiaab_common.h
index 10b4d7b..1d1e0c3 100644 (file)
--- a/source/tests/xestiaab_common.h
+++ b/source/tests/xestiaab_common.h
@@ -18,6 +18,9 @@
 
 #include <gtest/gtest.h>
 #include "../common/dirs.h"
+#include "../common/filename.h"
+#include "../common/mime.h"
+#include "../common/svrblist.h"
 
 TEST(CommonFunctions, GetUserDir){
        
@@ -108,4 +111,137 @@ TEST(CommonFunctions, GetAccountDir){
        
 #endif
        
+}
+
+TEST(CommonFunctions, GetAccountDirCertificate){
+       
+       // Run the unit tests depending on the operating system being used.
+       
+#if defined(__WIN32__)
+       
+#elif defined(__APPLE__)
+       
+#else
+       
+       // Setup for comparison purposes. Should be /home/<user>/.xestiaab/preferences
+       
+       wxString AccountDirCheck1 = "";
+       wxString AccountDirCheck2 = "";
+       wxString AccountDirCheck3 = "";
+       
+       AccountDirCheck1.Append(wxString::FromUTF8(getenv("HOME")));
+       AccountDirCheck1.Append(wxT("/.xestiaab/accounts/"));
+       AccountDirCheck1.Append("Example1");
+       AccountDirCheck1.Append(wxT("/server.crt"));
+
+       AccountDirCheck2.Append(wxString::FromUTF8(getenv("HOME")));
+       AccountDirCheck2.Append(wxT("/.xestiaab/accounts/"));
+       AccountDirCheck2.Append("Example2");
+       AccountDirCheck2.Append(wxT("/server.crt"));
+       
+       AccountDirCheck3.Append(wxString::FromUTF8(getenv("HOME")));
+       AccountDirCheck3.Append(wxT("/.xestiaab/accounts/"));
+       AccountDirCheck3.Append("Example3");
+       AccountDirCheck3.Append(wxT("/server.crt"));
+       
+       wxString AccountDir1 = GetAccountDir("Example1", true);
+       wxString AccountDir2 = GetAccountDir("Example2", true);
+       wxString AccountDir3 = GetAccountDir("Example3", true);
+
+       ASSERT_EQ(AccountDir1, AccountDirCheck1);
+       ASSERT_EQ(AccountDir2, AccountDirCheck2);
+       ASSERT_EQ(AccountDir3, AccountDirCheck3);
+       
+#endif
+       
+}
+
+TEST(CommonFunctions, GetAccountsFile){
+       
+       // Run the unit tests depending on the operating system being used.
+       
+#if defined(__WIN32__)
+       
+#elif defined(__APPLE__)
+       
+#else
+       
+       // Setup for comparison purposes. Should be 
+       // /home/<user>/.xestiaab/preferences/accounts
+       
+       wxString AccountsFileCheck = "";
+       AccountsFileCheck.Clear();
+       AccountsFileCheck.Append(wxString::FromUTF8(getenv("HOME")));
+       AccountsFileCheck.Append(wxT("/.xestiaab/preferences/accounts"));
+       
+       wxString AccountsFile = GetAccountsFile();
+       
+       ASSERT_EQ(AccountsFile, AccountsFileCheck);
+       
+#endif
+       
+}
+
+TEST(CommonFunctions, GetSettingsFile){
+       
+       // Run the unit tests depending on the operating system being used.
+       
+#if defined(__WIN32__)
+       
+#elif defined(__APPLE__)
+       
+#else
+       
+       // Setup for comparison purposes. Should be 
+       // /home/<user>/.xestiaab/preferences/accounts
+       
+       wxString SettingsFileCheck = "";
+       SettingsFileCheck.Clear();
+       SettingsFileCheck.Append(wxString::FromUTF8(getenv("HOME")));
+       SettingsFileCheck.Append(wxT("/.xestiaab/preferences/settings"));
+       
+       wxString SettingsFile = GetSettingsFile();
+       
+       ASSERT_EQ(SettingsFile, SettingsFileCheck);
+       
+#endif
+       
+}
+
+TEST(CommonFunctions, CreateFilenamePath){
+       
+       // Run the unit tests depending on the operating system being used.
+       
+#if defined(__WIN32__)
+       
+#elif defined(__APPLE__)
+       
+#else
+               
+       ASSERT_EQ("/test", CreateFilenamePath("", "test"));
+       ASSERT_EQ("/example/test", CreateFilenamePath("/example", "test"));
+       ASSERT_EQ("/home/meep/test", CreateFilenamePath("/home/meep", "test"));
+       ASSERT_EQ("/home/meep/moo/test", CreateFilenamePath("/home/meep/moo", "test"));
+       ASSERT_EQ("/home/meep/moo/yargh/test", CreateFilenamePath("/home/meep/moo/yargh", "test"));
+       
+#endif
+       
+}
+
+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