Added PHPUnit test cases

Author: ip2location

composer.json

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

phpunit.xml

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

src/Controller/IP2ProxyCoresController.php

@@ -30,10 +30,15 @@ public function index()
         //
     }
 
-    public function get($ip, $query = array())
+    public function get($ip, $db = '')
     {
         $obj = new \IP2Proxy\Database();
-        $obj->open(ROOT . DS . 'vendor' . DS . 'ip2location' . DS . 'ip2proxy-cakephp' . DS . 'src' . DS . 'Data' . DS . 'IP2PROXY.BIN', \IP2Proxy\Database::FILE_IO);
+
+        if($db == '') {
+            $obj->open(ROOT . DS . 'vendor' . DS . 'ip2location' . DS . 'ip2proxy-cakephp' . DS . 'src' . DS . 'Data' . DS . 'IP2PROXY.BIN', \IP2Proxy\Database::FILE_IO);
+        } else {
+            $obj->open($db, \IP2Proxy\Database::FILE_IO);
+        }
 
         try {
             $records = $obj->getAll($ip);

tests/ControllerTest.php

@@ -0,0 +1,30 @@
+<?php
+
+declare(strict_types=1);
+
+use PHPUnit\Framework\TestCase;
+use IP2ProxyCakePHP\Controller\IP2ProxyCoresController;
+
+class ControllerTest extends TestCase
+{
+	public function testGetDb() {
+		$IP2Proxy = new IP2ProxyCoresController();
+		$db = './src/Data/IP2PROXY.BIN';
+		$record = $IP2Proxy->get('1.0.241.135', $db);
+
+		$this->assertEquals(
+			'TH',
+			$record['countryCode'],
+		);
+	}
+
+	public function testGetWebService() {
+		$IP2Proxy = new IP2ProxyCoresController();
+		$record = $IP2Proxy->getWebService('1.0.241.135');
+
+		$this->assertEquals(
+			'TH',
+			$record['countryCode'],
+		);
+	}
+}

tests/bootstrap.php

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