Added tests

tests/Fixtures/inc/ThirdParty/Plugins/PageBuilder/Elementor/maybeClearCacheChangeNotice.php

@@ -1,22 +1,32 @@
 <?php
+
+$input = file_get_contents(__DIR__ . '/HTML/input.html');
+$output = file_get_contents(__DIR__ . '/HTML/output.html');
+$output_added = file_get_contents(__DIR__ . '/HTML/output_added.html');
 return [
-    '' => [
-        'config' => [
-			'boxes' => [],
-			'user_id' => 10,
-			'can' => true,
-			'transient' => true,
-			'notice' => true,
-        ],
-		'expected' => [
-			'notice' => [
-				'status'         => 'warning',
-				'dismissible'    => '',
-				'dismiss_button' => 'maybe_clear_cache_change_notice',
-				'message'        => '<strong>WP Rocket:</strong> Your Elementor template was updated. Clear the Used CSS if the layout, design or CSS styles were changed.',
-				'action'         => 'clear_cache',
-			]
-		]
-    ],
+	'AddScript' => [
+		'config' => [
+			'html' => $input,
+			'script' => 'script',
+			'is_allowed' => true,
+		],
+		'expected' => $output_added
+	],
+	'NotAllowedShouldReturnSame' => [
+		'config' => [
+			'html' => $input,
+			'script' => 'script',
+			'is_allowed' => false,
+		],
+		'expected' => $output
+	],
+	'NoScriptShould' => [
+		'config' => [
+			'html' => $input,
+			'script' => false,
+			'is_allowed' => true,
+		],
+		'expected' => $output
+	],
 
 ];

tests/Fixtures/inc/ThirdParty/Plugins/PageBuilder/Elementor/setupTransient.php

@@ -10,7 +10,7 @@
 			'remove_unused_css' => true,
 			'user_id' => 10,
 			'boxes' => [
-				'maybe_clear_cache_change_notice' => true
+				'maybe_clear_cache_change_notice'
 			],
 			'change' => true,
 		],
@@ -31,15 +31,15 @@
 			'remove_unused_css' => true,
 			'user_id' => 10,
 			'boxes' => [
-				'maybe_clear_cache_change_notice' => true
+				'maybe_clear_cache_change_notice'
 			],
 			'change' => false,
 		],
 		'expected' => [
 			'user_id' => 10,
 			'transient' => false,
 			'boxes' => [
-				'maybe_clear_cache_change_notice' => true
+				'maybe_clear_cache_change_notice'
 			]
 		]
     ],
@@ -53,15 +53,15 @@
 			'remove_unused_css' => false,
 			'user_id' => 10,
 			'boxes' => [
-				'maybe_clear_cache_change_notice' => true
+				'maybe_clear_cache_change_notice'
 			],
 			'change' => false,
 		],
 		'expected' => [
 			'user_id' => 10,
 			'transient' => false,
 			'boxes' => [
-				'maybe_clear_cache_change_notice' => true
+				'maybe_clear_cache_change_notice'
 			]
 		]
 	],
@@ -75,15 +75,15 @@
 			'remove_unused_css' => true,
 			'user_id' => 10,
 			'boxes' => [
-				'maybe_clear_cache_change_notice' => true
+				'maybe_clear_cache_change_notice'
 			],
 			'change' => false,
 		],
 		'expected' => [
 			'user_id' => 10,
 			'transient' => false,
 			'boxes' => [
-				'maybe_clear_cache_change_notice' => true
+				'maybe_clear_cache_change_notice'
 			]
 		]
 	],

tests/Unit/inc/ThirdParty/Plugins/PageBuilder/Elementor/ElementorTestTrait.php

@@ -0,0 +1,51 @@
+<?php
+
+namespace ThirdParty\Plugins\PageBuilder\Elementor;
+use Mockery;
+use WP_Filesystem_Direct;
+use WP_Rocket\Admin\Options_Data;
+use WP_Rocket\Engine\Common\Ajax\AjaxHandler;
+use WP_Rocket\Engine\Optimization\DelayJS\HTML;
+use WP_Rocket\Engine\Optimization\RUCSS\Controller\UsedCSS;
+use WP_Rocket\ThirdParty\Plugins\PageBuilder\Elementor;
+
+trait ElementorTestTrait
+{
+	/**
+	 * @var Options_Data
+	 */
+	protected $options;
+
+	protected $filesystem;
+
+	/**
+	 * @var HTML
+	 */
+	protected $delayjs_html;
+
+	/**
+	 * @var UsedCSS
+	 */
+	protected $used_css;
+
+	/**
+	 * @var AjaxHandler
+	 */
+	protected $ajax_handler;
+
+	/**
+	 * @var Elementor
+	 */
+	protected $elementor;
+
+	public function set_up() {
+		parent::set_up();
+		$this->options = Mockery::mock(Options_Data::class);
+		$this->filesystem = Mockery::mock(WP_Filesystem_Direct::class);
+		$this->delayjs_html = Mockery::mock(HTML::class);
+		$this->used_css = Mockery::mock(UsedCSS::class);
+		$this->ajax_handler = Mockery::mock(AjaxHandler::class);
+
+		$this->elementor = new Elementor($this->options, $this->filesystem, $this->delayjs_html, $this->used_css, $this->ajax_handler);
+	}
+}

tests/Unit/inc/ThirdParty/Plugins/PageBuilder/Elementor/addFixAnimationScript.php

@@ -3,6 +3,7 @@
 namespace WP_Rocket\Tests\Unit\inc\ThirdParty\Plugins\PageBuilder\Elementor;
 
 use Mockery;
+use ThirdParty\Plugins\PageBuilder\Elementor\ElementorTestTrait;
 use WP_Filesystem_Direct;
 use WP_Rocket\ThirdParty\Plugins\PageBuilder\Elementor;
 use WP_Rocket\Admin\Options_Data;
@@ -18,37 +19,7 @@
  */
 class Test_addFixAnimationScript extends TestCase {
 
-    /**
-     * @var Options_Data
-     */
-    protected $options;
-
-    protected $filesystem;
-
-    /**
-     * @var HTML
-     */
-    protected $delayjs_html;
-
-    /**
-     * @var UsedCSS
-     */
-    protected $used_css;
-
-    /**
-     * @var Elementor
-     */
-    protected $elementor;
-
-    public function set_up() {
-        parent::set_up();
-        $this->options = Mockery::mock(Options_Data::class);
-        $this->filesystem = Mockery::mock(WP_Filesystem_Direct::class);
-        $this->delayjs_html = Mockery::mock(HTML::class);
-        $this->used_css = Mockery::mock(UsedCSS::class);
-
-        $this->elementor = new Elementor($this->options, $this->filesystem, $this->delayjs_html, $this->used_css);
-    }
+   use ElementorTestTrait;
 
     /**
      * @dataProvider configTestData

tests/Unit/inc/ThirdParty/Plugins/PageBuilder/Elementor/clearCache.php

@@ -3,6 +3,7 @@
 
 use Mockery;
 use Brain\Monkey\Functions;
+use ThirdParty\Plugins\PageBuilder\Elementor\ElementorTestTrait;
 use WP_Rocket\Admin\Options_Data;
 use WP_Rocket\Engine\Optimization\DelayJS\HTML;
 use WP_Rocket\Tests\Unit\TestCase;
@@ -15,13 +16,7 @@
  * @group ThirdParty
  */
 class Test_ClearCache extends TestCase {
-	private $elementor;
-
-	public function setUp() : void {
-		parent::setUp();
-
-		$this->elementor = new Elementor( Mockery::mock( Options_Data::class ), null, Mockery::mock( HTML::class ), Mockery::mock( UsedCSS::class ) );
-	}
+	use ElementorTestTrait;
 
 	public function testShouldDoNothingWhenNotExternal() {
 		Functions\when( 'get_option' )->justReturn( 'internal' );

tests/Unit/inc/ThirdParty/Plugins/PageBuilder/Elementor/clearRelatedPostCache.php

@@ -3,6 +3,7 @@
 
 use Mockery;
 use Brain\Monkey\Functions;
+use ThirdParty\Plugins\PageBuilder\Elementor\ElementorTestTrait;
 use WP_Rocket\Tests\Unit\TestCase;
 use WP_Rocket\Admin\Options_Data;
 use WP_Rocket\Engine\Optimization\DelayJS\HTML;
@@ -15,10 +16,7 @@
  * @group  HealthCheck
  */
 class Test_ClearRelatedPostCache extends TestCase {
-    private $elementor;
-	private $wpdb;
-    private $options;
-    private $used_css;
+	use ElementorTestTrait;
 
 	public static function setUpBeforeClass(): void {
 		parent::setUpBeforeClass();
@@ -36,9 +34,6 @@ protected function setUp(): void {
 			require_once WP_ROCKET_PLUGIN_ROOT . 'inc/common/purge.php';
 		}
 
-        $this->options = Mockery::mock( Options_Data::class );
-        $this->used_css = Mockery::mock( UsedCSS::class );
-		$this->elementor = new Elementor( $this->options, null, Mockery::mock( HTML::class ), $this->used_css );
 		$GLOBALS['wpdb'] = $this->wpdb = new wpdb();
 	}
 
@@ -64,7 +59,7 @@ public function testShouldDoAsExpected( $config ) {
             Functions\expect( 'get_post_status' )
                 ->twice()
                 ->andReturn( $config['post_status'] );
-                
+
             foreach ( $config['results'] as $result ) {
                 if ( 'publish' === $config['post_status'] ) {
                     $this->options->shouldReceive( 'get' )

tests/Unit/inc/ThirdParty/Plugins/PageBuilder/Elementor/excludeJs.php

@@ -3,6 +3,7 @@
 
 use Mockery;
 use Brain\Monkey\Functions;
+use ThirdParty\Plugins\PageBuilder\Elementor\ElementorTestTrait;
 use WP_Rocket\Tests\Unit\TestCase;
 use WP_Rocket\Admin\Options_Data;
 use WP_Rocket\Engine\Optimization\DelayJS\HTML;
@@ -16,15 +17,7 @@
  * @group ThirdParty
  */
 class Test_ExcludeJs extends TestCase {
-	private $options;
-	private $elementor;
-
-	public function setUp(): void {
-		parent::setUp();
-
-		$this->options   = Mockery::mock( Options_Data::class );
-		$this->elementor = new Elementor( $this->options, null, Mockery::mock( HTML::class ), Mockery::mock( UsedCSS::class ) );
-	}
+	use ElementorTestTrait;
 
 	/**
 	 * @dataProvider configTestData