From bd7c6a3321def52dae5798138b60188625bad234 Mon Sep 17 00:00:00 2001
From: Kkevsterrr <user@example.com>
Date: Thu, 12 Dec 2019 23:09:07 -0500
Subject: [PATCH] Additional engine coverage

---
 engine.py            |  2 --
 tests/test_engine.py | 22 ++++++++++++++++++++++
 2 files changed, 22 insertions(+), 2 deletions(-)

diff --git a/engine.py b/engine.py
index 39fb4bc..f76cdc3 100644
--- a/engine.py
+++ b/engine.py
@@ -34,8 +34,6 @@ class Engine():
         self.server_port = server_port
         self.seen_packets = []
         # Set up the directory and ID for logging
-        if not output_directory:
-            output_directory = "trials"
         actions.utils.setup_dirs(output_directory)
         if not environment_id:
             environment_id = actions.utils.get_id()
diff --git a/tests/test_engine.py b/tests/test_engine.py
index e5cc148..1f68376 100644
--- a/tests/test_engine.py
+++ b/tests/test_engine.py
@@ -34,6 +34,14 @@ def test_engine_sleep():
     with engine.Engine(port, strategy, log_level="info") as eng:
         os.system("curl http://example.com?q=ultrasurf")
 
+    # Strategy to use in opposite direction
+    strategy = "\/ [TCP:flags:SA]-sleep{1}-|"
+
+    # Create the engine in debug mode
+    with engine.Engine(port, strategy, log_level="debug") as eng:
+        os.system("curl http://example.com?q=ultrasurf")
+
+
 
 def test_engine_trace():
     """
@@ -48,3 +56,17 @@ def test_engine_trace():
     with engine.Engine(port, strategy, log_level="debug") as eng:
         os.system("curl -m 5 http://example.com?q=ultrasurf")
 
+
+def test_engine_trace():
+    """
+    Basic engine test with trace
+    """
+    # Port to run the engine on
+    port = 80
+    # Strategy to use
+    strategy = "\/ [TCP:flags:SA]-drop-|"
+
+    # Create the engine in debug mode
+    with engine.Engine(port, strategy, log_level="debug") as eng:
+        os.system("curl -m 3 http://example.com?q=ultrasurf")
+