em = $this->prophesize(EntityManager::class); $this->visitService = new VisitService($this->em->reveal()); } /** @test */ public function locateVisitsIteratesAndLocatesUnlocatedVisits(): void { $unlocatedVisits = map( range(1, 200), fn (int $i) => new Visit(new ShortUrl(sprintf('short_code_%s', $i)), Visitor::emptyInstance()) ); $repo = $this->prophesize(VisitRepository::class); $findUnlocatedVisits = $repo->findUnlocatedVisits(false)->willReturn($unlocatedVisits); $getRepo = $this->em->getRepository(Visit::class)->willReturn($repo->reveal()); $persist = $this->em->persist(Argument::type(Visit::class))->will(function () { }); $flush = $this->em->flush()->will(function () { }); $clear = $this->em->clear()->will(function () { }); $this->visitService->locateUnlocatedVisits(fn () => Location::emptyInstance(), function () { $args = func_get_args(); $this->assertInstanceOf(VisitLocation::class, array_shift($args)); $this->assertInstanceOf(Visit::class, array_shift($args)); }); $findUnlocatedVisits->shouldHaveBeenCalledOnce(); $getRepo->shouldHaveBeenCalledOnce(); $persist->shouldHaveBeenCalledTimes(count($unlocatedVisits)); $flush->shouldHaveBeenCalledTimes(floor(count($unlocatedVisits) / 200) + 1); $clear->shouldHaveBeenCalledTimes(floor(count($unlocatedVisits) / 200) + 1); } /** * @test * @dataProvider provideIsNonLocatableAddress */ public function visitsWhichCannotBeLocatedAreIgnoredOrLocatedAsEmpty(bool $isNonLocatableAddress): void { $unlocatedVisits = [ new Visit(new ShortUrl('foo'), Visitor::emptyInstance()), ]; $repo = $this->prophesize(VisitRepository::class); $findUnlocatedVisits = $repo->findUnlocatedVisits(false)->willReturn($unlocatedVisits); $getRepo = $this->em->getRepository(Visit::class)->willReturn($repo->reveal()); $persist = $this->em->persist(Argument::type(Visit::class))->will(function () { }); $flush = $this->em->flush()->will(function () { }); $clear = $this->em->clear()->will(function () { }); $this->visitService->locateUnlocatedVisits(function () use ($isNonLocatableAddress) { throw $isNonLocatableAddress ? new IpCannotBeLocatedException('Cannot be located') : IpCannotBeLocatedException::forError(new Exception('')); }); $findUnlocatedVisits->shouldHaveBeenCalledOnce(); $getRepo->shouldHaveBeenCalledOnce(); $persist->shouldHaveBeenCalledTimes($isNonLocatableAddress ? 1 : 0); $flush->shouldHaveBeenCalledOnce(); $clear->shouldHaveBeenCalledOnce(); } public function provideIsNonLocatableAddress(): iterable { yield 'The address is locatable' => [false]; yield 'The address is non-locatable' => [true]; } }