[OutputEscaper] Add some tests

symfony · Oct 18, 2010 · dbfa06c · dbfa06c
1 parent 71ace34
commit dbfa06c
Show file tree

Hide file tree

Showing 2 changed files with 43 additions and 0 deletions.
diff --git a/tests/Symfony/Tests/Component/OutputEscaper/ObjectDecoratorTest.php b/tests/Symfony/Tests/Component/OutputEscaper/ObjectDecoratorTest.php
@@ -30,6 +30,9 @@ public function testGenericBehavior()
 
         $array = self::$escaped->getTitles();
         $this->assertEquals('&lt;strong&gt;escaped!&lt;/strong&gt;', $array[2], 'The escaped object behaves like the real object');
+
+        $this->assertEquals('Hello &lt;strong&gt;Fabien&lt;/strong&gt;', self::$escaped->sayHello('Fabien'), 'The escaped object behaves like the real object');
+        $this->assertEquals('Hello <strong>Fabien</strong>', self::$escaped->sayHello('Fabien', 'esc_raw'), 'The escaped object behaves like the real object');
     }
 
     public function testMagicToString()
@@ -47,6 +50,21 @@ public function testMagicIsset()
         $this->assertTrue(isset(self::$escaped->someMember), 'The escaped object behaves like the real object');
         $this->assertFalse(isset(self::$escaped->invalidMember), 'The escaped object behaves like the real object');
     }
+
+    public function testGetRaw()
+    {
+        $this->assertEquals('<em>escape me</em>', self::$escaped->getRaw('someMember'), '->getRaw() returns result with any escaping');
+    }
+
+    /**
+     * @expectedException LogicException
+     */
+    public function testGetRawException()
+    {
+        $object = new \stdClass();
+        $escaped = Escaper::escape('entities', $object);
+        $escaped->getRaw('something');
+    }
 }
 
 class OutputEscaperTest
@@ -63,8 +81,18 @@ public function getTitle()
         return '<strong>escaped!</strong>';
     }
 
+    public function sayHello($name)
+    {
+        return sprintf('Hello <strong>%s</strong>', $name);
+    }
+
     public function getTitles()
     {
         return array(1, 2, '<strong>escaped!</strong>');
     }
+
+    public function get($key)
+    {
+        return isset($this->{$key}) ? $this->{$key} : null;
+    }
 }
diff --git a/tests/Symfony/Tests/Component/OutputEscaper/SafeDecoratorTest.php b/tests/Symfony/Tests/Component/OutputEscaper/SafeDecoratorTest.php
@@ -37,6 +37,13 @@ public function testMagicCall()
         $this->assertEquals('ok', $safe->doSomething(), '->__call() invokes the embedded method');
     }
 
+    public function testMagicToString()
+    {
+        $safe = new SafeDecorator(new TestClass4());
+
+        $this->assertEquals('TestClass4', (string)$safe, '->__toString() invokes the embedded __toString method');
+    }
+
     public function testMagicIssetAndUnset()
     {
         $safe = new SafeDecorator(new TestClass3());
@@ -93,3 +100,11 @@ class TestClass3
         $boolValue = true,
         $nullValue = null;
 }
+
+class TestClass4
+{
+    public function __toString()
+    {
+        return 'TestClass4';
+    }
+}