From 3bfea90f5d6f15d2cf24da67f2b7c8815fae5ded Mon Sep 17 00:00:00 2001
From: Josh Cunningham <josh@joshcanhelp.com>
Date: Thu, 1 Aug 2019 09:18:18 -0700
Subject: [PATCH] Tests for new functionality

---
 lib/WP_Auth0_LoginManager.php       |  2 +-
 tests/testLoginManagerInitAuth0.php | 45 +++++++++++++++++++++++++++++
 2 files changed, 46 insertions(+), 1 deletion(-)

diff --git a/lib/WP_Auth0_LoginManager.php b/lib/WP_Auth0_LoginManager.php
index 60c70c36..f228f1b7 100755
--- a/lib/WP_Auth0_LoginManager.php
+++ b/lib/WP_Auth0_LoginManager.php
@@ -625,7 +625,7 @@ public static function build_authorize_url( array $params = array() ) {
 	}
 
 	/**
-	 * Get a value from query_vars or $_REQUEST global.
+	 * Get a value from query_vars or request global.
 	 *
 	 * @param string $key - query var key to return.
 	 *
diff --git a/tests/testLoginManagerInitAuth0.php b/tests/testLoginManagerInitAuth0.php
index d8bc6536..b1746ede 100644
--- a/tests/testLoginManagerInitAuth0.php
+++ b/tests/testLoginManagerInitAuth0.php
@@ -110,6 +110,51 @@ public function testThatErrorInUrlStopsCallback() {
 		$this->assertContains( '<a href="https://test.auth0.com/v2/logout?client_id=__test_client_id__', $output );
 	}
 
+	public function testThatQueryVarIsFound() {
+		self::auth0Ready();
+		set_query_var( 'auth0', 1 );
+
+		$output = '';
+		try {
+			$this->login->init_auth0();
+		} catch ( Exception $e ) {
+			$output = $e->getMessage();
+		}
+
+		// This error means that the callback was triggered, which is what we are testing for.
+		$this->assertContains( 'There was a problem with your log in: Invalid state', $output );
+	}
+
+	public function testThatGetVarIsFound() {
+		self::auth0Ready();
+		$_GET['auth0'] = 1;
+
+		$output = '';
+		try {
+			$this->login->init_auth0();
+		} catch ( Exception $e ) {
+			$output = $e->getMessage();
+		}
+
+		// This error means that the callback was triggered, which is what we are testing for.
+		$this->assertContains( 'There was a problem with your log in: Invalid state', $output );
+	}
+
+	public function testThatPostVarIsFound() {
+		self::auth0Ready();
+		$_POST['auth0'] = 1;
+
+		$output = '';
+		try {
+			$this->login->init_auth0();
+		} catch ( Exception $e ) {
+			$output = $e->getMessage();
+		}
+
+		// This error means that the callback was triggered, which is what we are testing for.
+		$this->assertContains( 'There was a problem with your log in: Invalid state', $output );
+	}
+
 	/**
 	 * Test that an error in the URL parameter logs the current user out.
 	 */