Chromium Code Reviews| 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..c895e343522a5dcbdc4fc41bf5f74cec29e326c9 |
| --- /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 specificed number of connection attempts to be seen. |
|
eroman
2017/05/16 20:21:20
specified
mmenke
2017/05/17 02:29:58
Done.
|
| +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); |
| +}; |
| + |
| +} // test_server |
| +} // net |
|
eroman
2017/05/16 20:21:20
// namespace net
mmenke
2017/05/17 02:29:58
Done.
|
| + |
| +#endif // NET_TEST_EMBEDDED_TEST_SERVER_SIMPLE_CONNECTION_LISTENER_H_ |