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()
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)
