src/unit/membuf.cpp


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

/*
 * Copyright (C) 2007 Xagasoft, All rights reserved.
 *
 * This file is part of the libbu++ library and is released under the
 * terms of the license contained in the file LICENSE.
 */

#include "bu/unitsuite.h"
#include "bu/membuf.h"

class Unit : public Bu::UnitSuite
{
public:
	Unit()
	{
		setName("MemBuf");
		addTest( Unit::testWriteRead01 );
	}

	virtual ~Unit()
	{
	}

	void testWriteRead01()
	{
		Bu::MemBuf mb;
		unitTest( mb.write("ab", 2 ) == 2 );
		unitTest( mb.write("cde", 3 ) == 3 );
		unitTest( mb.write("FG", 2 ) == 2 );

		mb.setPos( 0 );

		char buf[8];
		buf[7] = '\0';
		unitTest( mb.read( buf, 7 ) == 7 );
		unitTest( !strncmp( buf, "abcdeFG", 7 ) );
		unitTest( mb.read( buf, 7 ) == 0 );
		mb.seek( -3 );
		unitTest( mb.read( buf, 7 ) == 3 );
		unitTest( !strncmp( buf, "eFG", 3 ) );
	}
};

int main( int argc, char *argv[] ){ return Unit().run( argc, argv ); }