test: Fix unit tests

Automattic · Aug 28, 2024 · 5bc6d4d · 5bc6d4d
1 parent 9b801b1
commit 5bc6d4d
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/tests/unit/test-class-wc-payments-onboarding-service.php b/tests/unit/test-class-wc-payments-onboarding-service.php
@@ -193,13 +193,13 @@ public function test_add_admin_body_classes_when_onboarding() {
 	public function test_set_test_mode() {
 		$this->onboarding_service->set_test_mode( true );
 
-		$this->assertTrue( get_option( 'wcpay_onboarding_test_mode' ) );
+		$this->assertEquals( 'yes', get_option( WC_Payments_Onboarding_Service::TEST_MODE_OPTION, 'no' ) );
 
 		$this->onboarding_service->set_test_mode( false );
 
-		$this->assertFalse( get_option( 'wcpay_onboarding_test_mode' ) );
+		$this->assertEquals( 'no', get_option( WC_Payments_Onboarding_Service::TEST_MODE_OPTION, 'no' ) );
 
-		delete_option( 'wcpay_onboarding_test_mode' );
+		delete_option( WC_Payments_Onboarding_Service::TEST_MODE_OPTION );
 	}
 
 	/**