GeoIP2-php/tests/GeoIp2/Test/Database/ReaderTest.php

<?php

namespace GeoIp2\Test\WebService;

use GeoIp2\Database\Reader;

class ReaderTest extends \PHPUnit_Framework_TestCase
{
    public function testDefaultLocale()
    {
        foreach (array('City', 'Country') as $type) {
            $reader = new Reader("maxmind-db/test-data/GeoIP2-$type-Test.mmdb");
            $method = lcfirst($type);
            $record = $reader->$method('81.2.69.160');
            $this->assertEquals('United Kingdom', $record->country->name);
        }
        $reader->close();
    }

    public function testLocaleList()
    {

        foreach (array('City', 'Country') as $type) {
            $reader = new Reader(
                "maxmind-db/test-data/GeoIP2-$type-Test.mmdb",
                array('xx', 'ru', 'pt-BR', 'es', 'en')
            );
            $method = lcfirst($type);

            $record = $reader->$method('81.2.69.160');
            $this->assertEquals('Великобритания', $record->country->name);
        }
        $reader->close();
    }

    public function testHasIpAddress()
    {
        foreach (array('City', 'Country') as $type) {
            $reader = new Reader("maxmind-db/test-data/GeoIP2-$type-Test.mmdb");
            $method = lcfirst($type);
            $record = $reader->$method('81.2.69.160');
            $this->assertEquals('81.2.69.160', $record->traits->ipAddress);
        }
        $reader->close();
    }

    /**
     * @expectedException GeoIp2\Exception\AddressNotFoundException
     * @expectedExceptionMessage The address 10.10.10.10 is not in the database.
     */
    public function testUnknownAddress()
    {
        $reader = new Reader('maxmind-db/test-data/GeoIP2-City-Test.mmdb');
        $reader->city('10.10.10.10');
        $reader->close();
    }

    /**
     * @expectedException BadMethodCallException
     * @expectedExceptionMessage The country method cannot be used to open a GeoIP2-City database
     */
    public function testIncorrectDatabase()
    {
        $reader = new Reader('maxmind-db/test-data/GeoIP2-City-Test.mmdb');
        $reader->country('10.10.10.10');
        $reader->close();
    }

    /**
     * @expectedException BadMethodCallException
     * @expectedExceptionMessage The domain method cannot be used to open a GeoIP2-City database
     */
    public function testIncorrectDatabaseFlat()
    {
        $reader = new Reader('maxmind-db/test-data/GeoIP2-City-Test.mmdb');
        $reader->domain('10.10.10.10');
        $reader->close();
    }

    /**
     * @expectedException InvalidArgumentException
     * @expectedExceptionMessage is not a valid IP address
     */
    public function testInvalidAddress()
    {
        $reader = new Reader('maxmind-db/test-data/GeoIP2-City-Test.mmdb');
        $reader->city('invalid');
        $reader->close();
    }


    public function testConnectionType()
    {
        $reader = new Reader('maxmind-db/test-data/GeoIP2-Connection-Type-Test.mmdb');
        $ipAddress = '1.0.1.0';

        $record = $reader->connectionType($ipAddress);
        $this->assertEquals('Cable/DSL', $record->connectionType);
        $this->assertEquals($ipAddress, $record->ipAddress);
        $reader->close();
    }

    public function testDomain()
    {
        $reader = new Reader('maxmind-db/test-data/GeoIP2-Domain-Test.mmdb');

        $ipAddress = '1.2.0.0';
        $record = $reader->domain($ipAddress);
        $this->assertEquals('maxmind.com', $record->domain);
        $this->assertEquals($ipAddress, $record->ipAddress);
        $reader->close();
    }

    public function testIsp()
    {
        $reader = new Reader('maxmind-db/test-data/GeoIP2-ISP-Test.mmdb');

        $ipAddress = '1.128.0.0';
        $record = $reader->isp($ipAddress);
        $this->assertEquals(1221, $record->autonomousSystemNumber);
        $this->assertEquals(
            'Telstra Pty Ltd',
            $record->autonomousSystemOrganization
        );

        $this->assertEquals('Telstra Internet', $record->isp);
        $this->assertEquals('Telstra Internet', $record->organization);

        $this->assertEquals($ipAddress, $record->ipAddress);
        $reader->close();
    }

    public function testMetadata()
    {
        $reader = new Reader('maxmind-db/test-data/GeoIP2-City-Test.mmdb');
        $this->assertEquals('GeoIP2-City', $reader->metadata()->databaseType);

        $reader->close();
    }
}