1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
|
// Copyright 2018 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#ifndef SERVICES_NETWORK_MOJO_SOCKET_TEST_UTIL_H_
#define SERVICES_NETWORK_MOJO_SOCKET_TEST_UTIL_H_
#include "base/run_loop.h"
#include "mojo/public/cpp/bindings/pending_remote.h"
#include "mojo/public/cpp/bindings/receiver.h"
#include "net/base/net_errors.h"
#include "services/network/public/mojom/tcp_socket.mojom.h"
namespace network {
// A mojom::SocketObserver implementation used in tests.
class TestSocketObserver : public mojom::SocketObserver {
public:
TestSocketObserver();
TestSocketObserver(const TestSocketObserver&) = delete;
TestSocketObserver& operator=(const TestSocketObserver&) = delete;
~TestSocketObserver() override;
// Returns a mojo pending remote. This can only be called once.
mojo::PendingRemote<mojom::SocketObserver> GetObserverRemote();
// Waits for Read and Write error. Returns the error observed.
int WaitForReadError();
int WaitForWriteError();
private:
// mojom::SocketObserver implementation.
void OnReadError(int net_error) override;
void OnWriteError(int net_error) override;
int read_error_ = net::OK;
int write_error_ = net::OK;
base::RunLoop read_loop_;
base::RunLoop write_loop_;
mojo::Receiver<mojom::SocketObserver> receiver_{this};
};
} // namespace network
#endif // SERVICES_NETWORK_MOJO_SOCKET_TEST_UTIL_H_
|
