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')
Example #2
0
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')