Added PHPUnit test cases

Author: ip2location

composer.json

@@ -15,9 +15,17 @@
         "cakephp/cakephp": ">=3.0",
         "ip2location/ip2location-php": "8.*"
     },
+    "require-dev": {
+        "phpunit/phpunit": "^9"
+    },
     "autoload": {
         "psr-4": {
             "IP2LocationCakePHP\\": "src"
         }
+    },
+    "autoload-dev": {
+        "psr-4": {
+            "IP2LocationCakePHP\\Test\\": "tests"
+        }
     }
 }

phpunit.xml

@@ -0,0 +1,7 @@
+<phpunit bootstrap="tests/bootstrap.php" colors="true">
+	<testsuites>
+		<testsuite name="IP2Location CakePHP Testcase">
+			<directory suffix="Test.php">./tests/</directory>
+		</testsuite>
+	</testsuites>
+</phpunit>

src/Controller/IP2LocationCoresController.php

@@ -3,23 +3,23 @@
 
 // Web Service Settings
 if(!defined('IP2LOCATION_API_KEY')) {
-	define('IP2LOCATION_API_KEY', 'demo');
+    define('IP2LOCATION_API_KEY', 'demo');
 }
 
 if(!defined('IP2LOCATION_PACKAGE')) {
-	define('IP2LOCATION_PACKAGE', 'WS1');
+    define('IP2LOCATION_PACKAGE', 'WS1');
 }
 
 if(!defined('IP2LOCATION_USESSL')) {
-	define('IP2LOCATION_USESSL', false);
+    define('IP2LOCATION_USESSL', false);
 }
 
 if(!defined('IP2LOCATION_ADDONS')) {
-	define('IP2LOCATION_ADDONS', []);
+    define('IP2LOCATION_ADDONS', []);
 }
 
 if(!defined('IP2LOCATION_LANGUAGE')) {
-	define('IP2LOCATION_LANGUAGE', 'en');
+    define('IP2LOCATION_LANGUAGE', 'en');
 }
 
 /**
@@ -38,9 +38,13 @@ public function index()
         //
     }
 
-    public function get($ip, $query = array())
+    public function get($ip, $db = '')
     {
-        $obj = new \IP2Location\Database(ROOT . DS . 'vendor' . DS . 'ip2location' . DS . 'ip2location-cakephp' . DS . 'src' . DS . 'Data' . DS . 'IP2LOCATION.BIN', \IP2Location\Database::FILE_IO);
+        if($db == '') {
+            $obj = new \IP2Location\Database(ROOT . DS . 'vendor' . DS . 'ip2location' . DS . 'ip2location-cakephp' . DS . 'src' . DS . 'Data' . DS . 'IP2LOCATION.BIN', \IP2Location\Database::FILE_IO);
+        } else {
+            $obj = $db;
+        }
 
         try {
             $records = $obj->lookup($ip, \IP2Location\Database::ALL);

tests/ControllerTest.php

@@ -0,0 +1,30 @@
+<?php
+
+declare(strict_types=1);
+
+use PHPUnit\Framework\TestCase;
+use IP2LocationCakePHP\Controller\IP2LocationCoresController;
+
+class ControllerTest extends TestCase
+{
+	public function testGetDb() {
+		$IP2Location = new IP2LocationCoresController();
+		$db = new \IP2Location\Database('./src/Data/IP2LOCATION.BIN', \IP2Location\Database::FILE_IO);
+		$record = $IP2Location->get('8.8.8.8', $db);
+
+		$this->assertEquals(
+			'US',
+			$record['countryCode'],
+		);
+	}
+
+	public function testGetWebService() {
+		$IP2Location = new IP2LocationCoresController();
+		$record = $IP2Location->getWebService('8.8.8.8');
+
+		$this->assertEquals(
+			'US',
+			$record['country_code'],
+		);
+	}
+}

tests/bootstrap.php

@@ -0,0 +1,9 @@
+<?php
+
+declare(strict_types=1);
+
+if (!$loader = @include './vendor/autoload.php') {
+    die('Project dependencies missing');
+}
+
+$loader->add('IP2LocationCakePHP\Test', __DIR__);