minios/mock_cgpt_util.h

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

#ifndef MINIOS_MOCK_CGPT_UTIL_H_
#define MINIOS_MOCK_CGPT_UTIL_H_

#include <string>

#include <gmock/gmock.h>

#include "minios/cgpt_util.h"

namespace minios {

class MockCgptUtil : public CgptUtilInterface {
 public:
  MockCgptUtil() = default;
  ~MockCgptUtil() override = default;

  MockCgptUtil(const MockCgptUtil&) = delete;
  MockCgptUtil& operator=(const MockCgptUtil&) = delete;

  MOCK_METHOD(std::optional<int>,
              GetPartitionNumber,
              (const std::string& label),
              (const, override));
  MOCK_METHOD(std::optional<uint64_t>,
              GetSize,
              (const uint32_t partition_number),
              (const, override));
};

}  // namespace minios

#endif  // MINIOS_MOCK_CGPT_UTIL_H_