def user_choice(): """ Presents a menu with 3 options and allows the user to choose one. Provides the result the user has chosen. """ choice = input("Please select the number of your choice: ") clear_screen() try: # Checks if the number is integer or not choice = int(choice) except ValueError: print("Please enter the number again\n") # Returns the main menu main_menu() user_choice() if choice == 1: # Writes tasks on the csvfile task = Task() task.add_task() # Returns the main menu print("The entry has been added.\n") play_again() elif choice == 2: # Searches entry search() elif choice == 3: # Quits program print("See you again!")
def test_task(self): delete_db.delete_data_in_db() user = self.create_user_and_login() project = Project(user.get_cookie(), user.get_user_id(), "1") project.create_project() req = Requirement(user.get_cookie(), project.get_id()) req.add_req(project.get_id()) task = Task(user.get_cookie(), req.get_id()) task.add_task() task.update_task() task.get_task() task.get_all_task() task.delete_task()
def test_task(self): delete_db.delete_data_in_db() user = self.create_user_and_login() project = Project(user.get_cookie(), user.get_user_id(),"1") project.create_project() req = Requirement(user.get_cookie(), project.get_id()) req.add_req(project.get_id()) task = Task(user.get_cookie(), req.get_id()) task.add_task() task.update_task() task.get_task() task.get_all_task() task.delete_task()
def test_project(self): delete_db.delete_data_in_db() user = self.create_user_and_login() user.add_user() project = Project(user.get_cookie(), user.get_user_id(), "1") project.create_project() project.create_project() project.update_project() project.add_dev_to_projects("2") project.get_users_per_project() project.get_users_ranking() project.remove_dev_to_projects() project.get_project() project.get_all_projects_for_user() sprint = Sprint(user.get_cookie(), project.get_id()) sprint.add_sprint() req = Requirement(user.get_cookie(), project.get_id()) req.add_req(project.get_id()) task = Task(user.get_cookie(), req.get_id()) task.add_task() project.get_current_sprint() project.get_progress() project.get_backlog() project.delete_project()
class WorkLogTests(unittest.TestCase): def setUp(self): self.task = Task() self.new = Entry.create(employee_name=DATA["employee_name"], task_name=DATA["task_name"], mins=DATA["mins"], notes=DATA["notes"], date=DATA["date"]) def test_get_employee_name(self): with mock.patch('builtins.input', side_effect=["Test employee"], return_value=DATA["employee_name"]): assert self.task.get_employee_name() == DATA["employee_name"] def test_edit_employee_name(self): with mock.patch('builtins.input', side_effect=["Test Name Edit"]): self.assertEqual( self.task.edit_employee_name(self.new).employee_name, 'Test Name Edit') def test_get_task_name(self): with mock.patch('builtins.input', side_effect=["Test task"], return_value=DATA["task_name"]): assert self.task.get_task_name() == DATA["task_name"] def test_edit_task_name(self): with mock.patch('builtins.input', side_effect=["Test Task Name Edit"]): self.assertEqual( self.task.edit_task_name(self.new).task_name, 'Test Task Name Edit') def test_get_time_spent(self): with mock.patch('builtins.input', side_effect=["6"], return_value=DATA["mins"]): assert self.task.get_time_spent() == DATA["mins"] def test_edit_time_spent(self): with mock.patch('builtins.input', side_effect=["5"]): self.assertEqual(self.task.edit_time_spent(self.new).mins, '5') def test_get_notes(self): with mock.patch('builtins.input', side_effect=["Test notes"], return_value=DATA["notes"]): assert self.task.get_notes() == DATA["notes"] def test_edit_notes(self): with mock.patch('builtins.input', side_effect=["Test Note Edit"]): self.assertEqual( self.task.edit_notes(self.new).notes, 'Test Note Edit') def test_get_date(self): with mock.patch('builtins.input', side_effect=["04-12-1991"], return_value=DATA["date"]): self.assertRaises(TypeError) assert self.task.get_date() == DATA["date"] def test_edit_date(self): with mock.patch('builtins.input', side_effect=["04-13-1991"]): self.assertEqual(self.task.edit_date(self.new).date, '04-13-1991') def test_add_task(self): test = { "employee_name": "Test employee", "task_name": "Test task", "mins": "6", "notes": "Test notes", "date": "04-12-1991" } assert self.task.add_task(test)["task_name"] == DATA["task_name"] def test_find_employee_name(self): employee_name = self.task.find_task('Employee Name', 'Test employee') self.assertTrue(employee_name) def test_find_task_name(self): task_name = self.task.find_task('Keyword', 'Test task') self.assertTrue(task_name) def test_find_date(self): date = self.task.find_task('Date', '04-12-1991') self.assertTrue(date) def test_find_date_range(self): date = self.task.find_task('Date Range', ['04-02-1991', '04-20-1991']) self.assertTrue(date)