Tests for track_total_hits

Author: pdphilip

tests/LargeRecordsTest.php

@@ -0,0 +1,42 @@
+<?php
+
+declare(strict_types=1);
+
+use PDPhilip\Elasticsearch\Tests\Models\Product;
+
+beforeEach(function () {
+    Product::executeSchema();
+
+});
+
+it('tests track total hits', function () {
+
+    Product::buildRecords(11_000);
+
+    class ProductWithDefaultTrackTotalHits extends Product
+    {
+        protected $connection = 'elasticsearch_with_default_track_total_hits';
+    }
+
+    $products = Product::limit(1)->get();
+    expect($products->getQueryMeta()->getTotalHits())->toBe(10000);
+
+    $products = Product::limit(1)->withTrackTotalHits()->get();
+    expect($products->getQueryMeta()->getTotalHits())->toBe(11000);
+
+    $products = Product::limit(1)->withTrackTotalHits(false)->get();
+    expect($products->getQueryMeta()->getTotalHits())->toBe(-1);
+
+    $products = Product::limit(1)->withTrackTotalHits(300)->get();
+    expect($products->getQueryMeta()->getTotalHits())->toBe(300);
+
+    $products = ProductWithDefaultTrackTotalHits::limit(1)->get();
+    expect($products->getQueryMeta()->getTotalHits())->toBe(11000);
+
+    $products = ProductWithDefaultTrackTotalHits::limit(1)->withoutTrackTotalHits()->get();
+    expect($products->getQueryMeta()->getTotalHits())->toBe(10000);
+
+    $products = ProductWithDefaultTrackTotalHits::limit(1)->withTrackTotalHits(300)->get();
+    expect($products->getQueryMeta()->getTotalHits())->toBe(300);
+
+});

tests/Models/Product.php

@@ -29,4 +29,17 @@ public static function executeSchema()
             $table->date('updated_at');
         });
     }
+
+    public static function buildRecords($limit = 100)
+    {
+        $records = [];
+        while ($limit) {
+            $records[] = [
+                'state' => rand(1, 100),
+            ];
+            $limit--;
+        }
+        Product::insert($records);
+        //        Product::withoutRefresh()->insert($records);
+    }
 }

tests/TestCase.php

@@ -56,5 +56,15 @@ protected function getEnvironmentSetUp($app): void
                 'logging' => true,
             ],
         ]);
+
+        $app['config']->set('database.connections.elasticsearch_with_default_track_total_hits', [
+            'driver' => 'elasticsearch',
+            'auth_type' => 'http',
+            'hosts' => ['http://localhost:9200'],
+            'options' => [
+                'track_total_hits' => true,
+                'logging' => true,
+            ],
+        ]);
     }
 }