WIP: Reorientation

test/znoise_chiper_state_tests.erl

@@ -0,0 +1,39 @@
+%%%-------------------------------------------------------------------
+%%% @copyright (C) 2018, Aeternity Anstalt
+%%%-------------------------------------------------------------------
+
+-module(znoise_chiper_state_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),
+
+    CS0 = znoise_cipher_state:init(Key, 'ChaChaPoly'),
+    CS1 = znoise_cipher_state:set_nonce(CS0, Nonce),
+
+    {ok, _CS2, <<CipherText0:CTLen/binary, MAC0:MACLen/binary>>} =
+        znoise_cipher_state:encrypt_with_ad(CS1, AD, PlainText),
+
+    ?assertMatch(CipherText, CipherText0),
+    ?assertMatch(MAC, MAC0),
+
+    {ok, _CS3, <<PlainText0:PTLen/binary>>} =
+        znoise_cipher_state:decrypt_with_ad(CS1, AD, <<CipherText/binary, MAC/binary>>),
+
+    ?assertMatch(PlainText, PlainText0),
+
+    % rekey test
+    CS4 = znoise_cipher_state:rekey(CS1),
+    {ok, _CS5, <<CipherText1:CTLen/binary, MAC1:MACLen/binary>>} =
+        znoise_cipher_state:encrypt_with_ad(CS4, AD, PlainText),
+    {ok, _CS6, <<PlainText1:PTLen/binary>>} =
+        znoise_cipher_state:decrypt_with_ad(CS4, AD, <<CipherText1/binary, MAC1/binary>>),
+    ?assertMatch(PlainText, PlainText1),
+
+    ok.
+