hopefully fix tests

bshaffer · bshaffer · commit 60fe1387b5e1 · 2022-04-20T13:45:00.000-06:00
diff --git a/BigQuery/tests/Unit/QueryResultsTest.php b/BigQuery/tests/Unit/QueryResultsTest.php
@@ -36,6 +36,7 @@ class QueryResultsTest extends TestCase
     public $jobId = 'myJobId';
     public $queryData = [
         'jobComplete' => true,
+        'jobReference' => ['location' => null],
         'rows' => [
             ['f' => [['v' => 'Alton'], ['v' => 1]]]
         ],
diff --git a/Firestore/tests/Unit/FirestoreSessionHandlerTest.php b/Firestore/tests/Unit/FirestoreSessionHandlerTest.php
@@ -56,7 +56,8 @@ public function setUp()
     public function testOpen()
     {
         $this->connection->beginTransaction(['database' => $this->dbName()])
-            ->shouldBeCalledTimes(1);
+            ->shouldBeCalledTimes(1)
+            ->willReturn(['transaction' => 123]);;
         $firestoreSessionHandler = new FirestoreSessionHandler(
             $this->connection->reveal(),
             $this->valueMapper->reveal(),
@@ -90,6 +91,9 @@ public function testOpenWithException()
      */
     public function testReadNotAllowed()
     {
+        $this->connection->beginTransaction(['database' => $this->dbName()])
+            ->shouldBeCalledTimes(1)
+            ->willThrow(new ServiceException(''));
         $firestoreSessionHandler = new FirestoreSessionHandler(
             $this->connection->reveal(),
             $this->valueMapper->reveal(),
@@ -124,7 +128,8 @@ public function testReadNothing()
             ->shouldBeCalledTimes(1)
             ->willReturn(null);
         $this->connection->beginTransaction(['database' => $this->dbName()])
-            ->shouldBeCalledTimes(1);
+            ->shouldBeCalledTimes(1)
+            ->willReturn(['transaction' => 123]);
         $this->connection->batchGetDocuments([
             'database' => $this->dbName(),
             'documents' => [$this->documentName()],
@@ -150,7 +155,8 @@ public function testReadNothing()
     public function testReadWithException()
     {
         $this->connection->beginTransaction(['database' => $this->dbName()])
-            ->shouldBeCalledTimes(1);
+            ->shouldBeCalledTimes(1)
+            ->willReturn(['transaction' => 123]);
         $this->connection->batchGetDocuments([
             'database' => $this->dbName(),
             'documents' => [$this->documentName()],
@@ -217,7 +223,8 @@ public function testWrite()
                 return ['data' => ['stringValue' => 'sessiondata']];
             });
         $this->connection->beginTransaction(['database' => $this->dbName()])
-            ->shouldBeCalledTimes(1);
+            ->shouldBeCalledTimes(1)
+            ->willReturn(['transaction' => 123]);
         $this->connection->commit([
             'database' => $this->dbName(),
             'writes' => [
