Grasscutters/mitmproxy
3
0
Fork 0
mirror of https://github.com/Grasscutters/mitmproxy.git synced 2024-11-25 09:37:37 +00:00
Code Issues Projects Releases Wiki Activity

Replace asynctest with stdlib mock

Browse Source 
This is an implementation of
https://github.com/mitmproxy/mitmproxy/issues/4020

Tested to work fine here with Python 3.8.6.
This commit is contained in:
Felix Yan 2020-11-14 05:27:35 +08:00
parent 83570c7cba
commit 0b0a6cfef6
No known key found for this signature in database
GPG Key ID: 786C63F330D7CB92
2 changed files with 11 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
setup.py
View File

@ -102,7 +102,6 @@ setup(
"dataclasses>=0.7",
],
'dev': [
"asynctest>=0.12.0",
"Flask>=1.0,<1.2",
"hypothesis>=5.8,<6",
"parver>=0.1,<2.0",

22
test/mitmproxy/addons/test_readfile.py
View File

@ -2,7 +2,7 @@ import asyncio
import io
import pytest
import asynctest
from unittest import mock
import mitmproxy.io
from mitmproxy import exceptions
@ -54,17 +54,17 @@ class TestReadFile:
tf = tmpdir.join("tfile")
with asynctest.patch('mitmproxy.master.Master.load_flow') as mck:
with mock.patch('mitmproxy.master.Master.load_flow') as mck:
tf.write(data.getvalue())
tctx.configure(
rf,
rfile = str(tf),
readfile_filter = ".*"
)
assert not mck.awaited
mck.assert_not_awaited()
rf.running()
await asyncio.sleep(0)
assert mck.awaited
mck.assert_awaited()
tf.write(corrupt_data.getvalue())
tctx.configure(rf, rfile=str(tf))
@ -93,16 +93,16 @@ class TestReadFile:
class TestReadFileStdin:
@asynctest.patch('sys.stdin')
@mock.patch('sys.stdin')
@pytest.mark.asyncio
async def test_stdin(self, stdin, data, corrupt_data):
rf = readfile.ReadFileStdin()
with taddons.context(rf):
with asynctest.patch('mitmproxy.master.Master.load_flow') as mck:
with mock.patch('mitmproxy.master.Master.load_flow') as mck:
stdin.buffer = data
assert not mck.awaited
mck.assert_not_awaited()
await rf.load_flows(stdin.buffer)
assert mck.awaited
mck.assert_awaited()
stdin.buffer = corrupt_data
with pytest.raises(exceptions.FlowReadException):
@ -113,10 +113,10 @@ class TestReadFileStdin:
rf = readfile.ReadFileStdin()
with taddons.context(rf) as tctx:
tf = tmpdir.join("tfile")
with asynctest.patch('mitmproxy.master.Master.load_flow') as mck:
with mock.patch('mitmproxy.master.Master.load_flow') as mck:
tf.write(data.getvalue())
tctx.configure(rf, rfile=str(tf))
assert not mck.awaited
mck.assert_not_awaited()
rf.running()
await asyncio.sleep(0)
assert mck.awaited
mck.assert_awaited()