From 019da28a8e620d4e2b5ac828af9c9118c0821c40 Mon Sep 17 00:00:00 2001
From: Yuval Adam <_@yuv.al>
Date: Mon, 19 Feb 2018 15:17:38 +0100
Subject: More structure changes

---
 tests/__init__.py | 28 ++++++++++++++++++++++++++++
 1 file changed, 28 insertions(+)
 create mode 100644 tests/__init__.py

(limited to 'tests/__init__.py')

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()
-- 
cgit v1.3.1