def test_gen_file_name_never_increases_the_counter_if_the_file_does_not_exist(self, mock_path, mock_date): mock_date.today.return_value = date(2009, 8, 31) mock_path.return_value = False self.assertEqual(next(get_file_name(PurePath('c:\\tmp'), 'fr0st')), 'c:\\tmp\\fr0st.2009.08.31.1.ts') self.assertEqual(next(get_file_name(PurePath('c:\\tmp'), 'fr0st')), 'c:\\tmp\\fr0st.2009.08.31.1.ts') self.assertEqual(next(get_file_name(PurePath('c:\\tmp'), 'fr0st')), 'c:\\tmp\\fr0st.2009.08.31.1.ts')
def next_try(retries, directory, alias): file_name_generator = get_file_name(directory, alias, retries) for attempt in range(0, retries): yield (attempt, next(file_name_generator))
def test_gen_file_name_iterates_till_the_first_non_existent_file(self, mock_path, mock_date): mock_date.today.return_value = date(2009, 8, 31) mock_path.side_effect = lambda x: not x.name == 'fr0st.2009.08.31.44.ts' self.assertEqual(next(get_file_name(PurePath('c:\\tmp'), 'fr0st', 50)), 'c:\\tmp\\fr0st.2009.08.31.44.ts')