aboutsummaryrefslogtreecommitdiffstats
path: root/tests/lib/Security/FeaturePolicy/FeaturePolicyManagerTest.php
blob: 7386aa023a9381a29b13579a54fa8953c42f7156 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
<?php

declare(strict_types=1);
/**
 * SPDX-FileCopyrightText: 2019 Nextcloud GmbH and Nextcloud contributors
 * SPDX-License-Identifier: AGPL-3.0-or-later
 */

namespace Test\Security\CSP;

use OC\Security\FeaturePolicy\FeaturePolicyManager;
use OCP\AppFramework\Http\FeaturePolicy;
use OCP\EventDispatcher\IEventDispatcher;
use OCP\Security\FeaturePolicy\AddFeaturePolicyEvent;
use Test\TestCase;

class FeaturePolicyManagerTest extends TestCase {
	/** @var IEventDispatcher */
	private $dispatcher;

	/** @var FeaturePolicyManager */
	private $manager;

	protected function setUp(): void {
		parent::setUp();
		$this->dispatcher = \OC::$server->query(IEventDispatcher::class);
		$this->manager = new FeaturePolicyManager($this->dispatcher);
	}

	public function testAddDefaultPolicy(): void {
		$this->manager->addDefaultPolicy(new FeaturePolicy());
		$this->addToAssertionCount(1);
	}

	public function testGetDefaultPolicyWithPoliciesViaEvent(): void {
		$this->dispatcher->addListener(AddFeaturePolicyEvent::class, function (AddFeaturePolicyEvent $e) {
			$policy = new FeaturePolicy();
			$policy->addAllowedMicrophoneDomain('mydomain.com');
			$policy->addAllowedPaymentDomain('mypaymentdomain.com');

			$e->addPolicy($policy);
		});

		$this->dispatcher->addListener(AddFeaturePolicyEvent::class, function (AddFeaturePolicyEvent $e) {
			$policy = new FeaturePolicy();
			$policy->addAllowedPaymentDomain('mydomainother.com');
			$policy->addAllowedGeoLocationDomain('mylocation.here');

			$e->addPolicy($policy);
		});

		$this->dispatcher->addListener(AddFeaturePolicyEvent::class, function (AddFeaturePolicyEvent $e) {
			$policy = new FeaturePolicy();
			$policy->addAllowedAutoplayDomain('youtube.com');

			$e->addPolicy($policy);
		});

		$expected = new \OC\Security\FeaturePolicy\FeaturePolicy();
		$expected->addAllowedMicrophoneDomain('mydomain.com');
		$expected->addAllowedPaymentDomain('mypaymentdomain.com');
		$expected->addAllowedPaymentDomain('mydomainother.com');
		$expected->addAllowedGeoLocationDomain('mylocation.here');
		$expected->addAllowedAutoplayDomain('youtube.com');

		$expectedStringPolicy = "autoplay 'self' youtube.com;camera 'none';fullscreen 'self';geolocation mylocation.here;microphone mydomain.com;payment mypaymentdomain.com mydomainother.com";

		$this->assertEquals($expected, $this->manager->getDefaultPolicy());
		$this->assertSame($expectedStringPolicy, $this->manager->getDefaultPolicy()->buildPolicy());
	}
}