tests: Adjusted tests for isAIfetureActive

Yoast · Sep 12, 2024 · aa8c112 · aa8c112
1 parent 54b0812
commit aa8c112
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 3 deletions.
diff --git a/tests/Unit/Editors/Framework/Site/Post_Site_Information_Test.php b/tests/Unit/Editors/Framework/Site/Post_Site_Information_Test.php
@@ -136,6 +136,7 @@ public function test_legacy_site_information() {
 			],
 			'pluginUrl'                  => '/location',
 			'wistiaEmbedPermission'      => true,
+			'isAiFeatureActive'          => null,
 
 		];
 
@@ -191,7 +192,7 @@ public function test_site_information() {
 			'isRtl'                          => false,
 			'isPremium'                      => true,
 			'siteIconUrl'                    => 'https://example.org',
-
+			'isAiFeatureActive'              => null,
 		];
 
 		Monkey\Functions\expect( 'admin_url' )->andReturn( 'https://example.org' );

diff --git a/tests/Unit/Editors/Framework/Site/Term_Site_Information_Test.php b/tests/Unit/Editors/Framework/Site/Term_Site_Information_Test.php
@@ -127,6 +127,7 @@ public function test_site_information() {
 			'isRtl'                 => false,
 			'isPremium'             => true,
 			'siteIconUrl'           => 'https://example.org',
+			'isAiFeatureActive'     => null,
 		];
 
 		Monkey\Functions\expect( 'admin_url' )->andReturn( 'https://example.org' );
@@ -168,7 +169,7 @@ public function test_legacy_site_information() {
 			],
 			'pluginUrl'             => '/location',
 			'wistiaEmbedPermission' => true,
-
+			'isAiFeatureActive'     => null,
 		];
 
 		Monkey\Functions\expect( 'admin_url' )->andReturn( 'https://example.org' );

diff --git a/tests/WP/Editors/Framework/Site/Post_Site_Information_Test.php b/tests/WP/Editors/Framework/Site/Post_Site_Information_Test.php
@@ -124,6 +124,7 @@ public function test_legacy_site_information() {
 			'linkParams'                 => $this->short_link_helper->get_query_params(),
 			'pluginUrl'                  => 'http://example.org/wp-content/plugins/wordpress-seo',
 			'wistiaEmbedPermission'      => true,
+			'isAiFeatureActive'          => true,
 		];
 
 		$this->assertSame( $expected, $this->instance->get_legacy_site_information() );
@@ -160,7 +161,7 @@ public function test_site_information() {
 			'isRtl'                      => false,
 			'isPremium'                  => false,
 			'siteIconUrl'                => '',
-
+			'isAiFeatureActive'          => true,
 		];
 
 		$this->assertSame( $expected, $this->instance->get_site_information() );