Used while loop for fetchColumn

microsoft · Aug 19, 2019 · ac50212 · ac50212
1 parent 3d8c68e
commit ac50212
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/test/functional/pdo_sqlsrv/PDO81_MemoryCheck.phpt b/test/functional/pdo_sqlsrv/PDO81_MemoryCheck.phpt
@@ -244,9 +244,11 @@ function RunTest($noPasses, $noRows, $tableName, $conn, $prepared, $mode)
             case 6:     // fetchColumn
                 $stmt = ExecuteQueryEx($conn, $tsql, ($prepared ? false : true));
                 $fldCount = $stmt->columnCount();
-                $result = $stmt->fetchColumn();
-                $rowCount = count($result);
-                unset($result);
+                // Check for "false" to terminate because fetchColumn may return NULL
+                while (($result = $stmt->fetchColumn()) !== false) {
+                    unset($result);
+                    $rowCount++;
+                }
                 $stmt->closeCursor();
                 unset($stmt);
                 if ($rowCount != $noRows) {