X-Git-Url: http://source.bookstackapp.com/bookstack/blobdiff_plain/fd3929e8099d45160f242bea18556ed78ef0eaf8..refs/pull/2023/head:/tests/TestCase.php diff --git a/tests/TestCase.php b/tests/TestCase.php index 3f7d846f7..1f1d5ece7 100644 --- a/tests/TestCase.php +++ b/tests/TestCase.php @@ -1,51 +1,67 @@ -make(Illuminate\Contracts\Console\Kernel::class)->bootstrap(); + $this->assertTrue(session()->has($key), "Session does not contain a [{$key}] entry"); + return $this; + } - return $app; + /** + * Override of the get method so we can get visibility of custom TestResponse methods. + * @param string $uri + * @param array $headers + * @return TestResponse + */ + public function get($uri, array $headers = []) + { + return parent::get($uri, $headers); } - public function asAdmin() + /** + * Create the test response instance from the given response. + * + * @param \Illuminate\Http\Response $response + * @return TestResponse + */ + protected function createTestResponse($response) { - if($this->admin === null) { - $this->admin = \BookStack\User::find(1); - } - return $this->actingAs($this->admin); + return TestResponse::fromBaseResponse($response); } /** - * Quickly sets an array of settings. - * @param $settingsArray + * Assert that an activity entry exists of the given key. + * Checks the activity belongs to the given entity if provided. */ - protected function setSettings($settingsArray) + protected function assertActivityExists(string $key, Entity $entity = null) { - $settings = app('BookStack\Services\SettingService'); - foreach ($settingsArray as $key => $value) { - $settings->put($key, $value); + $detailsToCheck = ['key' => $key]; + + if ($entity) { + $detailsToCheck['entity_type'] = $entity->getMorphClass(); + $detailsToCheck['entity_id'] = $entity->id; } + + $this->assertDatabaseHas('activities', $detailsToCheck); } -} +} \ No newline at end of file