OLD | NEW |
1 // Copyright (c) 2010 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2010 The Chromium Authors. All rights reserved. |
2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
4 | 4 |
5 #ifndef NET_SOCKET_SOCKET_TEST_UTIL_H_ | 5 #ifndef NET_SOCKET_SOCKET_TEST_UTIL_H_ |
6 #define NET_SOCKET_SOCKET_TEST_UTIL_H_ | 6 #define NET_SOCKET_SOCKET_TEST_UTIL_H_ |
7 #pragma once | 7 #pragma once |
8 | 8 |
9 #include <cstring> | 9 #include <cstring> |
10 #include <deque> | 10 #include <deque> |
(...skipping 150 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
161 | 161 |
162 // SocketDataProvider which responds based on static tables of mock reads and | 162 // SocketDataProvider which responds based on static tables of mock reads and |
163 // writes. | 163 // writes. |
164 class StaticSocketDataProvider : public SocketDataProvider { | 164 class StaticSocketDataProvider : public SocketDataProvider { |
165 public: | 165 public: |
166 StaticSocketDataProvider(); | 166 StaticSocketDataProvider(); |
167 StaticSocketDataProvider(MockRead* reads, size_t reads_count, | 167 StaticSocketDataProvider(MockRead* reads, size_t reads_count, |
168 MockWrite* writes, size_t writes_count); | 168 MockWrite* writes, size_t writes_count); |
169 virtual ~StaticSocketDataProvider(); | 169 virtual ~StaticSocketDataProvider(); |
170 | 170 |
171 // SocketDataProvider methods: | |
172 virtual MockRead GetNextRead(); | |
173 virtual MockWriteResult OnWrite(const std::string& data); | |
174 virtual void Reset(); | |
175 virtual void CompleteRead() {} | |
176 | |
177 // These functions get access to the next available read and write data. | 171 // These functions get access to the next available read and write data. |
178 const MockRead& PeekRead() const; | 172 const MockRead& PeekRead() const; |
179 const MockWrite& PeekWrite() const; | 173 const MockWrite& PeekWrite() const; |
180 // These functions get random access to the read and write data, for timing. | 174 // These functions get random access to the read and write data, for timing. |
181 const MockRead& PeekRead(size_t index) const; | 175 const MockRead& PeekRead(size_t index) const; |
182 const MockWrite& PeekWrite(size_t index) const; | 176 const MockWrite& PeekWrite(size_t index) const; |
183 size_t read_index() const { return read_index_; } | 177 size_t read_index() const { return read_index_; } |
184 size_t write_index() const { return write_index_; } | 178 size_t write_index() const { return write_index_; } |
185 size_t read_count() const { return read_count_; } | 179 size_t read_count() const { return read_count_; } |
186 size_t write_count() const { return write_count_; } | 180 size_t write_count() const { return write_count_; } |
187 | 181 |
188 bool at_read_eof() const { return read_index_ >= read_count_; } | 182 bool at_read_eof() const { return read_index_ >= read_count_; } |
189 bool at_write_eof() const { return write_index_ >= write_count_; } | 183 bool at_write_eof() const { return write_index_ >= write_count_; } |
190 | 184 |
| 185 // SocketDataProvider methods: |
| 186 virtual MockRead GetNextRead(); |
| 187 virtual MockWriteResult OnWrite(const std::string& data); |
| 188 virtual void Reset(); |
| 189 virtual void CompleteRead() {} |
| 190 |
191 private: | 191 private: |
192 MockRead* reads_; | 192 MockRead* reads_; |
193 size_t read_index_; | 193 size_t read_index_; |
194 size_t read_count_; | 194 size_t read_count_; |
195 MockWrite* writes_; | 195 MockWrite* writes_; |
196 size_t write_index_; | 196 size_t write_index_; |
197 size_t write_count_; | 197 size_t write_count_; |
198 | 198 |
199 DISALLOW_COPY_AND_ASSIGN(StaticSocketDataProvider); | 199 DISALLOW_COPY_AND_ASSIGN(StaticSocketDataProvider); |
200 }; | 200 }; |
(...skipping 739 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
940 | 940 |
941 extern const char kSOCKS5OkRequest[]; | 941 extern const char kSOCKS5OkRequest[]; |
942 extern const int kSOCKS5OkRequestLength; | 942 extern const int kSOCKS5OkRequestLength; |
943 | 943 |
944 extern const char kSOCKS5OkResponse[]; | 944 extern const char kSOCKS5OkResponse[]; |
945 extern const int kSOCKS5OkResponseLength; | 945 extern const int kSOCKS5OkResponseLength; |
946 | 946 |
947 } // namespace net | 947 } // namespace net |
948 | 948 |
949 #endif // NET_SOCKET_SOCKET_TEST_UTIL_H_ | 949 #endif // NET_SOCKET_SOCKET_TEST_UTIL_H_ |
OLD | NEW |