aboutsummaryrefslogtreecommitdiff
path: root/test/common/bufferedreadstream.h
blob: c10fbf4b1d0d20e8ade7227f1f691689b3c5b908 (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
#include <cxxtest/TestSuite.h>

#include "common/memstream.h"
#include "common/bufferedstream.h"

class BufferedReadStreamTestSuite : public CxxTest::TestSuite {
	public:
	void test_traverse() {
		byte contents[10] = { 0, 1, 2, 3, 4, 5, 6, 7, 8, 9 };
		Common::MemoryReadStream ms(contents, 10);

		// Use a buffer size of 4 -- note that 10 % 4 != 0,
		// so we test what happens if the cache can't be completely
		// refilled.
		Common::ReadStream &srs = *Common::wrapBufferedReadStream(&ms, 4, DisposeAfterUse::NO);

		byte i, b;
		for (i = 0; i < 10; ++i) {
			TS_ASSERT(!srs.eos());

			b = srs.readByte();
			TS_ASSERT_EQUALS(i, b);
		}

		TS_ASSERT(!srs.eos());

		b = srs.readByte();

		TS_ASSERT(srs.eos());

		delete &srs;
	}

	void test_traverse2() {
		byte contents[9] = { 0, 1, 2, 3, 4, 5, 6, 7, 8 };
		Common::MemoryReadStream ms(contents, 9);

		Common::ReadStream &brs = *Common::wrapBufferedReadStream(&ms, 4, DisposeAfterUse::NO);

		// Traverse the stream with reads of 2 bytes. The size is not
		// a multiple of 2, so we can test the final partial read.

		byte i, b[2];
		for (i = 0; i < 4; ++i) {
			TS_ASSERT(!brs.eos());

			int n = brs.read(b, 2);
			TS_ASSERT_EQUALS(n, 2);
		}

		TS_ASSERT(!brs.eos());

		int n = brs.read(b, 2);
		TS_ASSERT_EQUALS(n, 1);

		TS_ASSERT(brs.eos());

		delete &brs;
	}
};