<section xmlns="http://docbook.org/ns/docbook"  xmlns:xlink="http://www.w3.org/1999/xlink" xmlns:xi="http://www.w3.org/2001/XInclude" xml:id="sec-writing-nixos-tests">
  <title>Writing Tests</title>
  <para>
    A NixOS test is a module that has the following structure:
  </para>
  <programlisting language="bash">
{

  # One or more machines:
  nodes =
    { machine =
        { config, pkgs, ... }: { … };
      machine2 =
        { config, pkgs, ... }: { … };
      …
    };

  testScript =
    ''
      Python code…
    '';
}
</programlisting>
  <para>
    We refer to the whole test above as a test module, whereas the
    values in
    <link linkend="test-opt-nodes"><literal>nodes.&lt;name&gt;</literal></link>
    are NixOS modules themselves.
  </para>
  <para>
    The option
    <link linkend="test-opt-testScript"><literal>testScript</literal></link>
    is a piece of Python code that executes the test (described below).
    During the test, it will start one or more virtual machines, the
    configuration of which is described by the option
    <link linkend="test-opt-nodes"><literal>nodes</literal></link>.
  </para>
  <para>
    An example of a single-node test is
    <link xlink:href="https://github.com/NixOS/nixpkgs/blob/master/nixos/tests/login.nix"><literal>login.nix</literal></link>.
    It only needs a single machine to test whether users can log in on
    the virtual console, whether device ownership is correctly
    maintained when switching between consoles, and so on. An
    interesting multi-node test is
    <link xlink:href="https://github.com/NixOS/nixpkgs/blob/master/nixos/tests/nfs/simple.nix"><literal>nfs/simple.nix</literal></link>.
    It uses two client nodes to test correct locking across server
    crashes.
  </para>
  <section xml:id="sec-calling-nixos-tests">
    <title>Calling a test</title>
    <para>
      Tests are invoked differently depending on whether the test is
      part of NixOS or lives in a different project.
    </para>
    <section xml:id="sec-call-nixos-test-in-nixos">
      <title>Testing within NixOS</title>
      <para>
        Tests that are part of NixOS are added to
        <link xlink:href="https://github.com/NixOS/nixpkgs/blob/master/nixos/tests/all-tests.nix"><literal>nixos/tests/all-tests.nix</literal></link>.
      </para>
      <programlisting language="bash">
  hostname = runTest ./hostname.nix;
</programlisting>
      <para>
        Overrides can be added by defining an anonymous module in
        <literal>all-tests.nix</literal>.
      </para>
      <programlisting language="bash">
  hostname = runTest {
    imports = [ ./hostname.nix ];
    defaults.networking.firewall.enable = false;
  };
</programlisting>
      <para>
        You can run a test with attribute name
        <literal>hostname</literal> in
        <literal>nixos/tests/all-tests.nix</literal> by invoking:
      </para>
      <programlisting>
cd /my/git/clone/of/nixpkgs
nix-build -A nixosTests.hostname
</programlisting>
    </section>
    <section xml:id="sec-call-nixos-test-outside-nixos">
      <title>Testing outside the NixOS project</title>
      <para>
        Outside the <literal>nixpkgs</literal> repository, you can
        instantiate the test by first importing the NixOS library,
      </para>
      <programlisting language="bash">
let nixos-lib = import (nixpkgs + &quot;/nixos/lib&quot;) { };
in

nixos-lib.runTest {
  imports = [ ./test.nix ];
  hostPkgs = pkgs;  # the Nixpkgs package set used outside the VMs
  defaults.services.foo.package = mypkg;
}
</programlisting>
      <para>
        <literal>runTest</literal> returns a derivation that runs the
        test.
      </para>
    </section>
  </section>
  <section xml:id="sec-nixos-test-nodes">
    <title>Configuring the nodes</title>
    <para>
      There are a few special NixOS options for test VMs:
    </para>
    <variablelist>
      <varlistentry>
        <term>
          <literal>virtualisation.memorySize</literal>
        </term>
        <listitem>
          <para>
            The memory of the VM in megabytes.
          </para>
        </listitem>
      </varlistentry>
      <varlistentry>
        <term>
          <literal>virtualisation.vlans</literal>
        </term>
        <listitem>
          <para>
            The virtual networks to which the VM is connected. See
            <link xlink:href="https://github.com/NixOS/nixpkgs/blob/master/nixos/tests/nat.nix"><literal>nat.nix</literal></link>
            for an example.
          </para>
        </listitem>
      </varlistentry>
      <varlistentry>
        <term>
          <literal>virtualisation.writableStore</literal>
        </term>
        <listitem>
          <para>
            By default, the Nix store in the VM is not writable. If you
            enable this option, a writable union file system is mounted
            on top of the Nix store to make it appear writable. This is
            necessary for tests that run Nix operations that modify the
            store.
          </para>
        </listitem>
      </varlistentry>
    </variablelist>
    <para>
      For more options, see the module
      <link xlink:href="https://github.com/NixOS/nixpkgs/blob/master/nixos/modules/virtualisation/qemu-vm.nix"><literal>qemu-vm.nix</literal></link>.
    </para>
    <para>
      The test script is a sequence of Python statements that perform
      various actions, such as starting VMs, executing commands in the
      VMs, and so on. Each virtual machine is represented as an object
      stored in the variable <literal>name</literal> if this is also the
      identifier of the machine in the declarative config. If you
      specified a node <literal>nodes.machine</literal>, the following
      example starts the machine, waits until it has finished booting,
      then executes a command and checks that the output is more-or-less
      correct:
    </para>
    <programlisting language="python">
machine.start()
machine.wait_for_unit(&quot;default.target&quot;)
if not &quot;Linux&quot; in machine.succeed(&quot;uname&quot;):
  raise Exception(&quot;Wrong OS&quot;)
</programlisting>
    <para>
      The first line is technically unnecessary; machines are implicitly
      started when you first execute an action on them (such as
      <literal>wait_for_unit</literal> or <literal>succeed</literal>).
      If you have multiple machines, you can speed up the test by
      starting them in parallel:
    </para>
    <programlisting language="python">
start_all()
</programlisting>
  </section>
  <section xml:id="ssec-machine-objects">
    <title>Machine objects</title>
    <para>
      The following methods are available on machine objects:
    </para>
    <variablelist>
      <varlistentry>
        <term>
          <literal>start</literal>
        </term>
        <listitem>
          <para>
            Start the virtual machine. This method is asynchronous — it
            does not wait for the machine to finish booting.
          </para>
        </listitem>
      </varlistentry>
      <varlistentry>
        <term>
          <literal>shutdown</literal>
        </term>
        <listitem>
          <para>
            Shut down the machine, waiting for the VM to exit.
          </para>
        </listitem>
      </varlistentry>
      <varlistentry>
        <term>
          <literal>crash</literal>
        </term>
        <listitem>
          <para>
            Simulate a sudden power failure, by telling the VM to exit
            immediately.
          </para>
        </listitem>
      </varlistentry>
      <varlistentry>
        <term>
          <literal>block</literal>
        </term>
        <listitem>
          <para>
            Simulate unplugging the Ethernet cable that connects the
            machine to the other machines.
          </para>
        </listitem>
      </varlistentry>
      <varlistentry>
        <term>
          <literal>unblock</literal>
        </term>
        <listitem>
          <para>
            Undo the effect of <literal>block</literal>.
          </para>
        </listitem>
      </varlistentry>
      <varlistentry>
        <term>
          <literal>screenshot</literal>
        </term>
        <listitem>
          <para>
            Take a picture of the display of the virtual machine, in PNG
            format. The screenshot is linked from the HTML log.
          </para>
        </listitem>
      </varlistentry>
      <varlistentry>
        <term>
          <literal>get_screen_text_variants</literal>
        </term>
        <listitem>
          <para>
            Return a list of different interpretations of what is
            currently visible on the machine's screen using optical
            character recognition. The number and order of the
            interpretations is not specified and is subject to change,
            but if no exception is raised at least one will be returned.
          </para>
          <note>
            <para>
              This requires
              <link linkend="test-opt-enableOCR"><literal>enableOCR</literal></link>
              to be set to <literal>true</literal>.
            </para>
          </note>
        </listitem>
      </varlistentry>
      <varlistentry>
        <term>
          <literal>get_screen_text</literal>
        </term>
        <listitem>
          <para>
            Return a textual representation of what is currently visible
            on the machine's screen using optical character recognition.
          </para>
          <note>
            <para>
              This requires
              <link linkend="test-opt-enableOCR"><literal>enableOCR</literal></link>
              to be set to <literal>true</literal>.
            </para>
          </note>
        </listitem>
      </varlistentry>
      <varlistentry>
        <term>
          <literal>send_monitor_command</literal>
        </term>
        <listitem>
          <para>
            Send a command to the QEMU monitor. This is rarely used, but
            allows doing stuff such as attaching virtual USB disks to a
            running machine.
          </para>
        </listitem>
      </varlistentry>
      <varlistentry>
        <term>
          <literal>send_key</literal>
        </term>
        <listitem>
          <para>
            Simulate pressing keys on the virtual keyboard, e.g.,
            <literal>send_key(&quot;ctrl-alt-delete&quot;)</literal>.
          </para>
        </listitem>
      </varlistentry>
      <varlistentry>
        <term>
          <literal>send_chars</literal>
        </term>
        <listitem>
          <para>
            Simulate typing a sequence of characters on the virtual
            keyboard, e.g.,
            <literal>send_chars(&quot;foobar\n&quot;)</literal> will
            type the string <literal>foobar</literal> followed by the
            Enter key.
          </para>
        </listitem>
      </varlistentry>
      <varlistentry>
        <term>
          <literal>send_console</literal>
        </term>
        <listitem>
          <para>
            Send keys to the kernel console. This allows interaction
            with the systemd emergency mode, for example. Takes a string
            that is sent, e.g.,
            <literal>send_console(&quot;\n\nsystemctl default\n&quot;)</literal>.
          </para>
        </listitem>
      </varlistentry>
      <varlistentry>
        <term>
          <literal>execute</literal>
        </term>
        <listitem>
          <para>
            Execute a shell command, returning a list
            <literal>(status, stdout)</literal>.
          </para>
          <para>
            Commands are run with <literal>set -euo pipefail</literal>
            set:
          </para>
          <itemizedlist>
            <listitem>
              <para>
                If several commands are separated by
                <literal>;</literal> and one fails, the command as a
                whole will fail.
              </para>
            </listitem>
            <listitem>
              <para>
                For pipelines, the last non-zero exit status will be
                returned (if there is one; otherwise zero will be
                returned).
              </para>
            </listitem>
            <listitem>
              <para>
                Dereferencing unset variables fails the command.
              </para>
            </listitem>
            <listitem>
              <para>
                It will wait for stdout to be closed.
              </para>
            </listitem>
          </itemizedlist>
          <para>
            If the command detaches, it must close stdout, as
            <literal>execute</literal> will wait for this to consume all
            output reliably. This can be achieved by redirecting stdout
            to stderr <literal>&gt;&amp;2</literal>, to
            <literal>/dev/console</literal>,
            <literal>/dev/null</literal> or a file. Examples of
            detaching commands are <literal>sleep 365d &amp;</literal>,
            where the shell forks a new process that can write to stdout
            and <literal>xclip -i</literal>, where the
            <literal>xclip</literal> command itself forks without
            closing stdout.
          </para>
          <para>
            Takes an optional parameter <literal>check_return</literal>
            that defaults to <literal>True</literal>. Setting this
            parameter to <literal>False</literal> will not check for the
            return code and return -1 instead. This can be used for
            commands that shut down the VM and would therefore break the
            pipe that would be used for retrieving the return code.
          </para>
          <para>
            A timeout for the command can be specified (in seconds)
            using the optional <literal>timeout</literal> parameter,
            e.g., <literal>execute(cmd, timeout=10)</literal> or
            <literal>execute(cmd, timeout=None)</literal>. The default
            is 900 seconds.
          </para>
        </listitem>
      </varlistentry>
      <varlistentry>
        <term>
          <literal>succeed</literal>
        </term>
        <listitem>
          <para>
            Execute a shell command, raising an exception if the exit
            status is not zero, otherwise returning the standard output.
            Similar to <literal>execute</literal>, except that the
            timeout is <literal>None</literal> by default. See
            <literal>execute</literal> for details on command execution.
          </para>
        </listitem>
      </varlistentry>
      <varlistentry>
        <term>
          <literal>fail</literal>
        </term>
        <listitem>
          <para>
            Like <literal>succeed</literal>, but raising an exception if
            the command returns a zero status.
          </para>
        </listitem>
      </varlistentry>
      <varlistentry>
        <term>
          <literal>wait_until_succeeds</literal>
        </term>
        <listitem>
          <para>
            Repeat a shell command with 1-second intervals until it
            succeeds. Has a default timeout of 900 seconds which can be
            modified, e.g.
            <literal>wait_until_succeeds(cmd, timeout=10)</literal>. See
            <literal>execute</literal> for details on command execution.
          </para>
        </listitem>
      </varlistentry>
      <varlistentry>
        <term>
          <literal>wait_until_fails</literal>
        </term>
        <listitem>
          <para>
            Like <literal>wait_until_succeeds</literal>, but repeating
            the command until it fails.
          </para>
        </listitem>
      </varlistentry>
      <varlistentry>
        <term>
          <literal>wait_for_unit</literal>
        </term>
        <listitem>
          <para>
            Wait until the specified systemd unit has reached the
            <quote>active</quote> state.
          </para>
        </listitem>
      </varlistentry>
      <varlistentry>
        <term>
          <literal>wait_for_file</literal>
        </term>
        <listitem>
          <para>
            Wait until the specified file exists.
          </para>
        </listitem>
      </varlistentry>
      <varlistentry>
        <term>
          <literal>wait_for_open_port</literal>
        </term>
        <listitem>
          <para>
            Wait until a process is listening on the given TCP port (on
            <literal>localhost</literal>, at least).
          </para>
        </listitem>
      </varlistentry>
      <varlistentry>
        <term>
          <literal>wait_for_closed_port</literal>
        </term>
        <listitem>
          <para>
            Wait until nobody is listening on the given TCP port.
          </para>
        </listitem>
      </varlistentry>
      <varlistentry>
        <term>
          <literal>wait_for_x</literal>
        </term>
        <listitem>
          <para>
            Wait until the X11 server is accepting connections.
          </para>
        </listitem>
      </varlistentry>
      <varlistentry>
        <term>
          <literal>wait_for_text</literal>
        </term>
        <listitem>
          <para>
            Wait until the supplied regular expressions matches the
            textual contents of the screen by using optical character
            recognition (see <literal>get_screen_text</literal> and
            <literal>get_screen_text_variants</literal>).
          </para>
          <note>
            <para>
              This requires
              <link linkend="test-opt-enableOCR"><literal>enableOCR</literal></link>
              to be set to <literal>true</literal>.
            </para>
          </note>
        </listitem>
      </varlistentry>
      <varlistentry>
        <term>
          <literal>wait_for_console_text</literal>
        </term>
        <listitem>
          <para>
            Wait until the supplied regular expressions match a line of
            the serial console output. This method is useful when OCR is
            not possibile or accurate enough.
          </para>
        </listitem>
      </varlistentry>
      <varlistentry>
        <term>
          <literal>wait_for_window</literal>
        </term>
        <listitem>
          <para>
            Wait until an X11 window has appeared whose name matches the
            given regular expression, e.g.,
            <literal>wait_for_window(&quot;Terminal&quot;)</literal>.
          </para>
        </listitem>
      </varlistentry>
      <varlistentry>
        <term>
          <literal>copy_from_host</literal>
        </term>
        <listitem>
          <para>
            Copies a file from host to machine, e.g.,
            <literal>copy_from_host(&quot;myfile&quot;, &quot;/etc/my/important/file&quot;)</literal>.
          </para>
          <para>
            The first argument is the file on the host. The file needs
            to be accessible while building the nix derivation. The
            second argument is the location of the file on the machine.
          </para>
        </listitem>
      </varlistentry>
      <varlistentry>
        <term>
          <literal>systemctl</literal>
        </term>
        <listitem>
          <para>
            Runs <literal>systemctl</literal> commands with optional
            support for <literal>systemctl --user</literal>
          </para>
          <programlisting language="python">
machine.systemctl(&quot;list-jobs --no-pager&quot;) # runs `systemctl list-jobs --no-pager`
machine.systemctl(&quot;list-jobs --no-pager&quot;, &quot;any-user&quot;) # spawns a shell for `any-user` and runs `systemctl --user list-jobs --no-pager`
</programlisting>
        </listitem>
      </varlistentry>
      <varlistentry>
        <term>
          <literal>shell_interact</literal>
        </term>
        <listitem>
          <para>
            Allows you to directly interact with the guest shell. This
            should only be used during test development, not in
            production tests. Killing the interactive session with
            <literal>Ctrl-d</literal> or <literal>Ctrl-c</literal> also
            ends the guest session.
          </para>
        </listitem>
      </varlistentry>
      <varlistentry>
        <term>
          <literal>console_interact</literal>
        </term>
        <listitem>
          <para>
            Allows you to directly interact with QEMU’s stdin. This
            should only be used during test development, not in
            production tests. Output from QEMU is only read line-wise.
            <literal>Ctrl-c</literal> kills QEMU and
            <literal>Ctrl-d</literal> closes console and returns to the
            test runner.
          </para>
        </listitem>
      </varlistentry>
    </variablelist>
    <para>
      To test user units declared by
      <literal>systemd.user.services</literal> the optional
      <literal>user</literal> argument can be used:
    </para>
    <programlisting language="python">
machine.start()
machine.wait_for_x()
machine.wait_for_unit(&quot;xautolock.service&quot;, &quot;x-session-user&quot;)
</programlisting>
    <para>
      This applies to <literal>systemctl</literal>,
      <literal>get_unit_info</literal>,
      <literal>wait_for_unit</literal>, <literal>start_job</literal> and
      <literal>stop_job</literal>.
    </para>
    <para>
      For faster dev cycles it's also possible to disable the
      code-linters (this shouldn't be commited though):
    </para>
    <programlisting language="bash">
{
  skipLint = true;
  nodes.machine =
    { config, pkgs, ... }:
    { configuration…
    };

  testScript =
    ''
      Python code…
    '';
}
</programlisting>
    <para>
      This will produce a Nix warning at evaluation time. To fully
      disable the linter, wrap the test script in comment directives to
      disable the Black linter directly (again, don't commit this within
      the Nixpkgs repository):
    </para>
    <programlisting language="bash">
  testScript =
    ''
      # fmt: off
      Python code…
      # fmt: on
    '';
</programlisting>
    <para>
      Similarly, the type checking of test scripts can be disabled in
      the following way:
    </para>
    <programlisting language="bash">
{
  skipTypeCheck = true;
  nodes.machine =
    { config, pkgs, ... }:
    { configuration…
    };
}
</programlisting>
  </section>
  <section xml:id="ssec-failing-tests-early">
    <title>Failing tests early</title>
    <para>
      To fail tests early when certain invariants are no longer met
      (instead of waiting for the build to time out), the decorator
      <literal>polling_condition</literal> is provided. For example, if
      we are testing a program <literal>foo</literal> that should not
      quit after being started, we might write the following:
    </para>
    <programlisting language="python">
@polling_condition
def foo_running():
    machine.succeed(&quot;pgrep -x foo&quot;)


machine.succeed(&quot;foo --start&quot;)
machine.wait_until_succeeds(&quot;pgrep -x foo&quot;)

with foo_running:
    ...  # Put `foo` through its paces
</programlisting>
    <para>
      <literal>polling_condition</literal> takes the following
      (optional) arguments:
    </para>
    <para>
      <literal>seconds_interval</literal>
    </para>
    <para>
      : specifies how often the condition should be polled:
    </para>
    <programlisting language="python">
@polling_condition(seconds_interval=10)
def foo_running():
    machine.succeed(&quot;pgrep -x foo&quot;)
</programlisting>
    <para>
      <literal>description</literal>
    </para>
    <para>
      : is used in the log when the condition is checked. If this is not
      provided, the description is pulled from the docstring of the
      function. These two are therefore equivalent:
    </para>
    <programlisting language="python">
@polling_condition
def foo_running():
    &quot;check that foo is running&quot;
    machine.succeed(&quot;pgrep -x foo&quot;)
</programlisting>
    <programlisting language="python">
@polling_condition(description=&quot;check that foo is running&quot;)
def foo_running():
    machine.succeed(&quot;pgrep -x foo&quot;)
</programlisting>
  </section>
  <section xml:id="ssec-python-packages-in-test-script">
    <title>Adding Python packages to the test script</title>
    <para>
      When additional Python libraries are required in the test script,
      they can be added using the parameter
      <literal>extraPythonPackages</literal>. For example, you could add
      <literal>numpy</literal> like this:
    </para>
    <programlisting language="bash">
{
  extraPythonPackages = p: [ p.numpy ];

  nodes = { };

  # Type checking on extra packages doesn't work yet
  skipTypeCheck = true;

  testScript = ''
    import numpy as np
    assert str(np.zeros(4) == &quot;array([0., 0., 0., 0.])&quot;)
  '';
}
</programlisting>
    <para>
      In that case, <literal>numpy</literal> is chosen from the generic
      <literal>python3Packages</literal>.
    </para>
  </section>
  <section xml:id="sec-test-options-reference">
    <title>Test Options Reference</title>
    <para>
      The following options can be used when writing tests.
    </para>
    <xi:include href="../../generated/test-options-db.xml" xpointer="test-options-list"/>
  </section>
</section>
