From 36da5a83b37f1def6471410ed1dfdc5896630774 Mon Sep 17 00:00:00 2001
From: Mohammad Alavi <Mohammad.alavi1990@gmail.com>
Date: Tue, 6 Feb 2024 17:54:28 +0330
Subject: [PATCH] API-1126-Add-test-assertions (#198)

* Update TestAssertionHelperTrait.php

* style: apply automated php-cs-fixer changes

* style: apply automated php-cs-fixer changes

* Update TestAssertionHelperTrait.php

---------

Co-authored-by: Mohammad-Alavi <Mohammad-Alavi@users.noreply.github.com>
---
 .../PhpUnit/TestAssertionHelperTrait.php      | 73 +++++++++++++++++++
 1 file changed, 73 insertions(+)

diff --git a/src/Traits/TestTraits/PhpUnit/TestAssertionHelperTrait.php b/src/Traits/TestTraits/PhpUnit/TestAssertionHelperTrait.php
index dd7e34e7..8530cd62 100644
--- a/src/Traits/TestTraits/PhpUnit/TestAssertionHelperTrait.php
+++ b/src/Traits/TestTraits/PhpUnit/TestAssertionHelperTrait.php
@@ -7,6 +7,7 @@
 use Illuminate\Support\Collection;
 use Illuminate\Support\Facades\Schema;
 use JetBrains\PhpStorm\Deprecated;
+use Mockery\MockInterface;
 use PHPUnit\Framework\MockObject\Exception;
 use PHPUnit\Framework\MockObject\MockObject;
 
@@ -94,4 +95,76 @@ protected function getInaccessiblePropertyValue(object $object, string $property
 
         return $property->getValue($object);
     }
+
+    /**
+     * Create a spy for an Action, SubAction or a Task that uses a repository.
+     *
+     * @param string $className the Action, SubAction or a Task class name
+     * @param string $repositoryClassName the repository class name
+     */
+    protected function createSpyWithRepository(string $className, string $repositoryClassName, bool $allowRun = true): MockInterface
+    {
+        /** @var MockInterface $taskSpy */
+        $taskSpy = \Mockery::mock($className, [app($repositoryClassName)])
+            ->shouldIgnoreMissing(null, true)
+            ->makePartial();
+
+        if ($allowRun) {
+            $taskSpy->allows('run')->andReturn();
+        }
+
+        $this->swap($className, $taskSpy);
+
+        return $taskSpy;
+    }
+
+    /**
+     * Mock a repository and assert that the given criteria is pushed to it.
+     *
+     * @param string $repositoryClassName the repository class name
+     * @param string $criteriaClassName the criteria class name
+     * @param array<string, mixed>|null $criteriaArgs the criteria constructor arguments
+     *
+     * @return MockInterface repository mock
+     *
+     * @example $this->
+     * assertCriteriaPushedToRepository(UserRepository::class, SearchUsersCriteria::class, ['parameterName' => 'value']);
+     */
+    protected function assertCriteriaPushedToRepository(string $repositoryClassName, string $criteriaClassName, array|null $criteriaArgs = null): MockInterface
+    {
+        $repositoryMock = $this->mock($repositoryClassName);
+
+        if (is_null($criteriaArgs)) {
+            $repositoryMock->expects('pushCriteria')->once();
+        } else {
+            $repositoryMock->expects('pushCriteriaWith')->once()->with($criteriaClassName, $criteriaArgs);
+        }
+
+        return $repositoryMock;
+    }
+
+    /**
+     * Assert that no criteria are pushed to the repository.
+     *
+     * @param string $repositoryClassName the repository class name
+     *
+     * @return MockInterface repository mock
+     */
+    protected function assertNoCriteriaPushedToRepository(string $repositoryClassName): MockInterface
+    {
+        $repositoryMock = $this->mock($repositoryClassName);
+        $repositoryMock->expects('pushCriteria')->never();
+
+        return $repositoryMock;
+    }
+
+    /**
+     * Allow "addRequestCriteria" method invocation on the repository mock.
+     * This is particularly useful when you want to test a repository that uses the RequestCriteria
+     * (e.g., for search and filtering).
+     */
+    protected function allowAddRequestCriteriaInvocation(MockInterface $repositoryMock): void
+    {
+        $repositoryMock->allows('addRequestCriteria')->andReturnSelf();
+    }
 }