transport_client_socket_test_util.h 1.3 KB

1234567891011121314151617181920212223242526272829303132333435363738394041
  1. // Copyright 2022 The Chromium Authors. All rights reserved.
  2. // Use of this source code is governed by a BSD-style license that can be
  3. // found in the LICENSE file.
  4. #ifndef NET_SOCKET_TRANSPORT_CLIENT_SOCKET_TEST_UTIL_H_
  5. #define NET_SOCKET_TRANSPORT_CLIENT_SOCKET_TEST_UTIL_H_
  6. #include <stdint.h>
  7. #include <string>
  8. #include "net/base/test_completion_callback.h"
  9. #include "net/socket/stream_socket.h"
  10. namespace net {
  11. class IOBuffer;
  12. // Sends a request from `socket` to `connected_socket`. Makes `connected_socket`
  13. // read the request and send a response.
  14. void SendRequestAndResponse(StreamSocket* socket,
  15. StreamSocket* connected_socket);
  16. // Reads `expected_bytes_read` bytes from `socket`. Returns the data
  17. // read as a string.
  18. std::string ReadDataOfExpectedLength(StreamSocket* socket,
  19. int expected_bytes_read);
  20. // Sends response from `socket`.
  21. void SendServerResponse(StreamSocket* socket);
  22. // `socket` reads `bytes_to_read` number of bytes into `buf`. Returns number of
  23. // bytes read.
  24. int DrainStreamSocket(StreamSocket* socket,
  25. IOBuffer* buf,
  26. uint32_t buf_len,
  27. uint32_t bytes_to_read,
  28. TestCompletionCallback* callback);
  29. } // namespace net
  30. #endif // NET_SOCKET_TRANSPORT_CLIENT_SOCKET_TEST_UTIL_H_