libhwsec/frontend/client/mock_frontend.h

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

#ifndef LIBHWSEC_FRONTEND_CLIENT_MOCK_FRONTEND_H_
#define LIBHWSEC_FRONTEND_CLIENT_MOCK_FRONTEND_H_

#include <optional>
#include <vector>

#include <brillo/secure_blob.h>
#include <gmock/gmock.h>

#include "libhwsec/backend/vendor.h"
#include "libhwsec/frontend/client/frontend.h"
#include "libhwsec/frontend/mock_frontend.h"

namespace hwsec {

class MockClientFrontend : public MockFrontend, public ClientFrontend {
 public:
  MockClientFrontend() = default;
  ~MockClientFrontend() override = default;

  MOCK_METHOD(StatusOr<brillo::Blob>,
              GetRandomBlob,
              (size_t size),
              (const override));
  MOCK_METHOD(StatusOr<bool>, IsSrkRocaVulnerable, (), (const override));
  MOCK_METHOD(StatusOr<uint32_t>, GetFamily, (), (const override));
  MOCK_METHOD(StatusOr<uint64_t>, GetSpecLevel, (), (const override));
  MOCK_METHOD(StatusOr<uint32_t>, GetManufacturer, (), (const override));
  MOCK_METHOD(StatusOr<uint32_t>, GetTpmModel, (), (const override));
  MOCK_METHOD(StatusOr<uint64_t>, GetFirmwareVersion, (), (const override));
  MOCK_METHOD(StatusOr<Vendor::RwVersion>, GetRwVersion, (), (const override));
  MOCK_METHOD(StatusOr<brillo::Blob>, GetVendorSpecific, (), (const override));
  MOCK_METHOD(StatusOr<IFXFieldUpgradeInfo>,
              GetIFXFieldUpgradeInfo,
              (),
              (const override));
};

}  // namespace hwsec

#endif  // LIBHWSEC_FRONTEND_CLIENT_MOCK_FRONTEND_H_