X-Git-Url: http://source.bookstackapp.com/bookstack/blobdiff_plain/0c70416b5cfe79210efadafbc47145a394c42755..refs/pull/494/head:/tests/UserProfileTest.php

diff --git a/tests/UserProfileTest.php b/tests/UserProfileTest.php
index ebec814c4..eabfce004 100644
--- a/tests/UserProfileTest.php
+++ b/tests/UserProfileTest.php
@@ -95,25 +95,26 @@ class UserProfileTest extends BrowserKitTest
             ->see('cannot delete the guest user');
     }
 
-    public function test_books_display_is_list()
+    public function test_books_view_is_list()
     {
-        $this->asAdmin()
-            ->visit('/settings/users/' . $this->user->id)
-            ->select('list', '#books_display')
-            ->press('Save')
+        $editor = $this->getEditor([
+          'books_view_type' => 'list'
+        ]);
+
+        $this->actingAs($editor)
             ->visit('/books')
-            ->pageNotHasElement('.gallery-item')
+            ->pageNotHasElement('.featured-image-container')
             ->pageHasElement('.entity-list-item');
     }
 
-    public function test_books_display_is_grid()
+    public function test_books_view_is_grid()
     {
-        $this->asAdmin()
-            ->visit('/settings/users/' . $this->user->id)
-            ->select('grid', '#books_display')
-            ->press('Save')
+        $editor = $this->getEditor([
+          'books_view_type' => 'grid'
+        ]);
+
+        $this->actingAs($editor)
             ->visit('/books')
-            ->pageNotHasElement('.entity-list-item')
-            ->pageHasElement('.gallery-item');
+            ->pageHasElement('.featured-image-container');
     }
 }