net/test/embedded_test_server/simple_connection_listener.cc

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

#include "net/test/embedded_test_server/simple_connection_listener.h"

#include "base/location.h"
#include "net/socket/stream_socket.h"
#include "testing/gtest/include/gtest/gtest.h"

namespace net {
namespace test_server {

SimpleConnectionListener::SimpleConnectionListener(
    int expected_connections,
    AllowAdditionalConnections allow_additional_connections)
    : expected_connections_(expected_connections),
      allow_additional_connections_(allow_additional_connections) {}

SimpleConnectionListener::~SimpleConnectionListener() = default;

std::unique_ptr<StreamSocket> SimpleConnectionListener::AcceptedSocket(
    std::unique_ptr<StreamSocket> socket) {
  ++seen_connections_;
  if (allow_additional_connections_ != ALLOW_ADDITIONAL_CONNECTIONS)
    EXPECT_LE(seen_connections_, expected_connections_);
  if (seen_connections_ == expected_connections_)
    run_loop_.Quit();
  return socket;
}

void SimpleConnectionListener::ReadFromSocket(const StreamSocket& socket,
                                              int rv) {}

void SimpleConnectionListener::WaitForConnections() {
  run_loop_.Run();
}

void SimpleConnectionListener::OnResponseCompletedSuccessfully(
    std::unique_ptr<StreamSocket> socket) {}

}  // namespace test_server
}  // namespace net