From 7d73645ed7a8ed0fe4ff8f145ce0ec64ce94b5f7 Mon Sep 17 00:00:00 2001
From: Ollie Read <code@ollieread.com>
Date: Tue, 19 Nov 2024 12:00:45 +0000
Subject: [PATCH] test: Add unit tests for the resolution helper

---
 tests/Unit/Support/ResolutionHelperTest.php | 297 ++++++++++++++++++++
 1 file changed, 297 insertions(+)

diff --git a/tests/Unit/Support/ResolutionHelperTest.php b/tests/Unit/Support/ResolutionHelperTest.php
index 81efd9c..f7d55f1 100644
--- a/tests/Unit/Support/ResolutionHelperTest.php
+++ b/tests/Unit/Support/ResolutionHelperTest.php
@@ -4,14 +4,31 @@
 namespace Sprout\Tests\Unit\Support;
 
 use Illuminate\Http\Request;
+use Illuminate\Routing\Route;
+use Mockery\MockInterface;
 use PHPUnit\Framework\Attributes\Test;
 use Sprout\Exceptions\MisconfigurationException;
+use Sprout\Exceptions\NoTenantFound;
+use Sprout\Managers\IdentityResolverManager;
+use Sprout\Managers\TenancyManager;
 use Sprout\Support\ResolutionHelper;
 use Sprout\Support\ResolutionHook;
 use Sprout\Tests\Unit\UnitTestCase;
+use Workbench\App\Models\TenantModel;
+use function Sprout\resolver;
+use function Sprout\sprout;
 
 class ResolutionHelperTest extends UnitTestCase
 {
+    protected function defineEnvironment($app): void
+    {
+        tap($app['config'], static function ($config) {
+            $config->set('multitenancy.defaults.resolver', 'path');
+            $config->set('multitenancy.providers.tenants.model', TenantModel::class);
+            $config->set('multitenancy.resolvers.subdomain.domain', 'localhost');
+        });
+    }
+
     #[Test]
     public function parsesMiddlewareOptions(): void
     {
@@ -45,4 +62,284 @@ public function throwsExceptionWhenHandlingResolutionForUnsupportedHook(): void
 
         ResolutionHelper::handleResolution($fakeRequest, ResolutionHook::Booting);
     }
+
+    #[Test]
+    public function returnsFalseIfThereIsAlreadyATenant(): void
+    {
+        /** @var \Sprout\Contracts\Tenancy<TenantModel> $tenancy */
+        $tenancy = app(TenancyManager::class)->get();
+
+        $tenancy->setTenant(TenantModel::factory()->createOne());
+
+        /** @var \Sprout\Contracts\IdentityResolver $resolver */
+        $resolver = app(IdentityResolverManager::class)->get('path');
+
+        /** @var \Illuminate\Http\Request $fakeRequest */
+        $fakeRequest = $this->mock(Request::class);
+
+        $this->assertTrue($tenancy->check());
+        $this->assertTrue($resolver->canResolve($fakeRequest, $tenancy, ResolutionHook::Routing));
+        $this->assertFalse(ResolutionHelper::handleResolution($fakeRequest, ResolutionHook::Routing, $resolver->getName(), $tenancy->getName()));
+    }
+
+    #[Test]
+    public function returnsFalseIfTheResolverCannotResolve(): void
+    {
+        /** @var \Sprout\Contracts\Tenancy<TenantModel> $tenancy */
+        $tenancy = app(TenancyManager::class)->get();
+
+        /** @var \Sprout\Contracts\IdentityResolver $resolver */
+        $resolver = app(IdentityResolverManager::class)->get('path');
+
+        $tenancy->setTenant(TenantModel::factory()->createOne())
+                ->resolvedVia($resolver)
+                ->resolvedAt(ResolutionHook::Routing);
+
+        /** @var \Illuminate\Http\Request $fakeRequest */
+        $fakeRequest = $this->mock(Request::class);
+
+        $this->assertTrue($tenancy->check());
+        $this->assertFalse($resolver->canResolve($fakeRequest, $tenancy, ResolutionHook::Routing));
+        $this->assertFalse(ResolutionHelper::handleResolution($fakeRequest, ResolutionHook::Routing, $resolver->getName(), $tenancy->getName()));
+    }
+
+    #[Test]
+    public function resolvesTenantUsingRouteParameters(): void
+    {
+        $tenant = TenantModel::factory()->createOne();
+
+        /** @var \Sprout\Contracts\Tenancy<TenantModel> $tenancy */
+        $tenancy = app(TenancyManager::class)->get();
+
+        /** @var \Sprout\Contracts\IdentityResolver&\Sprout\Contracts\IdentityResolverUsesParameters $resolver */
+        $resolver = app(IdentityResolverManager::class)->get('path');
+
+        /** @var \Illuminate\Routing\Route $fakeRoute */
+        $fakeRoute = $this->mock(Route::class, function (MockInterface $mock) use ($tenant, $tenancy, $resolver) {
+            $parameterName = $resolver->getRouteParameterName($tenancy);
+
+            $mock->shouldReceive('hasParameter')
+                 ->with($parameterName)
+                 ->andReturn(true);
+
+            $mock->shouldReceive('parameter')
+                 ->with($parameterName)
+                 ->andReturn($tenant->getTenantIdentifier());
+
+            $mock->shouldReceive('forgetParameter')
+                 ->with($parameterName);
+        });
+
+        /** @var \Illuminate\Http\Request $fakeRequest */
+        $fakeRequest = $this->mock(Request::class, function (MockInterface $mock) use ($fakeRoute) {
+            $mock->shouldReceive('route')->andReturn($fakeRoute);
+        });
+
+        $this->assertTrue(ResolutionHelper::handleResolution($fakeRequest, ResolutionHook::Routing, $resolver->getName(), $tenancy->getName()));
+        $this->assertSame($tenancy, sprout()->getCurrentTenancy());
+        $this->assertTrue($tenancy->check());
+        $this->assertTrue($tenant->is($tenancy->tenant()));
+        $this->assertTrue($tenancy->wasResolved());
+        $this->assertSame($resolver, $tenancy->resolver());
+        $this->assertSame(ResolutionHook::Routing, $tenancy->hook());
+
+        $tenancy->setTenant(null);
+
+        $this->assertTrue(ResolutionHelper::handleResolution($fakeRequest, ResolutionHook::Routing));
+        $this->assertSame($tenancy, sprout()->getCurrentTenancy());
+        $this->assertTrue($tenancy->check());
+        $this->assertTrue($tenant->is($tenancy->tenant()));
+        $this->assertTrue($tenancy->wasResolved());
+        $this->assertSame($resolver, $tenancy->resolver());
+        $this->assertSame(ResolutionHook::Routing, $tenancy->hook());
+    }
+
+    #[Test]
+    public function throwsAnExceptionWhenUnableToIdentifyATenantFromTheRoute(): void
+    {
+        /** @var \Sprout\Contracts\Tenancy<TenantModel> $tenancy */
+        $tenancy = app(TenancyManager::class)->get();
+
+        /** @var \Sprout\Contracts\IdentityResolver&\Sprout\Contracts\IdentityResolverUsesParameters $resolver */
+        $resolver = app(IdentityResolverManager::class)->get('path');
+
+        /** @var \Illuminate\Routing\Route $fakeRoute */
+        $fakeRoute = $this->mock(Route::class, function (MockInterface $mock) use ($tenancy, $resolver) {
+            $parameterName = $resolver->getRouteParameterName($tenancy);
+
+            $mock->shouldReceive('hasParameter')
+                 ->with($parameterName)
+                 ->andReturn(true);
+
+            $mock->shouldReceive('parameter')
+                 ->with($parameterName)
+                 ->andReturn('fake-identifier');
+
+            $mock->shouldReceive('forgetParameter')
+                 ->with($parameterName);
+        });
+
+        /** @var \Illuminate\Http\Request $fakeRequest */
+        $fakeRequest = $this->mock(Request::class, function (MockInterface $mock) use ($fakeRoute) {
+            $mock->shouldReceive('route')->andReturn($fakeRoute);
+        });
+
+        $this->expectException(NoTenantFound::class);
+        $this->expectExceptionMessage('No valid tenant [' . $tenancy->getName() . '] found [' . $resolver->getName() . ']');
+
+        ResolutionHelper::handleResolution($fakeRequest, ResolutionHook::Routing, $resolver->getName(), $tenancy->getName());
+
+        $this->expectException(NoTenantFound::class);
+        $this->expectExceptionMessage('No valid tenant [' . $tenancy->getName() . '] found [subdomain]');
+
+        ResolutionHelper::handleResolution($fakeRequest, ResolutionHook::Routing);
+    }
+
+    #[Test]
+    public function returnsFalseWhenUnableToIdentifyATenantFromTheRouteAndToldNotToThrow(): void
+    {
+        /** @var \Sprout\Contracts\Tenancy<TenantModel> $tenancy */
+        $tenancy = app(TenancyManager::class)->get();
+
+        /** @var \Sprout\Contracts\IdentityResolver&\Sprout\Contracts\IdentityResolverUsesParameters $resolver */
+        $resolver = app(IdentityResolverManager::class)->get('path');
+
+        /** @var \Illuminate\Routing\Route $fakeRoute */
+        $fakeRoute = $this->mock(Route::class, function (MockInterface $mock) use ($tenancy, $resolver) {
+            $parameterName = $resolver->getRouteParameterName($tenancy);
+
+            $mock->shouldReceive('hasParameter')
+                 ->with($parameterName)
+                 ->andReturn(true);
+
+            $mock->shouldReceive('parameter')
+                 ->with($parameterName)
+                 ->andReturn('fake-identifier');
+
+            $mock->shouldReceive('forgetParameter')
+                 ->with($parameterName);
+        });
+
+        /** @var \Illuminate\Http\Request $fakeRequest */
+        $fakeRequest = $this->mock(Request::class, function (MockInterface $mock) use ($fakeRoute) {
+            $mock->shouldReceive('route')->andReturn($fakeRoute);
+        });
+
+        $this->assertFalse(ResolutionHelper::handleResolution($fakeRequest, ResolutionHook::Routing, $resolver->getName(), $tenancy->getName(), false));
+        $this->assertSame($tenancy, sprout()->getCurrentTenancy());
+        $this->assertFalse($tenancy->check());
+        $this->assertFalse($tenancy->wasResolved());
+        $this->assertNull($tenancy->resolver());
+        $this->assertNull($tenancy->hook());
+
+        $this->assertFalse(ResolutionHelper::handleResolution($fakeRequest, ResolutionHook::Routing, throw: false));
+        $this->assertSame($tenancy, sprout()->getCurrentTenancy());
+        $this->assertFalse($tenancy->check());
+        $this->assertFalse($tenancy->wasResolved());
+        $this->assertNull($tenancy->resolver());
+        $this->assertNull($tenancy->hook());
+    }
+
+    #[Test]
+    public function resolvesTenantWithoutRouteParameters(): void
+    {
+        $tenant = TenantModel::factory()->createOne();
+
+        /** @var \Sprout\Contracts\Tenancy<TenantModel> $tenancy */
+        $tenancy = app(TenancyManager::class)->get();
+
+        /** @var \Sprout\Contracts\IdentityResolver&\Sprout\Contracts\IdentityResolverUsesParameters $resolver */
+        $resolver = app(IdentityResolverManager::class)->get('path');
+
+        /** @var \Illuminate\Http\Request $fakeRequest */
+        $fakeRequest = $this->mock(Request::class, function (MockInterface $mock) use ($tenant) {
+            $mock->shouldReceive('route')->andReturnNull();
+
+            $mock->shouldReceive('segment')
+                 ->with(1)
+                 ->andReturn($tenant->getTenantIdentifier());
+        });
+
+        $this->assertTrue(ResolutionHelper::handleResolution($fakeRequest, ResolutionHook::Routing, $resolver->getName(), $tenancy->getName()));
+        $this->assertSame($tenancy, sprout()->getCurrentTenancy());
+        $this->assertTrue($tenancy->check());
+        $this->assertTrue($tenant->is($tenancy->tenant()));
+        $this->assertTrue($tenancy->wasResolved());
+        $this->assertSame($resolver, $tenancy->resolver());
+        $this->assertSame(ResolutionHook::Routing, $tenancy->hook());
+
+        $tenancy->setTenant(null);
+
+        $this->assertTrue(ResolutionHelper::handleResolution($fakeRequest, ResolutionHook::Routing));
+        $this->assertSame($tenancy, sprout()->getCurrentTenancy());
+        $this->assertTrue($tenancy->check());
+        $this->assertTrue($tenant->is($tenancy->tenant()));
+        $this->assertTrue($tenancy->wasResolved());
+        $this->assertSame($resolver, $tenancy->resolver());
+        $this->assertSame(ResolutionHook::Routing, $tenancy->hook());
+    }
+
+    #[Test]
+    public function throwsAnExceptionWhenUnableToIdentifyATenantFromTheRequest(): void
+    {
+        /** @var \Sprout\Contracts\Tenancy<TenantModel> $tenancy */
+        $tenancy = app(TenancyManager::class)->get();
+
+        /** @var \Sprout\Contracts\IdentityResolver&\Sprout\Contracts\IdentityResolverUsesParameters $resolver */
+        $resolver = app(IdentityResolverManager::class)->get('path');
+
+        /** @var \Illuminate\Http\Request $fakeRequest */
+        $fakeRequest = $this->mock(Request::class, function (MockInterface $mock) {
+            $mock->shouldReceive('route')->andReturnNull();
+
+            $mock->shouldReceive('segment')
+                 ->with(1)
+                 ->andReturn('fake-identifier');
+        });
+
+        $this->expectException(NoTenantFound::class);
+        $this->expectExceptionMessage('No valid tenant [' . $tenancy->getName() . '] found [' . $resolver->getName() . ']');
+
+        ResolutionHelper::handleResolution($fakeRequest, ResolutionHook::Routing, $resolver->getName(), $tenancy->getName());
+
+        $this->expectException(NoTenantFound::class);
+        $this->expectExceptionMessage('No valid tenant [' . $tenancy->getName() . '] found [subdomain]');
+
+        ResolutionHelper::handleResolution($fakeRequest, ResolutionHook::Routing);
+    }
+
+    #[Test]
+    public function returnsFalseWhenUnableToIdentifyATenantFromTheRequestAndToldNotToThrow(): void
+    {
+        /** @var \Sprout\Contracts\Tenancy<TenantModel> $tenancy */
+        $tenancy = app(TenancyManager::class)->get();
+
+        /** @var \Sprout\Contracts\IdentityResolver&\Sprout\Contracts\IdentityResolverUsesParameters $resolver */
+        $resolver = app(IdentityResolverManager::class)->get('path');
+
+        /** @var \Illuminate\Http\Request $fakeRequest */
+        $fakeRequest = $this->mock(Request::class, function (MockInterface $mock) {
+            $mock->shouldReceive('route')->andReturnNull();
+
+            $mock->shouldReceive('segment')
+                 ->with(1)
+                 ->andReturn('fake-identifier');
+        });
+
+        $this->assertFalse(ResolutionHelper::handleResolution($fakeRequest, ResolutionHook::Routing, $resolver->getName(), $tenancy->getName(), false));
+        $this->assertSame($tenancy, sprout()->getCurrentTenancy());
+        $this->assertFalse($tenancy->check());
+        $this->assertFalse($tenancy->wasResolved());
+        $this->assertNull($tenancy->resolver());
+        $this->assertNull($tenancy->hook());
+
+        $tenancy->setTenant(null);
+
+        $this->assertFalse(ResolutionHelper::handleResolution($fakeRequest, ResolutionHook::Routing, throw: false));
+        $this->assertSame($tenancy, sprout()->getCurrentTenancy());
+        $this->assertFalse($tenancy->check());
+        $this->assertFalse($tenancy->wasResolved());
+        $this->assertNull($tenancy->resolver());
+        $this->assertNull($tenancy->hook());
+    }
 }