Assertions

tomodachi_testcontainers.assertions

DEFAULT_DATETIME_RANGE module-attribute

DEFAULT_DATETIME_RANGE = timedelta(seconds=10)

assert_datetime_within_range

assert_datetime_within_range(
    value, range=DEFAULT_DATETIME_RANGE
)

Source code in src/tomodachi_testcontainers/assertions.py

def assert_datetime_within_range(value: datetime, range: timedelta = DEFAULT_DATETIME_RANGE) -> None:
    current_datetime = datetime.now(UTC)
    start_datetime = current_datetime - range
    end_datetime = current_datetime + range
    assert start_datetime <= value <= end_datetime  # nosec: B101

assert_logs_contain

assert_logs_contain(container, contains)

Source code in src/tomodachi_testcontainers/assertions.py

def assert_logs_contain(container: DockerContainer, contains: str) -> None:
    stdout_and_stderr_logs = cast("tuple[bytes, bytes]", container.get_logs())
    for log in stdout_and_stderr_logs:
        if contains in log.decode():
            return
    raise AssertionError(f"Expected logs to contain: '{contains}'; logs: {stdout_and_stderr_logs}")

assert_logs_not_contain

assert_logs_not_contain(container, contains)

Source code in src/tomodachi_testcontainers/assertions.py

def assert_logs_not_contain(container: DockerContainer, contains: str) -> None:
    stdout_and_stderr_logs = cast("tuple[bytes, bytes]", container.get_logs())
    for log in stdout_and_stderr_logs:
        if contains in log.decode():
            raise AssertionError(f"Expected logs not to contain: '{contains}'; logs: {stdout_and_stderr_logs}")

assert_logs_match_line_count

assert_logs_match_line_count(container, contains, count)

Source code in src/tomodachi_testcontainers/assertions.py

def assert_logs_match_line_count(container: DockerContainer, contains: str, count: int) -> None:
    stdout_logs = cast("bytes", container.get_logs()[0])
    stderr_logs = cast("bytes", container.get_logs()[1])
    stdout_and_stderr_logs = f"{stdout_logs.decode()}\n{stderr_logs.decode()}"

    matched_lines = [log for log in stdout_and_stderr_logs.splitlines() if contains in log]
    error_msg = (
        f"Expected '{contains}' to be contained in {count} lines, found {len(matched_lines)} lines"
        f"; logs: {stdout_and_stderr_logs}"
    )
    assert len(matched_lines) == count, error_msg  # nosec: B101