Colorized console output

examples/run_tests.php

@@ -7,5 +7,9 @@
 
 $suite->require_all(dirname(__FILE__) . '/test');
 $suite->auto_fill();
-$suite->run(new ztest\ConsoleReporter);
+
+$reporter = new ztest\ConsoleReporter;
+$reporter->enable_color();
+
+$suite->run($reporter);
 ?>

inc/ConsoleReporter.php

@@ -3,16 +3,30 @@
 
 class ConsoleReporter extends Reporter
 {
+    private static $colors = array(
+        'red'       => "\033[31m",
+        'green'     => "\033[32m",
+        'blue'      => "\033[34m",
+        'yellow'    => "\033[33m",
+        'default'   => "\033[0m"
+    );
+
+    private $use_color = false;
+
+    public function enable_color() {
+        $this->use_color = true;
+    }
+
 	protected function report_test_pass() {
-		echo ".";
+		echo $this->wrap('.', 'green');
 	}
 
 	protected function report_test_fail(AssertionFailed $eaf) {
-		echo "F";
+		echo $this->wrap('F', 'red');
 	}
 
 	protected function report_test_error(\Exception $e) {
-		echo "E";	
+		echo $this->wrap('E', 'red');	
 	}
 
 	protected function report_summary() {
@@ -31,9 +45,30 @@ protected function report_summary() {
 		}
 
 		echo "\nSummary: ";
-		echo "{$this->test_passes}/{$this->test_total} tests passed";
-		echo ", {$this->assert_passes}/{$this->assert_total} assertions";
+
+		echo $this->wrap_by_success(
+		    "{$this->test_passes}/{$this->test_total} tests passed",
+		    $this->test_passes,
+		    $this->test_total);
+		echo ", ";
+		echo $this->wrap_by_success(
+		    "{$this->assert_passes}/{$this->assert_total} assertions",
+		    $this->assert_passes,
+		    $this->assert_total);
+
 		echo sprintf("\n%fs\n", $this->exec_time);
 	}
+
+	private function wrap($string, $color) {
+	    if ($this->use_color) {
+	        return self::$colors[$color] . $string . self::$colors['default'];
+	    } else {
+	        return $string;
+	    }
+	}
+
+	private function wrap_by_success($string, $success, $total) {
+	    return $this->wrap($string, $success == $total ? 'green' : 'yellow');
+	}
 }
 ?>

template/run_tests.php

@@ -18,6 +18,10 @@
 // Add non-abstract subclasses of ztest\TestCase as test-cases to be run
 $suite->auto_fill();
 
+// Create a reporter and enable color output
+$reporter = new ztest\ConsoleReporter;
+$reporter->enable_color();
+
 // Go, go, go
-$suite->run(new ztest\ConsoleReporter);
+$suite->run($reporter);
 ?>