diff options
-rw-r--r-- | test/mitmproxy/addons/test_evenstore.py | 32 | ||||
-rw-r--r-- | test/mitmproxy/addons/test_eventstore.py | 33 |
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 |