aboutsummaryrefslogtreecommitdiffstats
path: root/test/test_tcp.py
diff options
context:
space:
mode:
authorAldo Cortesi <aldo@corte.si>2013-01-16 19:15:39 -0800
committerAldo Cortesi <aldo@corte.si>2013-01-16 19:15:39 -0800
commitad92d37147e667068ed3458cf1256c187fd0f994 (patch)
treee9110e1a686f64d4567ae4bd44842d70c9aab4f4 /test/test_tcp.py
parentf673cfed634a1a1cc38dcca9a460d6054555f330 (diff)
parent04048b4c73f477f11d41788366eddffaae6bbb20 (diff)
downloadmitmproxy-ad92d37147e667068ed3458cf1256c187fd0f994.tar.gz
mitmproxy-ad92d37147e667068ed3458cf1256c187fd0f994.tar.bz2
mitmproxy-ad92d37147e667068ed3458cf1256c187fd0f994.zip
Merge pull request #8 from rouli/master
Improving timestamp feature
Diffstat (limited to 'test/test_tcp.py')
-rw-r--r--test/test_tcp.py24
1 files changed, 24 insertions, 0 deletions
diff --git a/test/test_tcp.py b/test/test_tcp.py
index 5a12da91..d27a678a 100644
--- a/test/test_tcp.py
+++ b/test/test_tcp.py
@@ -313,3 +313,27 @@ class TestFileLike:
s.write("x")
assert s.get_log() == "xx"
+ def test_reset_timestamps(self):
+ s = cStringIO.StringIO("foobar\nfoobar")
+ s = tcp.Reader(s)
+ s.first_byte_timestamp = 500
+ s.reset_timestamps()
+ assert not s.first_byte_timestamp
+
+ def test_first_byte_timestamp_updated_on_read(self):
+ s = cStringIO.StringIO("foobar\nfoobar")
+ s = tcp.Reader(s)
+ s.read(1)
+ assert s.first_byte_timestamp
+ expected = s.first_byte_timestamp
+ s.read(5)
+ assert s.first_byte_timestamp == expected
+
+ def test_first_byte_timestamp_updated_on_readline(self):
+ s = cStringIO.StringIO("foobar\nfoobar\nfoobar")
+ s = tcp.Reader(s)
+ s.readline()
+ assert s.first_byte_timestamp
+ expected = s.first_byte_timestamp
+ s.readline()
+ assert s.first_byte_timestamp == expected