def test_get_habits_with_period(self):
# Arrange
connection = connect_to_database("testing.db")
habit_list = get_all_habits(connection)
habit_collection = HabitCollection(habit_list)
# Assert
self.assertEqual(habit_collection.get_habits_with_period(3), "Workout")
self.assertEqual(habit_collection.get_habits_with_period(1), "Learn")
def test_get_current_longest_streak(self):
# Arrange
connection = connect_to_database("testing.db")
habit_list = get_all_habits(connection)
habit_collection = HabitCollection(habit_list)
# Assert
habit_name, habit_streak = habit_collection.get_current_longest_streak(
connection)
self.assertEqual(habit_name, "Learn")
self.assertEqual(habit_streak, 29)
def test_get_habit_by_name(self):
test_habit_1 = Habit("test_habit_create", 4, datetime.now())
test_habit_2 = Habit("Call mum", 4, datetime.now())
habit_collection = HabitCollection([test_habit_1, test_habit_2])
self.assertEqual(
habit_collection._get_habit_by_name("test_habit_create"),
test_habit_1)
self.assertEqual(habit_collection._get_habit_by_name("Call mum"),
test_habit_2)
self.assertEqual(habit_collection._get_habit_by_name("Wash car"),
False)
def test_get_tracked_habits(self):
# Arrange
connection = connect_to_database("testing.db")
habit_list = get_all_habits(connection)
habit_collection = HabitCollection(habit_list)
active_habits = """Workout
Learn
Wash dishes
Change bedding
Call mum"""
# Assert
self.assertEqual(habit_collection.get_tracked_habits(), active_habits)
def test_get_longest_streak_for_habit(self):
# Arrange
connection = connect_to_database("testing.db")
habit_list = get_all_habits(connection)
habit_collection = HabitCollection(habit_list)
# Assert
self.assertEqual(
habit_collection.get_longest_streak_for_habit(
"Workout", connection), 9)
self.assertEqual(
habit_collection.get_longest_streak_for_habit(
"Wash dishes", connection), 13)
def test_delete_habit(self):
connection = connect_to_database("testing.db")
habit_list = get_all_habits(connection)
habit_collection = HabitCollection(habit_list)
self.assertEqual(
delete_habit(connection, habit_collection, "Call son"), False)
self.assertEqual(
delete_habit(connection, habit_collection, "Call dad"), True)
close_connection_to_database(connection)
def test_habit_in_habit_collection(self):
new_habit = Habit("number_one", 1, datetime.now())
new_habit_2 = Habit("number_two", 2, datetime.now())
habit_collection = HabitCollection([new_habit, new_habit_2])
self.assertEqual(
habit_in_habit_collection("Call son", habit_collection), False)
self.assertEqual(
habit_in_habit_collection("number_one", habit_collection),
new_habit)
self.assertEqual(
habit_in_habit_collection("number_two", habit_collection),
new_habit_2)
def test_get_open_tasks(self):
# Arrange
connection = connect_to_database("testing.db")
habit_list = get_all_habits(connection)
habit_collection = HabitCollection(habit_list)
open_tasks_1 = """Workout
Wash dishes
Change bedding
Call mum"""
open_tasks_2 = """Workout
Wash dishes
Change bedding
Call mum
test_habit_open"""
open_tasks_3 = """Workout
Wash dishes
Change bedding
test_habit_open"""
# Assert 1
self.assertEqual(habit_collection.get_open_tasks(), open_tasks_1)
# Arrange for 2
create_new_habit(connection, "test_habit_open", 1,
datetime.now() - timedelta(days=2))
habit_list = get_all_habits(connection)
habit_collection = HabitCollection(habit_list)
# Assert for 2
self.assertEqual(habit_collection.get_open_tasks(), open_tasks_2)
# Arrange for 3
call_mum_habit = habit_collection._get_habit_by_name("Call mum")
call_mum_habit.confirm_task(connection)
# Assert for 3
self.assertEqual(habit_collection.get_open_tasks(), open_tasks_3)
# Close
open_habit = habit_collection._get_habit_by_name("test_habit_open")
open_habit.delete_habit_in_database(connection)
close_connection_to_database(connection)