diff options
author | Aldo Cortesi <aldo@nullcube.com> | 2011-08-19 21:30:24 +1200 |
---|---|---|
committer | Aldo Cortesi <aldo@nullcube.com> | 2011-08-19 21:30:24 +1200 |
commit | a566684e3280ebbe15dd397710ee1b26bf8bd571 (patch) | |
tree | 6099d0213e8b2778823140d2e96e827743bd59ab /test | |
parent | 34adc83c717d82b8c9fc7fecc690b8b570a04644 (diff) | |
download | mitmproxy-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.py | 65 |
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() -] - |