private function createGeolocator(OutputInterface $output, InputInterface $input) { $providerId = $input->getOption(self::PROVIDER_ARGUMENT); $geolocator = new VisitorGeolocator(LocationProvider::getProviderById($providerId) ?: null); $usedProvider = $geolocator->getProvider(); if (!$usedProvider->isAvailable()) { throw new \InvalidArgumentException("The provider '{$providerId}' is not currently available, please make sure it is configured correctly."); } $isWorkingOrErrorMessage = $usedProvider->isWorking(); if ($isWorkingOrErrorMessage !== true) { $errorMessage = "The provider '{$providerId}' does not appear to be working correctly. Details: {$isWorkingOrErrorMessage}"; $forceGeolocation = $input->getOption(self::FORCE_OPTION); if ($forceGeolocation) { $output->writeln("<error>{$errorMessage}</error>"); $output->writeln("<comment>Ignoring location provider issue, geolocating anyway due to --force option.</comment>"); } else { throw new \InvalidArgumentException($errorMessage); } } return $geolocator; }
public function test_reattributeVisitLogs_ReattributesVisitsInDateRangeAndFromSite_AndCallsCallbackWithEveryProcessedRow() { foreach (array(1, 2) as $idSite) { foreach (array('2012-01-01', '2012-01-02', '2012-01-03', '2012-01-04') as $date) { $this->insertVisit(array('visit_last_action_time' => $date, 'idsite' => $idSite)); } } $mockLocationProvider = $this->getProviderMockThatGeolocates(array('location_country' => 'US', 'location_region' => 'rg', 'location_city' => 'the city')); $geolocator = new VisitorGeolocator($mockLocationProvider); $reattributedVisits = array(); $geolocator->reattributeVisitLogs('2012-01-02', '2012-01-04', 2, $segmentLimit = 1000, function ($row) use(&$reattributedVisits) { $reattributedVisits[] = $row['idvisit']; }); sort($reattributedVisits); $expectedVisitsVisited = array(6, 7); $this->assertEquals($expectedVisitsVisited, $reattributedVisits); // check that no visits were re-attributed for site 1 $actualVisits = Db::fetchAll("SELECT visit_last_action_time, idsite, location_country, location_region, location_city FROM " . Common::prefixTable('log_visit') . " ORDER BY idsite ASC, visit_last_action_time ASC"); $expectedVisits = array(array('visit_last_action_time' => '2012-01-01 00:00:00', 'idsite' => '1', 'location_country' => 'xx', 'location_region' => null, 'location_city' => null), array('visit_last_action_time' => '2012-01-02 00:00:00', 'idsite' => '1', 'location_country' => 'xx', 'location_region' => null, 'location_city' => null), array('visit_last_action_time' => '2012-01-03 00:00:00', 'idsite' => '1', 'location_country' => 'xx', 'location_region' => null, 'location_city' => null), array('visit_last_action_time' => '2012-01-04 00:00:00', 'idsite' => '1', 'location_country' => 'xx', 'location_region' => null, 'location_city' => null), array('visit_last_action_time' => '2012-01-01 00:00:00', 'idsite' => '2', 'location_country' => 'xx', 'location_region' => null, 'location_city' => null), array('visit_last_action_time' => '2012-01-02 00:00:00', 'idsite' => '2', 'location_country' => 'us', 'location_region' => 'rg', 'location_city' => 'the city'), array('visit_last_action_time' => '2012-01-03 00:00:00', 'idsite' => '2', 'location_country' => 'us', 'location_region' => 'rg', 'location_city' => 'the city'), array('visit_last_action_time' => '2012-01-04 00:00:00', 'idsite' => '2', 'location_country' => 'xx', 'location_region' => null, 'location_city' => null)); $this->assertEquals($expectedVisits, $actualVisits); }