commit 3780b9e69caae0c5e69836e1e823a111b52264d3 · pyrox.dev/nixpkgs

+35 -1

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

···

       4
       4
        
       from xml.sax.saxutils import XMLGenerator

     

       5
       5
        
       import _thread

     

       6
       6
        
       import atexit

     

       7
       7
       +
       import json

     

       7
       8
        
       import os

     

       9
       9
       +
       import ptpython.repl

     

       8
       10
        
       import pty

     

       9
       11
        
       import queue

     

       10
       12
        
       import re

     
···

       15
       17
        
       import tempfile

     

       16
       18
        
       import time

     

       17
       19
        
       import unicodedata

     

       18
       18
       -
       import ptpython.repl

     

       19
       20
        
       

     

       20
       21
        
       CHAR_TO_KEY = {

     

       21
       22
        
           "A": "shift-a",

     
···

       344
       345
        
                   )

     

       345
       346
        
               return self.execute("systemctl {}".format(q))

     

       346
       347
        
       

     

       348
       348
       +
           def require_unit_state(self, unit, require_state="active"):

     

       349
       349
       +
               with self.nested(

     

       350
       350
       +
                   "checking if unit ‘{}’ has reached state '{}'".format(unit, require_state)

     

       351
       351
       +
               ):

     

       352
       352
       +
                   info = self.get_unit_info(unit)

     

       353
       353
       +
                   state = info["ActiveState"]

     

       354
       354
       +
                   if state != require_state:

     

       355
       355
       +
                       raise Exception(

     

       356
       356
       +
                           "Expected unit ‘{}’ to to be in state ".format(unit)

     

       357
       357
       +
                           + "'active' but it is in state ‘{}’".format(state)

     

       358
       358
       +
                       )

     

       359
       359
       +
       

     

       347
       360
        
           def execute(self, command):

     

       348
       361
        
               self.connect()

     

       349
       362
        
       

     
···

       641
       654
        
                       status, _ = self.execute("[ -e /tmp/.X11-unix/X0 ]")

     

       642
       655
        
                       if status == 0:

     

       643
       656
        
                           return

     

       657
       657
       +
       

     

       658
       658
       +
           def get_window_names(self):

     

       659
       659
       +
               return self.succeed(

     

       660
       660
       +
                   r"xwininfo -root -tree | sed 's/.*0x[0-9a-f]* \"\([^\"]*\)\".*/\1/; t; d'"

     

       661
       661
       +
               ).splitlines()

     

       662
       662
       +
       

     

       663
       663
       +
           def wait_for_window(self, regexp):

     

       664
       664
       +
               pattern = re.compile(regexp)

     

       665
       665
       +
       

     

       666
       666
       +
               def window_is_visible(last_try):

     

       667
       667
       +
                   names = self.get_window_names()

     

       668
       668
       +
                   if last_try:

     

       669
       669
       +
                       self.log(

     

       670
       670
       +
                           "Last chance to match {} on the window list,".format(regexp)

     

       671
       671
       +
                           + " which currently contains: "

     

       672
       672
       +
                           + ", ".join(names)

     

       673
       673
       +
                       )

     

       674
       674
       +
                   return any(pattern.search(name) for name in names)

     

       675
       675
       +
       

     

       676
       676
       +
               with self.nested("Waiting for a window to appear"):

     

       677
       677
       +
                   retry(window_is_visible)

     

       644
       678
        
       

     

       645
       679
        
           def sleep(self, secs):

     

       646
       680
        
               time.sleep(secs)

+22 -14

nixos/tests/firefox.nix

···

       1
       1
       -
       import ./make-test.nix ({ pkgs, ... }: {

     

       1
       1
       +
       import ./make-test-python.nix ({ pkgs, ... }: {

     

       2
       2
        
         name = "firefox";

     

       3
       3
        
         meta = with pkgs.stdenv.lib.maintainers; {

     

       4
       4
        
           maintainers = [ eelco shlevy ];

     
···

       11
       11
        
             environment.systemPackages = [ pkgs.firefox pkgs.xdotool ];

     

       12
       12
        
           };

     

       13
       13
        
       

     

       14
       14
       -
         testScript =

     

       15
       15
       -
           ''

     

       16
       16
       -
             $machine->waitForX;

     

       17
       17
       -
             $machine->execute("xterm -e 'firefox file://${pkgs.valgrind.doc}/share/doc/valgrind/html/index.html' &");

     

       18
       18
       -
             $machine->waitForWindow(qr/Valgrind/);

     

       19
       19
       -
             $machine->sleep(40); # wait until Firefox has finished loading the page

     

       20
       20
       -
             $machine->execute("xdotool key space"); # do I want to make Firefox the

     

       21
       21
       -
                                    # default browser? I just want to close the dialog

     

       22
       22
       -
             $machine->sleep(2); # wait until Firefox hides the default browser window

     

       23
       23
       -
             $machine->execute("xdotool key F12");

     

       24
       24
       -
             $machine->sleep(10); # wait until Firefox draws the developer tool panel

     

       25
       25
       -
             $machine->succeed("xwininfo -root -tree | grep Valgrind");

     

       26
       26
       -
             $machine->screenshot("screen");

     

       14
       14
       +
         testScript = ''

     

       15
       15
       +
             machine.wait_for_x()

     

       16
       16
       +
       

     

       17
       17
       +
             with subtest("wait until Firefox has finished loading the Valgrind docs page"):

     

       18
       18
       +
                 machine.execute(

     

       19
       19
       +
                     "xterm -e 'firefox file://${pkgs.valgrind.doc}/share/doc/valgrind/html/index.html' &"

     

       20
       20
       +
                 )

     

       21
       21
       +
                 machine.wait_for_window("Valgrind")

     

       22
       22
       +
                 machine.sleep(40)

     

       23
       23
       +
       

     

       24
       24
       +
             with subtest("Close default browser prompt"):

     

       25
       25
       +
                 machine.execute("xdotool key space")

     

       26
       26
       +
       

     

       27
       27
       +
             with subtest("Hide default browser window"):

     

       28
       28
       +
                 machine.sleep(2)

     

       29
       29
       +
                 machine.execute("xdotool key F12")

     

       30
       30
       +
       

     

       31
       31
       +
             with subtest("wait until Firefox draws the developer tool panel"):

     

       32
       32
       +
                 machine.sleep(10)

     

       33
       33
       +
                 machine.succeed("xwininfo -root -tree | grep Valgrind")

     

       34
       34
       +
                 machine.screenshot("screen")

     

       27
       35
        
           '';

     

       28
       36
        
       

     

       29
       37
        
       })

+12 -8

nixos/tests/matrix-synapse.nix

···

       1
       1
       -
       import ./make-test.nix ({ pkgs, ... } : let

     

       1
       1
       +
       import ./make-test-python.nix ({ pkgs, ... } : let

     

       2
       2
        
       

     

       3
       3
        
       

     

       4
       4
        
         runWithOpenSSL = file: cmd: pkgs.runCommand file {

     
···

       55
       55
        
         };

     

       56
       56
        
       

     

       57
       57
        
         testScript = ''

     

       58
       58
       -
           startAll;

     

       59
       59
       -
           $serverpostgres->waitForUnit("matrix-synapse.service");

     

       60
       60
       -
           $serverpostgres->waitUntilSucceeds("curl -L --cacert ${ca_pem} https://localhost:8448/");

     

       61
       61
       -
           $serverpostgres->requireActiveUnit("postgresql.service");

     

       62
       62
       -
           $serversqlite->waitForUnit("matrix-synapse.service");

     

       63
       63
       -
           $serversqlite->waitUntilSucceeds("curl -L --cacert ${ca_pem} https://localhost:8448/");

     

       64
       64
       -
           $serversqlite->mustSucceed("[ -e /var/lib/matrix-synapse/homeserver.db ]");

     

       58
       58
       +
           start_all()

     

       59
       59
       +
           serverpostgres.wait_for_unit("matrix-synapse.service")

     

       60
       60
       +
           serverpostgres.wait_until_succeeds(

     

       61
       61
       +
               "curl -L --cacert ${ca_pem} https://localhost:8448/"

     

       62
       62
       +
           )

     

       63
       63
       +
           serverpostgres.require_unit_state("postgresql.service")

     

       64
       64
       +
           serversqlite.wait_for_unit("matrix-synapse.service")

     

       65
       65
       +
           serversqlite.wait_until_succeeds(

     

       66
       66
       +
               "curl -L --cacert ${ca_pem} https://localhost:8448/"

     

       67
       67
       +
           )

     

       68
       68
       +
           serversqlite.succeed("[ -e /var/lib/matrix-synapse/homeserver.db ]")

     

       65
       69
        
         '';

     

       66
       70
        
       

     

       67
       71
        
       })