| <?php |
|
|
| |
| |
| |
| |
| |
| |
|
|
| namespace Piwik\Plugins\GeoIp2\tests\Integration; |
|
|
| use Piwik\Config; |
| use Piwik\Option; |
| use Piwik\Plugins\GeoIp2\Controller; |
| use Piwik\Plugins\GeoIp2\GeoIP2AutoUpdater; |
| use Piwik\Tests\Framework\TestCase\IntegrationTestCase; |
|
|
| |
| |
| |
| |
| |
| class ControllerTest extends IntegrationTestCase |
| { |
| private $controller; |
| private $server = []; |
| private $get = []; |
| private $post = []; |
| private $request = []; |
|
|
| public function setUp(): void |
| { |
| parent::setUp(); |
|
|
| Config::getInstance()->General['enable_geolocation_admin'] = 1; |
|
|
| $this->controller = $this->getMockBuilder(Controller::class) |
| ->disableOriginalConstructor() |
| ->setMethods(['checkTokenInUrl']) |
| ->getMock(); |
| $this->controller->expects($this->any()) |
| ->method('checkTokenInUrl'); |
|
|
| $this->server = $_SERVER; |
| $this->get = $_GET; |
| $this->post = $_POST; |
| $this->request = $_REQUEST; |
|
|
| Option::delete(GeoIP2AutoUpdater::LOC_URL_OPTION_NAME); |
| Option::delete(GeoIP2AutoUpdater::ISP_URL_OPTION_NAME); |
| Option::delete('geoip2.download_url.loc'); |
| Option::delete('geoip2.download_url.isp'); |
| } |
|
|
| public function tearDown(): void |
| { |
| $_SERVER = $this->server; |
| $_GET = $this->get; |
| $_POST = $this->post; |
| $_REQUEST = $this->request; |
|
|
| Option::delete(GeoIP2AutoUpdater::LOC_URL_OPTION_NAME); |
| Option::delete(GeoIP2AutoUpdater::ISP_URL_OPTION_NAME); |
| Option::delete('geoip2.download_url.loc'); |
| Option::delete('geoip2.download_url.isp'); |
|
|
| parent::tearDown(); |
| } |
|
|
| public function testDownloadMissingGeoIpDbShouldAbortIfConfiguredUrlChangesMidDownloadAndDeleteChunks() |
| { |
| $oldLocUrl = 'https://download.db-ip.com/free/dbip-city-lite-2020-01.mmdb.gz'; |
| $newLocUrl = 'https://download.maxmind.com/app/geoip_download?edition_id=GeoLite2-City&license_key=test&suffix=tar.gz'; |
| $ispUrl = 'https://download.db-ip.com/free/dbip-asn-lite-2020-01.mmdb.gz'; |
|
|
| Option::set(GeoIP2AutoUpdater::LOC_URL_OPTION_NAME, $oldLocUrl); |
| Option::set(GeoIP2AutoUpdater::ISP_URL_OPTION_NAME, $ispUrl); |
| Option::set('geoip2.download_url.loc', $oldLocUrl); |
| Option::set('geoip2.download_url.isp', $ispUrl); |
|
|
| $locDownloadChunk = GeoIP2AutoUpdater::getTemporaryFolder( |
| GeoIP2AutoUpdater::getZippedFilenameToDownloadTo( |
| $oldLocUrl, |
| 'loc', |
| GeoIP2AutoUpdater::getGeoIPUrlExtension($oldLocUrl) |
| ), |
| true |
| ); |
| $ispDownloadChunk = GeoIP2AutoUpdater::getTemporaryFolder( |
| GeoIP2AutoUpdater::getZippedFilenameToDownloadTo( |
| $ispUrl, |
| 'isp', |
| GeoIP2AutoUpdater::getGeoIPUrlExtension($ispUrl) |
| ), |
| true |
| ); |
|
|
| file_put_contents($locDownloadChunk, 'loc'); |
| file_put_contents($ispDownloadChunk, 'isp'); |
| Option::set($locDownloadChunk . '_expectedDownloadSize', '12'); |
| Option::set($ispDownloadChunk . '_expectedDownloadSize', '12'); |
|
|
| Option::set(GeoIP2AutoUpdater::LOC_URL_OPTION_NAME, $newLocUrl); |
|
|
| $this->setPostRequest(1, 'loc'); |
| $continueResponse = $this->decodeJsonResponse($this->controller->downloadMissingGeoIpDb()); |
|
|
| $this->assertArrayHasKey('error', $continueResponse); |
| $this->assertFileNotExists($locDownloadChunk); |
| $this->assertFileExists($ispDownloadChunk); |
| $this->assertFalse(Option::get($locDownloadChunk . '_expectedDownloadSize')); |
| $this->assertSame('12', Option::get($ispDownloadChunk . '_expectedDownloadSize')); |
| $this->assertFalse(Option::get('geoip2.download_url.loc')); |
|
|
| @unlink($ispDownloadChunk); |
| Option::delete($ispDownloadChunk . '_expectedDownloadSize'); |
| } |
|
|
| private function setPostRequest(int $continue, string $key): void |
| { |
| $_SERVER['REQUEST_METHOD'] = 'POST'; |
| $_GET = [ |
| 'continue' => $continue, |
| 'key' => $key, |
| ]; |
| $_POST = $_GET; |
| $_REQUEST = $_GET; |
| } |
|
|
| private function decodeJsonResponse($response): array |
| { |
| return (array) json_decode((string) $response, true); |
| } |
| } |
|
|
