patchpanel/mock_multicast_counters_service.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 PATCHPANEL_MOCK_MULTICAST_COUNTERS_SERVICE_H_
#define PATCHPANEL_MOCK_MULTICAST_COUNTERS_SERVICE_H_

#include <map>
#include <optional>
#include <string>
#include <string_view>

#include <gmock/gmock.h>

#include "patchpanel/multicast_counters_service.h"
#include "patchpanel/shill_client.h"

namespace patchpanel {

class MockMulticastCountersService : public MulticastCountersService {
 public:
  explicit MockMulticastCountersService(Datapath* datapath);
  ~MockMulticastCountersService() override;

  MOCK_METHOD(void,
              OnPhysicalDeviceAdded,
              (const ShillClient::Device& device),
              (override));
  MOCK_METHOD((std::optional<std::map<CounterKey, uint64_t>>),
              GetCounters,
              (),
              (override));
};

}  // namespace patchpanel

#endif  // PATCHPANEL_MOCK_MULTICAST_COUNTERS_SERVICE_H_