summaryrefslogtreecommitdiff
path: root/test.py
blob: 2cda602d62f5f844e4d5fffbfde679533e81212b (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
import unittest

from viewstate import ViewState

class TestViewState(unittest.TestCase):

    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()

if __name__ == '__main__':
    unittest.main()