1 files changed, 148 insertions, 0 deletions
diff --git a/plugins/UserCountry/tests/System/AttributeHistoricalDataWithLocationsTest.php b/plugins/UserCountry/tests/System/AttributeHistoricalDataWithLocationsTest.php
new file mode 100644
index 0000000000..c9ceaa3791
--- /dev/null
+++ b/plugins/UserCountry/tests/System/AttributeHistoricalDataWithLocationsTest.php
@@ -0,0 +1,148 @@
+<?php
+/**
+ * Piwik - free/libre analytics platform
+ *
+ * @link http://piwik.org
+ * @license http://www.gnu.org/licenses/gpl-3.0.html GPL v3 or later
+ *
+ */
+namespace Piwik\Plugins\UserCountry\Test\Integration;
+
+use Piwik\Common;
+use Piwik\Db;
+use Piwik\Plugins\UserCountry\Commands\AttributeHistoricalDataWithLocations;
+use Piwik\Tests\Fixtures\ManyVisitsWithGeoIP;
+use Piwik\Tests\Framework\TestCase\IntegrationTestCase;
+use Symfony\Component\Console\Application;
+use Symfony\Component\Console\Tester\CommandTester;
+
+/**
+ * Class AttributeHistoricalDataWithLocationsTest
+ * @package Piwik\Plugins\UserCountry\Test\Integration
+ *
+ * @group UserCountry
+ */
+class AttributeHistoricalDataWithLocationsTest extends IntegrationTestCase
+{
+    /**
+     * @var ManyVisitsWithGeoIP
+     */
+    public static $fixture = null;
+
+    public function setUp()
+    {
+        parent::setUp();
+
+        $tablesToUpdate = array('log_visit', 'log_conversion');
+        $columnsToUpdate = array(
+            'location_country' => '"xx"',
+            'location_region' => 'NULL',
+            'location_city' => 'NULL',
+            'location_latitude' => 'NULL',
+            'location_longitude' => 'NULL'
+        );
+
+        foreach ($tablesToUpdate as $table) {
+            $sql = "UPDATE `" . Common::prefixTable($table) . "` SET ";
+
+            $sets = array();
+            foreach ($columnsToUpdate as $column => $defaultValue) {
+                $sets[] = $column . ' = ' . $defaultValue;
+            }
+
+            $sql .= implode(', ', $sets);
+
+            Db::query($sql);
+        }
+
+        self::$fixture->setLocationProvider('GeoIPCity.dat');
+    }
+
+    /**
+     * @expectedException \RuntimeException
+     * @expectedExceptionMessage  Not enough arguments
+     */
+    public function testExecute_ShouldThrowException_IfArgumentIsMissing()
+    {
+        $this->executeCommand(null);
+    }
+
+    /**
+     * @expectedException \Exception
+     * @expectedExceptionMessage General_ExceptionInvalidDateFormat
+     */
+    public function testExecute_ShouldReturnMessage_IfDatesAreInvalid()
+    {
+        $this->executeCommand('test');
+    }
+
+    public function testExecute_ShouldReturnEmptyWorkingProcessLogs_IfThereIsNoData()
+    {
+        $this->assertRegExp(
+            '/Re-attribution for date range: 2014-06-01 to 2014-06-06. 0 visits to process with provider "geoip_php"./',
+            $this->executeCommand('2014-06-01,2014-06-06')
+        );
+    }
+
+    public function testExecute_ShouldReturnLogAfterWorkingWithSomeData()
+    {
+        $result = $this->executeCommand('2010-01-03,2010-06-03');
+
+        $this->assertContains(
+            'Re-attribution for date range: 2010-01-03 to 2010-06-03. 35 visits to process with provider "geoip_php".',
+            $result
+        );
+
+        $this->assertRegExp('/100% processed. Time elapsed: [0-9.]+s/', $result);
+
+        $queryParams = array(
+            'idSite'  => self::$fixture->idSite,
+            'date'    => self::$fixture->dateTime,
+            'period'  => 'month'
+        );
+
+        $this->assertApiResponseEqualsExpected("UserCountry.getCountry", $queryParams);
+        $this->assertApiResponseEqualsExpected("UserCountry.getContinent", $queryParams);
+        $this->assertApiResponseEqualsExpected("UserCountry.getRegion", $queryParams);
+        $this->assertApiResponseEqualsExpected("UserCountry.getCity", $queryParams);
+    }
+
+    /**
+     * @param string|null $dates
+     *
+     * @return string
+     */
+    private function executeCommand($dates)
+    {
+        $command = new AttributeHistoricalDataWithLocations();
+
+        $application = new Application();
+        $application->add($command);
+
+        $commandTester = new CommandTester($command);
+
+        if (is_null($dates)) {
+            $params = array();
+        } else {
+            $params = array(AttributeHistoricalDataWithLocations::DATES_RANGE_ARGUMENT => $dates);
+        }
+
+        $params['command'] = $command->getName();
+        $commandTester->execute($params);
+        $result = $commandTester->getDisplay();
+
+        return $result;
+    }
+
+    public static function configureFixture($fixture)
+    {
+        // empty (undo IntegrationTestCase configuring)
+    }
+
+    public static function getPathToTestDirectory()
+    {
+        return __DIR__;
+    }
+}
+
+AttributeHistoricalDataWithLocationsTest::$fixture = new ManyVisitsWithGeoIP();