diff options
| -rw-r--r-- | test.py | 16 | ||||
| -rw-r--r-- | viewstate.py | 6 |
2 files changed, 15 insertions, 7 deletions
@@ -4,15 +4,23 @@ 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('\x01\x02') - vs.decode() + # def test_invalid_decode(self): + # with self.assertRaises(Exception): + # vs = ViewState(b'\x01\x02', decoded=True) + # vs.decode() + + # def test_parse(self): + # vs = ViewState(b'\xff\x01\x67', decoded=True) + # assertTrue(vs.decode()) if __name__ == '__main__': unittest.main() diff --git a/viewstate.py b/viewstate.py index d1c9cf8..b4c162c 100644 --- a/viewstate.py +++ b/viewstate.py @@ -15,9 +15,9 @@ def parse(): class ViewState(object): - def __init__(self, base64_encoded): - self.base64_encoded = base64_encoded - self.raw = b64decode(self.base64_encoded) + def __init__(self, base64=''): + self.base64 = base64 + self.raw = b64decode(self.base64) self.decoded = None @property |