from functional2.testlib.terminal_code_eater import eat_terminal_codes


def test_eats_color():
    assert eat_terminal_codes(b'\x1b[7mfoo blah bar\x1b[0m') == b'foo blah bar'

def test_eats_osc():
    assert eat_terminal_codes(b'\x1b]8;;http://example.com\x1b\\This is a link\x1b]8;;\x1b\\') == b'This is a link'