From 96b10ca486c88e6918aabdf8d90cde3a91cf8e1a Mon Sep 17 00:00:00 2001
From: Aldo Cortesi <aldo@nullcube.com>
Date: Sat, 12 Mar 2011 15:08:06 +1300
Subject: [PATCH] console.py 100% test coverage

---
 libmproxy/console.py |  3 ---
 test/test_console.py | 17 ++++++++++++++++-
 2 files changed, 16 insertions(+), 4 deletions(-)

diff --git a/libmproxy/console.py b/libmproxy/console.py
index f95d028e7..13c314351 100644
--- a/libmproxy/console.py
+++ b/libmproxy/console.py
@@ -699,7 +699,6 @@ class ConsoleState(flow.State):
         flow.State.__init__(self)
         self.focus = None
         self.beep = None
-        self.store = None
 
         self.view_body_mode = VIEW_BODY_RAW
         self.view_flow_mode = VIEW_FLOW_REQUEST
@@ -716,8 +715,6 @@ class ConsoleState(flow.State):
         return f
 
     def add_response(self, resp):
-        if self.store is not None:
-            self.store.save_response(resp)
         f = flow.State.add_response(self, resp)
         if self.focus is None:
             self.set_focus(0)
diff --git a/test/test_console.py b/test/test_console.py
index 90ebdc559..034f8ea17 100644
--- a/test/test_console.py
+++ b/test/test_console.py
@@ -105,6 +105,15 @@ class uformat_flow(libpry.AutoTree):
         assert ('method', '[replay] ') in console.format_flow(f, True)
         assert ('method', '[replay] ') in console.format_flow(f, True, True)
 
+        f.response.code = 404
+        assert ('error', '404') in console.format_flow(f, True, True)
+        f.response.headers["content-type"] = ["text/html"]
+        assert ('text', ' text/html') in console.format_flow(f, True, True)
+
+        f.response =None
+        f.error = proxy.Error(f.request, "error")
+        assert ('error', 'error') in console.format_flow(f, True, True)
+
 
 
 class uPathCompleter(libpry.AutoTree):
@@ -147,11 +156,17 @@ class uPathCompleter(libpry.AutoTree):
         assert c.final == s
 
 
+class uOptions(libpry.AutoTree):
+    def test_all(self):
+        assert console.Options(beep=True)
+
+
 
 
 tests = [
     uformat_keyvals(),
     uformat_flow(),
     uState(), 
-    uPathCompleter()
+    uPathCompleter(),
+    uOptions()
 ]