Add test for stored procedure

harrygulliford · Nov 2, 2021 · 1f81b69 · 1f81b69
1 parent 69aeb1d
commit 1f81b69
Show file tree

Hide file tree

Showing 2 changed files with 40 additions and 1 deletion.
diff --git a/tests/QueryTest.php b/tests/QueryTest.php
@@ -1217,6 +1217,16 @@ public function it_can_limit_results()
     /** @test */
     public function it_can_execute_stored_procedures()
     {
-        $this->markTestSkipped('This test needs to be written.');
+        $firstNumber = random_int(1, 10);
+        $secondNumber = random_int(1, 10);
+
+        $result = DB::query()
+            ->fromProcedure('MULTIPLY', [
+                $firstNumber, $secondNumber,
+            ])
+            ->first()
+            ->RESULT;
+
+        $this->assertEquals($firstNumber * $secondNumber, $result);
     }
 }
diff --git a/tests/Support/MigrateDatabase.php b/tests/Support/MigrateDatabase.php
@@ -18,6 +18,9 @@ public function setUp(): void
             $this->dropTables();
             $this->createTables();
 
+            $this->dropProcedure();
+            $this->createProcedure();
+
             MigrationState::$migrated = true;
         }
     }
@@ -71,4 +74,30 @@ public function dropTables(): void
             }
         }
     }
+
+    public function createProcedure()
+    {
+        DB::select(
+            'CREATE PROCEDURE MULTIPLY (a INTEGER, b INTEGER)
+                RETURNS (result INTEGER)
+            AS BEGIN
+                result = a * b;
+                SUSPEND;
+            END'
+        );
+    }
+
+    public function dropProcedure()
+    {
+        try {
+            DB::select('DROP PROCEDURE MULTIPLY');
+        } catch (QueryException $e) {
+            // Suppress the "table does not exist" exception, as we want to
+            // replicate dropIfExists() functionality without using the Schema
+            // class.
+            if (! Str::contains($e->getMessage(), 'not found')) {
+                throw $e;
+            }
+        }
+    }
 }