// Copyright 2018 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 SHILL_WIFI_MOCK_WIFI_SERVICE_H_
#define SHILL_WIFI_MOCK_WIFI_SERVICE_H_

#include <string>
#include <vector>

#include <gmock/gmock.h>

#include "shill/wifi/wifi_service.h"

namespace shill {

class MockWiFiService : public WiFiService {
 public:
  MockWiFiService(Manager* manager,
                  WiFiProvider* provider,
                  const std::vector<uint8_t>& ssid,
                  const std::string& mode,
                  const std::string& security,
                  bool hidden_ssid);
  ~MockWiFiService() override;

  MOCK_METHOD(void, Configure, (const KeyValueStore&, Error*), (override));
  MOCK_METHOD(void, SetFailure, (ConnectFailure), (override));
  MOCK_METHOD(void, SetFailureSilent, (ConnectFailure), (override));
  MOCK_METHOD(void, SetState, (ConnectState), (override));
  MOCK_METHOD(bool,
              AddEAPCertification,
              (const std::string&, size_t),
              (override));
  MOCK_METHOD(bool, HasRecentConnectionIssues, (), (override));
  MOCK_METHOD(bool, AddSuspectedCredentialFailure, (), (override));
  MOCK_METHOD(void, ResetSuspectedCredentialFailures, (), (override));
  MOCK_METHOD(void, AddEndpoint, (const WiFiEndpointConstRefPtr&), (override));
  MOCK_METHOD(void,
              RemoveEndpoint,
              (const WiFiEndpointConstRefPtr&),
              (override));
  MOCK_METHOD(void,
              NotifyCurrentEndpoint,
              (const WiFiEndpointConstRefPtr&),
              (override));
  MOCK_METHOD(void,
              NotifyEndpointUpdated,
              (const WiFiEndpointConstRefPtr&),
              (override));
  MOCK_METHOD(void,
              DisconnectWithFailure,
              (ConnectFailure, Error*, const char*),
              (override));
  MOCK_METHOD(bool, IsActive, (Error*), (override));
  MOCK_METHOD(bool, IsConnected, (Error*), (const, override));
  MOCK_METHOD(bool, IsConnecting, (), (const, override));
  MOCK_METHOD(int, GetEndpointCount, (), (const, override));
  MOCK_METHOD(bool, HasEndpoints, (), (const, override));
  MOCK_METHOD(bool, IsRemembered, (), (const, override));
  MOCK_METHOD(void, ResetWiFi, (), (override));
  MOCK_METHOD(KeyValueStore,
              GetSupplicantConfigurationParameters,
              (),
              (const, override));
  MOCK_METHOD(bool, IsAutoConnectable, (const char**), (const, override));
  MOCK_METHOD(bool, HasStaticIPAddress, (), (const, override));

 private:
  DISALLOW_COPY_AND_ASSIGN(MockWiFiService);
};

}  // namespace shill

#endif  // SHILL_WIFI_MOCK_WIFI_SERVICE_H_