diff --git a/ChaosTests/Cipher/Arc4CryptTests.cpp b/ChaosTests/Cipher/Arc4CryptTests.cpp index 9f98c6e..05842c7 100644 --- a/ChaosTests/Cipher/Arc4CryptTests.cpp +++ b/ChaosTests/Cipher/Arc4CryptTests.cpp @@ -1,8 +1,10 @@ #include #include #include +#include #include "Cipher/Arc4/Arc4Crypt.hpp" +#include "Service/ChaosException.hpp" using namespace Chaos::Cipher::Arc4; @@ -63,3 +65,19 @@ TEST(Arc4CryptTests, BasicTest) ASSERT_EQ(data, recoveredData); } + +TEST(Arc4CryptTests, UninitializedArc4CryptTest) +{ + Arc4Crypt arc4; + + { + std::array in; + in.fill(0); + + std::array out; + out.fill(0); + + ASSERT_THROW(arc4.Encrypt(out.begin(), in.begin(), in.size()), Chaos::Service::ChaosException); + ASSERT_THROW(arc4.Decrypt(out.begin(), in.begin(), in.size()), Chaos::Service::ChaosException); + } +}