lokinet/test/pq_unittest.cpp

#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