X-Git-Url: http://source.bookstackapp.com/bookstack/blobdiff_plain/3ac34b584930aad5625d29ca6c2e59b85e9d2ef8..refs/pull/2023/head:/tests/TestCase.php diff --git a/tests/TestCase.php b/tests/TestCase.php index e0f160eed..1f1d5ece7 100644 --- a/tests/TestCase.php +++ b/tests/TestCase.php @@ -1,14 +1,15 @@ assertTrue(session()->has($key), "Session does not contain a [{$key}] entry"); + return $this; + } + + /** + * 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); + } + + /** + * Create the test response instance from the given response. + * + * @param \Illuminate\Http\Response $response + * @return TestResponse */ - protected function assertPermissionError(TestResponse $response) + protected function createTestResponse($response) { - $response->assertRedirect('/'); - $this->assertTrue(session()->has('error')); - session()->remove('error'); + return TestResponse::fromBaseResponse($response); + } + + /** + * Assert that an activity entry exists of the given key. + * Checks the activity belongs to the given entity if provided. + */ + protected function assertActivityExists(string $key, Entity $entity = null) + { + $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