tests/test_viewstate.py


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47

import pytest

from os import walk
from os.path import join
from viewstate import ViewState, ViewStateException


class TestViewState(object):
    def test_blank(self):
        vs = ViewState()
        assert not vs.is_valid()

    def test_is_valid(self):
        for root, dirs, files in walk("tests/samples"):
            for f in files:
                with open(join(root, f), "r") as t:
                    vs = ViewState(t.read())
                    assert vs.is_valid() is True

    def test_invalid_base64(self):
        with pytest.raises(ViewStateException):
            ViewState("hello")

    def test_invalid_decode(self):
        with pytest.raises(ViewStateException):
            vs = ViewState(raw=b"\x01\x02")
            vs.decode()

    def test_no_signature(self):
        vs = ViewState(raw=b"\xff\x01e")
        vs.decode()
        assert vs.mac is None
        assert vs.signature is None

    def test_macs(self):
        MAC_LENGTHS = {
            "hmac_sha1": 20,
            "hmac_sha256": 32,
            "unknown": 5,  # could be any other value
        }

        for mac, n in MAC_LENGTHS.items():
            sig = b"\x55" * n  # not a real signature, just testing length
            vs = ViewState(raw=b"\xff\x01d" + sig)
            vs.decode()
            assert vs.mac == mac
            assert vs.signature == sig