services/network/udp_socket_test_util.h

// 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 SERVICES_NETWORK_UDP_SOCKET_TEST_UTIL_H_
#define SERVICES_NETWORK_UDP_SOCKET_TEST_UTIL_H_

#include <stdint.h>

#include <memory>
#include <vector>

#include "base/containers/span.h"
#include "base/macros.h"
#include "base/optional.h"
#include "base/run_loop.h"
#include "net/base/ip_endpoint.h"
#include "net/base/net_errors.h"
#include "services/network/public/interfaces/udp_socket.mojom.h"

namespace network {

namespace test {

// Helper functions to invoke the corresponding mojo APIs and wait for
// completion.
class UDPSocketTestHelper {
 public:
  // TODO(xunjieli): Make the helper takes in a UDPSocketPtr* and these into
  // member methods.
  static int OpenSync(mojom::UDPSocketPtr* socket,
                      net::AddressFamily address_family);

  static int ConnectSync(mojom::UDPSocketPtr* socket,
                         const net::IPEndPoint& remote_addr,
                         net::IPEndPoint* local_addr_out);

  static int BindSync(mojom::UDPSocketPtr* socket,
                      const net::IPEndPoint& local_addr,
                      net::IPEndPoint* local_addr_out);

  static int SendToSync(mojom::UDPSocketPtr* socket,
                        const net::IPEndPoint& remote_addr,
                        const std::vector<uint8_t>& input);

  static int SendSync(mojom::UDPSocketPtr* socket,
                      const std::vector<uint8_t>& input);

  static int SetSendBufferSizeSync(mojom::UDPSocketPtr* socket, size_t size);

  static int SetReceiveBufferSizeSync(mojom::UDPSocketPtr* socket, size_t size);
};

// An implementation of mojom::UDPSocketReceiver that records received results.
class UDPSocketReceiverImpl : public mojom::UDPSocketReceiver {
 public:
  struct ReceivedResult {
    ReceivedResult(int net_error_arg,
                   const base::Optional<net::IPEndPoint>& src_addr_arg,
                   base::Optional<std::vector<uint8_t>> data_arg);
    ReceivedResult(const ReceivedResult& other);
    ~ReceivedResult();

    int net_error;
    base::Optional<net::IPEndPoint> src_addr;
    base::Optional<std::vector<uint8_t>> data;
  };

  UDPSocketReceiverImpl();
  ~UDPSocketReceiverImpl() override;

  const std::vector<ReceivedResult>& results() const { return results_; }

  void WaitForReceivedResults(size_t count);

 private:
  void OnReceived(int32_t result,
                  const base::Optional<net::IPEndPoint>& src_addr,
                  base::Optional<base::span<const uint8_t>> data) override;
  std::unique_ptr<base::RunLoop> run_loop_;
  std::vector<ReceivedResult> results_;
  size_t expected_receive_count_;

  DISALLOW_COPY_AND_ASSIGN(UDPSocketReceiverImpl);
};

}  // namespace test

}  // namespace network

#endif  // SERVICES_NETWORK_UDP_SOCKET_TEST_UTIL_H_