Some tests for cipher_state

2018-03-01 11:44:42 +01:00
parent d78c774b55
commit 0d563884eb
4 changed files with 112 additions and 0 deletions
@@ -0,0 +1,30 @@
+%%%-------------------------------------------------------------------
+%%% @copyright (C) 2018, Aeternity Anstalt
+%%%-------------------------------------------------------------------
+
+-module(enoise_crypto_tests).
+
+-include_lib("eunit/include/eunit.hrl").
+
+chachapoly_test() ->
+    #{ key := Key, nonce := Nonce, ad := AD, mac := MAC,
+       pt := PlainText, ct := CipherText } = test_utils:chacha_data(),
+    PTLen  = byte_size(PlainText),
+    CTLen  = byte_size(CipherText),
+    MACLen = byte_size(MAC),
+
+    %% Sanity check
+    ?assert(PTLen == CTLen),
+
+    <<CipherText0:CTLen/binary, MAC0:MACLen/binary>> =
+        enoise_crypto:encrypt('ChaChaPoly', Key, Nonce, AD, PlainText),
+
+    ?assertMatch(CipherText, CipherText0),
+    ?assertMatch(MAC, MAC0),
+
+    <<PlainText0:PTLen/binary>> =
+        enoise_crypto:decrypt('ChaChaPoly', Key, Nonce, AD, <<CipherText/binary, MAC/binary>>),
+
+    ?assertMatch(PlainText, PlainText0),
+    ok.
+