aboutsummaryrefslogtreecommitdiffstats
path: root/test/tutils.py
diff options
context:
space:
mode:
Diffstat (limited to 'test/tutils.py')
-rw-r--r--test/tutils.py10
1 files changed, 10 insertions, 0 deletions
diff --git a/test/tutils.py b/test/tutils.py
index 4cd7b7f8..afc1fb51 100644
--- a/test/tutils.py
+++ b/test/tutils.py
@@ -1,8 +1,11 @@
import os, shutil, tempfile
from contextlib import contextmanager
from libmproxy import flow, utils, controller
+from libmproxy.console.flowview import FlowView
+from libmproxy.console import ConsoleState
from netlib import certutils
from nose.plugins.skip import SkipTest
+from mock import Mock
def _SkipWindows():
raise SkipTest("Skipped on Windows.")
@@ -57,6 +60,13 @@ def tflow_err():
f.error = terr(f.request)
return f
+def tflowview():
+ m = Mock()
+ cs = ConsoleState()
+ flow = tflow()
+ fv = FlowView(m, cs, flow)
+ return fv
+
@contextmanager
def tmpdir(*args, **kwargs):