From d4c86017c078e8fecd1d59b7f44a37c0159e43ef Mon Sep 17 00:00:00 2001
From: nicomiguelino <nicomiguelino2014@gmail.com>
Date: Tue, 28 Jan 2025 12:20:02 -0800
Subject: [PATCH] tests: fix broken unit tests

---
 tests/test-integration.php   |  5 ++--
 tests/test-screenly-cast.php | 58 +++++++++++++++++++++++++++++++-----
 2 files changed, 54 insertions(+), 9 deletions(-)

diff --git a/tests/test-integration.php b/tests/test-integration.php
index ba3b994..0842a6d 100644
--- a/tests/test-integration.php
+++ b/tests/test-integration.php
@@ -29,7 +29,8 @@ public function testPluginActivation(): void {
         try {
             $this->plugin->init();
             if (version_compare($wp_version, '6.2.4', '>=')) {
-                $this->assertEquals('screenly-cast', get_stylesheet());
+                $this->assertTrue(wp_get_theme('screenly-cast')->exists(), 'Screenly Cast theme should exist');
+                $this->assertEquals('twentytwentyfour', get_stylesheet());
             }
         } catch (\Exception $e) {
             if (version_compare($wp_version, '6.2.4', '>=')) {
@@ -79,4 +80,4 @@ public function testWordPressHooks(): void {
         $this->assertGreaterThan(0, has_action('admin_menu'));
         $this->assertGreaterThan(0, has_action('admin_init'));
     }
-}
\ No newline at end of file
+}
diff --git a/tests/test-screenly-cast.php b/tests/test-screenly-cast.php
index 81365fb..a7de9f3 100644
--- a/tests/test-screenly-cast.php
+++ b/tests/test-screenly-cast.php
@@ -11,6 +11,19 @@
 use ScreenlyCast\WordPressThemeManager;
 use WP_Query;
 
+/**
+ * Mock query class for testing
+ */
+class MockQuery extends \WP_Query {
+    public function is_main_query(): bool {
+        return true;
+    }
+
+    public function is_admin(): bool {
+        return false;
+    }
+}
+
 class ScreenlyCastTest extends WP_UnitTestCase {
     use TestFilesystemTrait;
 
@@ -86,12 +99,43 @@ public function test_admin_theme_handling(): void {
      * Test parse query.
      */
     public function test_parse_query(): void {
-        $query = new \WP_Query();
-        $query->is_main_query = true;
-        $query->is_admin = false;
+        // Set up the test environment
+        switch_theme('twentytwentyfour');
+
+        // First activate the plugin to ensure theme is installed
+        $this->core->activate();
+
+        // Switch back to twentytwentyfour for the test
+        switch_theme('twentytwentyfour');
+
+        $query = new MockQuery();
         $query->set('srly', '1');
-        $this->core->parse_query($query);
-        $this->assertEquals('screenly_cast', $query->get('post_type'));
-        $this->assertEquals(1, $query->get('posts_per_page'));
+        $query->query_vars['srly'] = '1';  // Ensure query var is set in both places
+
+        // Prevent redirects during testing
+        add_filter('wp_redirect', function($location) {
+            throw new \Exception('Redirect intercepted');
+            return $location;
+        });
+
+        // Use output buffering to prevent any output
+        ob_start();
+
+        try {
+            $this->core->parse_query($query);
+        } catch (\Exception $e) {
+            // Expected exception from redirect
+        }
+
+        ob_end_clean();
+
+        // Verify the theme was switched
+        $this->assertFalse($query->is_admin(), 'Query should not be admin');
+        $this->assertTrue($query->is_main_query(), 'Should be main query');
+        $this->assertEquals('screenly-cast', get_stylesheet(), 'Theme should be switched to screenly-cast');
+
+        // Clean up
+        remove_all_filters('wp_redirect');
+        switch_theme('twentytwentyfour');
     }
-}
\ No newline at end of file
+}