"""Tests for tmuxp's helper and utility functions."""

from __future__ import annotations

import typing as t

import pytest

from libtmux.test.random import get_test_session_name

from libtmux.test.temporary import temp_session

if t.TYPE_CHECKING:

from libtmux.server import Server

def test_temp_session_kills_session_on_exit(server: Server) -> None:

"""Test temp_session() context manager kills session on exit."""

server = server

session_name = get_test_session_name(server=server)

with temp_session(server=server, session_name=session_name):

result = server.has_session(session_name)

assert result

assert not server.has_session(session_name)

@pytest.mark.flaky(reruns=5)

def test_temp_session_if_session_killed_before_exit(server: Server) -> None:

"""Handles situation where session already closed within context."""

server = server

session_name = get_test_session_name(server=server)

with temp_session(server=server, session_name=session_name):

# an error or an exception within a temp_session kills the session

server.kill_session(session_name)

result = server.has_session(session_name)

assert not result

# really dead?

assert not server.has_session(session_name)