Function test_session_log_bytesio

tests/test_netmiko_session_log.py:245–271

Verify session_log matches expected content, but when channel writes are also logged.

(device_slog_test_name, commands, expected_responses)

def test_session_log_bytesio(device_slog_test_name, commands, expected_responses):
    """Verify session_log matches expected content, but when channel writes are also logged."""
    device_slog = device_slog_test_name[0]
    test_name = device_slog_test_name[1]

    s_log = io.BytesIO()

    # The netmiko connection has not been established yet.
    device_slog["session_log"] = s_log
    device_slog["session_log_file_mode"] = "write"
    device_slog["session_log_record_writes"] = True

    conn = ConnectHandler(**device_slog)
    command = commands["basic"]
    session_action(conn, command)
    conn.disconnect()

    compare_file = expected_responses["compare_log"]
    compare_file = add_test_name_to_file_name(compare_file, test_name)

    with open(compare_file, "rb") as f:
        compare_contents = filter_bare_prompts(f.read(), conn.base_prompt)
    compare_log_md5 = calc_md5(contents=compare_contents)

    log_content = filter_bare_prompts(s_log.getvalue(), conn.base_prompt)
    session_log_md5 = calc_md5(contents=log_content)
    assert session_log_md5 == compare_log_md5


def test_session_log_no_log(device_slog_test_name):