Fix method resetSequence(). (#130)

terabytesoftw · web-flow · commit 76e0fe1a7e16 · 2022-10-14T11:06:36.000-03:00
* Fix method resetSequence().
diff --git a/src/DMLQueryBuilder.php b/src/DMLQueryBuilder.php
@@ -35,10 +35,8 @@ public function __construct(
 
     /**
      * @throws Exception|Throwable
-     *
-     * @psalm-suppress RiskyCast
      */
-    public function resetSequence(string $tableName, mixed $value = null): string
+    public function resetSequence(string $tableName, int|string $value = null): string
     {
         $table = $this->schema->getTableSchema($tableName);
 
@@ -47,11 +45,13 @@ public function resetSequence(string $tableName, mixed $value = null): string
         }
 
         $sequenceName = $table->getSequenceName();
+
         if ($sequenceName === null) {
             throw new InvalidArgumentException("There is not sequence associated with table '$tableName'.'");
         }
 
         $tableName = $this->quoter->quoteTableName($tableName);
+
         if ($value !== null) {
             $value = "'" . ((int) $value - 1) . "'";
         } else {
diff --git a/tests/QueryBuilderTest.php b/tests/QueryBuilderTest.php
@@ -278,6 +278,10 @@ public function testResetSequence(): void
         $this->assertEquals(1, $result);
 
         // change up
+        $expected = "UPDATE sqlite_sequence SET seq='0' WHERE name='testCreateTable'";
+        $sql = $qb->resetSequence('testCreateTable', '1');
+        $this->assertEquals($expected, $sql);
+
         $expected = "UPDATE sqlite_sequence SET seq='3' WHERE name='testCreateTable'";
         $sql = $qb->resetSequence('testCreateTable', 4);
         $this->assertEquals($expected, $sql);
