From 5dcc3b4ff832f817fc621d9116f1b399eb2ff137 Mon Sep 17 00:00:00 2001 From: kira0204 Date: Mon, 12 Mar 2018 00:28:43 +0530 Subject: [PATCH] Testing using addon --- test/mitmproxy/test_command.py | 28 +++++++++++++++++----------- 1 file changed, 17 insertions(+), 11 deletions(-) diff --git a/test/mitmproxy/test_command.py b/test/mitmproxy/test_command.py index 0c8d7208c..324985784 100644 --- a/test/mitmproxy/test_command.py +++ b/test/mitmproxy/test_command.py @@ -1,5 +1,4 @@ import typing -from unittest import mock from mitmproxy import command from mitmproxy import flow from mitmproxy import exceptions @@ -310,20 +309,27 @@ class TDec: pass +class TAttr: + def __getattr__(self, item): + raise IOError + + +class TCmds(TAttr): + def __init__(self): + self.TAttr = TAttr() + + def test_collect_commands(): """ 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: - mock_hasattr.side_effect = OSError - c = command.CommandManager(tctx.master) - a = TDec() - c.collect_commands(a) - assert "cmd1" not in c.commands - assert "cmd2" not in c.commands - assert "empty" not in c.commands + with taddons.context() as tctx: + c = command.CommandManager(tctx.master) + a = TCmds() + c.collect_commands(a) + assert "cmd1" not in c.commands + assert "cmd2" not in c.commands + assert "empty" not in c.commands def test_decorator():