use container name in dictionary instead of pod's to return logs

Author: Myroslav Rozum

chaosk8s/pod/probes.py

@@ -59,24 +59,37 @@ def read_pod_logs(name: str = None, last: Union[str, None] = None,
         now = datetime.now()
         since = int((now - dateparser.parse(last)).total_seconds())
 
-    params = dict(
-        namespace=ns,
-        follow=False,
-        previous=from_previous,
-        timestamps=True,
-        container=container_name or "",  # None is not a valid value
-        _preload_content=False
-    )
-
-    if since:
-        params["since_seconds"] = since
-
     logs = {}
+
     for p in ret.items:
-        name = p.metadata.name
-        logger.debug("Fetching logs for pod '{n}'".format(n=name))
-        r = v1.read_namespaced_pod_log(name, **params)
-        logs[name] = r.read().decode('utf-8')
+        # Retreive either logs for specific container
+        # or for all containers at once if not ptovided.
+        container_names = []
+        if container_name:
+            container_names.append(container_name)
+        else:
+            container_names = [c.name
+                               for i, c
+                               in enumerate(p.status.container_statuses)]
+
+        for cname in container_names:
+            params = dict(
+                namespace=ns,
+                follow=False,
+                previous=from_previous,
+                timestamps=True,
+                container=cname,
+                _preload_content=False
+            )
+            if since:
+                params["since_seconds"] = since
+
+            name = p.metadata.name
+            logger.debug("Fetching logs for pod '{n}' container {c}"
+                         .format(n=name, c=cname))
+
+            r = v1.read_namespaced_pod_log(name, **params)
+            logs[cname] = r.read().decode('utf-8')
 
     return logs
 

tests/test_probes.py

@@ -323,11 +323,11 @@ def test_fetch_last_logs(cl, client, has_conf):
 
     v1.read_namespaced_pod_log.return_value = io.BytesIO(b"hello")
 
-    logs = read_microservices_logs("myapp")
-
-    assert pod.metadata.name in logs
-    assert logs[pod.metadata.name] == "hello"
+    container_name = "mycontainer"
+    logs = read_microservices_logs(name="myapp", container_name=container_name)
 
+    assert container_name in logs
+    assert logs[container_name] == "hello"
 
 @patch('chaosk8s.has_local_config_file', autospec=True)
 @patch('chaosk8s.service.probes.client', autospec=True)