1 files changed, 36 insertions, 0 deletions
diff --git a/tests/UnitTests/framework/common.php b/tests/UnitTests/framework/common.php
index 71d6949e..bcff7a89 100644
--- a/tests/UnitTests/framework/common.php
+++ b/tests/UnitTests/framework/common.php
@@ -23,4 +23,40 @@ function __autoload($className)
 error_reporting(E_ALL);
 restore_error_handler();
 
+
+/**
+ * PradoTestCase class.
+ *
+ * Extends the simpletest UnitTestCase class to provide some fairly generic extra functionality.
+ *
+ * @author Alex Flint <alex@linium.net>
+ */
+ 
+class PradoUnitTestCase extends UnitTestCase {
+	
+	/**
+	 * Tests whether the given code results in an appropriate exception being raised.
+	 * @param string the PHP code to execute. must end with a semi-colon.
+	 * @param string the type of exception that should be raised.
+	 * @return boolean true
+	 */
+	public function assertException(string $code, string $exception) {
+		$pass = false;
+		$code = "
+try {
+	$code
+} catch ($exception \$e) {
+	\$pass = true;
+}";
+		eval($code);
+		if ($pass) {
+			$this->pass();
+		} else {
+			$this->fail("Code did not produce correct exception (wanted $exception, got something else");
+		}
+	}
+}
+		
+		
+
 ?>
 \ No newline at end of file