added routes for zip export

[bookstack] / tests / Auth / MfaConfigurationTest.php

diff --git a/tests/Auth/MfaConfigurationTest.php b/tests/Auth/MfaConfigurationTest.php
index adeb66189011ec59f4db38f9de081e100dad88ea..1f359b41a103622ea2e807871ed2b84539f84713 100644 (file)
--- a/tests/Auth/MfaConfigurationTest.php
+++ b/tests/Auth/MfaConfigurationTest.php
@@ -2,46 +2,49 @@
 
 namespace Tests\Auth;
 
 
 namespace Tests\Auth;
 

-use BookStack\Auth\Access\Mfa\MfaValue;
-use BookStack\Auth\User;
+use BookStack\Access\Mfa\MfaValue;
+use BookStack\Activity\ActivityType;
+use BookStack\Users\Models\Role;
+use BookStack\Users\Models\User;

 use PragmaRX\Google2FA\Google2FA;
 use Tests\TestCase;
 
 class MfaConfigurationTest extends TestCase
 {
 use PragmaRX\Google2FA\Google2FA;
 use Tests\TestCase;
 
 class MfaConfigurationTest extends TestCase
 {

-

     public function test_totp_setup()
     {
     public function test_totp_setup()
     {

-        $editor = $this->getEditor();
+        $editor = $this->users->editor();

         $this->assertDatabaseMissing('mfa_values', ['user_id' => $editor->id]);
 
         // Setup page state
         $resp = $this->actingAs($editor)->get('/mfa/setup');
         $this->assertDatabaseMissing('mfa_values', ['user_id' => $editor->id]);
 
         // Setup page state
         $resp = $this->actingAs($editor)->get('/mfa/setup');

-        $resp->assertElementContains('a[href$="/mfa/totp-generate"]', 'Setup');
+        $this->withHtml($resp)->assertElementContains('a[href$="/mfa/totp/generate"]', 'Setup');

 
         // Generate page access
 
         // Generate page access

-        $resp = $this->get('/mfa/totp-generate');
+        $resp = $this->get('/mfa/totp/generate');

         $resp->assertSee('Mobile App Setup');
         $resp->assertSee('Verify Setup');
         $resp->assertSee('Mobile App Setup');
         $resp->assertSee('Verify Setup');

-        $resp->assertElementExists('form[action$="/mfa/totp-confirm"] button');
+        $this->withHtml($resp)->assertElementExists('form[action$="/mfa/totp/confirm"] button');

         $this->assertSessionHas('mfa-setup-totp-secret');
         $this->assertSessionHas('mfa-setup-totp-secret');

-        $svg = $resp->getElementHtml('#main-content .card svg');
+        $svg = $this->withHtml($resp)->getOuterHtml('#main-content .card svg');

 
         // Validation error, code should remain the same
 
         // Validation error, code should remain the same

-        $resp = $this->post('/mfa/totp-confirm', [
+        $resp = $this->post('/mfa/totp/confirm', [

             'code' => 'abc123',
         ]);
             'code' => 'abc123',
         ]);

-        $resp->assertRedirect('/mfa/totp-generate');
+        $resp->assertRedirect('/mfa/totp/generate');

         $resp = $this->followRedirects($resp);
         $resp->assertSee('The provided code is not valid or has expired.');
         $resp = $this->followRedirects($resp);
         $resp->assertSee('The provided code is not valid or has expired.');

-        $revisitSvg = $resp->getElementHtml('#main-content .card svg');
+        $revisitSvg = $this->withHtml($resp)->getOuterHtml('#main-content .card svg');

         $this->assertTrue($svg === $revisitSvg);
         $this->assertTrue($svg === $revisitSvg);

+        $secret = decrypt(session()->get('mfa-setup-totp-secret'));
+
+        $resp->assertSee("?secret={$secret}&issuer=BookStack&algorithm=SHA1&digits=6&period=30");

 
         // Successful confirmation
         $google2fa = new Google2FA();
 
         // Successful confirmation
         $google2fa = new Google2FA();

-        $secret = decrypt(session()->get('mfa-setup-totp-secret'));

         $otp = $google2fa->getCurrentOtp($secret);
         $otp = $google2fa->getCurrentOtp($secret);

-        $resp = $this->post('/mfa/totp-confirm', [
+        $resp = $this->post('/mfa/totp/confirm', [

             'code' => $otp,
         ]);
         $resp->assertRedirect('/mfa/setup');
             'code' => $otp,
         ]);
         $resp->assertRedirect('/mfa/setup');


@@ -49,11 +52,11 @@ class MfaConfigurationTest extends TestCase
         // Confirmation of setup
         $resp = $this->followRedirects($resp);
         $resp->assertSee('Multi-factor method successfully configured');
         // Confirmation of setup
         $resp = $this->followRedirects($resp);
         $resp->assertSee('Multi-factor method successfully configured');

-        $resp->assertElementContains('a[href$="/mfa/totp-generate"]', 'Reconfigure');
+        $this->withHtml($resp)->assertElementContains('a[href$="/mfa/totp/generate"]', 'Reconfigure');

 
         $this->assertDatabaseHas('mfa_values', [
             'user_id' => $editor->id,
 
         $this->assertDatabaseHas('mfa_values', [
             'user_id' => $editor->id,

-            'method' => 'totp',
+            'method'  => 'totp',

         ]);
         $this->assertFalse(session()->has('mfa-setup-totp-secret'));
         $value = MfaValue::query()->where('user_id', '=', $editor->id)
         ]);
         $this->assertFalse(session()->has('mfa-setup-totp-secret'));
         $value = MfaValue::query()->where('user_id', '=', $editor->id)


@@ -63,37 +66,37 @@ class MfaConfigurationTest extends TestCase
 
     public function test_backup_codes_setup()
     {
 
     public function test_backup_codes_setup()
     {

-        $editor = $this->getEditor();
+        $editor = $this->users->editor();

         $this->assertDatabaseMissing('mfa_values', ['user_id' => $editor->id]);
 
         // Setup page state
         $resp = $this->actingAs($editor)->get('/mfa/setup');
         $this->assertDatabaseMissing('mfa_values', ['user_id' => $editor->id]);
 
         // Setup page state
         $resp = $this->actingAs($editor)->get('/mfa/setup');

-        $resp->assertElementContains('a[href$="/mfa/backup-codes-generate"]', 'Setup');
+        $this->withHtml($resp)->assertElementContains('a[href$="/mfa/backup_codes/generate"]', 'Setup');

 
         // Generate page access
 
         // Generate page access

-        $resp = $this->get('/mfa/backup-codes-generate');
+        $resp = $this->get('/mfa/backup_codes/generate');

         $resp->assertSee('Backup Codes');
         $resp->assertSee('Backup Codes');

-        $resp->assertElementContains('form[action$="/mfa/backup-codes-confirm"]', 'Confirm and Enable');
+        $this->withHtml($resp)->assertElementContains('form[action$="/mfa/backup_codes/confirm"]', 'Confirm and Enable');

         $this->assertSessionHas('mfa-setup-backup-codes');
         $codes = decrypt(session()->get('mfa-setup-backup-codes'));
         // Check code format
         $this->assertCount(16, $codes);
         $this->assertSessionHas('mfa-setup-backup-codes');
         $codes = decrypt(session()->get('mfa-setup-backup-codes'));
         // Check code format
         $this->assertCount(16, $codes);

-        $this->assertEquals(16*11, strlen(implode('', $codes)));
+        $this->assertEquals(16 * 11, strlen(implode('', $codes)));

         // Check download link
         $resp->assertSee(base64_encode(implode("\n\n", $codes)));
 
         // Confirm submit
         // Check download link
         $resp->assertSee(base64_encode(implode("\n\n", $codes)));
 
         // Confirm submit

-        $resp = $this->post('/mfa/backup-codes-confirm');
+        $resp = $this->post('/mfa/backup_codes/confirm');

         $resp->assertRedirect('/mfa/setup');
 
         // Confirmation of setup
         $resp = $this->followRedirects($resp);
         $resp->assertSee('Multi-factor method successfully configured');
         $resp->assertRedirect('/mfa/setup');
 
         // Confirmation of setup
         $resp = $this->followRedirects($resp);
         $resp->assertSee('Multi-factor method successfully configured');

-        $resp->assertElementContains('a[href$="/mfa/backup-codes-generate"]', 'Reconfigure');
+        $this->withHtml($resp)->assertElementContains('a[href$="/mfa/backup_codes/generate"]', 'Reconfigure');

 
         $this->assertDatabaseHas('mfa_values', [
             'user_id' => $editor->id,
 
         $this->assertDatabaseHas('mfa_values', [
             'user_id' => $editor->id,

-            'method' => 'backup_codes',
+            'method'  => 'backup_codes',

         ]);
         $this->assertFalse(session()->has('mfa-setup-backup-codes'));
         $value = MfaValue::query()->where('user_id', '=', $editor->id)
         ]);
         $this->assertFalse(session()->has('mfa-setup-backup-codes'));
         $value = MfaValue::query()->where('user_id', '=', $editor->id)


@@ -103,14 +106,14 @@ class MfaConfigurationTest extends TestCase
 
     public function test_backup_codes_cannot_be_confirmed_if_not_previously_generated()
     {
 
     public function test_backup_codes_cannot_be_confirmed_if_not_previously_generated()
     {

-        $resp = $this->asEditor()->post('/mfa/backup-codes-confirm');
+        $resp = $this->asEditor()->post('/mfa/backup_codes/confirm');

         $resp->assertStatus(500);
     }
 
     public function test_mfa_method_count_is_visible_on_user_edit_page()
     {
         $resp->assertStatus(500);
     }
 
     public function test_mfa_method_count_is_visible_on_user_edit_page()
     {

-        $user = $this->getEditor();
-        $resp = $this->actingAs($this->getAdmin())->get($user->getEditUrl());
+        $user = $this->users->editor();
+        $resp = $this->actingAs($this->users->admin())->get($user->getEditUrl());

         $resp->assertSee('0 methods configured');
 
         MfaValue::upsertWithValue($user, MfaValue::METHOD_TOTP, 'test');
         $resp->assertSee('0 methods configured');
 
         MfaValue::upsertWithValue($user, MfaValue::METHOD_TOTP, 'test');


@@ -124,25 +127,60 @@ class MfaConfigurationTest extends TestCase
 
     public function test_mfa_setup_link_only_shown_when_viewing_own_user_edit_page()
     {
 
     public function test_mfa_setup_link_only_shown_when_viewing_own_user_edit_page()
     {

-        $admin = $this->getAdmin();
+        $admin = $this->users->admin();

         $resp = $this->actingAs($admin)->get($admin->getEditUrl());
         $resp = $this->actingAs($admin)->get($admin->getEditUrl());

-        $resp->assertElementExists('a[href$="/mfa/setup"]');
+        $this->withHtml($resp)->assertElementExists('a[href$="/mfa/setup"]');

 
 

-        $resp = $this->actingAs($admin)->get($this->getEditor()->getEditUrl());
-        $resp->assertElementNotExists('a[href$="/mfa/setup"]');
+        $resp = $this->actingAs($admin)->get($this->users->editor()->getEditUrl());
+        $this->withHtml($resp)->assertElementNotExists('a[href$="/mfa/setup"]');

     }
 
     public function test_mfa_indicator_shows_in_user_list()
     {
     }
 
     public function test_mfa_indicator_shows_in_user_list()
     {

-        $admin = $this->getAdmin();
+        $admin = $this->users->admin();

         User::query()->where('id', '!=', $admin->id)->delete();
 
         $resp = $this->actingAs($admin)->get('/settings/users');
         User::query()->where('id', '!=', $admin->id)->delete();
 
         $resp = $this->actingAs($admin)->get('/settings/users');

-        $resp->assertElementNotExists('[title="MFA Configured"] svg');
+        $this->withHtml($resp)->assertElementNotExists('[title="MFA Configured"] svg');

 
         MfaValue::upsertWithValue($admin, MfaValue::METHOD_TOTP, 'test');
         $resp = $this->actingAs($admin)->get('/settings/users');
 
         MfaValue::upsertWithValue($admin, MfaValue::METHOD_TOTP, 'test');
         $resp = $this->actingAs($admin)->get('/settings/users');

-        $resp->assertElementExists('[title="MFA Configured"] svg');
+        $this->withHtml($resp)->assertElementExists('[title="MFA Configured"] svg');

     }
 
     }
 

-}
\ No newline at end of file
+    public function test_remove_mfa_method()
+    {
+        $admin = $this->users->admin();
+
+        MfaValue::upsertWithValue($admin, MfaValue::METHOD_TOTP, 'test');
+        $this->assertEquals(1, $admin->mfaValues()->count());
+        $resp = $this->actingAs($admin)->get('/mfa/setup');
+        $this->withHtml($resp)->assertElementExists('form[action$="/mfa/totp/remove"]');
+
+        $resp = $this->delete('/mfa/totp/remove');
+        $resp->assertRedirect('/mfa/setup');
+        $resp = $this->followRedirects($resp);
+        $resp->assertSee('Multi-factor method successfully removed');
+
+        $this->assertActivityExists(ActivityType::MFA_REMOVE_METHOD);
+        $this->assertEquals(0, $admin->mfaValues()->count());
+    }
+
+    public function test_totp_setup_url_shows_correct_user_when_setup_forced_upon_login()
+    {
+        $admin = $this->users->admin();
+        /** @var Role $role */
+        $role = $admin->roles()->first();
+        $role->mfa_enforced = true;
+        $role->save();
+
+        $resp = $this->post('/login', ['email' => $admin->email, 'password' => 'password']);
+        $this->assertFalse(auth()->check());
+        $resp->assertRedirect('/mfa/verify');
+
+        $resp = $this->get('/mfa/totp/generate');
+        $resp->assertSeeText('Mobile App Setup');
+        $resp->assertDontSee('otpauth://totp/BookStack:guest%40example.com', false);
+        $resp->assertSee('otpauth://totp/BookStack:admin%40admin.com', false);
+    }
+}