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_ |