Fix body and preamble properties

1 files changed, 8 insertions, 6 deletions

diff --git a/test.py b/test.py
index 5ef245f..1f84bc2 100644
--- a/test.py
+++ b/test.py
@@ -13,14 +13,16 @@ class TestViewState(unittest.TestCase):
             vs = ViewState(f.read())
             self.assertTrue(vs.is_valid())
 
-    # def test_invalid_decode(self):
-    #     with self.assertRaises(Exception):
-    #         vs = ViewState(b'\x01\x02', decoded=True)
-    #         vs.decode()
+    def test_invalid_decode(self):
+        with self.assertRaises(Exception):
+            vs = ViewState()
+            vs.raw = b'\x01\x02'
+            vs.decode()
 
     # def test_parse(self):
-    #     vs = ViewState(b'\xff\x01\x67', decoded=True)
-    #     assertTrue(vs.decode())
+    #     vs = ViewState()
+    #     vs.raw = b'\xff\x01\x67'
+    #     self.assertTrue(vs.decode())
 
 if __name__ == '__main__':
     unittest.main()