From 9bfb18e66f70d02c764950cf17ccedadcd84fefa Mon Sep 17 00:00:00 2001
From: Monika Kairaityte <monika@kibit.lt>
Date: Fri, 27 Sep 2024 13:37:53 +0300
Subject: [PATCH] tests/integration: Automate manual 'nets_test2' test

Signed-off-by: Monika Kairaityte <monika@kibit.lt>
---
 .../test_podman_compose_nets_test2.py         | 95 +++++++++++++++++++
 1 file changed, 95 insertions(+)
 create mode 100644 tests/integration/test_podman_compose_nets_test2.py

diff --git a/tests/integration/test_podman_compose_nets_test2.py b/tests/integration/test_podman_compose_nets_test2.py
new file mode 100644
index 0000000..1f1ce96
--- /dev/null
+++ b/tests/integration/test_podman_compose_nets_test2.py
@@ -0,0 +1,95 @@
+# SPDX-License-Identifier: GPL-2.0
+
+import json
+import os
+import unittest
+
+import requests
+
+from tests.integration.test_podman_compose import podman_compose_path
+from tests.integration.test_podman_compose import test_path
+from tests.integration.test_utils import RunSubprocessMixin
+
+
+def compose_yaml_path():
+    return os.path.join(os.path.join(test_path(), "nets_test2"), "docker-compose.yml")
+
+
+class TestComposeNetsTest2(unittest.TestCase, RunSubprocessMixin):
+    # test if port mapping works as expected with networks top-level element
+    def test_nets_test2(self):
+        try:
+            self.run_subprocess_assert_returncode(
+                [
+                    podman_compose_path(),
+                    "-f",
+                    compose_yaml_path(),
+                    "up",
+                    "-d",
+                ],
+            )
+            output, _ = self.run_subprocess_assert_returncode([
+                podman_compose_path(),
+                "-f",
+                compose_yaml_path(),
+                "ps",
+            ])
+            self.assertIn(b"nets_test2_web1_1", output)
+            self.assertIn(b"nets_test2_web2_1", output)
+
+            response = requests.get('http://localhost:8001/index.txt')
+            self.assertEqual(response.ok, True)
+            self.assertEqual(response.text, "test1\n")
+
+            response = requests.get('http://localhost:8002/index.txt')
+            self.assertEqual(response.ok, True)
+            self.assertEqual(response.text, "test2\n")
+
+            # inspect 1st container
+            output, _ = self.run_subprocess_assert_returncode([
+                "podman",
+                "inspect",
+                "nets_test2_web1_1",
+            ])
+            container_info = json.loads(output.decode('utf-8'))[0]
+
+            # check if network got specific name from networks top-level element
+            self.assertEqual(
+                list(container_info["NetworkSettings"]["Networks"].keys())[0], "nets_test2_mystack"
+            )
+
+            # check if Host port is the same as prodvided by the service port
+            self.assertEqual(
+                container_info['NetworkSettings']["Ports"],
+                {"8001/tcp": [{"HostIp": "", "HostPort": "8001"}]},
+            )
+
+            self.assertEqual(container_info["Config"]["Hostname"], "web1")
+
+            # inspect 2nd container
+            output, _ = self.run_subprocess_assert_returncode([
+                "podman",
+                "inspect",
+                "nets_test2_web2_1",
+            ])
+            container_info = json.loads(output.decode('utf-8'))[0]
+
+            self.assertEqual(
+                list(container_info["NetworkSettings"]["Networks"].keys())[0], "nets_test2_mystack"
+            )
+
+            self.assertEqual(
+                container_info['NetworkSettings']["Ports"],
+                {"8001/tcp": [{"HostIp": "", "HostPort": "8002"}]},
+            )
+
+            self.assertEqual(container_info["Config"]["Hostname"], "web2")
+        finally:
+            self.run_subprocess_assert_returncode([
+                podman_compose_path(),
+                "-f",
+                compose_yaml_path(),
+                "down",
+                "-t",
+                "0",
+            ])