tlbmc/hal/system_registry_mock.h

#ifndef THIRD_PARTY_MILOTIC_EXTERNAL_CC_TLBMC_HAL_SYSTEM_REGISTRY_MOCK_H_
#define THIRD_PARTY_MILOTIC_EXTERNAL_CC_TLBMC_HAL_SYSTEM_REGISTRY_MOCK_H_

#include <cstdint>
#include <string>
#include <string_view>

#include <gmock/gmock.h>
#include "absl/status/status.h"
#include "absl/status/statusor.h"
#include "tlbmc/hal/system_registry.h"

namespace milotic_tlbmc {

class MockSystemRegistry : public SystemRegistry {
 public:
  MOCK_METHOD(absl::StatusOr<uint32_t>, GetCoreCount, (std::string_view),
              (override));
  MOCK_METHOD(absl::Status, SetCoreCount, (std::string_view, uint32_t),
              (override));
  MOCK_METHOD(absl::Status, SetInitialCoreCounts, (), (override));
  MOCK_METHOD(absl::StatusOr<std::string>, GetBase64EncodedIfsPattern,
              (std::string_view), (override));
  MOCK_METHOD(absl::Status, SetIfsPattern, (std::string_view, std::string_view),
              (override));
  MOCK_METHOD(absl::Status, ResetIfsPattern, (std::string_view), (override));
  MOCK_METHOD(absl::StatusOr<std::string>, GetBootNext, (std::string_view),
              (override));
  MOCK_METHOD(absl::Status, SetBootNext, (std::string_view, std::string_view),
              (override));
  MOCK_METHOD(absl::StatusOr<uint32_t>, GetMemoryCapacityGib,
              (std::string_view), (override));
  MOCK_METHOD(absl::Status, SetMemoryCapacityGib, (std::string_view, uint32_t),
              (override));
};

}  // namespace milotic_tlbmc

#endif  // THIRD_PARTY_MILOTIC_EXTERNAL_CC_TLBMC_HAL_SYSTEM_REGISTRY_MOCK_H_