| Index: net/test/embedded_test_server/simple_connection_listener.h
|
| diff --git a/net/test/embedded_test_server/simple_connection_listener.h b/net/test/embedded_test_server/simple_connection_listener.h
|
| new file mode 100644
|
| index 0000000000000000000000000000000000000000..3370cf30238d01bc34994bb5144aec06d946e6f9
|
| --- /dev/null
|
| +++ b/net/test/embedded_test_server/simple_connection_listener.h
|
| @@ -0,0 +1,67 @@
|
| +// 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.
|
| +
|
| +#ifndef NET_TEST_EMBEDDED_TEST_SERVER_SIMPLE_CONNECTION_LISTENER_H_
|
| +#define NET_TEST_EMBEDDED_TEST_SERVER_SIMPLE_CONNECTION_LISTENER_H_
|
| +
|
| +#include "base/macros.h"
|
| +#include "base/memory/ref_counted.h"
|
| +#include "base/run_loop.h"
|
| +#include "net/test/embedded_test_server/embedded_test_server_connection_listener.h"
|
| +
|
| +namespace base {
|
| +class SequencedTaskRunner;
|
| +}
|
| +
|
| +namespace net {
|
| +
|
| +class StreamSocket;
|
| +
|
| +namespace test_server {
|
| +
|
| +// Waits for a specified number of connection attempts to be seen.
|
| +class SimpleConnectionListener : public EmbeddedTestServerConnectionListener {
|
| + public:
|
| + enum AllowAdditionalConnections {
|
| + // Add an expect failure if more than the specified number of connections
|
| + // are seen.
|
| + FAIL_ON_ADDITIONAL_CONNECTIONS,
|
| + // Silently ignores extra connection attempts.
|
| + ALLOW_ADDITIONAL_CONNECTIONS
|
| + };
|
| +
|
| + // A connection listener that waits for the specified number of total
|
| + // connections when WaitForConnections() is called. Must be created on a
|
| + // thread with a SingleThreadedTaskRunner.
|
| + SimpleConnectionListener(
|
| + int expected_connections,
|
| + AllowAdditionalConnections allow_additional_connections);
|
| +
|
| + // Must be torn down only after the EmbeddedTestServer it's attached to is
|
| + // shut down.
|
| + ~SimpleConnectionListener() override;
|
| +
|
| + void AcceptedSocket(const StreamSocket& socket) override;
|
| + void ReadFromSocket(const StreamSocket& socket, int rv) override;
|
| +
|
| + // Wait until the expected number of connections have been seen.
|
| + void WaitForConnections();
|
| +
|
| + private:
|
| + int seen_connections_ = 0;
|
| +
|
| + const int expected_connections_;
|
| + const AllowAdditionalConnections allow_additional_connections_;
|
| +
|
| + const scoped_refptr<base::SequencedTaskRunner> run_loop_task_runner_;
|
| +
|
| + base::RunLoop run_loop_;
|
| +
|
| + DISALLOW_COPY_AND_ASSIGN(SimpleConnectionListener);
|
| +};
|
| +
|
| +} // namespace test_server
|
| +} // namespace net
|
| +
|
| +#endif // NET_TEST_EMBEDDED_TEST_SERVER_SIMPLE_CONNECTION_LISTENER_H_
|
|
|