aboutsummaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
-rw-r--r--test/mitmproxy/addons/test_evenstore.py32
-rw-r--r--test/mitmproxy/addons/test_eventstore.py33
2 files changed, 32 insertions, 33 deletions
diff --git a/test/mitmproxy/addons/test_evenstore.py b/test/mitmproxy/addons/test_evenstore.py
deleted file mode 100644
index f54b9980..00000000
--- a/test/mitmproxy/addons/test_evenstore.py
+++ /dev/null
@@ -1,32 +0,0 @@
-from unittest import mock
-from mitmproxy import log
-from mitmproxy.addons import eventstore
-
-
-def test_simple():
- store = eventstore.EventStore()
- assert not store.data
-
- sig_add = mock.Mock(spec=lambda: 42)
- sig_refresh = mock.Mock(spec=lambda: 42)
- store.sig_add.connect(sig_add)
- store.sig_refresh.connect(sig_refresh)
-
- assert not sig_add.called
- assert not sig_refresh.called
-
- # test .log()
- store.log(log.LogEntry("test", "info"))
- assert store.data
-
- assert sig_add.called
- assert not sig_refresh.called
-
- # test .clear()
- sig_add.reset_mock()
-
- store.clear()
- assert not store.data
-
- assert not sig_add.called
- assert sig_refresh.called
diff --git a/test/mitmproxy/addons/test_eventstore.py b/test/mitmproxy/addons/test_eventstore.py
index 777ab4dd..f54b9980 100644
--- a/test/mitmproxy/addons/test_eventstore.py
+++ b/test/mitmproxy/addons/test_eventstore.py
@@ -1 +1,32 @@
-# TODO: write tests
+from unittest import mock
+from mitmproxy import log
+from mitmproxy.addons import eventstore
+
+
+def test_simple():
+ store = eventstore.EventStore()
+ assert not store.data
+
+ sig_add = mock.Mock(spec=lambda: 42)
+ sig_refresh = mock.Mock(spec=lambda: 42)
+ store.sig_add.connect(sig_add)
+ store.sig_refresh.connect(sig_refresh)
+
+ assert not sig_add.called
+ assert not sig_refresh.called
+
+ # test .log()
+ store.log(log.LogEntry("test", "info"))
+ assert store.data
+
+ assert sig_add.called
+ assert not sig_refresh.called
+
+ # test .clear()
+ sig_add.reset_mock()
+
+ store.clear()
+ assert not store.data
+
+ assert not sig_add.called
+ assert sig_refresh.called