def test_works_for_1_and_2_attempts(self, mock_date):
mock_date.today.return_value = date(1945, 8, 6)
results = [fname for fname in file_name_generator('john', 1)]
self.assertEqual(results[0], 'john.1945.08.06.1.ts')
self.assertEqual(results[-1], 'john.1945.08.06.9.ts')
results = [fname for fname in file_name_generator('jack', 2)]
self.assertEqual(results[0], 'jack.1945.08.06.1.ts')
self.assertEqual(results[-1], 'jack.1945.08.06.9.ts')
def test_filenames_for_number_of_attempts_greater_than_10_should_have_leading_zeroes(self, mock_date):
mock_date.today.return_value = date(1980, 7, 5)
for fname in file_name_generator('fido', 45):
pass
self.assertEqual(fname, 'fido.1980.07.05.99.ts')
file_name_gen = file_name_generator('fido', 105)
self.assertEqual(next(file_name_gen), 'fido.1980.07.05.001.ts')
self.assertEqual(next(file_name_gen), 'fido.1980.07.05.002.ts')
def test_get_last_filename(self, mock_date):
mock_date.today.return_value = date(1980, 7, 5)
for fname in file_name_generator('fido', 10):
pass
self.assertEqual(fname, 'fido.1980.07.05.9.ts')
def test_get_two_filenames(self, mock_date):
file_name_gen = file_name_generator('fr0st', 10)
mock_date.today.return_value = date(2009, 8, 31)
self.assertEqual(next(file_name_gen), 'fr0st.2009.08.31.1.ts')
self.assertEqual(next(file_name_gen), 'fr0st.2009.08.31.2.ts')
