]> BookStack Code Mirror - bookstack/blob - tests/ImageTest.php
Updated Spanish translation
[bookstack] / 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      * Get the path to our basic test image.
10      * @return string
11      */
12     protected function getTestImageFilePath()
13     {
14         return base_path('tests/test-data/test-image.png');
15     }
16
17     /**
18      * Get a test image that can be uploaded
19      * @param $fileName
20      * @return \Illuminate\Http\UploadedFile
21      */
22     protected function getTestImage($fileName)
23     {
24         return new \Illuminate\Http\UploadedFile($this->getTestImageFilePath(), $fileName, 'image/png', 5238);
25     }
26
27     /**
28      * Get the path for a test image.
29      * @param $type
30      * @param $fileName
31      * @return string
32      */
33     protected function getTestImagePath($type, $fileName)
34     {
35         return '/uploads/images/' . $type . '/' . Date('Y-m-M') . '/' . $fileName;
36     }
37
38     /**
39      * Uploads an image with the given name.
40      * @param $name
41      * @param int $uploadedTo
42      * @return \Illuminate\Foundation\Testing\TestResponse
43      */
44     protected function uploadImage($name, $uploadedTo = 0)
45     {
46         $file = $this->getTestImage($name);
47         return $this->call('POST', '/images/gallery/upload', ['uploaded_to' => $uploadedTo], [], ['file' => $file], []);
48     }
49
50     /**
51      * Delete an uploaded image.
52      * @param $relPath
53      */
54     protected function deleteImage($relPath)
55     {
56         $path = public_path($relPath);
57         if (file_exists($path)) {
58             unlink($path);
59         }
60     }
61
62
63     public function test_image_upload()
64     {
65         $page = Page::first();
66         $admin = $this->getAdmin();
67         $this->actingAs($admin);
68
69         $imageName = 'first-image.png';
70         $relPath = $this->getTestImagePath('gallery', $imageName);
71         $this->deleteImage($relPath);
72
73         $upload = $this->uploadImage($imageName, $page->id);
74         $upload->assertStatus(200);
75
76         $this->assertTrue(file_exists(public_path($relPath)), 'Uploaded image not found at path: '. public_path($relPath));
77
78         $this->deleteImage($relPath);
79
80         $this->assertDatabaseHas('images', [
81             'url' => $this->baseUrl . $relPath,
82             'type' => 'gallery',
83             'uploaded_to' => $page->id,
84             'path' => $relPath,
85             'created_by' => $admin->id,
86             'updated_by' => $admin->id,
87             'name' => $imageName
88         ]);
89     }
90
91     public function test_secure_images_uploads_to_correct_place()
92     {
93         config()->set('filesystems.default', 'local_secure');
94         $this->asEditor();
95         $galleryFile = $this->getTestImage('my-secure-test-upload');
96         $page = Page::first();
97         $expectedPath = storage_path('uploads/images/gallery/' . Date('Y-m-M') . '/my-secure-test-upload');
98
99         $upload = $this->call('POST', '/images/gallery/upload', ['uploaded_to' => $page->id], [], ['file' => $galleryFile], []);
100         $upload->assertStatus(200);
101
102         $this->assertTrue(file_exists($expectedPath), 'Uploaded image not found at path: '. $expectedPath);
103
104         if (file_exists($expectedPath)) {
105             unlink($expectedPath);
106         }
107     }
108
109     public function test_system_images_remain_public()
110     {
111         config()->set('filesystems.default', 'local_secure');
112         $this->asEditor();
113         $galleryFile = $this->getTestImage('my-system-test-upload');
114         $page = Page::first();
115         $expectedPath = public_path('uploads/images/system/' . Date('Y-m-M') . '/my-system-test-upload');
116
117         $upload = $this->call('POST', '/images/system/upload', ['uploaded_to' => $page->id], [], ['file' => $galleryFile], []);
118         $upload->assertStatus(200);
119
120         $this->assertTrue(file_exists($expectedPath), 'Uploaded image not found at path: '. $expectedPath);
121
122         if (file_exists($expectedPath)) {
123             unlink($expectedPath);
124         }
125     }
126
127     public function test_image_delete()
128     {
129         $page = Page::first();
130         $this->asAdmin();
131         $imageName = 'first-image.png';
132
133         $this->uploadImage($imageName, $page->id);
134         $image = Image::first();
135         $relPath = $this->getTestImagePath('gallery', $imageName);
136
137         $delete = $this->delete( '/images/' . $image->id);
138         $delete->assertStatus(200);
139
140         $this->assertDatabaseMissing('images', [
141             'url' => $this->baseUrl . $relPath,
142             'type' => 'gallery'
143         ]);
144
145         $this->assertFalse(file_exists(public_path($relPath)), 'Uploaded image has not been deleted as expected');
146     }
147
148     public function testBase64Get()
149     {
150         $page = Page::first();
151         $this->asAdmin();
152         $imageName = 'first-image.png';
153
154         $this->uploadImage($imageName, $page->id);
155         $image = Image::first();
156
157         $imageGet = $this->getJson("/images/base64/{$image->id}");
158         $imageGet->assertJson([
159             'content' => 'iVBORw0KGgoAAAANSUhEUgAAAAUAAAAFCAIAAAACDbGyAAAACXBIWXMAAAsTAAALEwEAmpwYAAAAB3RJTUUH4gEcDCo5iYNs+gAAAB1pVFh0Q29tbWVudAAAAAAAQ3JlYXRlZCB3aXRoIEdJTVBkLmUHAAAAFElEQVQI12O0jN/KgASYGFABqXwAZtoBV6Sl3hIAAAAASUVORK5CYII='
160         ]);
161     }
162
163     public function test_drawing_base64_upload()
164     {
165         $page = Page::first();
166         $editor = $this->getEditor();
167         $this->actingAs($editor);
168
169         $upload = $this->postJson('images/drawing/upload', [
170             'uploaded_to' => $page->id,
171             'image' => 'image/png;base64,iVBORw0KGgoAAAANSUhEUgAAAAUAAAAFCAIAAAACDbGyAAAACXBIWXMAAAsTAAALEwEAmpwYAAAAB3RJTUUH4gEcDCo5iYNs+gAAAB1pVFh0Q29tbWVudAAAAAAAQ3JlYXRlZCB3aXRoIEdJTVBkLmUHAAAAFElEQVQI12O0jN/KgASYGFABqXwAZtoBV6Sl3hIAAAAASUVORK5CYII='
172         ]);
173
174         $upload->assertStatus(200);
175         $upload->assertJson([
176             'type' => 'drawio',
177             'uploaded_to' => $page->id,
178             'created_by' => $editor->id,
179             'updated_by' => $editor->id,
180         ]);
181
182         $image = Image::where('type', '=', 'drawio')->first();
183         $this->assertTrue(file_exists(public_path($image->path)), 'Uploaded image not found at path: '. public_path($image->path));
184
185         $testImageData = file_get_contents($this->getTestImageFilePath());
186         $uploadedImageData = file_get_contents(public_path($image->path));
187         $this->assertTrue($testImageData === $uploadedImageData, "Uploaded image file data does not match our test image as expected");
188     }
189
190     public function test_drawing_replacing()
191     {
192         $page = Page::first();
193         $editor = $this->getEditor();
194         $this->actingAs($editor);
195
196         $this->postJson('images/drawing/upload', [
197             'uploaded_to' => $page->id,
198             'image' => 'image/png;base64,iVBORw0KGgoAAAANSUhEUgAAAAUAAAAFCAIAAAACDbGyAAAACXBIWXMAAAsTAAALEwEAmpwYAAAAB3RJTUUH4gEcDQ4S1RUeKwAAAB1pVFh0Q29tbWVudAAAAAAAQ3JlYXRlZCB3aXRoIEdJTVBkLmUHAAAAFElEQVQI12NctNWSAQkwMaACUvkAfCkBmjyhGl4AAAAASUVORK5CYII='
199         ]);
200
201         $image = Image::where('type', '=', 'drawio')->first();
202
203         $replace = $this->putJson("images/drawing/upload/{$image->id}", [
204             'image' => 'image/png;base64,iVBORw0KGgoAAAANSUhEUgAAAAUAAAAFCAIAAAACDbGyAAAACXBIWXMAAAsTAAALEwEAmpwYAAAAB3RJTUUH4gEcDCo5iYNs+gAAAB1pVFh0Q29tbWVudAAAAAAAQ3JlYXRlZCB3aXRoIEdJTVBkLmUHAAAAFElEQVQI12O0jN/KgASYGFABqXwAZtoBV6Sl3hIAAAAASUVORK5CYII='
205         ]);
206
207         $replace->assertStatus(200);
208         $replace->assertJson([
209             'type' => 'drawio',
210             'uploaded_to' => $page->id,
211             'created_by' => $editor->id,
212             'updated_by' => $editor->id,
213         ]);
214
215         $this->assertTrue(file_exists(public_path($image->path)), 'Uploaded image not found at path: '. public_path($image->path));
216
217         $testImageData = file_get_contents($this->getTestImageFilePath());
218         $uploadedImageData = file_get_contents(public_path($image->path));
219         $this->assertTrue($testImageData === $uploadedImageData, "Uploaded image file data does not match our test image as expected");
220     }
221
222     public function test_user_images_deleted_on_user_deletion()
223     {
224         $editor = $this->getEditor();
225         $this->actingAs($editor);
226
227         $imageName = 'profile.png';
228         $relPath = $this->getTestImagePath('gallery', $imageName);
229         $this->deleteImage($relPath);
230
231         $file = $this->getTestImage($imageName);
232         $this->call('POST', '/images/user/upload', [], [], ['file' => $file], []);
233         $this->call('POST', '/images/user/upload', [], [], ['file' => $file], []);
234
235         $profileImages = Image::where('type', '=', 'user')->where('created_by', '=', $editor->id)->get();
236         $this->assertTrue($profileImages->count() === 2, "Found profile images does not match upload count");
237
238         $userDelete = $this->asAdmin()->delete("/settings/users/{$editor->id}");
239         $userDelete->assertStatus(302);
240         $this->assertDatabaseMissing('images', [
241             'type' => 'user',
242             'created_by' => $editor->id
243         ]);
244     }
245
246 }
Morty Proxy This is a proxified and sanitized view of the page, visit original site.