Add tests for decode

Author: whtsky

tests/test_decode.py

@@ -0,0 +1,32 @@
+# -*- coding: utf-8 -*-
+
+from bencoder import bdecode
+
+
+def test_decode_str():
+    assert bdecode(b'6:WWWWWW') == "WWWWWW"
+
+
+def test_decode_int():
+    assert bdecode(b'i233e') == 233
+
+
+def test_decode_list():
+    assert bdecode(b'l1:a1:bi3ee') == ['a', 'b', 3]
+
+
+def test_decode_dict():
+    od = dict()
+    od['ka'] = 'va'
+    od['kb'] = 2
+    assert bdecode(b'd2:ka2:va2:kbi2ee') == od
+
+
+def test_encode_complex():
+    od = dict()
+    od['KeyA'] = ['listitemA', {'k': 'v'}, 3]
+    od['KeyB'] = {'k': 'v'}
+    od['KeyC'] = 3
+    od['KeyD'] = 'AString'
+    expected_result = b'd4:KeyAl9:listitemAd1:k1:vei3ee4:KeyBd1:k1:ve4:KeyCi3e4:KeyD7:AStringe'
+    assert bdecode(expected_result) == od

tests/test_encode.py

@@ -14,7 +14,7 @@ def test_encode_int():
 def test_encode_bytes():
     b = b"TheseAreSomeBytes"
     coded = bencode(b)
-    l = bytes(str(len(b)))
+    l = str(len(b)).encode()
     assert coded == l + b':' + b