lokinet/test/pq_unittest.cpp
2019-01-02 01:04:06 +00:00

42 lines
803 B
C++

#include <gtest/gtest.h>
#include <crypto.hpp>
#include <iostream>
namespace llarp
{
struct PQCryptoTest : public ::testing::Test
{
llarp::Crypto crypto;
PQKeyPair keys;
PQCryptoTest() : crypto(llarp::Crypto::sodium{})
{
}
llarp::Crypto*
Crypto()
{
return &crypto;
}
void
SetUp()
{
crypto.pqe_keygen(keys);
}
};
TEST_F(PQCryptoTest, TestCrypto)
{
PQCipherBlock block;
SharedSecret shared, otherShared;
auto c = Crypto();
ASSERT_TRUE(keys.size() == PQ_KEYPAIRSIZE);
ASSERT_TRUE(
c->pqe_encrypt(block, shared, PQPubKey(pq_keypair_to_public(keys))));
ASSERT_TRUE(c->pqe_decrypt(block, otherShared, pq_keypair_to_secret(keys)));
ASSERT_TRUE(otherShared == shared);
}
} // namespace llarp