def test_format_data_truncate_info_level(mocked_logger):
data = ''
for i in range(0, 100001):
data = data + str(i)
mocked_logger.getLogger().level = 20
truncated = format_data_to_log_string(data)
assert truncated == data[:10000] + '... (set the log level to DEBUG or TRACE to see the full content)'
def test_format_data_not_truncate_trace_level(mocked_logger):
data = ''
for i in range(0, 100001):
data = data + str(i)
mocked_logger.getLogger().level = 0
truncated = format_data_to_log_string(data)
assert truncated == data
def test_format_with_file_descriptor():
with open(os.path.join(SCRIPT_DIR, '../atests/randombytes.bin'),
'rb') as f:
data_str = format_data_to_log_string(f)
assert data_str == repr(f)
def test_format_with_utf_encoded_data():
with open(os.path.join(SCRIPT_DIR, '../atests/data.json'), 'rb') as f:
data = f.read()
data_str = format_data_to_log_string(data)
assert data_str == data
def test_format_with_binary_data():
with open(os.path.join(SCRIPT_DIR, '../atests/randombytes.bin'),
'rb') as f:
data = f.read()
data_str = format_data_to_log_string(data)
assert data_str == data
def test_format_with_data_string():
data = "<xml>text</xml>"
data_str = format_data_to_log_string(data)
assert data_str == data
def test_format_with_data_json():
data = json.dumps({'key': 'value'})
data_str = format_data_to_log_string(data)
assert data_str == data
def test_format_data_to_log_string_truncated_3():
data = ''
for i in range(0, 10):
data = data + str(i)
truncated = format_data_to_log_string(data, 10)
assert truncated == data
def test_format_data_to_log_string_truncated_2():
data = ''
for i in range(0, 10):
data = data + str(i)
truncated = format_data_to_log_string(data, 0)
assert truncated == '... (set the log level to DEBUG or TRACE to see the full content)'
def test_format_with_data_and_headers_none():
data_str = format_data_to_log_string(None)
assert data_str is None
