From 469bbcf0701e1eb8a6670c23145b0da87357e178 Mon Sep 17 00:00:00 2001
From: Mike Buland <eichlan@xagasoft.com>
Date: Sun, 25 Mar 2012 20:00:08 +0000
Subject: Code is all reorganized.  We're about ready to release.  I should
 write up a little explenation of the arrangement.

---
 src/stable/unitsuite.h | 139 +++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 139 insertions(+)
 create mode 100644 src/stable/unitsuite.h

(limited to 'src/stable/unitsuite.h')

diff --git a/src/stable/unitsuite.h b/src/stable/unitsuite.h
new file mode 100644
index 0000000..2250a4d
--- /dev/null
+++ b/src/stable/unitsuite.h
@@ -0,0 +1,139 @@
+/*
+ * Copyright (C) 2007-2011 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.
+ */
+
+#ifndef BU_UNIT_SUITE_H
+#define BU_UNIT_SUITE_H
+
+#include <stdint.h>
+#include "bu/list.h"
+#include "bu/string.h"
+#include "bu/file.h"
+#include "bu/array.h"
+
+namespace Bu
+{
+	/**
+	 * Provides a unit testing framework.  This is pretty easy to use, probably
+	 * the best way to get started is to use ch to generate a template, or use
+	 * the code below with appropriate tweaks:
+	 *@code
+	 * #include "unitsuite.h"
+	 * 
+	 * class Unit : public Bu::UnitSuite
+	 * {
+	 * public:
+	 * 	Unit()
+	 * 	{
+	 * 		setName("Example");
+	 * 		addTest( Unit::test );
+	 * 	}
+	 * 
+	 * 	virtual ~Unit() { }
+	 * 
+	 * 	//
+	 * 	// Tests go here
+	 * 	//
+	 * 	void test()
+	 * 	{
+	 * 		unitTest( 1 == 1 );
+	 * 	}
+	 * };
+	 * 
+	 * int main( int argc, char *argv[] )
+	 * {
+	 * 	return Unit().run( argc, argv );
+	 * }
+	 *
+	 @endcode
+	 * The main function can contain other things, but using this one exactly
+	 * makes all of the test suites work exactly the same.  Using the optional
+	 * setName at the top of the constructor replaces the class name with the
+	 * chosen name when printing out stats and info.
+	 */
+	class UnitSuite
+	{
+	public:
+		UnitSuite();
+		UnitSuite( int iOptions );
+		virtual ~UnitSuite();
+
+		int run( int argc=0, char *argv[]=NULL );
+
+		Bu::File tempFile( Bu::String &sFileName );
+
+		typedef void (UnitSuite::*Test)();
+
+		class Failed
+		{
+		public:
+			Failed() : str(""), bFile( false ) { }
+			Failed( const String &s ) : str( s ), bFile( false ) { }
+			Failed( const String &s, const String &sFile, int nLine ) :
+				str( s ), sFile( sFile ), nLine( nLine ), bFile( true ) { }
+
+			String str;
+			String sFile;
+			int nLine;
+			bool bFile;
+		};
+
+		enum
+		{
+			optStopOnError		= 0x000001
+		};
+
+		enum Expect
+		{
+			expectPass,
+			expectFail
+		};
+		
+	protected:
+		void add( Test fTest, const Bu::String &sName, Expect e=expectPass );
+		void setName( const String &sName );
+
+		void dispProgress();
+		void setStepCount( int iSteps );
+		void incProgress( int iAmnt = 1 );
+		void setProgress( int iAmnt );
+
+	private:
+		int onListCases( Bu::Array<Bu::String> aParam );
+
+	private:
+		typedef struct TestInfo
+		{
+			String sName;
+			Test fTest;
+			Expect eExpect;
+		} TestInfo;
+
+		typedef Bu::List<TestInfo> TestList;
+		TestList lTests;
+		String sSuiteName;
+
+		int iOptions;
+
+		typedef Bu::List<Bu::String> StrList;
+		StrList lFileCleanup;
+		int iNameWidth;
+		int iStepCount;
+		int iProgress;
+		time_t tLastUpdate;
+	};
+
+Bu::Formatter &operator<<( Bu::Formatter &f, const Bu::UnitSuite::Expect &e );
+}
+
+#define addTest( fn ) add( static_cast<Bu::UnitSuite::Test>(&fn), #fn )
+#define unitTest( tst ) if( !(tst) )							\
+{																\
+	throw Bu::UnitSuite::Failed( #tst, __FILE__, __LINE__ );	\
+} else (void)0
+#define unitFailed( msg ) throw Bu::UnitSuite::Failed(msg, __FILE__, __LINE__)
+
+#endif
-- 
cgit v1.2.3