fix(models): handle query() returning false in result-consuming methods (#4734)

* fix(models): handle query() returning false in result-consuming methods

As of CI 4.1.2, Connection::query() returns false when a query fails
(previously an empty result object). Guard the model/helper methods that
consume the result so a failed query degrades to an empty result instead
of throwing "Call to member function getResultArray() on false".

- Tax::get_taxes()
- Sale::get_all_suspended()
- Reports\Inventory_low::getData()
- migration_helper: dropAllForeignKeyConstraints(), indexExists(),
  primaryKeyExists()

Adds QueryFalseGuardTest forcing query() to return false. Closes #3634.

* test(query-false): mock concrete driver instead of abstract BaseConnection

onlyMethods() does not stub the 16 abstract methods on BaseConnection, so
mocking it produced an invalid double. Mock the concrete driver class
(resolved from the live test connection) instead.

* fix(3634): fail loudly when migration metadata queries return false

Model read methods degrade to an empty result on a failed query, but the
migration metadata helpers (indexExists, primaryKeyExists,
dropAllForeignKeyConstraints) must not mistake a failed query for 'not
found' and proceed with schema changes. They now throw a DatabaseException
so the migration aborts loudly.

- Add an optional injectable BaseConnection param for testability.
- Add missing docstrings (docstring-coverage check).
- Add regression tests asserting the helpers throw on a failed query.

* fix(3634): use supplied connection for FK drop; stub error() in tests

Address CodeRabbit review: dropAllForeignKeyConstraints now builds its
Forge from the supplied connection (consistent with the lookup) instead of
always the default one. Test mock also stubs error() to avoid a warning on
the constructor-disabled connection.

* fix(3634): revert FK drop to Database::forge() (driver-specific forge)

new Forge($db) creates the base Forge which does not implement
dropForeignKey for the driver ('This feature is not available'), breaking
migrations. Restore Database::forge(), which loads the driver-specific
Forge via the factory. The optional connection param is retained solely
for testing the fail-loud behavior on the lookup path.

* test(3634): add regression test for Inventory_low::getData query-failure guard

Covers the low-stock report's degrade-to-empty behavior on a failed query,
addressing the CodeRabbit note that it lacked a focused test.

* test(3634): adopt test{Method}_{Purpose} naming convention

Align QueryFalseGuardTest method names with the convention from #4730
(testGetTaxes_ReturnsEmptyArrayWhenQueryFails, etc.), per review.

* test: split QueryFalseGuardTest into per-component test files

Apply CodeRabbit review: extract the shared failing-connection helpers
into tests/Support/FailedQueryConnectionTrait.php and move each test to
its component's test file (TaxTest, SaleTest, Inventory_lowTest,
migration_helperTest). Use fresh model instances instead of the shared
model() singleton so the injected failing connection does not leak into
other tests.

* style(tests): remove redundant docblocks from query-failure regression tests

---------

Co-authored-by: objecttothis <17935339+objecttothis@users.noreply.github.com>
This commit is contained in:
jekkosandobjecttothis authored and GitHub committed 2026-10-08 22:30:16 +02:00
1 parent 5e87ba7bfd
commit 9fef0bd88f
9 files changed
+255 -12

No files matched your search

+85 -9
View File
@@ -1,5 +1,7 @@
<?php
use CodeIgniter\Database\BaseConnection;
use CodeIgniter\Database\Exceptions\DatabaseException;
use Config\Database;
/**
@@ -118,13 +120,20 @@ function createPrimaryKey(string $table, string $index): void {
/**
* Drops all foreign key constraints that reference the provided table and column.
*
* A failed metadata query is treated as an error (thrown) rather than mistaken
* for "no constraints found", so callers never proceed with schema changes on a
* wrong assumption.
*
* @param string $table
* @param string $column
* @param BaseConnection|null $db Connection to use (defaults to the default connection).
*
* @return array containing the deleted constraints in case they need to be recreated after.
*
* @throws DatabaseException If the metadata query fails.
*/
function dropAllForeignKeyConstraints(string $table, string $column): array {
$db = Database::connect();
function dropAllForeignKeyConstraints(string $table, string $column, ?BaseConnection $db = null): array {
$db ??= Database::connect();
$result = $db->query("
SELECT DISTINCT
kcu.CONSTRAINT_NAME,
@@ -144,6 +153,23 @@ function dropAllForeignKeyConstraints(string $table, string $column): array {
AND rc.CONSTRAINT_NAME IS NOT NULL
");
if ($result === false) {
$detail = 'unknown error';
try {
$error = $db->error();
if (is_array($error) && !empty($error['message'])) {
$detail = (string) $error['message'];
}
} catch (\Throwable $e) {
// A disconnected or mocked connection may not support error(); keep the default.
}
throw new DatabaseException(sprintf(
'Failed to look up foreign key constraints referencing table "%s" column "%s": %s',
$table, $column, $detail
));
}
$deletedConstraints = [];
foreach ($result->getResultArray() as $constraint) {
@@ -187,22 +213,72 @@ function deleteIndex(string $table, string $index): void {
*
* @param string $table
* @param string $index
* @param BaseConnection|null $db Connection to use (defaults to the default connection).
*
* @return bool
*
* @throws DatabaseException If the metadata query fails.
*/
function indexExists(string $table, string $index): bool {
$db = Database::connect();
function indexExists(string $table, string $index, ?BaseConnection $db = null): bool {
$db ??= Database::connect();
$result = $db->query('SELECT COUNT(*) FROM information_schema.statistics WHERE table_schema = DATABASE() AND table_name = \'' . $db->getPrefix() . "$table' AND index_name = '$index'");
if ($result === false) {
$detail = 'unknown error';
try {
$error = $db->error();
if (is_array($error) && !empty($error['message'])) {
$detail = (string) $error['message'];
}
} catch (\Throwable $e) {
// A disconnected or mocked connection may not support error(); keep the default.
}
throw new DatabaseException(sprintf(
'Failed to check whether index "%s" exists on table "%s": %s',
$index, $table, $detail
));
}
$row_array = $result->getRowArray();
return $row_array && $row_array['COUNT(*)'] > 0;
return $row_array !== null && $row_array['COUNT(*)'] > 0;
}
function primaryKeyExists(string $table): bool {
$db = Database::connect();
/**
* Checks if the specified table has a primary key.
*
* @param string $table
* @param BaseConnection|null $db Connection to use (defaults to the default connection).
*
* @return bool
*
* @throws DatabaseException If the metadata query fails.
*/
function primaryKeyExists(string $table, ?BaseConnection $db = null): bool {
$db ??= Database::connect();
$result = $db->query('SELECT COUNT(*) FROM information_schema.table_constraints WHERE table_schema = DATABASE() AND table_name = \'' . $db->getPrefix() . "$table' AND constraint_type = 'PRIMARY KEY'");
if ($result === false) {
$detail = 'unknown error';
try {
$error = $db->error();
if (is_array($error) && !empty($error['message'])) {
$detail = (string) $error['message'];
}
} catch (\Throwable $e) {
// A disconnected or mocked connection may not support error(); keep the default.
}
throw new DatabaseException(sprintf(
'Failed to check whether table "%s" has a primary key: %s',
$table, $detail
));
}
$row_array = $result->getRowArray();
return $row_array && $row_array['COUNT(*)'] > 0;
return $row_array !== null && $row_array['COUNT(*)'] > 0;
}
function recreateForeignKeyConstraints(array $constraints): void {
+1 -1
View File
@@ -47,7 +47,7 @@ class Inventory_low extends Report
AND stock_locations.deleted = 0
ORDER BY items.name");
return $query->getResultArray() ?: [];
return $query ? $query->getResultArray() : [];
}
/**
+1 -1
View File
@@ -1193,7 +1193,7 @@ class Sale extends Model
. $this->db->prefixTable('sales') . ' where sale_status = ' . SUSPENDED . ' AND customer_id = ' . $customer_id);
}
return $query->getResultArray() ?: [];
return $query ? $query->getResultArray() : [];
}
/**
+1 -1
View File
@@ -127,7 +127,7 @@ class Tax extends Model
$query = $this->db->query($sql);
return $query->getResultArray() ?: [];
return $query ? $query->getResultArray() : [];
}
/**
@@ -0,0 +1,34 @@
<?php
namespace Tests\Models\Reports;
use App\Models\Reports\Inventory_low;
use CodeIgniter\Test\CIUnitTestCase;
use CodeIgniter\Test\DatabaseTestTrait;
use Tests\Support\FailedQueryConnectionTrait;
/**
* Regression test for Issue #3634: Inventory_low::getData() must degrade to an
* empty result (not throw "Call to member function ... on false") when the
* query fails.
*/
class Inventory_lowTest extends CIUnitTestCase
{
use DatabaseTestTrait;
use FailedQueryConnectionTrait;
protected $migrate = true;
protected $migrateOnce = true;
protected $refresh = false;
protected $namespace = null;
/**
* Inventory_low::getData() must return an empty array, not throw, when the query fails.
*/
public function testInventoryLowGetData_ReturnsEmptyArrayWhenQueryFails(): void
{
$report = $this->modelWithFailingQuery(Inventory_low::class);
$this->assertSame([], $report->getData([]));
}
}
+12
View File
@@ -7,6 +7,7 @@ use CodeIgniter\Database\Config;
use CodeIgniter\Test\CIUnitTestCase;
use CodeIgniter\Test\DatabaseTestTrait;
use Tests\Support\EmployeeFixtureTrait;
use Tests\Support\FailedQueryConnectionTrait;
use Tests\Support\ItemFixtureTrait;
/**
@@ -18,6 +19,7 @@ class SaleTest extends CIUnitTestCase
{
use DatabaseTestTrait;
use EmployeeFixtureTrait;
use FailedQueryConnectionTrait;
use ItemFixtureTrait;
protected $migrate = true;
@@ -434,4 +436,14 @@ class SaleTest extends CIUnitTestCase
$this->assertTrue($deleteResult);
$this->assertEqualsWithDelta(10.0, $this->getItemQuantity($itemId, self::LOCATION_ID), 0.001);
}
public function testGetAllSuspended_ReturnsEmptyArrayWhenQueryFails(): void
{
$model = $this->modelWithFailingQuery(Sale::class);
// Exercises both the NEW_ENTRY branch and the customer-id branch.
$this->assertSame([], $model->get_all_suspended());
$this->assertSame([], $model->get_all_suspended(NEW_ENTRY));
$this->assertSame([], $model->get_all_suspended(123));
}
}
+27
View File
@@ -0,0 +1,27 @@
<?php
namespace Tests\Models;
use App\Models\Tax;
use CodeIgniter\Test\CIUnitTestCase;
use CodeIgniter\Test\DatabaseTestTrait;
use Tests\Support\FailedQueryConnectionTrait;
class TaxTest extends CIUnitTestCase
{
use DatabaseTestTrait;
use FailedQueryConnectionTrait;
protected $migrate = true;
protected $migrateOnce = true;
protected $seedOnce = true;
protected $refresh = false;
protected $namespace = null;
public function testGetTaxes_ReturnsEmptyArrayWhenQueryFails(): void
{
$model = $this->modelWithFailingQuery(Tax::class);
$this->assertSame([], $model->get_taxes(1, 1));
}
}
@@ -0,0 +1,41 @@
<?php
namespace Tests\Support;
use CodeIgniter\BaseModel;
use CodeIgniter\Database\BaseConnection;
trait FailedQueryConnectionTrait
{
protected function failingConnection(): BaseConnection
{
$driverClass = get_class(\Config\Database::connect($this->DBGroup));
$db = $this->getMockBuilder($driverClass)
->disableOriginalConstructor()
->onlyMethods(['query', 'error'])
->getMock();
$db->method('query')->willReturn(false);
$db->method('error')->willReturn(['code' => 1, 'message' => 'simulated query failure']);
return $db;
}
/**
* @template T of object
*
* @param class-string<T> $class
*
* @return T
*/
protected function modelWithFailingQuery(string $class)
{
$model = new $class();
$prop = new \ReflectionProperty(BaseModel::class, 'db');
$prop->setAccessible(true);
$prop->setValue($model, $this->failingConnection());
return $model;
}
}
+53
View File
@@ -0,0 +1,53 @@
<?php
namespace Tests\helpers;
use CodeIgniter\Database\Exceptions\DatabaseException;
use CodeIgniter\Test\CIUnitTestCase;
use CodeIgniter\Test\DatabaseTestTrait;
use Tests\Support\FailedQueryConnectionTrait;
class migration_helperTest extends CIUnitTestCase
{
use DatabaseTestTrait;
use FailedQueryConnectionTrait;
protected $migrate = true;
protected $migrateOnce = true;
protected $refresh = false;
protected $namespace = null;
public function testIndexExists_ThrowsWhenQueryFails(): void
{
$this->requireMigrationHelper();
$this->expectException(DatabaseException::class);
indexExists('some_table', 'some_index', $this->failingConnection());
}
public function testPrimaryKeyExists_ThrowsWhenQueryFails(): void
{
$this->requireMigrationHelper();
$this->expectException(DatabaseException::class);
primaryKeyExists('some_table', $this->failingConnection());
}
public function testDropAllForeignKeyConstraints_ThrowsWhenQueryFails(): void
{
$this->requireMigrationHelper();
$this->expectException(DatabaseException::class);
dropAllForeignKeyConstraints('some_table', 'some_column', $this->failingConnection());
}
/**
* Ensure the migration helper functions are loaded (they are global).
*/
private function requireMigrationHelper(): void
{
if (! function_exists('indexExists')) {
require APPPATH . 'Helpers/migration_helper.php';
}
}
}