From 515dff331bb8f690ffd75e2338c4ef9ff7ccb141 Mon Sep 17 00:00:00 2001
From: Ryan Faircloth <rfaircloth@splunk.com>
Date: Sat, 29 Jun 2019 19:29:41 -0400
Subject: [PATCH] handle other containers not yet ready

---
 tests/test_poc.py | 21 +++++++++++++++++++--
 1 file changed, 19 insertions(+), 2 deletions(-)

diff --git a/tests/test_poc.py b/tests/test_poc.py
index cafce3e..4e7f486 100644
--- a/tests/test_poc.py
+++ b/tests/test_poc.py
@@ -24,13 +24,30 @@ def setup_wordlist():
 
 @pytest.fixture
 def setup_splunk():
-    return client.connect(username="admin", password="Changed@11", host="splunk", port="8089")
+    tried=0
+    while true:
+        try:
+            c = client.connect(username="admin", password="Changed@11", host="splunk", port="8089")
+        except ConnectionRefusedError:
+            tried +=1
+            if tried>90:
+                raise
+            sleep(1)
+    return c
 
 def sendsingle(message):
     sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
     server_address = ('sc4s', 514)
 
-    sock.connect(server_address)
+    tried = 0
+    try
+
+        sock.connect(server_address)
+    except:
+        if tried > 90:
+            raise
+        sleep(1)
+        
     sock.sendall(str.encode(message))
     sock.close()