tests: Update checks

e2e/sedge/checks.go

@@ -85,9 +85,18 @@ func checkContainerRunning(t *testing.T, containerNames ...string) {
 
 	for _, containerName := range containerNames {
 		t.Logf("Checking %s container is running", containerName)
-		isRunning, err := dockerServiceManager.IsRunning(containerName)
-		require.NoError(t, err)
-		assert.True(t, isRunning, "%s container should be running", containerName)
+		if containerName == "oncall_db_migration" {
+			// For DB migration, we just check if it exists and completed
+			status, err := dockerServiceManager.ContainerStatus(containerName)
+			require.NoError(t, err)
+			assert.Contains(t, []int{1, 6}, int(status),
+				"%s container should be exited or created", containerName)
+		} else {
+			// For all other containers, check if they're running
+			isRunning, err := dockerServiceManager.IsRunning(containerName)
+			require.NoError(t, err)
+			assert.True(t, isRunning, "%s container should be running", containerName)
+		}
 	}
 }
 

e2e/sedge/monitoring_stack_test.go

@@ -30,7 +30,7 @@ func skipIfNotAMD64(t *testing.T) {
 	}
 }
 
-var grafanaOnCallContainers = []string{"engine", "celery", "redis"}
+var grafanaOnCallContainers = []string{"engine", "celery", "redis", "oncall_db_migration"}
 
 // TestMonitoringStack_Init tests that the monitoring stack is not initialized if the user does not run the init-monitoring command
 func TestE2E_MonitoringStack_NotInitialized(t *testing.T) {