diff options
author | Jacek Galowicz <jacek@galowicz.de> | 2019-11-05 14:35:10 +0100 |
---|---|---|
committer | Jacek Galowicz <jacek@galowicz.de> | 2019-11-05 18:07:00 +0100 |
commit | 9c54658226e99b4edfbb0d570d547a147d7d9eb4 (patch) | |
tree | 5550dde527d23fae2508dd8da580030f164d6a64 /nixos/lib/test-driver | |
parent | fa2909bf51c29d568dd9ce5c27cdbf7393644909 (diff) | |
download | nixlib-9c54658226e99b4edfbb0d570d547a147d7d9eb4.tar nixlib-9c54658226e99b4edfbb0d570d547a147d7d9eb4.tar.gz nixlib-9c54658226e99b4edfbb0d570d547a147d7d9eb4.tar.bz2 nixlib-9c54658226e99b4edfbb0d570d547a147d7d9eb4.tar.lz nixlib-9c54658226e99b4edfbb0d570d547a147d7d9eb4.tar.xz nixlib-9c54658226e99b4edfbb0d570d547a147d7d9eb4.tar.zst nixlib-9c54658226e99b4edfbb0d570d547a147d7d9eb4.zip |
nixos/tests: Implement python test-driver version of requireActiveUnit
Diffstat (limited to 'nixos/lib/test-driver')
-rw-r--r-- | nixos/lib/test-driver/test-driver.py | 15 |
1 files changed, 14 insertions, 1 deletions
diff --git a/nixos/lib/test-driver/test-driver.py b/nixos/lib/test-driver/test-driver.py index 45b7e229a5c6..2c137ab38372 100644 --- a/nixos/lib/test-driver/test-driver.py +++ b/nixos/lib/test-driver/test-driver.py @@ -4,7 +4,9 @@ from contextlib import contextmanager from xml.sax.saxutils import XMLGenerator import _thread import atexit +import json import os +import ptpython.repl import pty import queue import re @@ -15,7 +17,6 @@ import sys import tempfile import time import unicodedata -import ptpython.repl CHAR_TO_KEY = { "A": "shift-a", @@ -344,6 +345,18 @@ class Machine: ) return self.execute("systemctl {}".format(q)) + def require_unit_state(self, unit, require_state="active"): + with self.nested( + "checking if unit ‘{}’ has reached state '{}'".format(unit, require_state) + ): + info = self.get_unit_info(unit) + state = info["ActiveState"] + if state != require_state: + raise Exception( + "Expected unit ‘{}’ to to be in state ".format(unit) + + "'active' but it is in state ‘{}’".format(state) + ) + def execute(self, command): self.connect() |