BookStack Code Mirror - bookstack/blob - tests/Uploads/DrawioTest.php

   1 <?php
   2
   3 namespace Tests\Uploads;
   4
   5 use BookStack\Entities\Models\Page;
   6 use BookStack\Uploads\Image;
   7 use Tests\TestCase;
   8
   9 class DrawioTest extends TestCase
  10 {
  11     use UsesImages;
  12
  13     public function test_get_image_as_base64()
  14     {
  15         $page = $this->entities->page();
  16         $this->asAdmin();
  17         $imageName = 'first-image.png';
  18
  19         $this->uploadImage($imageName, $page->id);
  20         /** @var Image $image */
  21         $image = Image::query()->first();
  22         $image->type = 'drawio';
  23         $image->save();
  24
  25         $imageGet = $this->getJson("/images/drawio/base64/{$image->id}");
  26         $imageGet->assertJson([
  27             'content' => 'iVBORw0KGgoAAAANSUhEUgAAAAUAAAAFCAIAAAACDbGyAAAACXBIWXMAAAsTAAALEwEAmpwYAAAAB3RJTUUH4gEcDCo5iYNs+gAAAB1pVFh0Q29tbWVudAAAAAAAQ3JlYXRlZCB3aXRoIEdJTVBkLmUHAAAAFElEQVQI12O0jN/KgASYGFABqXwAZtoBV6Sl3hIAAAAASUVORK5CYII=',
  28         ]);
  29     }
  30
  31     public function test_non_accessible_image_returns_404_error_and_message()
  32     {
  33         $page = $this->entities->page();
  34         $this->asEditor();
  35         $imageName = 'non-accessible-image.png';
  36
  37         $this->uploadImage($imageName, $page->id);
  38         /** @var Image $image */
  39         $image = Image::query()->first();
  40         $image->type = 'drawio';
  41         $image->save();
  42         $this->permissions->disableEntityInheritedPermissions($page);
  43
  44         $imageGet = $this->getJson("/images/drawio/base64/{$image->id}");
  45         $imageGet->assertNotFound();
  46         $imageGet->assertJson([
  47             'message' => 'Drawing data could not be loaded. The drawing file might no longer exist or you may not have permission to access it.',
  48         ]);
  49     }
  50
  51     public function test_drawing_base64_upload()
  52     {
  53         $page = $this->entities->page();
  54         $editor = $this->users->editor();
  55         $this->actingAs($editor);
  56
  57         $upload = $this->postJson('images/drawio', [
  58             'uploaded_to' => $page->id,
  59             'image'       => 'image/png;base64,iVBORw0KGgoAAAANSUhEUgAAAAUAAAAFCAIAAAACDbGyAAAACXBIWXMAAAsTAAALEwEAmpwYAAAAB3RJTUUH4gEcDCo5iYNs+gAAAB1pVFh0Q29tbWVudAAAAAAAQ3JlYXRlZCB3aXRoIEdJTVBkLmUHAAAAFElEQVQI12O0jN/KgASYGFABqXwAZtoBV6Sl3hIAAAAASUVORK5CYII=',
  60         ]);
  61
  62         $upload->assertStatus(200);
  63         $upload->assertJson([
  64             'type'        => 'drawio',
  65             'uploaded_to' => $page->id,
  66             'created_by'  => $editor->id,
  67             'updated_by'  => $editor->id,
  68         ]);
  69
  70         $image = Image::where('type', '=', 'drawio')->first();
  71         $this->assertTrue(file_exists(public_path($image->path)), 'Uploaded image not found at path: ' . public_path($image->path));
  72
  73         $testImageData = file_get_contents($this->getTestImageFilePath());
  74         $uploadedImageData = file_get_contents(public_path($image->path));
  75         $this->assertTrue($testImageData === $uploadedImageData, 'Uploaded image file data does not match our test image as expected');
  76     }
  77
  78     public function test_drawio_url_can_be_configured()
  79     {
  80         config()->set('services.drawio', 'http://cats.com?dog=tree');
  81         $page = $this->entities->page();
  82         $editor = $this->users->editor();
  83
  84         $resp = $this->actingAs($editor)->get($page->getUrl('/edit'));
  85         $resp->assertSee('drawio-url="http://cats.com?dog=tree"', false);
  86     }
  87
  88     public function test_drawio_url_can_be_disabled()
  89     {
  90         config()->set('services.drawio', true);
  91         $page = $this->entities->page();
  92         $editor = $this->users->editor();
  93
  94         $resp = $this->actingAs($editor)->get($page->getUrl('/edit'));
  95         $resp->assertSee('drawio-url="https://embed.diagrams.net/?embed=1&amp;proto=json&amp;spin=1&amp;configure=1"', false);
  96
  97         config()->set('services.drawio', false);
  98         $resp = $this->actingAs($editor)->get($page->getUrl('/edit'));
  99         $resp->assertDontSee('drawio-url', false);
 100     }
 101 }