Merge pull request #425 from Limmen/integration_test

cli test: ls containers is added.

Author: Limmen

simulation-system/libs/csle-cli/tests/test_cli.py

@@ -1,13 +1,16 @@
+from typing import List
 from click.testing import CliRunner
 from csle_cli.cli import ls
 
+from csle_cluster.cluster_manager.cluster_manager_pb2 import DockerContainerDTO
+
 
 class TestCSLECliSuite:
     """
     Test suite for cli.py
     """
 
-    def test_ls(self) -> None:
+    def test_ls_all(self) -> None:
         """
         Tests the ls command
 
@@ -16,3 +19,31 @@ def test_ls(self) -> None:
         runner = CliRunner()
         result = runner.invoke(ls, ["--all"])
         assert result.exit_code == 0
+
+    def test_ls_running_containers(self) -> None:
+        """
+        Tests the ls command
+
+        :return: None
+        """
+        import csle_common.constants.constants as constants
+        from csle_common.metastore.metastore_facade import MetastoreFacade
+        from csle_cluster.cluster_manager.cluster_controller import ClusterController
+        config = MetastoreFacade.get_config(id=1)
+        running_containers: List[DockerContainerDTO] = []
+        for node in config.cluster_config.cluster_nodes:
+            running_containers_dto = ClusterController.list_all_running_containers(
+                ip=node.ip, port=constants.GRPC_SERVERS.CLUSTER_MANAGER_PORT)
+            running_containers_dtos = list(running_containers_dto.runningContainers)
+            running_containers = running_containers + running_containers_dtos
+        runner = CliRunner()
+        result = runner.invoke(ls, ["containers"])
+        cli_output = result.stdout_bytes.decode('utf-8')
+        cli_output_lines = cli_output.strip().split('\n')
+        number_of_containers_from_cli = 0
+        for line in cli_output_lines:
+            if "[running]" in line:
+                number_of_containers_from_cli += 1
+
+        assert number_of_containers_from_cli == len(running_containers)
+        assert result.exit_code == 0