Rolled tri-layout to page edit and book-create

[bookstack] / tests / ImageTest.php

diff --git a/tests/ImageTest.php b/tests/ImageTest.php
index 0f1e772e937e86660110f1ca171642872b33d693..38bac2ccacb07e9f6514e6167c9aceb00ea9d716 100644 (file)
--- a/tests/ImageTest.php
+++ b/tests/ImageTest.php
@@ -1,9 +1,10 @@
 <?php namespace Tests;
 
-use BookStack\Image;
-use BookStack\Page;
-use BookStack\Repos\EntityRepo;
-use BookStack\Services\ImageService;
+use BookStack\Entities\Repos\PageRepo;
+use BookStack\Uploads\Image;
+use BookStack\Entities\Page;
+use BookStack\Entities\Repos\EntityRepo;
+use BookStack\Uploads\ImageService;
 
 class ImageTest extends TestCase
 {
@@ -250,8 +251,8 @@ class ImageTest extends TestCase
         $upload->assertStatus(200);
         $image = Image::where('type', '=', 'gallery')->first();
 
-        $entityRepo = app(EntityRepo::class);
-        $entityRepo->updatePage($page, $page->book_id, [
+        $pageRepo = app(PageRepo::class);
+        $pageRepo->updatePage($page, $page->book_id, [
             'name' => $page->name,
             'html' => $page->html . "<img src=\"{$image->url}\">",
             'summary' => ''
@@ -263,7 +264,7 @@ class ImageTest extends TestCase
         $this->assertCount(0, $toDelete);
 
         // Save a revision of our page without the image;
-        $entityRepo->updatePage($page, $page->book_id, [
+        $pageRepo->updatePage($page, $page->book_id, [
             'name' => $page->name,
             'html' => "<p>Hello</p>",
             'summary' => ''