// Copyright 2020 Google LLC. All Rights Reserved.

#ifndef WHITEBOX_API_LICENSE_WHITEBOX_BENCHMARK_H_
#define WHITEBOX_API_LICENSE_WHITEBOX_BENCHMARK_H_

#include <stdint.h>

#include <memory>
#include <vector>

#include "api/license_whitebox.h"
#include "api/test_license_builder.h"
#include "benchmarking/data_source.h"
#include "crypto_utils/rsa_key.h"
#include "testing/gtest/include/gtest/gtest.h"

namespace widevine {
class LicenseWhiteboxBenchmark : public ::testing::Test {
 protected:
  virtual void SetUp() override;

  License CreateLicense() const;

  std::vector<uint8_t> SignAsServer(const std::vector<uint8_t>& message) const;

  DataSource& Data() { return data_source_; }

  const RsaPublicKey* PublicKey() const { return public_key_.get(); }

  const std::vector<uint8_t>& ContentKeyId() const { return key_id_; }

  const std::vector<uint8_t>& ContentKey() const { return key_; }

  const std::vector<uint8_t>& ContentIV() const { return iv_; }

 private:
  DataSource data_source_;
  std::unique_ptr<RsaPublicKey> public_key_;

  std::vector<uint8_t> key_id_;
  std::vector<uint8_t> key_;
  std::vector<uint8_t> iv_;
};

}  // namespace widevine

#endif  // WHITEBOX_API_LICENSE_WHITEBOX_BENCHMARK_H_