mirror of
https://github.com/NixOS/nixpkgs.git
synced 2026-08-26 18:24:53 +00:00
d9befc6484
We have a lot of changes in the style of
def foo(
self,
new: dt.timedelta | None = None,
old: float | None = None,
)
which is OK for cases where the API is used as
foo(old=23.5)
however for named arguments that actually falls short in the `ty`-stage:
foo(23.5)
As a workaround I flipped the order of old/new and changed the type of
`old` to accept both a timedelta and the old type, i.e.
def foo(
self,
old: float | dt.timedelta | None = None,
new: dt.timedelta | None = None,
)
and only give a deprecation warning if `old` is not of type `dt.timedelta | None`.
That way, both
foo(old=23.5)
foo(23.5)
are still accepted, at the same time, both
foo(new=timedelta(...))
foo(timedelta(...))
are OK.
143 lines
3.0 KiB
Python
143 lines
3.0 KiB
Python
# This file contains type hints that can be prepended to Nix test scripts so they can be type
|
|
# checked.
|
|
|
|
import datetime as dt
|
|
from contextlib import contextmanager
|
|
from typing import Any, Callable, ContextManager, Generator, List, Optional, Union
|
|
from unittest import TestCase
|
|
|
|
from test_driver.debug import DebugAbstract, DebugNop
|
|
from test_driver.driver import Driver
|
|
from test_driver.logger import AbstractLogger, CompositeLogger
|
|
from typing_extensions import Protocol
|
|
|
|
from test_driver.machine import BaseMachine, NspawnMachine, QemuMachine
|
|
from test_driver.vlan import VLan
|
|
|
|
|
|
# Protocols
|
|
|
|
|
|
class CreateMachineProtocol(Protocol):
|
|
def __call__(
|
|
self,
|
|
start_command: str | dict,
|
|
*,
|
|
name: Optional[str] = None,
|
|
keep_machine_state: bool = False,
|
|
**kwargs: Any, # to allow usage of deprecated keep_vm_state
|
|
) -> QemuMachine:
|
|
raise Exception("This is just type information for the Nix test driver")
|
|
|
|
|
|
class PollingConditionProtocol(Protocol):
|
|
def __call__(
|
|
self,
|
|
fun_: Callable | None = None,
|
|
*,
|
|
seconds_interval: dt.timedelta | float | None = None,
|
|
interval: dt.timedelta | None = None,
|
|
description: str | None = None,
|
|
) -> Union[Callable[[Callable], ContextManager], ContextManager]:
|
|
raise Exception("This is just type information for the Nix test driver")
|
|
|
|
|
|
# Classes
|
|
|
|
|
|
class AssertionTester(TestCase):
|
|
pass
|
|
|
|
|
|
# Global Variables
|
|
|
|
debug: DebugAbstract = DebugNop()
|
|
machines: List[BaseMachine] = []
|
|
machines_nspawn: List[NspawnMachine] = []
|
|
machines_qemu: List[QemuMachine] = []
|
|
t = AssertionTester()
|
|
vlans: List[VLan] = []
|
|
|
|
|
|
def create_fake_driver() -> Driver:
|
|
raise Exception("fake driver")
|
|
|
|
|
|
driver = create_fake_driver()
|
|
|
|
|
|
# Functions
|
|
|
|
|
|
# these are going to be called by the testScriptWithTypes in driver.nix
|
|
def create_fake_qemu_machine() -> QemuMachine:
|
|
raise Exception("fake qemu machine")
|
|
|
|
|
|
def create_fake_nspawn_machine() -> NspawnMachine:
|
|
raise Exception("fake nspawn machine")
|
|
|
|
|
|
def create_fake_vlan() -> VLan:
|
|
raise Exception("fake vlan")
|
|
|
|
|
|
def create_machine(
|
|
start_command: str, name: str | None = None, keep_machine_state: bool = False
|
|
) -> QemuMachine:
|
|
raise Exception("fake machine")
|
|
|
|
|
|
def dump_machine_ssh() -> None:
|
|
return None
|
|
|
|
|
|
def join_all() -> None:
|
|
return None
|
|
|
|
|
|
log: AbstractLogger = CompositeLogger([])
|
|
|
|
|
|
def polling_condition(
|
|
fun_: Callable | None = None,
|
|
*,
|
|
seconds_interval: float | dt.timedelta | None = None,
|
|
interval: dt.timedelta | None = None,
|
|
description: str | None = None,
|
|
):
|
|
pass
|
|
|
|
|
|
def retry(
|
|
fn: Callable,
|
|
timeout_seconds: int | dt.timedelta | None = None,
|
|
timeout: dt.timedelta | None = None,
|
|
) -> None:
|
|
pass
|
|
|
|
|
|
def run_tests() -> None:
|
|
return
|
|
|
|
|
|
def serial_stdout_off() -> None:
|
|
return None
|
|
|
|
|
|
def serial_stdout_on() -> None:
|
|
return None
|
|
|
|
|
|
def start_all() -> None:
|
|
return
|
|
|
|
|
|
def test_script() -> None:
|
|
return
|
|
|
|
|
|
@contextmanager
|
|
def subtest(str: str) -> Generator[None, None, None]:
|
|
yield
|