// $Id: tut_reporter.h 1820 2006-09-06 16:54:23Z mloskot $
// 
// TUT: C++ Unit Test Framework
// http://tut-framework.sourceforge.net/
//
// NOTE: This file is a customized version of the official tut_reporter.h.
// The TUT's version of tut_reporter.h file is only a sample of
// how to write TUT visualizer.
//
#ifndef TUT_REPORTER
#define TUT_REPORTER

#include <tut.h>

/**
* Template Unit Tests Framework for C++.
* http://tut.dozen.ru
*
* @author dozen, tut@dozen.ru
*/
namespace
{
	std::ostream& operator << (std::ostream& os,const tut::test_result& tr)
	{
		switch(tr.result)
		{
		case tut::test_result::ok: 
			os << '.'; 
			break;

		case tut::test_result::fail: 
			os << '[' << tr.test << "=F]";
			break;

		case tut::test_result::ex_ctor: 
			os << '[' << tr.test << "=C]";
			break;

		case tut::test_result::ex: 
			os << '[' << tr.test << "=X]";
			break;

		case tut::test_result::warn: 
			os << '[' << tr.test << "=W]";
			break;

		case tut::test_result::term: 
			os << '[' << tr.test << "=T]";
			break;
		case tut::test_result::inform: 
			break; // do nothing
		}

		return os;
	}
} // namespace

namespace tut
{
	/**
	* Default TUT callback handler.
	*/
	class reporter : public tut::callback
	{
		std::string current_group;
		typedef std::vector<tut::test_result> not_passed_list;
		not_passed_list not_passed;
		std::ostream& os;

	public:
		int ok_count;
		int ok_group_count;
		int exceptions_count;
		int failures_count;
		int terminations_count;
		int warnings_count;
		int informs_count;

		reporter() : os(std::cout)
		{
			init();
		}

		reporter(std::ostream& out) : os(out)
		{
			init();
		}

		void run_started()
		{
			init();
		}

		void test_completed(const tut::test_result& tr)
		{
			if( tr.group != current_group )
			{
				os << std::endl << tr.group << ": " << std::flush;
				current_group = tr.group;
			}

			os << tr << std::flush;
			
			// inform result is neutral: not failure naither success
			if (tr.result == tut::test_result::inform)
			{
				informs_count++;
			}

			// Treat inform test results as passed
			if( tr.result == tut::test_result::ok
				|| tr.result == tut::test_result::inform )
			{
				ok_group_count++;
				ok_count++;
			}
			else if( tr.result == tut::test_result::ex )
				exceptions_count++;
			else if( tr.result == tut::test_result::ex_ctor )
				exceptions_count++;
			else if( tr.result == tut::test_result::fail )
				failures_count++;
			else if( tr.result == tut::test_result::warn )
				warnings_count++;
			else
				terminations_count++;

			if( tr.result != tut::test_result::ok )
			{
				not_passed.push_back(tr);
			}
		}

		void group_started(const std::string& name)
		{
			ok_group_count = 0;
		}

		void group_completed(const std::string& name)
		{
			os << " " << ok_group_count << std::endl;
		}

		void run_completed()
		{
			os << std::endl;

			if( not_passed.size() > 0 )
			{
				not_passed_list::const_iterator i = not_passed.begin();
				while( i != not_passed.end() )
				{
					tut::test_result tr = *i;

					os << std::endl;
					os << "---> " << "group: " << tr.group
						<< ", test: test<" << tr.test
						<< ">" << std::endl;
					os << "     problem: ";

					switch(tr.result)
					{
					case test_result::fail: 
						os << "assertion failed" << std::endl; 
						break;
					case test_result::ex: 
					case test_result::ex_ctor: 
						os << "unexpected exception" << std::endl;
						if( tr.exception_typeid != "" )
						{ 
							os << "     exception typeid: " 
								<< tr.exception_typeid << std::endl;
						}
						break;
					case test_result::term: 
						os << "would be terminated" << std::endl; 
						break;
					case test_result::warn: 
						os << "test passed, but cleanup code (destructor) \
								raised an exception" << std::endl;
						break;
					case test_result::inform: 
						os << "test sends neutral information message" << std::endl; 
						break;
					default:
						break;
					}

					if( tr.message != "" )
					{
						if( tr.result == test_result::fail )
						{
							os << "     failed assertion: \""
								<< tr.message << "\"" << std::endl;
						}
						else
						{
							os << "     message: \""
								<< tr.message << "\"" << std::endl;
						}
					}

					++i;
				}
			}

			os << std::endl;
			os << "Tests summary:" << std::endl;
			if( terminations_count > 0 )
				os << " - terminations:" << terminations_count << std::endl;
			if( exceptions_count > 0 )
				os << " - exceptions:" << exceptions_count << std::endl;
			if( failures_count > 0 )
				os << " - failures:" << failures_count << std::endl;
			if( warnings_count > 0 )
				os << " - warnings:" << warnings_count << std::endl;
			if( informs_count > 0 )
				os << " - informing:" << informs_count << std::endl;
			
			os << " - passed: " << ok_count;
			os << std::endl;
		}

		bool all_ok() const
		{
			return not_passed.size() == 0;
		}

		int get_failures_count() const
		{
			return failures_count;
		}

	private:
		void init()
		{
			ok_count = 0;
			ok_group_count = 0;
			exceptions_count = 0;  
			failures_count = 0;
			terminations_count = 0;
			warnings_count = 0;
			informs_count = 0;

			not_passed.clear();
		}    
	};

} // namespace tut

#endif // #ifndef TUT_REPORTER