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')