Grasscutters/mitmproxy
3
0
Fork 0
mirror of https://github.com/Grasscutters/mitmproxy.git synced 2024-12-02 12:01:17 +00:00
Code Issues Projects Releases Wiki Activity

Testing using addon

Browse Source
This commit is contained in:
kira0204 2018-03-12 00:28:43 +05:30
parent 167e01acdf
commit 5dcc3b4ff8
1 changed files with 17 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
test/mitmproxy/test_command.py
View File

@ -1,5 +1,4 @@
import typing import typing
from unittest import mock
from mitmproxy import command from mitmproxy import command
from mitmproxy import flow from mitmproxy import flow
from mitmproxy import exceptions from mitmproxy import exceptions
@ -310,16 +309,23 @@ class TDec:
pass pass
class TAttr:
def __getattr__(self, item):
raise IOError
class TCmds(TAttr):
def __init__(self):
self.TAttr = TAttr()
def test_collect_commands(): def test_collect_commands():
""" """
This tests for the error thrown by hasattr() This tests for the error thrown by hasattr()
""" """
with mock.patch("mitmproxy.command.hasattr") as mock_hasattr:
mock_hasattr.return_value = False
with taddons.context() as tctx: with taddons.context() as tctx:
mock_hasattr.side_effect = OSError
c = command.CommandManager(tctx.master) c = command.CommandManager(tctx.master)
a = TDec() a = TCmds()
c.collect_commands(a) c.collect_commands(a)
assert "cmd1" not in c.commands assert "cmd1" not in c.commands
assert "cmd2" not in c.commands assert "cmd2" not in c.commands