From 210accdb3919c90e4add3a43313f0d0bca86ec34 Mon Sep 17 00:00:00 2001
From: willcl-ark <will@256k1.dev>
Date: Thu, 12 Sep 2024 20:49:18 +0100
Subject: [PATCH] test for binary file

---
 .github/workflows/test.yml                |  1 +
 test/binary_test.py                       | 68 +++++++++++++++++++++++
 test/data/small_2_node/network.yaml       |  5 ++
 test/data/small_2_node/node-defaults.yaml |  4 ++
 4 files changed, 78 insertions(+)
 create mode 100755 test/binary_test.py
 create mode 100644 test/data/small_2_node/network.yaml
 create mode 100644 test/data/small_2_node/node-defaults.yaml

diff --git a/.github/workflows/test.yml b/.github/workflows/test.yml
index c717bddd4..e51dee5e8 100644
--- a/.github/workflows/test.yml
+++ b/.github/workflows/test.yml
@@ -69,6 +69,7 @@ jobs:
           - services_test.py
           - signet_test.py
           - scenarios_test.py
+          - binary_test.py
     steps:
       - uses: actions/checkout@v4
       - uses: azure/setup-helm@v4.2.0
diff --git a/test/binary_test.py b/test/binary_test.py
new file mode 100755
index 000000000..9e84c0b21
--- /dev/null
+++ b/test/binary_test.py
@@ -0,0 +1,68 @@
+#!/usr/bin/env python3
+
+import json
+import os
+import tempfile
+import time
+from pathlib import Path
+
+from test_base import TestBase
+
+from warnet.k8s import get_default_namespace
+from warnet.process import run_command
+
+
+class BinaryTest(TestBase):
+    TEST_STRING = "Hello, World!"
+
+    def __init__(self):
+        super().__init__()
+        self.network_dir = Path(os.path.dirname(__file__)) / "data" / "small_2_node"
+
+    def run_test(self):
+        try:
+            self.setup_network()
+            self.test_generic_binary()
+        finally:
+            self.cleanup()
+
+    def setup_network(self):
+        self.log.info("Setting up network")
+        self.log.info(self.warnet(f"deploy {self.network_dir}"))
+        self.wait_for_all_tanks_status(target="running")
+        self.wait_for_all_edges()
+
+    def test_generic_binary(self):
+        self.log.info("Launching binary")
+        temp_file_content = f"""#!/usr/bin/env sh
+echo {self.TEST_STRING}"""
+
+        with tempfile.NamedTemporaryFile(mode="w", delete=False, suffix=".sh") as temp_file:
+            temp_file.write(temp_file_content)
+            temp_file_path = temp_file.name
+
+        os.chmod(temp_file_path, 0o755)
+        self.warnet(f"run-binary {temp_file_path}")
+
+        # Get the commander pod name
+        pods = run_command(f"kubectl get pods -n {get_default_namespace()} -o json")
+        pods = json.loads(pods)
+        pod_list = [item["metadata"]["name"] for item in pods["items"]]
+        binary_pod = next((pod for pod in pod_list if pod.startswith("binary")), None)
+        if binary_pod is None:
+            raise ValueError("No pod found starting with 'binary'")
+        self.log.info(f"Got pod: {binary_pod}")
+
+        def g_log():
+            logs = self.warnet(f"logs {binary_pod}")
+            return self.TEST_STRING in logs
+
+        time.sleep(5)
+        self.wait_for_predicate(g_log, timeout=60, interval=5)
+
+        os.unlink(temp_file_path)
+
+
+if __name__ == "__main__":
+    test = BinaryTest()
+    test.run_test()
diff --git a/test/data/small_2_node/network.yaml b/test/data/small_2_node/network.yaml
new file mode 100644
index 000000000..3d51aef3f
--- /dev/null
+++ b/test/data/small_2_node/network.yaml
@@ -0,0 +1,5 @@
+nodes:
+  - name: tank-0000
+    connect:
+      - tank-0001
+  - name: tank-0001
diff --git a/test/data/small_2_node/node-defaults.yaml b/test/data/small_2_node/node-defaults.yaml
new file mode 100644
index 000000000..7e021cad1
--- /dev/null
+++ b/test/data/small_2_node/node-defaults.yaml
@@ -0,0 +1,4 @@
+image:
+  repository: bitcoindevproject/bitcoin
+  pullPolicy: IfNotPresent
+  tag: "27.0"