BookStack Code Mirror - bookstack/blob - tests/ImageTest.php

   1 <?php namespace Tests;
   2
   3 use BookStack\Image;
   4 use BookStack\Page;
   5
   6 class ImageTest extends TestCase
   7 {
   8
   9     /**
  10      * Get the path to our basic test image.
  11      * @return string
  12      */
  13     protected function getTestImageFilePath()
  14     {
  15         return base_path('tests/test-data/test-image.png');
  16     }
  17
  18     /**
  19      * Get a test image that can be uploaded
  20      * @param $fileName
  21      * @return \Illuminate\Http\UploadedFile
  22      */
  23     protected function getTestImage($fileName)
  24     {
  25         return new \Illuminate\Http\UploadedFile($this->getTestImageFilePath(), $fileName, 'image/jpeg', 5238);
  26     }
  27
  28     /**
  29      * Get the path for a test image.
  30      * @param $type
  31      * @param $fileName
  32      * @return string
  33      */
  34     protected function getTestImagePath($type, $fileName)
  35     {
  36         return '/uploads/images/' . $type . '/' . Date('Y-m-M') . '/' . $fileName;
  37     }
  38
  39     /**
  40      * Uploads an image with the given name.
  41      * @param $name
  42      * @param int $uploadedTo
  43      * @return \Illuminate\Foundation\Testing\TestResponse
  44      */
  45     protected function uploadImage($name, $uploadedTo = 0)
  46     {
  47         $file = $this->getTestImage($name);
  48         return $this->call('POST', '/images/gallery/upload', ['uploaded_to' => $uploadedTo], [], ['file' => $file], []);
  49     }
  50
  51     /**
  52      * Delete an uploaded image.
  53      * @param $relPath
  54      */
  55     protected function deleteImage($relPath)
  56     {
  57         unlink(public_path($relPath));
  58     }
  59
  60
  61     public function test_image_upload()
  62     {
  63         $page = Page::first();
  64         $this->asAdmin();
  65         $admin = $this->getAdmin();
  66         $imageName = 'first-image.png';
  67
  68         $upload = $this->uploadImage($imageName, $page->id);
  69         $upload->assertStatus(200);
  70         $relPath = $this->getTestImagePath('gallery', $imageName);
  71
  72         $this->assertTrue(file_exists(public_path($relPath)), 'Uploaded image not found at path: '. public_path($relPath));
  73
  74         $this->deleteImage($relPath);
  75
  76         $this->assertDatabaseHas('images', [
  77             'url' => $this->baseUrl . $relPath,
  78             'type' => 'gallery',
  79             'uploaded_to' => $page->id,
  80             'path' => $relPath,
  81             'created_by' => $admin->id,
  82             'updated_by' => $admin->id,
  83             'name' => $imageName
  84         ]);
  85
  86     }
  87
  88     public function test_image_delete()
  89     {
  90         $page = Page::first();
  91         $this->asAdmin();
  92         $imageName = 'first-image.png';
  93
  94         $this->uploadImage($imageName, $page->id);
  95         $image = Image::first();
  96         $relPath = $this->getTestImagePath('gallery', $imageName);
  97
  98         $delete = $this->delete( '/images/' . $image->id);
  99         $delete->assertStatus(200);
 100
 101         $this->assertDatabaseMissing('images', [
 102             'url' => $this->baseUrl . $relPath,
 103             'type' => 'gallery'
 104         ]);
 105
 106         $this->assertFalse(file_exists(public_path($relPath)), 'Uploaded image has not been deleted as expected');
 107     }
 108
 109     public function testBase64Get()
 110     {
 111         $page = Page::first();
 112         $this->asAdmin();
 113         $imageName = 'first-image.png';
 114
 115         $this->uploadImage($imageName, $page->id);
 116         $image = Image::first();
 117
 118         $imageGet = $this->getJson("/images/base64/{$image->id}");
 119         $imageGet->assertJson([
 120             'content' => 'iVBORw0KGgoAAAANSUhEUgAAAAUAAAAFCAIAAAACDbGyAAAACXBIWXMAAAsTAAALEwEAmpwYAAAAB3RJTUUH4gEcDCo5iYNs+gAAAB1pVFh0Q29tbWVudAAAAAAAQ3JlYXRlZCB3aXRoIEdJTVBkLmUHAAAAFElEQVQI12O0jN/KgASYGFABqXwAZtoBV6Sl3hIAAAAASUVORK5CYII='
 121         ]);
 122     }
 123
 124     public function test_drawing_base64_upload()
 125     {
 126         $page = Page::first();
 127         $editor = $this->getEditor();
 128         $this->actingAs($editor);
 129
 130         $upload = $this->postJson('images/drawing/upload', [
 131             'uploaded_to' => $page->id,
 132             'image' => 'image/png;base64,iVBORw0KGgoAAAANSUhEUgAAAAUAAAAFCAIAAAACDbGyAAAACXBIWXMAAAsTAAALEwEAmpwYAAAAB3RJTUUH4gEcDCo5iYNs+gAAAB1pVFh0Q29tbWVudAAAAAAAQ3JlYXRlZCB3aXRoIEdJTVBkLmUHAAAAFElEQVQI12O0jN/KgASYGFABqXwAZtoBV6Sl3hIAAAAASUVORK5CYII='
 133         ]);
 134
 135         $upload->assertStatus(200);
 136         $upload->assertJson([
 137             'type' => 'drawio',
 138             'uploaded_to' => $page->id,
 139             'created_by' => $editor->id,
 140             'updated_by' => $editor->id,
 141         ]);
 142
 143         $image = Image::where('type', '=', 'drawio')->first();
 144         $this->assertTrue(file_exists(public_path($image->path)), 'Uploaded image not found at path: '. public_path($image->path));
 145
 146         $testImageData = file_get_contents($this->getTestImageFilePath());
 147         $uploadedImageData = file_get_contents(public_path($image->path));
 148         $this->assertTrue($testImageData === $uploadedImageData, "Uploaded image file data does not match our test image as expected");
 149     }
 150
 151     public function test_drawing_replacing()
 152     {
 153         $page = Page::first();
 154         $editor = $this->getEditor();
 155         $this->actingAs($editor);
 156
 157         $this->postJson('images/drawing/upload', [
 158             'uploaded_to' => $page->id,
 159             'image' => 'image/png;base64,iVBORw0KGgoAAAANSUhEUgAAAAUAAAAFCAIAAAACDbGyAAAACXBIWXMAAAsTAAALEwEAmpwYAAAAB3RJTUUH4gEcDQ4S1RUeKwAAAB1pVFh0Q29tbWVudAAAAAAAQ3JlYXRlZCB3aXRoIEdJTVBkLmUHAAAAFElEQVQI12NctNWSAQkwMaACUvkAfCkBmjyhGl4AAAAASUVORK5CYII='
 160         ]);
 161
 162         $image = Image::where('type', '=', 'drawio')->first();
 163
 164         $replace = $this->putJson("images/drawing/upload/{$image->id}", [
 165             'image' => 'image/png;base64,iVBORw0KGgoAAAANSUhEUgAAAAUAAAAFCAIAAAACDbGyAAAACXBIWXMAAAsTAAALEwEAmpwYAAAAB3RJTUUH4gEcDCo5iYNs+gAAAB1pVFh0Q29tbWVudAAAAAAAQ3JlYXRlZCB3aXRoIEdJTVBkLmUHAAAAFElEQVQI12O0jN/KgASYGFABqXwAZtoBV6Sl3hIAAAAASUVORK5CYII='
 166         ]);
 167
 168         $replace->assertStatus(200);
 169         $replace->assertJson([
 170             'type' => 'drawio',
 171             'uploaded_to' => $page->id,
 172             'created_by' => $editor->id,
 173             'updated_by' => $editor->id,
 174         ]);
 175
 176         $this->assertTrue(file_exists(public_path($image->path)), 'Uploaded image not found at path: '. public_path($image->path));
 177
 178         $testImageData = file_get_contents($this->getTestImageFilePath());
 179         $uploadedImageData = file_get_contents(public_path($image->path));
 180         $this->assertTrue($testImageData === $uploadedImageData, "Uploaded image file data does not match our test image as expected");
 181     }
 182
 183 }