def test_should_delete_the_goal_if_all_data_is_OK_and_the_admin_is_logged_in( database): database.executescript( "INSERT INTO admin_users VALUES('admin-1', 'user-1');") user_repository = UserRepository( None, database, get_current_user_id=lambda: "admin-1", ) goal_repository = GoalRepository(None, database) goal_interactor = GoalInteractor(None, goal_repository, user_repository) data = { "id": "test-id", "date": "2020-03-15", "title": "test-title", "category": "test-category", "status": 0, "user_id": "user-1" } goal_interactor.save_goal(data) goals = goal_interactor.get_goals_by_date_and_assigned_user_id( "2020-03-15", "user-1") assert len(goals) == 1 assert goals[0].id == "test-id" goal_interactor.delete_goal_by_id("test-id") goals = goal_interactor.get_goals_by_date_and_assigned_user_id( "2020-03-15", "user-1") assert goals == []
def test_should_update_the_goal_if_the_goal_exists_and_all_data_is_OK_and_the_admin_is_logged_in(database): database.executescript( """ INSERT INTO admin_users VALUES ('admin-1', 'user-1'); INSERT INTO goals VALUES ("test-id-1", "2020-03-15", "test-title", "test-category", 1, "user-1"), ("test-id-2", "2020-03-15", "test-title", "test-category", 1, "user-1"), ("test-id-3", "2020-03-15", "test-title", "test-category", 1, "user-1"), ("test-id-4", "2020-03-16", "test-title", "test-category", 0, "user-1"), ("test-id-5", "2020-03-17", "test-title", "test-category", 1, "user-2"); """) user_repository = UserRepository( None, database, get_current_user_id=lambda: "admin-1", ) goal_repository = GoalRepository(None, database) goal_interactor = GoalInteractor(None, goal_repository, user_repository) data = { "id": "test-id-1", "date": "2020-03-18", "title": "test-title", "category": "test-category", "status": 0, "user_id": "user-1" } goal_interactor.save_goal(data) goals = goal_interactor.get_goals_by_date_and_assigned_user_id( "2020-03-18", "user-1") assert len(goals) == 1 assert goals[0].id == "test-id-1" assert goals[0].date == "2020-03-18" assert goals[0].title == "test-title" assert goals[0].category == "test-category" assert goals[0].status == False assert goals[0].user_id == "user-1"