shlink/module/Common/test/IpGeolocation/ChainIpLocationResolverTest.php

<?php
declare(strict_types=1);

namespace ShlinkioTest\Shlink\Common\IpGeolocation;

use PHPUnit\Framework\TestCase;
use Prophecy\Prophecy\ObjectProphecy;
use Shlinkio\Shlink\Common\Exception\WrongIpException;
use Shlinkio\Shlink\Common\IpGeolocation\ChainIpLocationResolver;
use Shlinkio\Shlink\Common\IpGeolocation\IpLocationResolverInterface;
use Shlinkio\Shlink\Common\IpGeolocation\Model\Location;

class ChainIpLocationResolverTest extends TestCase
{
    /** @var ChainIpLocationResolver */
    private $resolver;
    /** @var ObjectProphecy */
    private $firstInnerResolver;
    /** @var ObjectProphecy */
    private $secondInnerResolver;

    public function setUp(): void
    {
        $this->firstInnerResolver = $this->prophesize(IpLocationResolverInterface::class);
        $this->secondInnerResolver = $this->prophesize(IpLocationResolverInterface::class);

        $this->resolver = new ChainIpLocationResolver(
            $this->firstInnerResolver->reveal(),
            $this->secondInnerResolver->reveal()
        );
    }

    /** @test */
    public function throwsExceptionWhenNoInnerResolverCanHandleTheResolution()
    {
        $ipAddress = '1.2.3.4';

        $firstResolve = $this->firstInnerResolver->resolveIpLocation($ipAddress)->willThrow(WrongIpException::class);
        $secondResolve = $this->secondInnerResolver->resolveIpLocation($ipAddress)->willThrow(WrongIpException::class);

        $this->expectException(WrongIpException::class);
        $firstResolve->shouldBeCalledOnce();
        $secondResolve->shouldBeCalledOnce();

        $this->resolver->resolveIpLocation($ipAddress);
    }

    /** @test */
    public function returnsResultOfFirstInnerResolver(): void
    {
        $ipAddress = '1.2.3.4';

        $firstResolve = $this->firstInnerResolver->resolveIpLocation($ipAddress)->willReturn(Location::emptyInstance());
        $secondResolve = $this->secondInnerResolver->resolveIpLocation($ipAddress)->willThrow(WrongIpException::class);

        $this->resolver->resolveIpLocation($ipAddress);

        $firstResolve->shouldHaveBeenCalledOnce();
        $secondResolve->shouldNotHaveBeenCalled();
    }

    /** @test */
    public function returnsResultOfSecondInnerResolver(): void
    {
        $ipAddress = '1.2.3.4';

        $firstResolve = $this->firstInnerResolver->resolveIpLocation($ipAddress)->willThrow(WrongIpException::class);
        $secondResolve = $this->secondInnerResolver->resolveIpLocation($ipAddress)->willReturn(
            Location::emptyInstance()
        );

        $this->resolver->resolveIpLocation($ipAddress);

        $firstResolve->shouldHaveBeenCalledOnce();
        $secondResolve->shouldHaveBeenCalledOnce();
    }
}