From e4d8ac34eac1ab810ca6579efa23035d407a64f3 Mon Sep 17 00:00:00 2001
From: Andre Kurait <akurait@amazon.com>
Date: Fri, 6 Sep 2024 21:54:50 -0500
Subject: [PATCH] Add logs to E2E github action

Signed-off-by: Andre Kurait <akurait@amazon.com>
---
 .github/workflows/CI.yml | 19 +++++++++++++++++++
 1 file changed, 19 insertions(+)

diff --git a/.github/workflows/CI.yml b/.github/workflows/CI.yml
index 688cd0af7..64f36e10c 100644
--- a/.github/workflows/CI.yml
+++ b/.github/workflows/CI.yml
@@ -156,6 +156,25 @@ jobs:
         working-directory: TrafficCapture/dockerSolution/src/main/docker/migrationConsole/lib/integ_test
         run: |
           docker exec $(docker ps --filter "name=migration-console" -q) pipenv run pytest /root/lib/integ_test/integ_test/replayer_tests.py --unique_id="testindex" -s
+      - name: Collect Docker, OpenSearch Benchmark, and Shared Logs
+        if: always()
+        run: |
+          mkdir -p logs/docker logs/opensearch_benchmark_logs logs/shared_logs_output
+          for container in $(docker ps -aq); do
+            container_name=$(docker inspect --format '{{.Name}}' $container | sed 's/\///')
+            docker logs $container > logs/docker/${container_name}_logs.txt 2>&1
+          done
+          docker cp $(docker ps --filter "name=migration-console" -q):/root/.benchmark/logs logs/opensearch_benchmark_logs
+          docker cp $(docker ps --filter "name=migration-console" -q):/shared-logs-output logs/shared_logs_output
+      - name: Upload Logs
+        if: always()
+        uses: actions/upload-artifact@v4
+        with:
+          name: e2e-test-logs
+          path: |
+            logs/docker
+            logs/opensearch_benchmark_logs
+            logs/shared_logs_output
       - name: Clean up migrations docker images before caching
         run: |
           docker stop $(docker ps -q) && docker system prune --volumes -f