Fixed incorrect digest calculation

[pygost.git] / pygost / test_gost34112012.py

diff --git a/pygost/test_gost34112012.py b/pygost/test_gost34112012.py
index 63232d602d4bb635e892310640d96b8efcf1d7be..c7c2df9220e4200200c53ccf8f7d52a175813ca5 100644 (file)
--- a/pygost/test_gost34112012.py
+++ b/pygost/test_gost34112012.py
@@ -14,6 +14,8 @@
 # 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 os import urandom
+from random import randint
 from unittest import skip
 from unittest import TestCase
 import hmac
@@ -37,6 +39,20 @@ class TestCopy(TestCase):
         self.assertSequenceEqual(m.digest(), c.digest())
 
 
+class TestSymmetric(TestCase):
+    def runTest(self):
+        chunks = []
+        for _ in range(randint(1, 10)):
+            chunks.append(urandom(randint(20, 80)))
+        m = GOST34112012256()
+        for chunk in chunks:
+            m.update(chunk)
+        self.assertSequenceEqual(
+            m.hexdigest(),
+            GOST34112012256(b"".join(chunks)).hexdigest(),
+        )
+
+
 class TestHMAC(TestCase):
     """RFC 7836
     """
@@ -88,7 +104,7 @@ class TestVectors(TestCase):
         )
 
     def test_habr144(self):
-        """Test vector from https://m.habr.com/ru/post/450024/
+        """Test vector from https://habr.com/ru/post/452200/
         """
         m = hexdec("d0cf11e0a1b11ae1000000000000000000000000000000003e000300feff0900060000000000000000000000010000000100000000000000001000002400000001000000feffffff0000000000000000ffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffff")
         self.assertSequenceEqual(