1 files changed, 8 insertions, 2 deletions
diff --git a/examples/cppunittest/TestSetUpTest.cpp b/examples/cppunittest/TestSetUpTest.cpp
index 3f8f7dd..20c4f24 100644
--- a/examples/cppunittest/TestSetUpTest.cpp
+++ b/examples/cppunittest/TestSetUpTest.cpp
@@ -1,6 +1,8 @@
 #include "ExtensionSuite.h"
 #include "TestSetUpTest.h"
 #include <cppunit/TestResult.h>
+#include "MockTestCase.h"
+
 
 CPPUNIT_TEST_SUITE_NAMED_REGISTRATION( TestSetUpTest,
                                        extensionSuiteName() );
@@ -32,10 +34,14 @@ void
 TestSetUpTest::testRun()
 {
   CPPUNIT_NS::TestResult result;
-  CPPUNIT_NS::TestCase test;
-  MockSetUp setUpTest( &test );
+  MockTestCase *test = new MockTestCase( "TestSetUpTest" );
+  test->setExpectedSetUpCall();
+  test->setExpectedRunTestCall();
+  test->setExpectedTearDownCall();
+  MockSetUp setUpTest( test );
   
   setUpTest.run( &result );
 
   setUpTest.verify();
+  test->verify();
 }