def test_get_stored_market_names__contains_files(self, mock_os): market = Market('market') DummyFile('a') filename_list = ['a', 'b', 'c', 'd'] dummyfile_list = [ DummyFile('a'), DummyFile('b'), DummyFile('c'), DummyFile('d') ] iterator_filelist = iter(dummyfile_list) mock_os.scandir.return_value = iterator_filelist self.assertEqual(market.get_stored_market_names(), filename_list)
def test_get_stored_market_names__empty(self, mock_os): market = Market('market') mock_os.scandir.return_value = [] self.assertEqual(market.get_stored_market_names(), [])
def test_get_stored_market_names__scandir_is_called(self, mock_os): market = Market('market') market.get_stored_market_names() mock_os.scandir.assert_called_with("Markets")