nixos/tests/installer: port to python

NixOS · Nov 11, 2019 · 496db01 · 496db01
1 parent 556a169
commit 496db01
Show file tree

Hide file tree

Showing 2 changed files with 271 additions and 246 deletions.
diff --git a/nixos/lib/test-driver/test-driver.py b/nixos/lib/test-driver/test-driver.py
@@ -4,6 +4,7 @@
 from xml.sax.saxutils import XMLGenerator
 import _thread
 import atexit
+import base64
 import json
 import os
 import ptpython.repl
@@ -710,6 +711,13 @@ def unblock(self):
         """
         self.send_monitor_command("set_link virtio-net-pci.1 on")
 
+    def copy_file_from_host(self, from_path, to_path):
+        """Copy a file to the machine.
+        """
+        with open(from_path, "rb") as f:
+            content = base64.b64encode(f.read())
+            self.succeed("echo '{}' | base64 -d > {}".format(content.decode(), to_path))
+
 
 def create_machine(args):
     global log