def test_trim_too_long_filename(self): """trim_too_long_filename trims filename if its longer than allowed""" filename = 'filename' extension = '.jpg' long_filename = '{}{}'.format(filename * 10, extension) trimmed_filename = trim_long_filename(long_filename) self.assertEqual(len(trimmed_filename), FILENAME_MAX_LEN) self.assertTrue(trimmed_filename.startswith(filename)) self.assertTrue(trimmed_filename.endswith(extension))
def test_trim_short_filename(self): """trim_too_long_filename returns short filename as it is""" filename = 'filename.jpg' trimmed_filename = trim_long_filename(filename) self.assertEqual(trimmed_filename, filename)