pygost/test_gost28147_mac.py

   1 # coding: utf-8
   2 # PyGOST -- Pure Python GOST cryptographic functions library
   3 # Copyright (C) 2015-2016 Sergey Matveev <stargrave@stargrave.org>
   4 #
   5 # This program is free software: you can redistribute it and/or modify
   6 # it under the terms of the GNU General Public License as published by
   7 # the Free Software Foundation, either version 3 of the License, or
   8 # (at your option) any later version.
   9 #
  10 # This program is distributed in the hope that it will be useful,
  11 # but WITHOUT ANY WARRANTY; without even the implied warranty of
  12 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  13 # GNU General Public License for more details.
  14 #
  15 # You should have received a copy of the GNU General Public License
  16 # along with this program.  If not, see <http://www.gnu.org/licenses/>.
  17
  18 from unittest import TestCase
  19
  20 from pygost.gost28147_mac import MAC
  21
  22
  23 class TestMAC(TestCase):
  24     """ Test vectors generated with libgcl3 library
  25     """
  26     k = b'This is message\xFF length\x0032 bytes'
  27
  28     def test_a(self):
  29         self.assertEqual(
  30             MAC(self.k, b'a').hexdigest(),
  31             'bd5d3b5b2b7b57af',
  32         )
  33
  34     def test_abc(self):
  35         self.assertEqual(
  36             MAC(self.k, b'abc').hexdigest(),
  37             '28661e40805b1ff9',
  38         )
  39
  40     def test_128U(self):
  41         self.assertEqual(
  42             MAC(self.k, 128 * b'U').hexdigest(),
  43             '1a06d1bad74580ef',
  44         )
  45
  46     def test_13x(self):
  47         self.assertEqual(
  48             MAC(self.k, 13 * b'x').hexdigest(),
  49             '917ee1f1a668fbd3',
  50         )
  51
  52     def test_parts(self):
  53         m = MAC(self.k)
  54         m.update(b'foo')
  55         m.update(b'bar')
  56         self.assertEqual(m.digest(), MAC(self.k, b'foobar').digest())
  57
  58     def test_copy(self):
  59         m = MAC(self.k, b'foo')
  60         c = m.copy()
  61         m.update(b'barbaz')
  62         c.update(b'bar')
  63         c.update(b'baz')
  64         self.assertEqual(m.digest(), c.digest())