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