From 8a48a061467cd3a7ab9dfca8052675323a0f5555 Mon Sep 17 00:00:00 2001
From: Ryan Faircloth <rfaircloth@splunk.com>
Date: Fri, 28 Jun 2019 15:21:12 -0400
Subject: [PATCH] Working default route test case

---
 tests/test_poc.py | 73 ++++++++++++++++++++++++-----------------------
 1 file changed, 37 insertions(+), 36 deletions(-)

diff --git a/tests/test_poc.py b/tests/test_poc.py
index b95ac2b..43e862c 100644
--- a/tests/test_poc.py
+++ b/tests/test_poc.py
@@ -12,13 +12,13 @@
 from time import sleep
 
 
-
 env = Environment(extensions=['jinja2_time.TimeExtension'])
 word_url = "http://svnweb.freebsd.org/csrg/share/dict/words?view=co&content-type=text/plain"
 response = urllib.request.urlopen(word_url)
 long_txt = response.read().decode()
 words = long_txt.splitlines()
 
+
 def sendsingle(message):
     # Create a TCP/IP socket
     sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
@@ -27,53 +27,54 @@ def sendsingle(message):
     sock.connect(server_address)
     sock.sendall(str.encode(message))
 
+
 def splunk_single(search):
-    service = client.connect(username="admin", password="Changed@11",host="splunk",port="8089")
+    service = client.connect(username="admin", password="Changed@11", host="splunk", port="8089")
     kwargs_normalsearch = {"exec_mode": "normal"}
-    job = service.jobs.create(search, **kwargs_normalsearch)
-
-    # A normal search returns the job's SID right away, so we need to poll for completion
+    tried = 0
     while True:
-        while not job.is_ready():
-            pass
-        stats = {"isDone": job["isDone"],
-                 "doneProgress": float(job["doneProgress"]) * 100,
-                 "scanCount": int(job["scanCount"]),
-                 "eventCount": int(job["eventCount"]),
-                 "resultCount": int(job["resultCount"])}
-
-        status = ("\r%(doneProgress)03.1f%%   %(scanCount)d scanned   "
-                  "%(eventCount)d matched   %(resultCount)d results") % stats
-
-        sys.stdout.write(status)
-        sys.stdout.flush()
-        if stats["isDone"] == "1":
-            sys.stdout.write("\n\nDone!\n\n")
+        job = service.jobs.create(search, **kwargs_normalsearch)
+
+        # A normal search returns the job's SID right away, so we need to poll for completion
+        while True:
+            while not job.is_ready():
+                pass
+            stats = {"isDone": job["isDone"],
+                     "doneProgress": float(job["doneProgress"]) * 100,
+                     "scanCount": int(job["scanCount"]),
+                     "eventCount": int(job["eventCount"]),
+                     "resultCount": int(job["resultCount"])}
+
+            if stats["isDone"] == "1":
+                break
+            sleep(2)
+
+        # Get the results and display them
+        resultCount = stats["resultCount"]
+        eventCount = stats["eventCount"]
+        if resultCount > 0 or tried>15:
             break
-        sleep(2)
-
-    # Get the results and display them
-    resultCount=stats["resultCount"]
-    eventCount=stats["eventCount"]
-    return resultCount,eventCount
-
-
+        else:
+            tried += 1
+            sleep(1)
+    return resultCount, eventCount
 
-def test_defaultroute():
-
-    ##mark## ##date##T##time##.000z "##name## bluecoat[0]:SPLV5
-
-    host = "{}-{}".format(random.choice(words),random.choice(words))
 
+def test_defaultroute(record_property):
+    host = "{}-{}".format(random.choice(words), random.choice(words))
 
     t = env.from_string("{{ mark }} {% now 'utc', '%b %d %H:%M:%S' %}.000z {{ host }} sc4s_default[0]: test")
     message = t.render(mark="<111>1", host=host)
 
     sendsingle(message)
 
-    sleep(15)
-
     search = 'search "{}" | head 10'.format(host)
 
-    resultCount ,eventCount = splunk_single(search)
+    resultCount, eventCount = splunk_single(search)
+    record_property("host", host)
+    record_property("resultCount", resultCount)
+    record_property("message", message)
+
     assert resultCount == 1
+
+