X-Git-Url: http://Server1/repobrowser/?a=blobdiff_plain;f=source%2Ftests%2Fxestiacalendar_test.cpp;h=c6a602f9a58621510b1a19f4fc8469feddf2d99e;hb=92f76e70da98ebf857ba003bd5e14947c497082b;hp=06c8f7c7b65f309c83fe041f92e4034f0efe0e7d;hpb=d4ab20f1e8459d9e30a088e3b2faa88cf8e3ef30;p=xestiacalendar%2F.git

diff --git a/source/tests/xestiacalendar_test.cpp b/source/tests/xestiacalendar_test.cpp
index 06c8f7c..c6a602f 100644
--- a/source/tests/xestiacalendar_test.cpp
+++ b/source/tests/xestiacalendar_test.cpp
@@ -16,30 +16,42 @@
 // You should have received a copy of the GNU General Public License along
 // with Xestia Calendar. If not, see <http://www.gnu.org/licenses/>
 
+#include <curl/curl.h>
 #include <gtest/gtest.h>
 #include <iostream>
 #include <string>
 #include <stdexcept>
 
-#include "xestiacalendar_icaleventload.h"
 #include "xestiacalendar_commonfunctions.h"
+#include "xestiacalendar_populate.h"
+#include "xestiacalendar_icaleventload.h"
+#include "xestiacalendar_icaleventsave.h"
 #include "xestiacalendar_icaltaskload.h"
 #include "xestiacalendar_icaljournalload.h"
 #include "xestiacalendar_icalfreebusyload.h"
+#include "xestiacalendar_icalfreebusysave.h"
 #include "xestiacalendar_icaltimezoneload.h"
 #include "xestiacalendar_caldav.h"
 
 enum MenuOpts {
 	TESTS_ICALLOADEVENT = 1,
+	TESTS_ICALSAVEEVENT,
 	TESTS_ICALLOADTODO,
 	TESTS_ICALLOADJOURNAL,
 	TESTS_ICALLOADFREEBUSY,
+	TESTS_ICALSAVEFREEBUSY,
 	TESTS_ICALLOADTIMEZONE,
 	TESTS_CALDAV,
 	TESTS_COMMONFUNCTIONS,
+	TESTS_EXTRA,
 	TESTS_QUIT
 };
 
+enum ExtraMenuOpts{
+	EXTRA_POPULATECALDAV = 1,
+	EXTRA_RETURN
+};
+
 void printn(std::string text){
 // printn: Print a line and end with a newline (\n).
 
@@ -52,21 +64,91 @@ void printmenu(){
 
 	std::cout << "Select an option:" << std::endl << std::endl;
 	std::cout << TESTS_ICALLOADEVENT << ". iCalendar Event Component Load" << std::endl;
+	std::cout << TESTS_ICALSAVEEVENT << ". iCalendar Event Component Save" << std::endl;
 	std::cout << TESTS_ICALLOADTODO << ". iCalendar Task Component Load" << std::endl;
 	std::cout << TESTS_ICALLOADJOURNAL << ". iCalendar Journal Component Load" << std::endl;
 	std::cout << TESTS_ICALLOADFREEBUSY << ". iCalendar FreeBusy Component Load" << std::endl;
+	std::cout << TESTS_ICALSAVEFREEBUSY << ". iCalendar FreeBusy Component Save" << std::endl;
 	std::cout << TESTS_ICALLOADTIMEZONE << ". iCalendar Timezone Component Load" << std::endl;
 	std::cout << TESTS_CALDAV << ". CalDAV Object" << std::endl;
 	std::cout << TESTS_COMMONFUNCTIONS << ". Common Functions" << std::endl;
+	std::cout << TESTS_EXTRA << ". Extra Functions Menu" << std::endl;
 	std::cout << TESTS_QUIT << ". Quit" << std::endl;
 	std::cout << std::endl;
 
 }
 
+void printextramenu(){
+
+	std::cout << "Extra Functions Menu" << std::endl << std::endl;
+	std::cout << "Select an option:" << std::endl << std::endl;	
+	std::cout << EXTRA_POPULATECALDAV << ". Populate CalDAV calendar" << std::endl;
+	std::cout << EXTRA_RETURN << ". Return to previous menu" << std::endl;
+	std::cout << std::endl;
+	
+}
+
+void runextramenu(){
+	
+	bool ExitEnabled = false;
+	std::string StringOption = "";
+	int TestResult = 0;
+
+	while(ExitEnabled == false){
+	
+		printextramenu();
+
+		// Get user input.
+
+		std::cout << "Select Option: ";	
+		std::cin >> StringOption;
+		
+		int IntOption = -1;
+		
+		// Check if input is a number.
+		
+		try{
+			IntOption = stoi(StringOption);
+		}
+		
+		// Return to the top of the while statement if input
+		// really isn't a number.
+		
+		catch(std::invalid_argument e){
+			printn("Error: Selected option is not a number.");
+			continue;
+		}
+		
+		// Find which option has been selected from the
+		// input.
+		
+		switch(IntOption){
+		
+			case EXTRA_POPULATECALDAV:
+				printn("Populating CalDAV calendar...");
+				populatecaldav();
+				break;
+			case EXTRA_RETURN:
+				return;
+				break;
+			default:
+				printn("Invalid menu number given."); 
+				break;
+		
+		}
+	
+	}
+	
+}
+
 int main(int argc, char* argv[]){
 
+	// Initialise the several libraries that have
+	// been included.
+	
 	::testing::InitGoogleTest(&argc, argv);
-
+	curl_global_init(CURL_GLOBAL_ALL);
+	
 	printn("Xestia Calendar Unit Testing Application");
 	printn("(c)2016 Xestia Software Development");
 	printn("Note: Unit testing is currently in development");
@@ -111,6 +193,11 @@ int main(int argc, char* argv[]){
 				::testing::GTEST_FLAG(filter) = "iCalendarEvent*";
 				TestResult = RUN_ALL_TESTS();
 				break;
+			case TESTS_ICALSAVEEVENT:
+				printn("Running iCalendar Event Component Save tests...");
+				::testing::GTEST_FLAG(filter) = "iCalendarSaveEvent*";
+				TestResult = RUN_ALL_TESTS();
+				break;
 			case TESTS_ICALLOADTODO:
 				printn("Running iCalendar Task Component tests...");
 				::testing::GTEST_FLAG(filter) = "iCalendarTask*";
@@ -126,8 +213,13 @@ int main(int argc, char* argv[]){
 				::testing::GTEST_FLAG(filter) = "iCalendarFreeBusy*";
 				TestResult = RUN_ALL_TESTS();
 				break;
+			case TESTS_ICALSAVEFREEBUSY:
+				printn("Running iCalendar Free Busy Component Save tests...");
+				::testing::GTEST_FLAG(filter) = "iCalendarSaveFreeBusy*";
+				TestResult = RUN_ALL_TESTS();
+				break;
 			case TESTS_ICALLOADTIMEZONE:
-				printn("Running iCalendar Free Busy Component tests...");
+				printn("Running iCalendar Timezone Component tests...");
 				::testing::GTEST_FLAG(filter) = "iCalendarTimezone*";
 				TestResult = RUN_ALL_TESTS();
 				break;
@@ -141,6 +233,9 @@ int main(int argc, char* argv[]){
 				::testing::GTEST_FLAG(filter) = "CommonFunctions*";
 				TestResult = RUN_ALL_TESTS();
 				break;
+			case TESTS_EXTRA:
+				runextramenu();
+				break;
 			case TESTS_QUIT:
 				return 0;
 				break;