1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
|
From: =?utf-8?q?S=C3=A9bastien_Delafond?= <sdelafond@gmail.com>
Date: Sun, 5 Aug 2018 14:02:47 +0200
Subject: Remove test_readfile.py
---
test/mitmproxy/addons/test_readfile.py | 114 ---------------------------------
1 file changed, 114 deletions(-)
delete mode 100644 test/mitmproxy/addons/test_readfile.py
diff --git a/test/mitmproxy/addons/test_readfile.py b/test/mitmproxy/addons/test_readfile.py
deleted file mode 100644
index 20e7721..0000000
--- a/test/mitmproxy/addons/test_readfile.py
+++ /dev/null
@@ -1,114 +0,0 @@
-import asyncio
-import io
-
-import pytest
-from unittest import mock
-
-import mitmproxy.io
-from mitmproxy import exceptions
-from mitmproxy.addons import readfile
-from mitmproxy.test import taddons
-from mitmproxy.test import tflow
-
-
-@pytest.fixture
-def data():
- f = io.BytesIO()
-
- w = mitmproxy.io.FlowWriter(f)
- flows = [
- tflow.tflow(resp=True),
- tflow.tflow(err=True),
- tflow.ttcpflow(),
- tflow.ttcpflow(err=True),
- ]
- for flow in flows:
- w.add(flow)
-
- f.seek(0)
- return f
-
-
-@pytest.fixture
-def corrupt_data(data):
- f = io.BytesIO(data.getvalue())
- f.seek(0, io.SEEK_END)
- f.write(b"qibble")
- f.seek(0)
- return f
-
-
-class TestReadFile:
- def test_configure(self):
- rf = readfile.ReadFile()
- with taddons.context(rf) as tctx:
- tctx.configure(rf, readfile_filter="~q")
- with pytest.raises(Exception, match="Invalid filter expression"):
- tctx.configure(rf, readfile_filter="~~")
- tctx.configure(rf, readfile_filter="")
-
- async def test_read(self, tmpdir, data, corrupt_data):
- rf = readfile.ReadFile()
- with taddons.context(rf) as tctx:
- assert not rf.reading()
-
- tf = tmpdir.join("tfile")
-
- with mock.patch("mitmproxy.master.Master.load_flow") as mck:
- tf.write(data.getvalue())
- tctx.configure(rf, rfile=str(tf), readfile_filter=".*")
- mck.assert_not_awaited()
- rf.running()
- await asyncio.sleep(0)
- mck.assert_awaited()
-
- tf.write(corrupt_data.getvalue())
- tctx.configure(rf, rfile=str(tf))
- rf.running()
- await tctx.master.await_log("corrupted")
-
- async def test_corrupt(self, corrupt_data):
- rf = readfile.ReadFile()
- with taddons.context(rf) as tctx:
- with pytest.raises(exceptions.FlowReadException):
- await rf.load_flows(io.BytesIO(b"qibble"))
-
- tctx.master.clear()
- with pytest.raises(exceptions.FlowReadException):
- await rf.load_flows(corrupt_data)
- await tctx.master.await_log("file corrupted")
-
- async def test_nonexistent_file(self):
- rf = readfile.ReadFile()
- with taddons.context(rf) as tctx:
- with pytest.raises(exceptions.FlowReadException):
- await rf.load_flows_from_path("nonexistent")
- await tctx.master.await_log("nonexistent")
-
-
-class TestReadFileStdin:
- @mock.patch("sys.stdin")
- async def test_stdin(self, stdin, data, corrupt_data):
- rf = readfile.ReadFileStdin()
- with taddons.context(rf):
- with mock.patch("mitmproxy.master.Master.load_flow") as mck:
- stdin.buffer = data
- mck.assert_not_awaited()
- await rf.load_flows(stdin.buffer)
- mck.assert_awaited()
-
- stdin.buffer = corrupt_data
- with pytest.raises(exceptions.FlowReadException):
- await rf.load_flows(stdin.buffer)
-
- async def test_normal(self, tmpdir, data):
- rf = readfile.ReadFileStdin()
- with taddons.context(rf) as tctx:
- tf = tmpdir.join("tfile")
- with mock.patch("mitmproxy.master.Master.load_flow") as mck:
- tf.write(data.getvalue())
- tctx.configure(rf, rfile=str(tf))
- mck.assert_not_awaited()
- rf.running()
- await asyncio.sleep(0)
- mck.assert_awaited()
|
