More structure changes

1 files changed, 28 insertions, 0 deletions

diff --git a/tests/__init__.py b/tests/__init__.py
new file mode 100644
index 0000000..f203615
--- /dev/null
+++ b/tests/__init__.py
@@ -0,0 +1,28 @@
+import unittest
+
+from viewstate import ViewState
+
+class TestViewState(unittest.TestCase):
+
+    def test_blank(self):
+        vs = ViewState()
+        self.assertFalse(vs.is_valid())
+
+    def test_is_valid(self):
+        with open('samples/ngcs.sample', 'r') as f:
+            vs = ViewState(f.read())
+            self.assertTrue(vs.is_valid())
+
+    def test_invalid_decode(self):
+        with self.assertRaises(Exception):
+            vs = ViewState()
+            vs.raw = b'\x01\x02'
+            vs.decode()
+
+    def test_parse_const_value(self):
+        vs = ViewState()
+        vs.raw = b'\xff\x01\x67'
+        self.assertEqual(vs.decode(), True)
+
+if __name__ == '__main__':
+    unittest.main()