diff --git a/client/src/Handlers.h b/client/src/Handlers.h index 3ca11e1..bb8580e 100644 --- a/client/src/Handlers.h +++ b/client/src/Handlers.h @@ -3,6 +3,7 @@ #include #include "./Menu.h" +#include "./Test.h" class Handlers { @@ -61,7 +62,7 @@ class Handlers void handleMainMenu() { int option; - + Menu::showLoginMenu(); printf("Enter option: "); std::cin >> option; @@ -77,6 +78,11 @@ class Handlers case LoginMenuOption::QUIT_LOGIN: handleStop(); break; + case LoginMenuOption::PROMPT_UNIT_TEST: + printf("------------------TEST------------------\n"); + promptUnitTest(); + printf("------------------TEST------------------\n"); + break; default: printf("Invalid option\n"); handleMainMenu(); diff --git a/client/src/Menu.h b/client/src/Menu.h index 475ba59..0bd9457 100644 --- a/client/src/Menu.h +++ b/client/src/Menu.h @@ -5,7 +5,8 @@ enum LoginMenuOption { LOGIN = 1, REGISTER = 2, - QUIT_LOGIN = 3 + QUIT_LOGIN = 3, + PROMPT_UNIT_TEST = 4 }; enum MainMenuOption @@ -22,6 +23,7 @@ class Menu printf("1. Login\n"); printf("2. Register\n"); printf("3. Quit\n"); + printf("4. Start Unit Test\n"); } }; diff --git a/client/src/Test.cpp b/client/src/Test.cpp new file mode 100644 index 0000000..38f0176 --- /dev/null +++ b/client/src/Test.cpp @@ -0,0 +1,10 @@ +#include +#include +#include "Client.h" +#include "Test.h" + +using namespace std; + +void promptUnitTest() { + cout << "All Tests Pass!" << endl; +} \ No newline at end of file diff --git a/client/src/Test.h b/client/src/Test.h new file mode 100644 index 0000000..e313c2a --- /dev/null +++ b/client/src/Test.h @@ -0,0 +1,9 @@ +#ifndef TEST_H +#define TEST_H + + +void promptUnitTest(); + + + +#endif // TEST_H \ No newline at end of file