From 6c89749f0a0c77e3a56a68df8709daf9c9f2f33c Mon Sep 17 00:00:00 2001
From: Aldo Cortesi <aldo@nullcube.com>
Date: Thu, 3 Feb 2011 13:30:47 +1300
Subject: Add timestamps to flows.

For now, these are only displayed on the connection view screen, with second
granularity.
---
 test/test_console.py | 5 +++++
 test/test_proxy.py   | 2 ++
 test/test_utils.py   | 8 +++++++-
 3 files changed, 14 insertions(+), 1 deletion(-)

(limited to 'test')

diff --git a/test/test_console.py b/test/test_console.py
index 6252d8de..9236baff 100644
--- a/test/test_console.py
+++ b/test/test_console.py
@@ -108,13 +108,18 @@ class uformat_flow(libpry.AutoTree):
         assert ('focus', '>> ') not in console.format_flow(f, False)
         assert ('focus', '>> ') in console.format_flow(f, True)
 
+        assert ('focus', '>> ') not in console.format_flow(f, False, True)
+        assert ('focus', '>> ') in console.format_flow(f, True, True)
+
         f.response = utils.tresp()
         f.request = f.response.request
         f.backup()
 
         assert ('method', '[edited] ') in console.format_flow(f, True)
+        assert ('method', '[edited] ') in console.format_flow(f, True, True)
         f.connection = flow.ReplayConnection()
         assert ('method', '[replay] ') in console.format_flow(f, True)
+        assert ('method', '[replay] ') in console.format_flow(f, True, True)
 
 
 class uPathCompleter(libpry.AutoTree):
diff --git a/test/test_proxy.py b/test/test_proxy.py
index ff9ec4bb..eaf73ee1 100644
--- a/test/test_proxy.py
+++ b/test/test_proxy.py
@@ -267,6 +267,8 @@ class uError(libpry.AutoTree):
         state = e.get_state()
         assert proxy.Error.from_state(state) == e
 
+        assert e.copy()
+
 
 class uProxyError(libpry.AutoTree):
     def test_simple(self):
diff --git a/test/test_utils.py b/test/test_utils.py
index f7c48c08..325664a4 100644
--- a/test/test_utils.py
+++ b/test/test_utils.py
@@ -1,8 +1,13 @@
-import textwrap, cStringIO, os
+import textwrap, cStringIO, os, time
 import libpry
 from libmproxy import utils
 
 
+class uformat_timestamp(libpry.AutoTree):
+    def test_simple(self):
+        assert utils.format_timestamp(time.time())
+
+
 class uisBin(libpry.AutoTree):
     def test_simple(self):
         assert not utils.isBin("testing\n\r")
@@ -234,6 +239,7 @@ class uprettybody(libpry.AutoTree):
 
     
 tests = [
+    uformat_timestamp(),
     umake_bogus_cert(),
     uisBin(),
     uhexdump(),
-- 
cgit v1.2.3