def test_19_truncate_comma_list(self):
r = truncate_comma_list("123456,234567,345678", 19)
self.assertEqual(len(r), 19)
self.assertEqual(r, "1234+,234567,345678")
r = truncate_comma_list("123456,234567,345678", 18)
self.assertEqual(len(r), 18)
self.assertEqual(r, "1234+,2345+,345678")
r = truncate_comma_list("123456,234567,345678", 16)
self.assertEqual(len(r), 16)
self.assertEqual(r, "123+,2345+,3456+")
# There are more entries than the max_len. We will not be able
# to shorten all entries, so we simply take the beginning of the string.
r = truncate_comma_list("12,234567,3456,989,123,234,234", 4)
self.assertEqual(len(r), 4)
self.assertEqual(r, "12,+")
def test_19_truncate_comma_list(self):
r = truncate_comma_list("123456,234567,345678", 19)
self.assertEqual(len(r), 19)
self.assertEqual(r, "1234+,234567,345678")
r = truncate_comma_list("123456,234567,345678", 18)
self.assertEqual(len(r), 18)
self.assertEqual(r, "1234+,2345+,345678")
r = truncate_comma_list("123456,234567,345678", 16)
self.assertEqual(len(r), 16)
self.assertEqual(r, "123+,2345+,3456+")
# There are more entries than the max_len. We will not be able
# to shorten all entries, so we simply take the beginning of the string.
r = truncate_comma_list("12,234567,3456,989,123,234,234", 4)
self.assertEqual(len(r), 4)
self.assertEqual(r, "12,+")
def _truncate_data(self):
"""
Truncate self.audit_data according to the self.custom_column_length.
:return: None
"""
for column, l in self.custom_column_length.items():
if column in self.audit_data:
data = self.audit_data[column]
if isinstance(data, string_types):
if column == "policies":
# The policies column is shortend per comma entry
data = truncate_comma_list(data, l)
else:
data = data[:l]
self.audit_data[column] = data
def _truncate_data(self):
"""
Truncate self.audit_data according to the column_length.
:return: None
"""
for column, l in column_length.items():
if column in self.audit_data:
data = self.audit_data[column]
if isinstance(data, string_types):
if column == "policies":
# The policies column is shortend per comma entry
data = truncate_comma_list(data, l)
else:
data = data[:l]
self.audit_data[column] = data
