rmad/utils/mock_vpd_utils.h

// Copyright 2021 The Chromium OS Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#ifndef RMAD_UTILS_MOCK_VPD_UTILS_H_
#define RMAD_UTILS_MOCK_VPD_UTILS_H_

#include "rmad/utils/vpd_utils.h"

#include <string>
#include <vector>

#include <gmock/gmock.h>

namespace rmad {

class MockVpdUtils : public VpdUtils {
 public:
  MockVpdUtils() = default;
  ~MockVpdUtils() override = default;

  MOCK_METHOD(bool, GetSerialNumber, (std::string*), (const, override));
  MOCK_METHOD(bool, GetWhitelabelTag, (std::string*), (const, override));
  MOCK_METHOD(bool, GetRegion, (std::string*), (const, override));
  MOCK_METHOD(bool,
              GetCalibbias,
              (const std::vector<std::string>&, std::vector<int>*),
              (const, override));
  MOCK_METHOD(bool, SetSerialNumber, (const std::string&), (override));
  MOCK_METHOD(bool, SetWhitelabelTag, (const std::string&), (override));
  MOCK_METHOD(bool, SetRegion, (const std::string&), (override));
  MOCK_METHOD(bool,
              SetCalibbias,
              (const std::vector<std::string>&, const std::vector<int>&),
              (override));

 protected:
  MOCK_METHOD(bool,
              SetRoVpd,
              (const std::string&, const std::string&),
              (override));
  MOCK_METHOD(bool,
              GetRoVpd,
              (const std::string&, std::string*),
              (const, override));
  MOCK_METHOD(bool,
              SetRwVpd,
              (const std::string&, const std::string&),
              (override));
  MOCK_METHOD(bool,
              GetRwVpd,
              (const std::string&, std::string*),
              (const, override));
};

}  // namespace rmad

#endif  // RMAD_UTILS_MOCK_VPD_UTILS_H_