diff options
-rw-r--r-- | nixos/lib/test-driver/test_driver/driver.py | 4 | ||||
-rw-r--r-- | nixos/lib/test-driver/test_driver/logger.py | 3 |
2 files changed, 5 insertions, 2 deletions
diff --git a/nixos/lib/test-driver/test_driver/driver.py b/nixos/lib/test-driver/test_driver/driver.py index 7dfbfdcf1c2aa..0650acb4d8975 100644 --- a/nixos/lib/test-driver/test_driver/driver.py +++ b/nixos/lib/test-driver/test_driver/driver.py @@ -249,10 +249,10 @@ class Driver: ) def serial_stdout_on(self) -> None: - rootlog._print_serial_logs = True + rootlog.print_serial_logs(True) def serial_stdout_off(self) -> None: - rootlog._print_serial_logs = False + rootlog.print_serial_logs(False) def check_polling_conditions(self) -> None: for condition in self.polling_conditions: diff --git a/nixos/lib/test-driver/test_driver/logger.py b/nixos/lib/test-driver/test_driver/logger.py index 1b802f992f640..77a9b81c5946c 100644 --- a/nixos/lib/test-driver/test_driver/logger.py +++ b/nixos/lib/test-driver/test_driver/logger.py @@ -24,6 +24,9 @@ class Logger: self._print_serial_logs = True + def print_serial_logs(self, enable: bool) -> None: + self._print_serial_logs = enable + @staticmethod def _eprint(*args: object, **kwargs: Any) -> None: print(*args, file=sys.stderr, **kwargs) |