2 files changed, 28 insertions, 1 deletions

diff --git a/src/unitsuite.cpp b/src/unitsuite.cpp
index 2a28eb5..b61f486 100644
--- a/src/unitsuite.cpp
+++ b/src/unitsuite.cpp
@@ -1,6 +1,12 @@
 #include "unitsuite.h"
 
-Bu::UnitSuite::UnitSuite()
+Bu::UnitSuite::UnitSuite() :
+        iOptions( 0 )
+{
+}
+
+Bu::UnitSuite::UnitSuite( int iOptions ) :
+        iOptions( iOptions )
 {
 }
 
@@ -35,10 +41,23 @@ int Bu::UnitSuite::run( int argc, char *argv[] )
                                         e.str.getStr()
                                         );
                         }
+
+                        if( (iOptions & optStopOnError) )
+                                return 0;
+                }
+                catch( std::exception &e )
+                {
+                        printf("failed with unknown exception.  what: %s\n", e.what() );
+
+                        if( (iOptions & optStopOnError) )
+                                return 0;
                 }
                 catch( ... )
                 {
                         printf("failed with external exception.\n");
+
+                        if( (iOptions & optStopOnError) )
+                                return 0;
                 }
         }
 
diff --git a/src/unitsuite.h b/src/unitsuite.h
index 578b4cc..aed7659 100644
--- a/src/unitsuite.h
+++ b/src/unitsuite.h
@@ -49,6 +49,7 @@ namespace Bu
         {
         public:
                 UnitSuite();
+                UnitSuite( int iOptions );
                 virtual ~UnitSuite();
 
                 int run( int argc=0, char *argv[]=NULL );
@@ -68,6 +69,11 @@ namespace Bu
                         int nLine;
                         bool bFile;
                 };
+
+                enum
+                {
+                        optStopOnError          = 0x000001
+                };
                 
         protected:
                 void add( Test fTest, Bu::FString sName );
@@ -83,6 +89,8 @@ namespace Bu
                 typedef std::list<TestInfo> TestList;
                 TestList lTests;
                 FString sSuiteName;
+
+                int iOptions;
         };
 }