aboutsummaryrefslogtreecommitdiffstats
path: root/test
diff options
context:
space:
mode:
authorAldo Cortesi <aldo@nullcube.com>2011-08-19 21:30:24 +1200
committerAldo Cortesi <aldo@nullcube.com>2011-08-19 21:30:24 +1200
commita566684e3280ebbe15dd397710ee1b26bf8bd571 (patch)
tree6099d0213e8b2778823140d2e96e827743bd59ab /test
parent34adc83c717d82b8c9fc7fecc690b8b570a04644 (diff)
downloadmitmproxy-a566684e3280ebbe15dd397710ee1b26bf8bd571.tar.gz
mitmproxy-a566684e3280ebbe15dd397710ee1b26bf8bd571.tar.bz2
mitmproxy-a566684e3280ebbe15dd397710ee1b26bf8bd571.zip
Move to typed netstrings for serialization.
This change is backwards incompatible with the old serialization format!
Diffstat (limited to 'test')
-rw-r--r--test/test_netstring.py65
1 files changed, 0 insertions, 65 deletions
diff --git a/test/test_netstring.py b/test/test_netstring.py
deleted file mode 100644
index 5c4f775a..00000000
--- a/test/test_netstring.py
+++ /dev/null
@@ -1,65 +0,0 @@
-from libmproxy import netstring
-from cStringIO import StringIO
-import libpry
-
-
-class uNetstring(libpry.AutoTree):
- def setUp(self):
- self.test_data = "Netstring module by Will McGugan"
- self.encoded_data = "9:Netstring,6:module,2:by,4:Will,7:McGugan,"
-
- def test_header(self):
- t = [ ("netstring", "9:"),
- ("Will McGugan", "12:"),
- ("", "0:") ]
- for test, result in t:
- assert netstring.header(test) == result
-
- def test_file_encoder(self):
- file_out = StringIO()
- data = self.test_data.split()
- encoder = netstring.FileEncoder(file_out)
- for s in data:
- encoder.write(s)
- encoded_data = file_out.getvalue()
- assert encoded_data == self.encoded_data
-
- def test_decode_file(self):
- data = self.test_data.split()
- for buffer_size in range(1, len(self.encoded_data)):
- file_in = StringIO(self.encoded_data[:])
- decoded_data = list(netstring.decode_file(file_in, buffer_size = buffer_size))
- assert decoded_data == data
-
- def test_decoder(self):
- encoded_data = self.encoded_data
- for step in range(1, len(encoded_data)):
- i = 0
- chunks = []
- while i < len(encoded_data):
- chunks.append(encoded_data[i:i+step])
- i += step
- decoder = netstring.Decoder()
- decoded_data = []
- for chunk in chunks:
- for s in decoder.feed(chunk):
- decoded_data.append(s)
- assert decoded_data == self.test_data.split()
-
- def test_errors(self):
- d = netstring.Decoder()
- libpry.raises("Illegal digit", list, d.feed("1:foo"))
- d = netstring.Decoder()
- libpry.raises("Preceding zero", list, d.feed("01:f"))
- d = netstring.Decoder(5)
- libpry.raises("Maximum size", list, d.feed("500:f"))
- d = netstring.Decoder()
- libpry.raises("Illegal digit", list, d.feed(":f"))
-
-
-
-
-tests = [
- uNetstring()
-]
-