[ticket/469] Add tests for displaying all pages using main controller

B3P-469

tests/mock/template.php

@@ -36,6 +36,16 @@ class template
 		}
 	}
 
+	public function assign_vars($vars)
+	{
+		$this->data = array_merge($this->data, $vars);
+	}
+
+	public function assign_var($key, $var)
+	{
+		$this->data[$key] = $var;
+	}
+
 	public function assert_equals($data, $row)
 	{
 		foreach ($data as $key => $value)
@@ -43,4 +53,9 @@ class template
 			$this->test_case->assertEquals($value, $this->data[$row][$key]);
 		}
 	}
+
+	public function assert_same($expected, $row)
+	{
+		$this->test_case->assertSame($expected, $this->data[$row]);
+	}
 }