hammerd/mock_dbus_wrapper.h

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

#ifndef HAMMERD_MOCK_DBUS_WRAPPER_H_
#define HAMMERD_MOCK_DBUS_WRAPPER_H_

#include <string>
#include <vector>

#include <gmock/gmock.h>

#include "hammerd/dbus_wrapper.h"

namespace hammerd {

class MockDBusWrapper : public DBusWrapperInterface {
 public:
  MockDBusWrapper() = default;

  MOCK_METHOD(void, SendSignal, (const std::string&), (override));
  MOCK_METHOD(void,
              SendSignalWithArgHelper,
              (const std::string&, const std::vector<uint8_t>));

  void SendSignalWithArg(const std::string& signal_name,
                         const uint8_t* values,
                         size_t length) override {
    // We only care about the argument value, instead of the address.
    std::vector<uint8_t> arg(values, values + length);
    SendSignalWithArgHelper(signal_name, arg);
  }
};

}  // namespace hammerd
#endif  // HAMMERD_MOCK_DBUS_WRAPPER_H_