#pragma once

#include "envoy/server/worker.h"

#include "absl/strings/string_view.h"

#include "gmock/gmock.h"

namespace Envoy {

namespace Server {

class MockWorker : public Worker {

public:

  MockWorker();

  ~MockWorker() override;

  void callAddCompletion() {

    EXPECT_NE(nullptr, add_listener_completion_);

    add_listener_completion_();

    add_listener_completion_ = nullptr;

  }

  void callRemovalCompletion() {

    EXPECT_NE(nullptr, remove_listener_completion_);

    remove_listener_completion_();

    remove_listener_completion_ = nullptr;

  }

  void callDrainFilterChainsComplete() {

    EXPECT_NE(nullptr, remove_filter_chains_completion_);

    remove_filter_chains_completion_();

    remove_filter_chains_completion_ = nullptr;

  }

  // Server::Worker

  MOCK_METHOD(void, addListener,

              (absl::optional<uint64_t> overridden_listener, Network::ListenerConfig& listener,

               AddListenerCompletion completion, Runtime::Loader&, Random::RandomGenerator&));

  MOCK_METHOD(uint64_t, numConnections, (), (const));

  MOCK_METHOD(void, removeListener,

              (Network::ListenerConfig & listener, std::function<void()> completion));

  MOCK_METHOD(void, start, (OptRef<GuardDog> guard_dog, const std::function<void()>& cb));

  MOCK_METHOD(void, initializeStats, (Stats::Scope & scope));

  MOCK_METHOD(void, stop, ());

  MOCK_METHOD(void, stopListener,

              (Network::ListenerConfig & listener,

               const Network::ExtraShutdownListenerOptions& options,

               std::function<void()> completion));

  MOCK_METHOD(void, removeFilterChains,

              (uint64_t listener_tag, const std::list<const Network::FilterChain*>& filter_chains,

               std::function<void()> completion));

  AddListenerCompletion add_listener_completion_;

  std::function<void()> remove_listener_completion_;

  std::function<void()> remove_filter_chains_completion_;

};

} // namespace Server

} // namespace Envoy