BookStack/tests/SecurityHeaderTest.php

<?php namespace Tests;


use Illuminate\Support\Str;

class SecurityHeaderTest extends TestCase
{

    public function test_cookies_samesite_lax_by_default()
    {
        $resp = $this->get("/");
        foreach ($resp->headers->getCookies() as $cookie) {
            $this->assertEquals("lax", $cookie->getSameSite());
        }
    }

    public function test_cookies_samesite_none_when_iframe_hosts_set()
    {
        $this->runWithEnv("ALLOWED_IFRAME_HOSTS", "http://example.com", function() {
            $resp = $this->get("/");
            foreach ($resp->headers->getCookies() as $cookie) {
                $this->assertEquals("none", $cookie->getSameSite());
            }
        });
    }

    public function test_secure_cookies_controlled_by_app_url()
    {
        $this->runWithEnv("APP_URL", "http://example.com", function() {
            $resp = $this->get("/");
            foreach ($resp->headers->getCookies() as $cookie) {
                $this->assertFalse($cookie->isSecure());
            }
        });

        $this->runWithEnv("APP_URL", "https://example.com", function() {
            $resp = $this->get("/");
            foreach ($resp->headers->getCookies() as $cookie) {
                $this->assertTrue($cookie->isSecure());
            }
        });
    }

    public function test_iframe_csp_self_only_by_default()
    {
        $resp = $this->get("/");
        $cspHeaders = collect($resp->headers->get('Content-Security-Policy'));
        $frameHeaders = $cspHeaders->filter(function ($val) {
            return Str::startsWith($val, 'frame-ancestors');
        });

        $this->assertTrue($frameHeaders->count() === 1);
        $this->assertEquals('frame-ancestors \'self\'', $frameHeaders->first());
    }

    public function test_iframe_csp_includes_extra_hosts_if_configured()
    {
        $this->runWithEnv("ALLOWED_IFRAME_HOSTS", "https://a.example.com https://b.example.com", function() {
            $resp = $this->get("/");
            $cspHeaders = collect($resp->headers->get('Content-Security-Policy'));
            $frameHeaders = $cspHeaders->filter(function($val) {
                return Str::startsWith($val, 'frame-ancestors');
            });

            $this->assertTrue($frameHeaders->count() === 1);
            $this->assertEquals('frame-ancestors \'self\' https://a.example.com https://b.example.com', $frameHeaders->first());
        });

    }

}