// Copyright 2020 Google LLC. All Rights Reserved.

#ifndef WHITEBOX_API_LICENSE_TEST_HELPER_H_
#define WHITEBOX_API_LICENSE_TEST_HELPER_H_

#include <stddef.h>
#include <stdint.h>
#include <string>
#include <vector>

namespace widevine {

// Returns a serialized LicenseRequest.
std::string CreateLicenseRequest();

// Returns a serialized License based on |serialized_license_request| and
// |session_key|. The license contains 3 content keys (key1, key2, key3)
// with levels SW_SECURE_CRYPTO, SW_SECURE_DECODE, and HW_SECURE_CRYPTO,
// respectively.
std::string CreateLicense(const std::string& serialized_license_request,
                          const std::string& session_key);

// Returns the key ID for each key.
std::vector<uint8_t> GetSoftwareCryptoKeyId();
std::vector<uint8_t> GetSoftwareDecodeKeyId();
std::vector<uint8_t> GetHardwareCryptoKeyId();

// Returns a key ID which is not included in the license.
std::vector<uint8_t> GetUnusedKeyId();

}  // namespace widevine

#endif  // WHITEBOX_API_LICENSE_TEST_HELPER_H_