X-Git-Url: http://source.bookstackapp.com/bookstack/blobdiff_plain/64455307b13d7eae3d40ab91a75a34ddec3ea686..refs/pull/2393/head:/tests/Api/ApiListingTest.php

diff --git a/tests/Api/ApiListingTest.php b/tests/Api/ApiListingTest.php
index 741b9664b..c3d9bc108 100644
--- a/tests/Api/ApiListingTest.php
+++ b/tests/Api/ApiListingTest.php
@@ -1,8 +1,7 @@
-<?php
+<?php namespace Tests\Api;
 
-namespace Tests;
-
-use BookStack\Entities\Book;
+use BookStack\Entities\Models\Book;
+use Tests\TestCase;
 
 class ApiListingTest extends TestCase
 {
@@ -82,4 +81,20 @@ class ApiListingTest extends TestCase
         }
     }
 
+    public function test_total_on_results_shows_correctly()
+    {
+        $this->actingAsApiEditor();
+        $bookCount = Book::query()->count();
+        $resp = $this->get($this->endpoint . '?count=1');
+        $resp->assertJson(['total' => $bookCount ]);
+    }
+
+    public function test_total_on_results_shows_correctly_when_offset_provided()
+    {
+        $this->actingAsApiEditor();
+        $bookCount = Book::query()->count();
+        $resp = $this->get($this->endpoint . '?count=1&offset=1');
+        $resp->assertJson(['total' => $bookCount ]);
+    }
+
 }
\ No newline at end of file