hammerd/mock_fmap_utils.h

// Copyright 2017 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 HAMMERD_MOCK_FMAP_UTILS_H_
#define HAMMERD_MOCK_FMAP_UTILS_H_

#include <string>

#include <gmock/gmock.h>

#include "hammerd/fmap_utils.h"

namespace hammerd {

class MockFmap : public FmapInterface {
 public:
  MOCK_METHOD2(Find, int64_t(const uint8_t* image, unsigned int len));
  MOCK_METHOD2(FindArea,
               const fmap_area*(const fmap* fmap, const std::string& name));
};

}  // namespace hammerd
#endif  // HAMMERD_MOCK_FMAP_UTILS_H_