summaryrefslogtreecommitdiff
path: root/pygost/test_gost28147_mac.py
blob: 2f54dd855bf43b44aea4c9392827f7524ab96b4b (plain)
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
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
# coding: utf-8
# PyGOST -- Pure Python GOST cryptographic functions library
# Copyright (C) 2015-2020 Sergey Matveev <stargrave@stargrave.org>
#
# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, version 3 of the License.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program.  If not, see <http://www.gnu.org/licenses/>.

from unittest import TestCase

from pygost.gost28147_mac import MAC


class TestMAC(TestCase):
    """ Test vectors generated with libgcl3 library
    """
    k = b"This is message\xFF length\x0032 bytes"

    def test_a(self):
        self.assertSequenceEqual(
            MAC(self.k, b"a").hexdigest(),
            "bd5d3b5b2b7b57af",
        )

    def test_abc(self):
        self.assertSequenceEqual(
            MAC(self.k, b"abc").hexdigest(),
            "28661e40805b1ff9",
        )

    def test_128U(self):
        self.assertSequenceEqual(
            MAC(self.k, 128 * b"U").hexdigest(),
            "1a06d1bad74580ef",
        )

    def test_13x(self):
        self.assertSequenceEqual(
            MAC(self.k, 13 * b"x").hexdigest(),
            "917ee1f1a668fbd3",
        )

    def test_parts(self):
        m = MAC(self.k)
        m.update(b"foo")
        m.update(b"bar")
        self.assertSequenceEqual(m.digest(), MAC(self.k, b"foobar").digest())

    def test_copy(self):
        m = MAC(self.k, b"foo")
        c = m.copy()
        m.update(b"barbaz")
        c.update(b"bar")
        c.update(b"baz")
        self.assertSequenceEqual(m.digest(), c.digest())