commit 814027378bcddb1f885a1a17f0b34b16d7593b54 · pyrox.dev/nixpkgs

nixos/lib/test-driver/default.nix

···

       25
        
         checkPhase = ''

     

       26
        
           mypy --disallow-untyped-defs \

     

       27
        
                 --no-implicit-optional \

     

       0
        
       
     

       0
        
       
     

       28
        
                 --ignore-missing-imports ${src}/test_driver

     

       29
        
           pylint --errors-only --enable=unused-import ${src}/test_driver

     

       30
        
           black --check --diff ${src}/test_driver

···

       25
        
         checkPhase = ''

     

       26
        
           mypy --disallow-untyped-defs \

     

       27
        
                 --no-implicit-optional \

     

       28
       +
                 --pretty \

     

       29
       +
                 --no-color-output \

     

       30
        
                 --ignore-missing-imports ${src}/test_driver

     

       31
        
           pylint --errors-only --enable=unused-import ${src}/test_driver

     

       32
        
           black --check --diff ${src}/test_driver

nixos/lib/test-driver/test_driver/py.typed

This is a binary file and will not be displayed.

+42

nixos/lib/test-script-prepend.py

···

       1
       +
       # This file contains type hints that can be prepended to Nix test scripts so they can be type

     

       2
       +
       # checked.

     

       3
       +
       

     

       4
       +
       from test_driver.driver import Driver

     

       5
       +
       from test_driver.vlan import VLan

     

       6
       +
       from test_driver.machine import Machine

     

       7
       +
       from test_driver.logger import Logger

     

       8
       +
       from typing import Callable, Iterator, ContextManager, Optional, List, Dict, Any, Union

     

       9
       +
       from typing_extensions import Protocol

     

       10
       +
       from pathlib import Path

     

       11
       +
       

     

       12
       +
       

     

       13
       +
       class RetryProtocol(Protocol):

     

       14
       +
           def __call__(self, fn: Callable, timeout: int = 900) -> None:

     

       15
       +
               raise Exception("This is just type information for the Nix test driver")

     

       16
       +
       

     

       17
       +
       

     

       18
       +
       class PollingConditionProtocol(Protocol):

     

       19
       +
           def __call__(

     

       20
       +
               self,

     

       21
       +
               fun_: Optional[Callable] = None,

     

       22
       +
               *,

     

       23
       +
               seconds_interval: float = 2.0,

     

       24
       +
               description: Optional[str] = None,

     

       25
       +
           ) -> Union[Callable[[Callable], ContextManager], ContextManager]:

     

       26
       +
               raise Exception("This is just type information for the Nix test driver")

     

       27
       +
       

     

       28
       +
       

     

       29
       +
       start_all: Callable[[], None]

     

       30
       +
       subtest: Callable[[str], ContextManager[None]]

     

       31
       +
       retry: RetryProtocol

     

       32
       +
       test_script: Callable[[], None]

     

       33
       +
       machines: List[Machine]

     

       34
       +
       vlans: List[VLan]

     

       35
       +
       driver: Driver

     

       36
       +
       log: Logger

     

       37
       +
       create_machine: Callable[[Dict[str, Any]], Machine]

     

       38
       +
       run_tests: Callable[[], None]

     

       39
       +
       join_all: Callable[[], None]

     

       40
       +
       serial_stdout_off: Callable[[], None]

     

       41
       +
       serial_stdout_on: Callable[[], None]

     

       42
       +
       polling_condition: PollingConditionProtocol

+22 -3

nixos/lib/testing-python.nix

···

       85
        
       

     

       86
        
             nodeHostNames = let

     

       87
        
               nodesList = map (c: c.config.system.name) (lib.attrValues nodes);

     

       88
       -
             in nodesList ++ lib.optional (lib.length nodesList == 1) "machine";

     

       89
        
       

     

       90
        
             # TODO: This is an implementation error and needs fixing

     

       91
        
             # the testing famework cannot legitimately restrict hostnames further

     
···

       100
        
               then testScript { inherit nodes; }

     

       101
        
               else testScript;

     

       102
        
       

     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       103
        
           in

     

       104
        
           if lib.length invalidNodeNames > 0 then

     

       105
        
             throw ''

     
···

       113
        
           else lib.warnIf skipLint "Linting is disabled" (runCommand testDriverName

     

       114
        
             {

     

       115
        
               inherit testName;

     

       116
       -
               nativeBuildInputs = [ makeWrapper ];

     

       117
        
               testScript = testScript';

     

       118
        
               preferLocalBuild = true;

     

       119
        
               passthru = passthru // {

     
···

       125
        
               mkdir -p $out/bin

     

       126
        
       

     

       127
        
               vmStartScripts=($(for i in ${toString vms}; do echo $i/bin/run-*-vm; done))

     

       128
       -
               echo -n "$testScript" > $out/test-script

     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       129
        
               ln -s ${testDriver}/bin/nixos-test-driver $out/bin/nixos-test-driver

     

       130
        
       

     

       131
        
               ${testDriver}/bin/generate-driver-symbols

···

       85
        
       

     

       86
        
             nodeHostNames = let

     

       87
        
               nodesList = map (c: c.config.system.name) (lib.attrValues nodes);

     

       88
       +
             in nodesList ++ lib.optional (lib.length nodesList == 1 && !lib.elem "machine" nodesList) "machine";

     

       89
        
       

     

       90
        
             # TODO: This is an implementation error and needs fixing

     

       91
        
             # the testing famework cannot legitimately restrict hostnames further

     
···

       100
        
               then testScript { inherit nodes; }

     

       101
        
               else testScript;

     

       102
        
       

     

       103
       +
             uniqueVlans = lib.unique (builtins.concatLists vlans);

     

       104
       +
             vlanNames = map (i: "vlan${toString i}: VLan;") uniqueVlans;

     

       105
       +
             machineNames = map (name: "${name}: Machine;") nodeHostNames;

     

       106
        
           in

     

       107
        
           if lib.length invalidNodeNames > 0 then

     

       108
        
             throw ''

     
···

       116
        
           else lib.warnIf skipLint "Linting is disabled" (runCommand testDriverName

     

       117
        
             {

     

       118
        
               inherit testName;

     

       119
       +
               nativeBuildInputs = [ makeWrapper mypy ];

     

       120
        
               testScript = testScript';

     

       121
        
               preferLocalBuild = true;

     

       122
        
               passthru = passthru // {

     
···

       128
        
               mkdir -p $out/bin

     

       129
        
       

     

       130
        
               vmStartScripts=($(for i in ${toString vms}; do echo $i/bin/run-*-vm; done))

     

       131
       +
       

     

       132
       +
               # prepend type hints so the test script can be type checked with mypy

     

       133
       +
               cat "${./test-script-prepend.py}" >> testScriptWithTypes

     

       134
       +
               echo "${builtins.toString machineNames}" >> testScriptWithTypes

     

       135
       +
               echo "${builtins.toString vlanNames}" >> testScriptWithTypes

     

       136
       +
               echo -n "$testScript" >> testScriptWithTypes

     

       137
       +
       

     

       138
       +
               # set pythonpath so mypy knows where to find the imports. this requires the py.typed file.

     

       139
       +
               export PYTHONPATH='${./test-driver}'

     

       140
       +
               mypy  --no-implicit-optional \

     

       141
       +
                     --pretty \

     

       142
       +
                     --no-color-output \

     

       143
       +
                     testScriptWithTypes

     

       144
       +
               unset PYTHONPATH

     

       145
       +
       

     

       146
       +
               echo -n "$testScript" >> $out/test-script

     

       147
       +
       

     

       148
        
               ln -s ${testDriver}/bin/nixos-test-driver $out/bin/nixos-test-driver

     

       149
        
       

     

       150
        
               ${testDriver}/bin/generate-driver-symbols