test(integration): add missing mocks

lengow · Sep 3, 2024 · e632fae · e632fae
1 parent 5f621b7
commit e632fae
Show file tree

Hide file tree

Showing 5 changed files with 59 additions and 19 deletions.
diff --git a/tests/bootstrap.php b/tests/bootstrap.php
@@ -67,6 +67,18 @@ function _manually_active_plugins( $plugins ): array {
 
 const WP_ADMIN = true;
 
+if ( file_exists( $file = dirname( __FILE__, 2 ) . '/config/marketplaces.json' ) ) {
+	if ( ! unlink( $file ) ) {
+		echo 'Fail to remove file marketplaces.json' . PHP_EOL;
+	}
+}
+
+if ( file_exists( $file = dirname( __FILE__, 2 ) . '/export/flux.csv' ) ) {
+	if ( ! unlink( $file ) ) {
+		echo 'Fail to remove file flux.csv' . PHP_EOL;
+	}
+}
+
 // Start up the WP testing environment.
 require "{$_tests_dir}/includes/bootstrap.php";
 

diff --git a/tests/integration/test-admin.php b/tests/integration/test-admin.php
@@ -46,6 +46,7 @@ public function test_admin_dashboard() {
 	public function test_admin_dashboard_end_free_trial() {
 		$this->init_with_mock_client();
 		$this->mock_on_access_token();
+		$this->mock_basic_stuff( [ Api\Restriction::API ] );
 
 		$json = json_decode( file_get_contents(
 			$this->sdk_mock_dir . 'restriction-restrictions.json'

diff --git a/tests/integration/test-connect-cms.php b/tests/integration/test-connect-cms.php
@@ -1,6 +1,8 @@
 <?php
 declare(strict_types=1);
 
+use Lengow\Sdk\Resource\Api;
+
 /**
  * Class Test_Connect_Cms
  *
@@ -50,6 +52,7 @@ public function test_will_connect() {
 	public function test_will_fail() {
 		$this->init_with_mock_client( true, true );
 		$this->mock_on_access_token_fail();
+		$this->mock_basic_stuff( [], [ Api\Plugin::API ] );
 
 		$post_data = [
 			'do_action' => 'connect_cms',

diff --git a/tests/integration/test-cron.php b/tests/integration/test-cron.php
@@ -33,6 +33,7 @@ protected function mock_marketplace() {
 	public function test_import_order() {
 		$this->init_with_mock_client();
 		$this->mock_on_access_token();
+		$this->mock_basic_stuff();
 		$this->mock_marketplace();
 		$this->mock_order_list();
 

diff --git a/tests/integration/trait-mock-client.php b/tests/integration/trait-mock-client.php
@@ -63,27 +63,50 @@ public function init_with_mock_client( bool $without_token = true, bool $without
 		}, true);
 	}
 
-	public function mock_basic_stuff() {
-		$this->mock_client->on(
-			new RequestMatcher( Api\Cms::API, null, [ 'GET' ] ),
-			new Response( 200, [], file_get_contents(
-				$this->sdk_mock_dir . 'cms-list.json'
-			) )
-		);
+	public function mock_basic_stuff( array $exclude = [], array $include = [] ) {
 
-		$this->mock_client->on(
-			new RequestMatcher( Api\Cms::API, null, [ 'PUT', 'POST' ] ),
-			new Response( 200, [], file_get_contents(
-				$this->sdk_mock_dir . 'cms.json'
-			) )
-		);
+		if (
+			! in_array( Api\Cms::API, $exclude )
+			&& ( empty( $include ) || in_array( Api\Cms::API, $include ) )
+		) {
+			$this->mock_client->on(
+				new RequestMatcher( Api\Cms::API, null, [ 'GET' ] ),
+				new Response( 200, [], file_get_contents(
+					$this->sdk_mock_dir . 'cms-list.json'
+				) )
+			);
 
-		$this->mock_client->on(
-			new RequestMatcher( Api\Restriction::API ),
-			new Response( 200, [], file_get_contents(
-				$this->sdk_mock_dir . 'restriction-restrictions.json'
-			) )
-		);
+			$this->mock_client->on(
+				new RequestMatcher( Api\Cms::API, null, [ 'PUT', 'POST' ] ),
+				new Response( 200, [], file_get_contents(
+					$this->sdk_mock_dir . 'cms.json'
+				) )
+			);
+		}
+
+		if (
+			! in_array( Api\Restriction::API, $exclude )
+			&& ( empty( $include ) || in_array( Api\Restriction::API, $include ) )
+		) {
+			$this->mock_client->on(
+				new RequestMatcher( Api\Restriction::API ),
+				new Response( 200, [], file_get_contents(
+					$this->sdk_mock_dir . 'restriction-restrictions.json'
+				) )
+			);
+		}
+
+		if (
+			! in_array( Api\Plugin::API, $exclude )
+			&& ( empty( $include ) || in_array( Api\Plugin::API, $include ) )
+		) {
+			$this->mock_client->on(
+				new RequestMatcher( Api\Plugin::API, null, [ 'GET' ] ),
+				new Response( 200, [], file_get_contents(
+					$this->sdk_mock_dir . 'plugin-list.json'
+				) )
+			);
+		}
 	}
 
 	public function mock_on_access_token() {