Covering the model classes by Unit Tests

Author: eduard13

app/code/Magento/Csp/Test/Unit/Model/Mode/ConfigManagerTest.php

@@ -0,0 +1,109 @@
+<?php
+/**
+ * Copyright © Magento, Inc. All rights reserved.
+ * See COPYING.txt for license details.
+ */
+
+declare(strict_types=1);
+
+namespace Magento\Csp\Test\Unit\Model;
+
+use Magento\Csp\Model\Mode\ConfigManager;
+use Magento\Csp\Model\Mode\Data\ModeConfigured;
+use Magento\Framework\App\Area;
+use Magento\Framework\App\Config\ScopeConfigInterface;
+use Magento\Framework\App\State;
+use Magento\Framework\TestFramework\Unit\Helper\ObjectManager;
+use Magento\Store\Model\Store;
+use PHPUnit\Framework\MockObject\MockObject;
+use PHPUnit\Framework\TestCase;
+use RuntimeException;
+
+/**
+ * Class ConfigManagerTest
+ *
+ * Test for \Magento\Csp\Model\Mode\ConfigManager
+ */
+class ConfigManagerTest extends TestCase
+{
+    const STUB_CRONTAB_AREA = 'crontab';
+
+    /**
+     * @var ConfigManager
+     */
+    private $model;
+
+    /**
+     * @var ScopeConfigInterface|MockObject
+     */
+    private $scopeConfigMock;
+
+    /**
+     * @var Store|MockObject
+     */
+    private $storeMock;
+
+    /**
+     * @var State|MockObject
+     */
+    private $stateMock;
+
+    /**
+     * Set Up
+     */
+    protected function setUp()
+    {
+        $objectManager = new ObjectManager($this);
+
+        $this->scopeConfigMock = $this->createMock(ScopeConfigInterface::class);
+        $this->storeMock = $this->createMock(Store::class);
+        $this->stateMock = $this->createMock(State::class);
+
+        $this->model = $objectManager->getObject(
+            ConfigManager::class,
+            [
+                'config' => $this->scopeConfigMock,
+                'storeModel' => $this->storeMock,
+                'state' => $this->stateMock
+            ]
+        );
+    }
+
+    /**
+     * Test throwing an exception for non storefront or admin areas
+     *
+     * @return void
+     *
+     * @expectedExceptionMessage CSP can only be configured for storefront or admin area
+     * @expectedException RuntimeException
+     */
+    public function testThrownExceptionForCrontabArea()
+    {
+        $this->stateMock->expects($this->any())
+            ->method('getAreaCode')
+            ->willReturn(static::STUB_CRONTAB_AREA);
+
+        $this->model->getConfigured();
+    }
+
+    /**
+     * Test returning the configured CSP for admin area
+     *
+     * @return void
+     */
+    public function testConfiguredCSPForAdminArea()
+    {
+        $this->stateMock->expects($this->any())
+            ->method('getAreaCode')
+            ->willReturn(Area::AREA_ADMINHTML);
+        $this->scopeConfigMock->expects($this->any())
+            ->method('isSetFlag')
+            ->willReturn(true);
+        $this->scopeConfigMock->expects($this->any())
+            ->method('getValue')
+            ->willReturn('testReportUri');
+        $result = $this->model->getConfigured();
+
+        $this->assertInstanceOf(ModeConfigured::class, $result);
+    }
+}

app/code/Magento/Csp/Test/Unit/Model/PolicyRendererPoolTest.php

@@ -0,0 +1,96 @@
+<?php
+/**
+ * Copyright © Magento, Inc. All rights reserved.
+ * See COPYING.txt for license details.
+ */
+
+declare(strict_types=1);
+
+namespace Magento\Csp\Test\Unit\Model;
+
+use Magento\Csp\Api\Data\PolicyInterface;
+use Magento\Csp\Model\Policy\Renderer\SimplePolicyHeaderRenderer;
+use Magento\Csp\Model\PolicyRendererPool;
+use Magento\Framework\TestFramework\Unit\Helper\ObjectManager;
+use PHPUnit\Framework\MockObject\MockObject;
+use PHPUnit\Framework\TestCase;
+use RuntimeException;
+
+/**
+ * Class PolicyRendererPoolTest
+ *
+ * Test for \Magento\Csp\Model\PolicyRendererPool
+ */
+class PolicyRendererPoolTest extends TestCase
+{
+    const STUB_POLICY_ID = 'header';
+
+    /**
+     * @var PolicyRendererPool
+     */
+    private $model;
+
+    /**
+     * @var SimplePolicyHeaderRenderer|MockObject
+     */
+    private $simplePolicyHeaderRendererMock;
+
+    /**
+     * @var PolicyInterface|MockObject
+     */
+    private $policyMock;
+
+    /**
+     * Set Up
+     */
+    protected function setUp()
+    {
+        $objectManager = new ObjectManager($this);
+        $this->simplePolicyHeaderRendererMock = $this->createPartialMock(
+            SimplePolicyHeaderRenderer::class,
+            ['canRender']
+        );
+        $this->policyMock = $this->createMock(PolicyInterface::class);
+
+        $this->model = $objectManager->getObject(
+            PolicyRendererPool::class,
+            [
+                'renderers' => [
+                    $this->simplePolicyHeaderRendererMock
+                ]
+            ]
+        );
+    }
+
+    /**
+     * Test throwing an exception for not found policy renders
+     *
+     * @return void
+     *
+     * @expectedExceptionMessage Failed to find a renderer for policy
+     * @expectedException RuntimeException
+     */
+    public function testThrownExceptionForNotFoundPolicyRenders()
+    {
+        $this->policyMock->expects($this->any())
+            ->method('getId')
+            ->willReturn(static::STUB_POLICY_ID);
+
+        $this->model->getRenderer($this->policyMock);
+    }
+
+    /**
+     * Test returning a renderer for the given policy
+     *
+     * @return void
+     */
+    public function testReturningThePolicyRender()
+    {
+        $this->simplePolicyHeaderRendererMock->expects($this->any())
+            ->method('canRender')
+            ->with($this->policyMock)
+            ->willReturn(true);
+
+        $this->model->getRenderer($this->policyMock);
+    }
+}