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)