X-Git-Url: http://source.bookstackapp.com/bookstack/blobdiff_plain/46d71a181ee2dc2eedb8df87ade25786d14b61e2..refs/pull/5721/head:/tests/Auth/LoginAutoInitiateTest.php

diff --git a/tests/Auth/LoginAutoInitiateTest.php b/tests/Auth/LoginAutoInitiateTest.php
index 86e1f93e2..fcb4431af 100644
--- a/tests/Auth/LoginAutoInitiateTest.php
+++ b/tests/Auth/LoginAutoInitiateTest.php
@@ -25,8 +25,8 @@ class LoginAutoInitiateTest extends TestCase
 
         $req = $this->get('/login');
         $req->assertSeeText('Attempting Login');
-        $req->assertElementExists('form[action$="/oidc/login"][method=POST][id="login-form"] button');
-        $req->assertElementExists('button[form="login-form"]');
+        $this->withHtml($req)->assertElementExists('form[action$="/oidc/login"][method=POST][id="login-form"] button');
+        $this->withHtml($req)->assertElementExists('button[form="login-form"]');
     }
 
     public function test_with_saml2()
@@ -37,8 +37,8 @@ class LoginAutoInitiateTest extends TestCase
 
         $req = $this->get('/login');
         $req->assertSeeText('Attempting Login');
-        $req->assertElementExists('form[action$="/saml2/login"][method=POST][id="login-form"] button');
-        $req->assertElementExists('button[form="login-form"]');
+        $this->withHtml($req)->assertElementExists('form[action$="/saml2/login"][method=POST][id="login-form"] button');
+        $this->withHtml($req)->assertElementExists('button[form="login-form"]');
     }
 
     public function test_it_does_not_run_if_social_provider_is_active()
@@ -70,7 +70,7 @@ class LoginAutoInitiateTest extends TestCase
         config()->set([
             'auth.method' => 'oidc',
         ]);
-        $this->actingAs($this->getEditor());
+        $this->actingAs($this->users->editor());
 
         $req = $this->post('/logout');
         $req->assertRedirect('/login?prevent_auto_init=true');