// Copyright 2020 Google LLC. All Rights Reserved.

#ifndef WHITEBOX_API_LICENSE_WHITEBOX_TEST_BASE_H_
#define WHITEBOX_API_LICENSE_WHITEBOX_TEST_BASE_H_

#include <memory>

#include "api/golden_data.h"
#include "api/license_whitebox.h"
#include "crypto_utils/rsa_key.h"
#include "testing/gtest/include/gtest/gtest.h"

namespace widevine {

class LicenseWhiteboxTestBase : public ::testing::Test {
 protected:
  void SetUp() override;

  void TearDown() override;

  // Modify a buffer so that it won't be the exact same as it was before. This
  // to make it easier to invalidate signatures.
  void Modify(std::vector<uint8_t>* data) const;

  std::unique_ptr<RsaPublicKey> public_key_;

  GoldenData golden_data_;

  WB_License_Whitebox* whitebox_;
};

}  // namespace widevine

#endif  // WHITEBOX_API_LICENSE_WHITEBOX_TEST_BASE_H_