From d81e1b3711610cb9589b101bd5e4a1ff89cedc66 Mon Sep 17 00:00:00 2001
From: naomi <naomi@happycatsin.space>
Date: Sun, 5 Jun 2022 12:25:28 +0100
Subject: [PATCH] Added test for OCActor::getCountOfCaseProvisions

---
 tests/src/Unit/OCActorTest.php | 31 +++++++++++++++++++++++++++++++
 1 file changed, 31 insertions(+)
 create mode 100644 tests/src/Unit/OCActorTest.php

diff --git a/tests/src/Unit/OCActorTest.php b/tests/src/Unit/OCActorTest.php
new file mode 100644
index 0000000..3e332a1
--- /dev/null
+++ b/tests/src/Unit/OCActorTest.php
@@ -0,0 +1,31 @@
+<?php declare(strict_types = 1);
+
+namespace Drupal\Tests\opencase\Unit;
+
+use Drupal\Tests\UnitTestCase;
+use Drupal\opencase_entities\Entity\OCActor;
+use Drupal\Core\DependencyInjection\ContainerBuilder;
+
+class OCActorTest extends UnitTestCase{
+    public function setUp(): void {
+        $this->entityTypeManager = $this->getMockBuilder('\\Drupal\\Core\\Entity\\EntityTypeManager')->disableOriginalConstructor()->getMock();
+        $container = new ContainerBuilder();
+        $container->set('entity_type.manager', $this->entityTypeManager);
+        \Drupal::setContainer($container);
+        $this->actor = $this->getMockBuilder(OCActor::class)->disableOriginalConstructor()
+            ->onlyMethods(['id'])
+            ->getMock();
+    }
+
+    public function testGetCountOfCaseProvisions(){  
+        $caseProvisionStorage = $this->getMockBuilder('\\Drupal\\Core\\Entity\\EntityStorageInterface')->disableOriginalConstructor()->getMock(); 
+        $this->entityTypeManager->expects($this->once())->method('getStorage')->with('oc_case_provision')->willReturn($caseProvisionStorage);     
+        $query = $this->getMockBuilder('\\Drupal\\Core\\Entity\\Query\\QueryInterface')->getMock();
+        $caseProvisionStorage->method('getQuery')->willReturn($query);
+        $this->actor->expects($this->once())->method('id')->willReturn(5);
+        $query->expects($this->once())->method('condition')->with('oc_provider', 5);
+        $query->expects($this->once())->method('execute')->willReturn([1,2,3,4]);
+        $count = $this->actor->getCountOfCaseProvisions();
+        $this->assertTrue($count == 4);
+    }
+}