aboutsummaryrefslogtreecommitdiff
path: root/python/tests
diff options
context:
space:
mode:
authorDamir Jelić <poljar@termina.org.uk>2019-04-09 10:57:36 +0200
committerDamir Jelić <poljar@termina.org.uk>2019-04-09 10:57:36 +0200
commit32b99b793572c109fe0f3507705a10aefe2d8cf7 (patch)
treee6b304a80a97c39ba82508c8766cbb86fa1f4a90 /python/tests
parent659eb34fa4a28a94dd18d7dd743f6a6c032f2089 (diff)
python: Add support for the long KDF MAC calculation.
Diffstat (limited to 'python/tests')
-rw-r--r--python/tests/sas_test.py17
1 files changed, 17 insertions, 0 deletions
diff --git a/python/tests/sas_test.py b/python/tests/sas_test.py
index 05013b7..9001e67 100644
--- a/python/tests/sas_test.py
+++ b/python/tests/sas_test.py
@@ -80,3 +80,20 @@ class TestClass(object):
alice_mac = sas_alice.calculate_mac(message, extra_info)
assert alice_mac == expected_mac
+
+ def test_long_mac_generating(self):
+ sas_alice = Sas()
+ sas_bob = Sas()
+
+ with pytest.raises(OlmSasError):
+ sas_alice.calculate_mac_long_kdf(MESSAGE, EXTRA_INFO)
+
+ sas_alice.set_their_pubkey(sas_bob.pubkey)
+ sas_bob.set_their_pubkey(sas_alice.pubkey)
+
+ alice_mac = sas_alice.calculate_mac_long_kdf(MESSAGE, EXTRA_INFO)
+ bob_mac = sas_bob.calculate_mac_long_kdf(MESSAGE, EXTRA_INFO)
+ bob_short_mac = sas_bob.calculate_mac(MESSAGE, EXTRA_INFO)
+
+ assert alice_mac == bob_mac
+ assert alice_mac != bob_short_mac