From 2652d18f24e2fb8c1f026ee2578e3cd673a7b4c0 Mon Sep 17 00:00:00 2001
From: Sergey Dudoladov <sergey.dudoladov@zalando.de>
Date: Tue, 30 Apr 2019 13:49:42 +0200
Subject: [PATCH] add timeouts

---
 e2e/tests/test_smoke.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/e2e/tests/test_smoke.py b/e2e/tests/test_smoke.py
index baeae7b13..7ad51cf1b 100755
--- a/e2e/tests/test_smoke.py
+++ b/e2e/tests/test_smoke.py
@@ -2,13 +2,13 @@
 import unittest, yaml, time
 from kubernetes import client, config, utils
 from pprint import pprint
+import timeout_decorator
 import subprocess
 
 class SmokeTestCase(unittest.TestCase):
     '''
     Test the most basic e2e functionality of the operator.
     '''
-
     @classmethod
     def setUpClass(cls):
         '''
@@ -56,6 +56,7 @@ class SmokeTestCase(unittest.TestCase):
             print("Waiting for the Spilo master pod to start. Current phase: " + str(pod_phase))
             time.sleep(5)
 
+    @timeout_decorator.timeout(60)
     def test_master_is_unique(self):
         """
            Check that there is a single pod in the k8s cluster with the label "spilo-role=master".
@@ -65,6 +66,7 @@ class SmokeTestCase(unittest.TestCase):
         master_pods = v1.list_namespaced_pod('default', label_selector='spilo-role=master,version=acid-minimal-cluster').items
         self.assertEqual(len(master_pods), 1, "Expected 1 master pod,found " + str(len(master_pods)))
 
+    @timeout_decorator.timeout(60)
     def test_scaling(self):
         """
            Scale up from 2 to 3 pods and back to 2 by updating the Postgres manifest at runtime.