plugins/UserCountry/tests/Integration/APITest.php


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85

<?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\tests;

use Piwik\Access;
use Piwik\Common;
use Piwik\Config;
use Piwik\Plugins\UserCountry\API;
use Piwik\Plugins\GeoIp2\LocationProvider\GeoIp2;
use Piwik\Plugins\UserCountry\LocationProvider;
use Piwik\Plugins\UserCountry\LocationProvider\DefaultProvider;
use Piwik\Tests\Framework\TestCase\IntegrationTestCase;

/**
 * @group UserCountry
 * @group APITest
 * @group Plugins
 */
class APITest extends IntegrationTestCase
{
    /**
     * @var API
     */
    private $api;
    
    public function setUp()
    {
        parent::setUp();

        $this->api = API::getInstance();

        // reset location providers as they might be manipulated by other tests
        LocationProvider::$providers = null;
        LocationProvider::getAllProviders();
    }

    public function test_setLocationProvider()
    {
        GeoIp2::$geoIPDatabaseDir = 'tests/lib/geoip-files';
        $locationProvider = GeoIp2\Php::ID;
        $this->api->setLocationProvider($locationProvider);
        $this->assertEquals($locationProvider, Common::getCurrentLocationProviderId());

        $locationProvider = DefaultProvider::ID;
        $this->api->setLocationProvider($locationProvider);
        $this->assertEquals($locationProvider, Common::getCurrentLocationProviderId());
    }

    /**
     * @expectedException \Exception
     */
    public function test_setLocationProviderInvalid()
    {
        $locationProvider = 'invalidProvider';
        $this->api->setLocationProvider($locationProvider);
    }

    /**
     * @expectedException \Exception
     */
    public function test_setLocationProviderNoSuperUser()
    {
        Access::getInstance()->setSuperUserAccess(false);

        $locationProvider = LocationProvider\GeoIp\Php::ID;
        $this->api->setLocationProvider($locationProvider);
    }

    /**
     * @expectedException \Exception
     */
    public function test_setLocationProviderDisabledInConfig()
    {
        Config::getInstance()->General['enable_geolocation_admin'] = 0;

        $locationProvider = LocationProvider\GeoIp\Php::ID;
        $this->api->setLocationProvider($locationProvider);
    }
}