Source code for tests.test_terminal_data

# Copyright Kevin Deldycke <kevin@deldycke.com> and contributors.
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
#     http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
"""Test all terminal definitions, detection and terminal-specific groups."""

from __future__ import annotations

import logging

import pytest

from extra_platforms import (
    UNKNOWN_TERMINAL,
    current_terminal,
    detection,
    invalidate_caches,
)




@pytest.mark.parametrize(
    ("term", "level"),
    [
        ("xterm-256color", logging.WARNING),
        # Dumb terminals, as non-interactive SSH sessions declare them.
        ("dumb", logging.INFO),
        ("su", logging.INFO),
        ("unknown", logging.INFO),
        ("", logging.INFO),
    ],
)
@pytest.mark.parametrize("ssh", [False, True])
def test_unrecognized_terminal_log_level(
    term, level, ssh, monkeypatch, caplog, fresh_caches
):
    """Only a ``TERM`` naming a terminal type makes an unrecognized one a warning.

    An SSH session forwards ``TERM`` alone, from the client running the emulator.
    """
    env = {"TERM": term}
    if ssh:
        env["SSH_CONNECTION"] = "192.0.2.10 50000 192.0.2.20 22"
        level = logging.INFO
    monkeypatch.setattr(detection, "environ", env)
    invalidate_caches()
    caplog.set_level(logging.INFO)
    assert current_terminal() is UNKNOWN_TERMINAL
    assert [record.levelno for record in caplog.records] == [level]