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