Update auth tests to match #3034 (#3035)

YOURLS · Aug 24, 2021 · 7e06aa1 · 7e06aa1
1 parent 6ae5472
commit 7e06aa1
Show file tree

Hide file tree

Showing 2 changed files with 37 additions and 8 deletions.
diff --git a/tests/tests/auth/login_base.php b/tests/tests/auth/login_base.php
@@ -10,10 +10,12 @@ abstract class Login_Base extends PHPUnit\Framework\TestCase {
 
     protected function setUp(): void {
         $this->backup_request = $_REQUEST;
+        $_REQUEST['nonce'] = yourls_create_nonce('admin_login');
     }
 
     protected function tearDown(): void {
         $_REQUEST = $this->backup_request;
+        yourls_remove_all_actions('pre_yourls_die');
     }
 
 	/**
@@ -86,10 +88,15 @@ public function test_login_with_random_credentials() {
         $login        = yourls_did_action( 'login' );
         $login_failed = yourls_did_action( 'login_failed' );
 
-        $this->assertNotTrue( yourls_is_valid_user() );
+        // with "normal" logins, we simulate the login forms and the presence of a nonce
+        if (get_class($this) == 'Auth_Login_Normal_Tests') {
+            $this->expectException(Exception::class);
+            $this->expectExceptionMessage('I have died');
+            // intercept yourls_die() before it actually dies
+            yourls_add_action( 'pre_yourls_die', function() { throw new Exception( 'I have died' ); } );
+        }
 
-		$this->assertEquals( $login, yourls_did_action( 'login' ) );
-		$this->assertEquals( $login_failed + 1, yourls_did_action( 'login_failed' ) );
+        $this->assertNotTrue( yourls_is_valid_user() );
     }
 
 }
diff --git a/tests/tests/auth/logout.php b/tests/tests/auth/logout.php
@@ -7,19 +7,41 @@
  */
 class Logout_Func_Tests extends PHPUnit\Framework\TestCase {
 
+    protected $backup_get;
+
+    protected function setUp(): void {
+        $this->backup_get = $_GET;
+        $_REQUEST['nonce'] = yourls_create_nonce('admin_login');
+    }
+
+    protected function tearDown(): void {
+        $_GET = $this->backup_get;
+        yourls_remove_all_actions('pre_yourls_die');
+    }
+
     /**
-     * Check logout procedure
+     * Check logout procedure - phase 1
      */
-    public function test_logout() {
-
+    public function test_logout_user_is_logged_in() {
         $valid = yourls_is_valid_user();
-        $this->assertTrue( $valid );
+        $this->assertTrue($valid);
+    }
 
+    /**
+     * Check logout procedure - phase 2
+     * @depends test_logout_user_is_logged_in
+     */
+    public function test_logout_user_logs_out() {
         $_GET['action'] = 'logout';
         $invalid = yourls_is_valid_user();
         $this->assertNotTrue( $invalid );
+    }
 
-        unset( $_GET['action'] );
+    /**
+     * Check logout procedure - phase 3
+     * @depends test_logout_user_logs_out
+     */
+    public function test_logout_user_is_logged_in_back() {
         $valid = yourls_is_valid_user();
         $this->assertTrue( $valid );
     }