Add unit tests.

paragonie · Jul 24, 2017 · 4c04e16 · 4c04e16
1 parent 030691e
commit 4c04e16
Show file tree

Hide file tree

Showing 2 changed files with 27 additions and 3 deletions.
diff --git a/src/CSPBuilder.php b/src/CSPBuilder.php
@@ -447,9 +447,9 @@ public function saveSnippet(
     /**
      * Send the compiled CSP as a header()
      *
-     * @param boolean $legacy Send legacy headers?
+     * @param bool $legacy Send legacy headers?
      *
-     * @return boolean
+     * @return bool
      * @throws \Exception
      */
     public function sendCSPHeader(bool $legacy = true): bool

diff --git a/test/BasicTest.php b/test/BasicTest.php
@@ -118,7 +118,7 @@ public function testSourceHttpsConversion()
     }
 
     /**
-     * @covers \ParagonIE\CSPBuilder\CSPBuilder
+     * @covers CSPBuilder::disableHttpsTransformOnHttpsConnections()
      */
     public function testUpgradeInsecureBeatsDisableHttpsConversionFlag()
     {
@@ -131,6 +131,30 @@ public function testUpgradeInsecureBeatsDisableHttpsConversionFlag()
         $this->assertNotContains('http://example.com', $compiled);
     }
 
+    /**
+     * @covers CSPBuilder::setDataAllowed()
+     */
+    public function testAllowDataUris()
+    {
+        $csp = new CSPBuilder();
+        $csp->setDataAllowed('img-src', true);
+        $compiled = $csp->compile();
+
+        $this->assertContains("data:", $compiled);
+    }
+
+    /**
+     * @covers CSPBuilder::setSelfAllowed()
+     */
+    public function testAllowSelfUris()
+    {
+        $csp = new CSPBuilder();
+        $csp->setSelfAllowed('img-src', true);
+        $compiled = $csp->compile();
+
+        $this->assertContains("'self'", $compiled);
+    }
+
     /*
     public function testInjectCSPHeaderWithoutLegacy()
     {