OilfieldToolsNet/shlink
4
0
Fork 0
mirror of https://github.com/shlinkio/shlink.git synced 2025-02-25 18:45:27 -06:00
Code Issues Packages Projects Releases Wiki Activity

Fix UpdateGeoLiteDbTest

Browse Source
This commit is contained in:
Alejandro Celaya 2024-12-11 08:35:24 +01:00
parent b8ac9f3673
commit 4f3c2c7d2d
1 changed files with 15 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

23
module/Core/test/EventDispatcher/UpdateGeoLiteDbTest.php
View File

@ -14,6 +14,7 @@ use RuntimeException;
use Shlinkio\Shlink\Core\EventDispatcher\Event\GeoLiteDbCreated;
use Shlinkio\Shlink\Core\EventDispatcher\UpdateGeoLiteDb;
use Shlinkio\Shlink\Core\Geolocation\GeolocationDbUpdaterInterface;
use Shlinkio\Shlink\Core\Geolocation\GeolocationDownloadProgressHandlerInterface;
use Shlinkio\Shlink\Core\Geolocation\GeolocationResult;
use function array_map;
@ -51,11 +52,11 @@ class UpdateGeoLiteDbTest extends TestCase
}
#[Test, DataProvider('provideFlags')]
public function noticeMessageIsPrintedWhenFirstCallbackIsInvoked(bool $oldDbExists, string $expectedMessage): void
public function noticeMessageIsPrintedWhenDownloadIsStarted(bool $oldDbExists, string $expectedMessage): void
{
$this->dbUpdater->expects($this->once())->method('checkDbUpdate')->withAnyParameters()->willReturnCallback(
function (callable $firstCallback) use ($oldDbExists): GeolocationResult {
$firstCallback($oldDbExists);
function (GeolocationDownloadProgressHandlerInterface $handler) use ($oldDbExists): GeolocationResult {
$handler->beforeDownload($oldDbExists);
return GeolocationResult::DB_IS_UP_TO_DATE;
},
);
@ -73,18 +74,24 @@ class UpdateGeoLiteDbTest extends TestCase
}
#[Test, DataProvider('provideDownloaded')]
public function noticeMessageIsPrintedWhenSecondCallbackIsInvoked(
public function noticeMessageIsPrintedWhenDownloadIsFinished(
int $total,
int $downloaded,
bool $oldDbExists,
string|null $expectedMessage,
): void {
$this->dbUpdater->expects($this->once())->method('checkDbUpdate')->withAnyParameters()->willReturnCallback(
function ($_, callable $secondCallback) use ($total, $downloaded, $oldDbExists): GeolocationResult {
function (
GeolocationDownloadProgressHandlerInterface $handler,
) use (
$total,
$downloaded,
$oldDbExists,
): GeolocationResult {
// Invoke several times to ensure the log is printed only once
$secondCallback($total, $downloaded, $oldDbExists);
$secondCallback($total, $downloaded, $oldDbExists);
$secondCallback($total, $downloaded, $oldDbExists);
$handler->handleProgress($total, $downloaded, $oldDbExists);
$handler->handleProgress($total, $downloaded, $oldDbExists);
$handler->handleProgress($total, $downloaded, $oldDbExists);
return GeolocationResult::DB_UPDATED;
},