def test_given_a_list_of_done_studies_when_retrieve_all_studies_called_then_message_is_shown(
self, ):
# given
study = StudyDTO("path")
study.done = True
study_list = [deepcopy(study), deepcopy(study)]
display_mock = mock.Mock(spec=IDisplay)
my_retriever = RetrieveController(
self.data_repo,
self.remote_env_mock,
self.file_manager,
display_mock,
self.state_updater_mock,
)
my_retriever.repo.get_list_of_studies = mock.Mock(
return_value=study_list)
display_mock.show_message = mock.Mock()
# when
output = my_retriever.retrieve_all_studies()
# then
expected_message1 = "Retrieving all studies"
expected_message2 = "Everything is done"
calls = [
call(expected_message1, mock.ANY),
call(expected_message2, mock.ANY),
] # , call(my_study3)]
display_mock.show_message.assert_has_calls(calls)
assert output is True
def test_given_a_list_of_done_studies_when_all_studies_done_called_then_return_true(
self, ):
# given
study = StudyDTO("path")
study.done = True
study_list = [deepcopy(study), deepcopy(study)]
my_retriever = RetrieveController(
self.data_repo,
self.remote_env_mock,
self.file_manager,
self.display,
self.state_updater_mock,
)
my_retriever.repo.get_list_of_studies = mock.Mock(
return_value=study_list)
# when
output = my_retriever.all_studies_done
# then
assert output is True
