aboutsummaryrefslogtreecommitdiffstats
path: root/test/mitmproxy/addons/test_browser.py
diff options
context:
space:
mode:
Diffstat (limited to 'test/mitmproxy/addons/test_browser.py')
-rw-r--r--test/mitmproxy/addons/test_browser.py14
1 files changed, 8 insertions, 6 deletions
diff --git a/test/mitmproxy/addons/test_browser.py b/test/mitmproxy/addons/test_browser.py
index 407a3fe6..b05996fe 100644
--- a/test/mitmproxy/addons/test_browser.py
+++ b/test/mitmproxy/addons/test_browser.py
@@ -1,31 +1,33 @@
from unittest import mock
+import pytest
from mitmproxy.addons import browser
from mitmproxy.test import taddons
-def test_browser():
+@pytest.mark.asyncio
+async def test_browser():
with mock.patch("subprocess.Popen") as po, mock.patch("shutil.which") as which:
which.return_value = "chrome"
b = browser.Browser()
with taddons.context() as tctx:
b.start()
assert po.called
- b.start()
- assert not tctx.master.has_log("already running")
+ b.start()
b.browser.poll = lambda: None
b.start()
- assert tctx.master.has_log("already running")
+ assert await tctx.master.await_log("already running")
b.done()
assert not b.browser
-def test_no_browser():
+@pytest.mark.asyncio
+async def test_no_browser():
with mock.patch("shutil.which") as which:
which.return_value = False
b = browser.Browser()
with taddons.context() as tctx:
b.start()
- assert tctx.master.has_log("platform is not supported")
+ assert await tctx.master.await_log("platform is not supported")