Add and clean up test

silinternational · Aug 16, 2023 · c5c32e4 · c5c32e4
1 parent e9becb1
commit c5c32e4
Show file tree

Hide file tree

Showing 2 changed files with 40 additions and 4 deletions.
diff --git a/SilMock/Google/Service/Directory/Resource/TwoStepVerification.php b/SilMock/Google/Service/Directory/Resource/TwoStepVerification.php
@@ -17,21 +17,27 @@ public function __construct(string $dbFile = '')
      * NOTE: This doesn't need to work. It just needs to exist.
      *
      * @param $userKey
-     * @param $optParams
+     * @param array $optParams
      * @return void
      */
     public function turnOff($userKey, $optParams = [])
     {
-        // Confirm verification codes exist.
+        // Grab the corresponding twoStepVerification record.
         $sqliteUtils = $this->getSqliteUtils();
         $twoStepVerificationRecord = $sqliteUtils->getAllRecordsByDataKey(
             $this->dataType,
             $this->dataClass,
             'twoStepVerification',
             $userKey
         );
-        $recordId = $twoStepVerificationRecord['id'];
+        // Update it
         $twoStepVerificationRecord['onOrOff'] = 'off';
-        $sqliteUtils->updateRecordById($recordId, json_encode($twoStepVerificationRecord));
+        // Get the record id and update it, as needed.
+        $recordId = $twoStepVerificationRecord['id'];
+        // If there was a recordId, then it probably would be functional
+        // However, we just need this method to exist, not actually work
+        if (! empty($recordId)) {
+            $sqliteUtils->updateRecordById($recordId, json_encode($twoStepVerificationRecord));
+        }
     }
 }
diff --git a/SilMock/tests/Google/Service/Directory/Resource/TwoStepVerificationTest.php b/SilMock/tests/Google/Service/Directory/Resource/TwoStepVerificationTest.php
@@ -0,0 +1,30 @@
+<?php
+
+namespace Service\Directory\Resource;
+
+use Exception;
+use PHPUnit\Framework\TestCase;
+use SilMock\Google\Service\Directory\Resource\TwoStepVerification;
+
+class TwoStepVerificationTest extends TestCase
+{
+    public $dataFile = DATAFILE2;
+
+    public function testTwoStepVerificationTurnOff()
+    {
+        $twoStepVerfication = new TwoStepVerification($this->dataFile);
+        $this->assertIsObject($twoStepVerfication, 'Unable to instantiate twoStepVerification Mock object');
+
+        try {
+            $twoStepVerfication->turnOff('[email protected]');
+        } catch (Exception $exception) {
+            $this->assertFalse(
+                true,
+                sprintf(
+                    'Was expecting the turnOff method to function, but got: %s',
+                    $exception->getMessage()
+                )
+            );
+        }
+    }
+}