Merge pull request #31 from vyos-contrib/release/v0.4.0-cleanupv0.4.0

Release v0.4.0 — cleanup and consolidation

1 files changed, 105 insertions, 0 deletions

diff --git a/examples/integration_smoke.py b/examples/integration_smoke.py
new file mode 100644
index 0000000..137ca9b
--- /dev/null
+++ b/examples/integration_smoke.py
@@ -0,0 +1,105 @@
+"""Integration smoke test for pyvyos.
+
+This script changes configuration on the target VyOS device and should only
+be used against a disposable lab device, such as the one provided by
+``examples/vagrant/``.
+
+It may create and delete dummy interfaces, generate temporary files on the
+device, and save/load configuration files.
+
+Run only against a lab device::
+
+    PYVYOS_ALLOW_MUTATING_EXAMPLE=1 \\
+    VYDEVICE_HOSTNAME=127.0.0.1 \\
+    VYDEVICE_APIKEY=secret \\
+    python examples/integration_smoke.py
+"""
+
+import os
+import pprint
+import random
+import string
+import sys
+
+from dotenv import load_dotenv
+
+from pyvyos import VyDevice
+
+load_dotenv()
+
+
+def env_bool(name: str, default: str = "true") -> bool:
+    return os.environ.get(name, default).lower() in ("1", "true", "yes")
+
+
+def make_device() -> VyDevice:
+    return VyDevice(
+        hostname=os.environ["VYDEVICE_HOSTNAME"],
+        apikey=os.environ["VYDEVICE_APIKEY"],
+        port=int(os.environ.get("VYDEVICE_PORT", "443")),
+        protocol=os.environ.get("VYDEVICE_PROTOCOL", "https"),
+        verify=env_bool("VYDEVICE_VERIFY_SSL"),
+        timeout=int(os.environ.get("VYDEVICE_TIMEOUT", "60")),
+    )
+
+
+def main() -> None:
+    if os.environ.get("PYVYOS_ALLOW_MUTATING_EXAMPLE") != "1":
+        sys.exit(
+            "This example mutates the target device. "
+            "Set PYVYOS_ALLOW_MUTATING_EXAMPLE=1 to run it."
+        )
+
+    device = make_device()
+
+    # Retrieve the running configuration for the system tree.
+    pprint.pprint(device.retrieve_show_config(path=["system"]))
+
+    # Configure a dummy interface, read it back, then delete it.
+    pprint.pprint(
+        device.configure_set(
+            path=["interfaces", "dummy", "dum1", "address", "192.168.56.100/24"]
+        )
+    )
+    pprint.pprint(
+        device.retrieve_return_values(
+            path=["interfaces", "dummy", "dum1", "address"]
+        )
+    )
+    pprint.pprint(device.configure_delete(path=["interfaces", "dummy", "dum1"]))
+
+    # Generate a one-shot SSH client key.
+    randstring = "".join(
+        random.choice(string.ascii_letters + string.digits) for _ in range(20)
+    )
+    pprint.pprint(
+        device.generate(path=["ssh", "client-key", f"/tmp/key_{randstring}"])
+    )
+
+    # Operational commands.
+    pprint.pprint(device.show(path=["system", "image"]))
+    pprint.pprint(device.reset(path=["conntrack-sync", "internal-cache"]))
+
+    # Save and reload the running configuration to/from a file on the device.
+    pprint.pprint(device.config_file_save(file="/config/test300.config"))
+    pprint.pprint(device.config_file_load(file="/config/test300.config"))
+
+    # Batch multiple configuration operations in a single request.
+    pprint.pprint(
+        device.configure_multiple_op(
+            op_path=[
+                {
+                    "op": "set",
+                    "path": [
+                        "interfaces", "dummy", "dum1", "address",
+                        "192.168.56.100/24",
+                    ],
+                },
+                {"op": "delete", "path": ["interfaces", "dummy", "dum1"]},
+            ]
+        )
+    )
+
+
+if __name__ == "__main__":
+    main()