blob: 4a0afe13359416f363f6ac1b709853c16caf625b (
plain)
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
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
|
// Copyright (c) 2012 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_BASE_TEST_COMPLETION_CALLBACK_H_
#define NET_BASE_TEST_COMPLETION_CALLBACK_H_
#include "base/compiler_specific.h"
#include "base/tuple.h"
#include "net/base/completion_callback.h"
#include "net/base/net_errors.h"
//-----------------------------------------------------------------------------
// completion callback helper
// A helper class for completion callbacks, designed to make it easy to run
// tests involving asynchronous operations. Just call WaitForResult to wait
// for the asynchronous operation to complete.
//
// NOTE: Since this runs a message loop to wait for the completion callback,
// there could be other side-effects resulting from WaitForResult. For this
// reason, this class is probably not ideal for a general application.
//
namespace net {
class IOBuffer;
namespace internal {
class TestCompletionCallbackBaseInternal {
public:
bool have_result() const { return have_result_; }
protected:
TestCompletionCallbackBaseInternal();
void DidSetResult();
void WaitForResult();
bool have_result_;
bool waiting_for_result_;
private:
DISALLOW_COPY_AND_ASSIGN(TestCompletionCallbackBaseInternal);
};
template <typename R>
class TestCompletionCallbackTemplate
: public TestCompletionCallbackBaseInternal {
public:
virtual ~TestCompletionCallbackTemplate() {}
R WaitForResult() {
TestCompletionCallbackBaseInternal::WaitForResult();
return result_;
}
R GetResult(R result) {
if (net::ERR_IO_PENDING != result)
return result;
return WaitForResult();
}
protected:
// Override this method to gain control as the callback is running.
virtual void SetResult(R result) {
result_ = result;
DidSetResult();
}
TestCompletionCallbackTemplate() : result_(R()) {}
R result_;
private:
DISALLOW_COPY_AND_ASSIGN(TestCompletionCallbackTemplate);
};
} // namespace internal
// Base class overridden by custom implementations of TestCompletionCallback.
typedef internal::TestCompletionCallbackTemplate<int>
TestCompletionCallbackBase;
typedef internal::TestCompletionCallbackTemplate<int64>
TestInt64CompletionCallbackBase;
class TestCompletionCallback : public TestCompletionCallbackBase {
public:
TestCompletionCallback();
virtual ~TestCompletionCallback();
const CompletionCallback& callback() const { return callback_; }
private:
const CompletionCallback callback_;
DISALLOW_COPY_AND_ASSIGN(TestCompletionCallback);
};
class TestInt64CompletionCallback : public TestInt64CompletionCallbackBase {
public:
TestInt64CompletionCallback();
virtual ~TestInt64CompletionCallback();
const Int64CompletionCallback& callback() const { return callback_; }
private:
const Int64CompletionCallback callback_;
DISALLOW_COPY_AND_ASSIGN(TestInt64CompletionCallback);
};
// Makes sure that the buffer is not referenced when the callback runs.
class ReleaseBufferCompletionCallback: public TestCompletionCallback {
public:
explicit ReleaseBufferCompletionCallback(IOBuffer* buffer);
virtual ~ReleaseBufferCompletionCallback();
private:
virtual void SetResult(int result) OVERRIDE;
IOBuffer* buffer_;
DISALLOW_COPY_AND_ASSIGN(ReleaseBufferCompletionCallback);
};
} // namespace net
#endif // NET_BASE_TEST_COMPLETION_CALLBACK_H_
|
