// Copyright 2024 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#include "components/enterprise/client_certificates/core/ec_private_key.h"

#include "base/memory/scoped_refptr.h"
#include "base/test/task_environment.h"
#include "base/test/test_future.h"
#include "components/enterprise/client_certificates/core/constants.h"
#include "components/enterprise/client_certificates/core/private_key.h"
#include "crypto/keypair.h"
#include "net/ssl/ssl_private_key.h"
#include "testing/gtest/include/gtest/gtest.h"

namespace client_certificates {

TEST(ECPrivateKeyTest, KeyWorksAsExpected) {
  base::test::TaskEnvironment task_environment;

  auto ec_private_key = base::MakeRefCounted<ECPrivateKey>(
      crypto::keypair::PrivateKey::GenerateEcP256());

  EXPECT_EQ(ec_private_key->GetAlgorithm(),
            crypto::SignatureVerifier::ECDSA_SHA256);

  auto spki_bytes = ec_private_key->GetSubjectPublicKeyInfo();
  EXPECT_GT(spki_bytes.size(), 0U);

  base::test::TestFuture<std::optional<std::vector<uint8_t>>> test_future;
  ec_private_key->Sign(spki_bytes, test_future.GetCallback());
  auto signature = test_future.Get();
  ASSERT_TRUE(signature.has_value());
  EXPECT_GT(signature->size(), 0U);

  auto proto_key = ec_private_key->ToProto();
  EXPECT_EQ(proto_key.source(),
            client_certificates_pb::PrivateKey::PRIVATE_SOFTWARE_KEY);
  EXPECT_GT(proto_key.wrapped_key().size(), 0U);

  auto dict_key = ec_private_key->ToDict();
  EXPECT_EQ(*dict_key.FindInt(kKeySource),
            static_cast<int>(PrivateKeySource::kSoftwareKey));
  EXPECT_GT(dict_key.FindString(kKey)->size(), 0U);
}

}  // namespace client_certificates