1// Copyright (c) 2011 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
5#ifndef NET_BASE_TEST_DATA_STREAM_H_
6#define NET_BASE_TEST_DATA_STREAM_H_
7
8#include <string.h>  // for memcpy().
9#include <algorithm>
10#include "net/base/net_export.h"
11
12// This is a class for generating an infinite stream of data which can be
13// verified independently to be the correct stream of data.
14
15namespace net {
16
17class NET_EXPORT TestDataStream {
18 public:
19  TestDataStream();
20
21  // Fill |buffer| with |length| bytes of data from the stream.
22  void GetBytes(char* buffer, int length);
23
24  // Verify that |buffer| contains the expected next |length| bytes from the
25  // stream.  Returns true if correct, false otherwise.
26  bool VerifyBytes(const char *buffer, int length);
27
28  // Resets all the data.
29  void Reset();
30
31 private:
32  // If there is no data spilled over from the previous index, advance the
33  // index and fill the buffer.
34  void AdvanceIndex();
35
36  // Consume data from the spill buffer.
37  void Consume(int bytes);
38
39  int index_;
40  int bytes_remaining_;
41  char buffer_[16];
42  char* buffer_ptr_;
43};
44
45}  // namespace net
46
47#endif  // NET_BASE_TEST_DATA_STREAM_H_
48