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_with_png_content()
  14     {
  15         $page = Page::first();
  16         $this->asAdmin();
  17         $imageName = 'first-image.png';
  18
  19         $this->uploadImage($imageName, $page->id);
  20         $image = Image::first();
  21         $image->type = 'drawio';
  22         $image->save();
  23
  24         $imageGet = $this->getJson("/images/drawio/base64/{$image->id}");
  25         $imageGet->assertJson([
  26             'content' => 'data:image/png;base64,iVBORw0KGgoAAAANSUhEUgAAAAUAAAAFCAIAAAACDbGyAAAACXBIWXMAAAsTAAALEwEAmpwYAAAAB3RJTUUH4gEcDCo5iYNs+gAAAB1pVFh0Q29tbWVudAAAAAAAQ3JlYXRlZCB3aXRoIEdJTVBkLmUHAAAAFElEQVQI12O0jN/KgASYGFABqXwAZtoBV6Sl3hIAAAAASUVORK5CYII=',
  27         ]);
  28     }
  29
  30     public function test_get_image_as_base64_with_svg_content()
  31     {
  32         $page = Page::first();
  33         $this->asAdmin();
  34
  35         $this->uploadImage('my-drawing.svg', $page->id, 'image/svg+xml', 'diagram.svg');
  36         $image = Image::first();
  37         $image->type = 'drawio';
  38         $image->save();
  39
  40         $imageGet = $this->getJson("/images/drawio/base64/{$image->id}");
  41         $imageGet->assertJson([
  42             'content' => 'data:image/svg+xml;base64,' . base64_encode(file_get_contents($this->getTestImageFilePath('diagram.svg'))),
  43         ]);
  44     }
  45
  46     public function test_drawing_base64_upload_with_png()
  47     {
  48         $page = Page::first();
  49         $editor = $this->getEditor();
  50         $this->actingAs($editor);
  51
  52         $upload = $this->postJson('images/drawio', [
  53             'uploaded_to' => $page->id,
  54             'image'       => 'data:image/png;base64,iVBORw0KGgoAAAANSUhEUgAAAAUAAAAFCAIAAAACDbGyAAAACXBIWXMAAAsTAAALEwEAmpwYAAAAB3RJTUUH4gEcDCo5iYNs+gAAAB1pVFh0Q29tbWVudAAAAAAAQ3JlYXRlZCB3aXRoIEdJTVBkLmUHAAAAFElEQVQI12O0jN/KgASYGFABqXwAZtoBV6Sl3hIAAAAASUVORK5CYII=',
  55         ]);
  56
  57         $upload->assertStatus(200);
  58         $upload->assertJson([
  59             'type'        => 'drawio',
  60             'uploaded_to' => $page->id,
  61             'created_by'  => $editor->id,
  62             'updated_by'  => $editor->id,
  63         ]);
  64
  65         $image = Image::where('type', '=', 'drawio')->first();
  66         $this->assertTrue(file_exists(public_path($image->path)), 'Uploaded image not found at path: ' . public_path($image->path));
  67
  68         $testImageData = file_get_contents($this->getTestImageFilePath());
  69         $uploadedImageData = file_get_contents(public_path($image->path));
  70         $this->assertTrue($testImageData === $uploadedImageData, 'Uploaded image file data does not match our test image as expected');
  71     }
  72
  73     public function test_drawing_base64_upload_with_svg()
  74     {
  75         $page = Page::first();
  76         $editor = $this->getEditor();
  77         $this->actingAs($editor);
  78
  79         $upload = $this->postJson('images/drawio', [
  80             'uploaded_to' => $page->id,
  81             'image'       => 'data:image/svg+xml;base64,' . base64_encode(file_get_contents($this->getTestImageFilePath('diagram.svg'))),
  82         ]);
  83
  84         $upload->assertStatus(200);
  85         $upload->assertJson([
  86             'type'        => 'drawio',
  87             'uploaded_to' => $page->id,
  88             'created_by'  => $editor->id,
  89             'updated_by'  => $editor->id,
  90         ]);
  91
  92         $image = Image::where('type', '=', 'drawio')->first();
  93         $this->assertStringEndsWith('.svg', $image->path);
  94         $this->assertTrue(file_exists(public_path($image->path)), 'Uploaded image not found at path: ' . public_path($image->path));
  95
  96         $testImageData = file_get_contents($this->getTestImageFilePath('diagram.svg'));
  97         $uploadedImageData = file_get_contents(public_path($image->path));
  98         $this->assertTrue($testImageData === $uploadedImageData, 'Uploaded image file data does not match our test image as expected');
  99     }
 100
 101     public function test_drawio_url_can_be_configured()
 102     {
 103         config()->set('services.drawio', 'http://cats.com?dog=tree');
 104         $page = Page::first();
 105         $editor = $this->getEditor();
 106
 107         $resp = $this->actingAs($editor)->get($page->getUrl('/edit'));
 108         $resp->assertSee('drawio-url="http://cats.com?dog=tree"', false);
 109     }
 110
 111     public function test_drawio_url_can_be_disabled()
 112     {
 113         config()->set('services.drawio', true);
 114         $page = Page::first();
 115         $editor = $this->getEditor();
 116
 117         $resp = $this->actingAs($editor)->get($page->getUrl('/edit'));
 118         $resp->assertSee('drawio-url="https://embed.diagrams.net/?embed=1&amp;proto=json&amp;spin=1&amp;configure=1"', false);
 119
 120         config()->set('services.drawio', false);
 121         $resp = $this->actingAs($editor)->get($page->getUrl('/edit'));
 122         $resp->assertDontSee('drawio-url', false);
 123     }
 124 }