From c4c6b4e7c17e5b96d86828a3bc809a1c6c9559df Mon Sep 17 00:00:00 2001
From: Yuval Adam <yuval@y3xz.com>
Date: Thu, 24 Mar 2016 22:24:43 +0200
Subject: Extract parse_const

---
 test.py      | 2 +-
 viewstate.py | 7 +++++--
 2 files changed, 6 insertions(+), 3 deletions(-)

diff --git a/test.py b/test.py
index c0cfc10..f203615 100644
--- a/test.py
+++ b/test.py
@@ -22,7 +22,7 @@ class TestViewState(unittest.TestCase):
     def test_parse_const_value(self):
         vs = ViewState()
         vs.raw = b'\xff\x01\x67'
-        self.assertEquals(vs.decode(), True)
+        self.assertEqual(vs.decode(), True)
 
 if __name__ == '__main__':
     unittest.main()
diff --git a/viewstate.py b/viewstate.py
index bf14005..b75cd37 100644
--- a/viewstate.py
+++ b/viewstate.py
@@ -9,10 +9,13 @@ CONSTS = {
     104: False
 }
 
+def parse_const(b):
+    return CONSTS.get(b, None)
+
 def parse(b):
     assert type(b) == bytes().__class__
-    if len(b) == 1 and b[0] in CONSTS:
-        return CONSTS[b[0]]
+    if len(b) == 1:
+        return parse_const(b[0])
     return None
 
 
-- 
cgit v1.3.1