From 502900e3e5f5ce8bfae2de9e78d3e6a5028e6d1f Mon Sep 17 00:00:00 2001 From: Kkevsterrr Date: Sun, 16 Feb 2020 22:50:24 -0500 Subject: [PATCH] Test now passes --- tests/test_sleep.py | 13 ++++++------- 1 file changed, 6 insertions(+), 7 deletions(-) diff --git a/tests/test_sleep.py b/tests/test_sleep.py index 3f4a3e2..2b5b2c0 100644 --- a/tests/test_sleep.py +++ b/tests/test_sleep.py @@ -1,5 +1,6 @@ +import logging + from scapy.all import IP, TCP -import evolve import actions.utils import actions.strategy import actions.packet @@ -9,12 +10,11 @@ import sys sys.path.append("..") -def test_basic_sleep(log_level): +def test_basic_sleep(): """ Tests the sleep action primitive """ - logger = evolve.logger - evolve.set_logger(log_level) + logger = logging.getLogger("test") sleep = actions.sleep.SleepAction(.5) assert str(sleep) == "sleep{0.5}", "Sleep returned incorrect string representation: %s" % str(sleep) @@ -24,12 +24,11 @@ def test_basic_sleep(log_level): assert packet1.sleep == .5, "Packet had wrong sleep value" -def test_sleep_str_parse(log_level): +def test_sleep_str_parse(): """ Tests stringing and parsing a sleep action with a float sleep time """ - logger = evolve.logger - evolve.set_logger(log_level) + logger = logging.getLogger("test") strat = actions.utils.parse("[TCP:flags:A]-sleep{0.5}-|", logger)