MockRequestBuilderCollection.php

<?php

declare(strict_types=1);

namespace Brainbits\FunctionalTestHelpers\HttpClientMock;

use Countable;
use IteratorAggregate;
use Symfony\Contracts\HttpClient\ResponseInterface;
use Traversable;

use function array_map;
use function count;
use function is_callable;

/** @implements IteratorAggregate<MockRequestBuilder> */
final class MockRequestBuilderCollection implements IteratorAggregate, Countable
{
    private RealRequestFactory $requestFactory;
    private MockRequestResolver $requestResolver;
    /** @var MockRequestBuilder[] */
    private array $requestBuilders = [];

    public function __construct()
    {
        $this->requestFactory = new RealRequestFactory();
        $this->requestResolver = new MockRequestResolver();
    }

    /** @param mixed[] $options */
    public function __invoke(string $method, string $url, array $options): ResponseInterface
    {
        $realRequest = ($this->requestFactory)($method, $url, $options);

        $requestBuilder = ($this->requestResolver)($this, $realRequest);
        $requestBuilder->assert($realRequest);
        $requestBuilder->called($realRequest);

        if ($requestBuilder->onMatch && is_callable($requestBuilder->onMatch)) {
            ($requestBuilder->onMatch)($realRequest);
        }

        return $requestBuilder->getResponse($realRequest);
    }

    public function addMockRequestBuilder(MockRequestBuilder $mockRequestBuilder): void
    {
        $this->requestBuilders[] = $mockRequestBuilder;
    }

    public function getCallStack(): CallStack
    {
        $callStacks = array_map(
            static fn ($requestBuilder) => $requestBuilder->getCallStack(),
            $this->requestBuilders,
        );

        return CallStack::fromCallStacks(...$callStacks);
    }

    /** @return Traversable<MockRequestBuilder> */
    public function getIterator(): Traversable
    {
        yield from $this->requestBuilders;
    }

    public function count(): int
    {
        return count($this->requestBuilders);
    }

    public function reset(): void
    {
        $this->requestBuilders = [];
    }
}