From dab0d83e159586bb376b9f6da4d67e7e4be1f689 Mon Sep 17 00:00:00 2001
From: Tobias Kremer <tobias.kremer@ecmwf.int>
Date: Tue, 11 Jun 2024 10:52:20 +0200
Subject: [PATCH] Added test case for pyflow file resource

---
 tests/test_resource.py | 44 ++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 44 insertions(+)
 create mode 100644 tests/test_resource.py

diff --git a/tests/test_resource.py b/tests/test_resource.py
new file mode 100644
index 0000000..d10fefc
--- /dev/null
+++ b/tests/test_resource.py
@@ -0,0 +1,44 @@
+from os import path
+from os.path import join
+
+from pyflow import FileResource, Notebook, Suite
+from pyflow.host import SSHHost
+
+
+def test_file_resource():
+
+    resouces_directory = "/resources_directory"
+
+    sshhost_1 = SSHHost("example_ssh_host_1", resources_directory=resouces_directory)
+    sshhost_2 = SSHHost("example_ssh_host_2", resources_directory=resouces_directory)
+    host_set = [sshhost_1, sshhost_2]
+
+    source_file = path.join(path.dirname(path.abspath(__file__)),  "file_resource.txt")
+    name = "file_resource"
+
+    with Suite("s", host=sshhost_1) as s:
+        s.resource_file = FileResource(name, hosts=host_set, source_file=source_file)
+
+    # Check that variables are set correctly
+    assert s.resource_file.host == sshhost_1
+    assert s.resource_file.location() == join(
+        str(sshhost_1.resources_directory), s.name, name
+    )
+    assert s.resource_file._hosts == host_set
+
+    # Check that the deployment scripts have been generated
+    s.check_definition()
+    s.generate_node()
+
+    s.deploy_suite(target=Notebook)
+
+    generate_file_resource_script_lines, _ = s.resource_file.generate_script()
+    assert any(sshhost_1.name in s for s in generate_file_resource_script_lines)
+    assert any(sshhost_2.name in s for s in generate_file_resource_script_lines)
+
+
+if __name__ == "__main__":
+
+    import pytest
+
+    pytest.main(path.abspath(__file__))