cyclicbuffertestcase.cpp

功能非常强的写日志程序

/*
 * Copyright 2003,2004 The Apache Software Foundation.
 * 
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 * 
 *      http://www.apache.org/licenses/LICENSE-2.0
 * 
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

#include <cppunit/TestFixture.h>
#include <cppunit/extensions/HelperMacros.h>

#include <log4cxx/logmanager.h>
#include <log4cxx/logger.h>
#include <log4cxx/helpers/cyclicbuffer.h>
#include <log4cxx/spi/loggingevent.h>

using namespace log4cxx;
using namespace log4cxx::helpers;
using namespace log4cxx::spi;

#define MAX 1000

class CyclicBufferTestCase : public CppUnit::TestFixture
{
	CPPUNIT_TEST_SUITE(CyclicBufferTestCase);
		CPPUNIT_TEST(test0);
		CPPUNIT_TEST(test1);
		CPPUNIT_TEST(testResize);
	CPPUNIT_TEST_SUITE_END();
	
	LoggerPtr logger;
	std::vector<LoggingEventPtr> e;
	
public:
	void setUp()
	{
  		logger = Logger::getLogger(_T("x"));
		e.reserve(1000);
		for (int i = 0; i < MAX; i++)
		{
			e.push_back(
			new LoggingEvent(_T(""), logger, Level::DEBUG, _T("e")));
		}
	}

	void tearDown()
	{
		LogManager::shutdown();
	}
	
	void test0()
	{
		int size = 2;

		CyclicBuffer cb(size);
		CPPUNIT_ASSERT_EQUAL(size, cb.getMaxSize());

		cb.add(e[0]);
		CPPUNIT_ASSERT_EQUAL(1, cb.length());
		CPPUNIT_ASSERT_EQUAL(e[0], cb.get());
		CPPUNIT_ASSERT_EQUAL(0, cb.length());
		CPPUNIT_ASSERT(cb.get() == 0);
		CPPUNIT_ASSERT_EQUAL(0, cb.length());

		CyclicBuffer cb2(size);
		cb2.add(e[0]);
		cb2.add(e[1]);
		CPPUNIT_ASSERT_EQUAL(2, cb2.length());
		CPPUNIT_ASSERT_EQUAL(e[0], cb2.get());
		CPPUNIT_ASSERT_EQUAL(1, cb2.length());
		CPPUNIT_ASSERT_EQUAL(e[1], cb2.get());
		CPPUNIT_ASSERT_EQUAL(0, cb2.length());
		CPPUNIT_ASSERT(cb2.get() == 0);
		CPPUNIT_ASSERT_EQUAL(0, cb2.length());
	}
	
	void test1()
	{
		for (int bufSize = 1; bufSize <= 128; bufSize *= 2)
			doTest1(bufSize);
	}
	
	void doTest1(int size)
	{
		//System.out.println("Doing test with size = "+size);
		CyclicBuffer cb(size);

		CPPUNIT_ASSERT_EQUAL(size, cb.getMaxSize());
		
		int i;

		for (i = -(size + 10); i < (size + 10); i++)
		{
			CPPUNIT_ASSERT(cb.get(i) == 0);
		}

		for (i = 0; i < MAX; i++)
		{
			cb.add(e[i]);

			int limit = (i < (size - 1)) ? i : (size - 1);

			//System.out.println("\nLimit is " + limit + ", i="+i);
			for (int j = limit; j >= 0; j--) 
			{
				//System.out.println("i= "+i+", j="+j);
				CPPUNIT_ASSERT_EQUAL(e[i - (limit - j)], cb.get(j));
			}

			CPPUNIT_ASSERT(cb.get(-1) == 0);
			CPPUNIT_ASSERT(cb.get(limit + 1) == 0);
		}
	}
	
	void testResize()
	{
		for (int isize = 1; isize <= 128; isize *= 2)
		{
			doTestResize(isize, (isize / 2) + 1, (isize / 2) + 1);
			doTestResize(isize, (isize / 2) + 1, isize + 10);
			doTestResize(isize, isize + 10, (isize / 2) + 1);
			doTestResize(isize, isize + 10, isize + 10);
		}
	
	}
	
	void doTestResize(int initialSize, int numberOfAdds, int newSize)
	{
		//System.out.println("initialSize = "+initialSize+", numberOfAdds="
		//	       +numberOfAdds+", newSize="+newSize);
		CyclicBuffer cb(initialSize);

		for (int i = 0; i < numberOfAdds; i++)
		{
			cb.add(e[i]);
		}

		cb.resize(newSize);

		int offset = numberOfAdds - initialSize;

		if (offset < 0)
		{
			offset = 0;
		}

		int len = (newSize < numberOfAdds) ? newSize : numberOfAdds;
		len = (len < initialSize) ? len : initialSize;

		//System.out.println("Len = "+len+", offset="+offset);
		for (int j = 0; j < len; j++)
		{
			CPPUNIT_ASSERT_EQUAL(e[offset + j], cb.get(j));
		}
	}		
};

CPPUNIT_TEST_SUITE_REGISTRATION(CyclicBufferTestCase);