[Fail Case] Add test for aggregator transforms

tests/Features/AggregatorTest.php

@@ -12,6 +12,8 @@
 use App\Thread;
 use Tests\TestCase;
 use Algolia\ScoutExtended\Searchable\AggregatorCollection;
+use Tests\Features\Fixtures\ThreadAggregatorUsingOnlyDateTransform;
+use Tests\Features\Fixtures\ThreadWithSearchableArrayUsingDateTransform;
 
 final class AggregatorTest extends TestCase
 {
@@ -219,4 +221,24 @@ public function testRelationLoad(): void
         $this->assertFalse(Wall::search()->get()->first()->relationLoaded('threads'));
         $this->assertTrue(News::search()->get()->first()->relationLoaded('threads'));
     }
+
+  public function testModelTransformsAreRespectedOnAggregators(): void
+  {
+        ThreadAggregatorUsingOnlyDateTransform::bootSearchable();
+
+        $threadsIndexMock = $this->mockIndex('threads');
+        $threadsAggregator = $this->mockIndex((new ThreadAggregatorUsingOnlyDateTransform)->searchableAs());
+
+        $threadsIndexMock->shouldReceive('saveObjects')->once()->with(Mockery::on(function ($argument) {
+            // Ensure the subscriber count on the single thread index has not been transformed
+            return $argument[0]['subscriber_count'] === '100';
+        }));
+
+        $threadsAggregator->shouldReceive('saveObjects')->once()->with(Mockery::on(function ($argument) {
+            // Ensure the subscriber count on the aggregator index has not been transformed
+            return $argument[0]['subscriber_count'] === '100';
+        }));
+
+        ThreadWithSearchableArrayUsingDateTransform::create(factory(Thread::class)->raw());
+    }
 }

tests/Features/Fixtures/ThreadAggregatorUsingOnlyDateTransform.php

@@ -0,0 +1,12 @@
+<?php
+
+namespace Tests\Features\Fixtures;
+
+use Algolia\ScoutExtended\Searchable\Aggregator;
+
+class ThreadAggregatorUsingOnlyDateTransform extends Aggregator
+{
+    protected $models = [
+        ThreadWithSearchableArrayUsingDateTransform::class,
+    ];
+}

tests/Features/Fixtures/ThreadWithSearchableArrayUsingDateTransform.php

@@ -0,0 +1,19 @@
+<?php
+
+namespace Tests\Features\Fixtures;
+
+use App\Thread;
+use Algolia\ScoutExtended\Transformers\ConvertDatesToTimestamps;
+
+class ThreadWithSearchableArrayUsingDateTransform extends Thread
+{
+    protected  $table = 'threads';
+
+    public function toSearchableArray(): array
+    {
+        $data = array_merge($this->toArray(), ['subscriber_count' => '100']);
+        return $this->transform($data, [
+            ConvertDatesToTimestamps::class,
+        ]);
+    }
+}

tests/Features/TransformersTest.php

@@ -13,6 +13,7 @@
 use Algolia\ScoutExtended\Transformers\ConvertDatesToTimestamps;
 use Tests\Features\Fixtures\ThreadWithSearchableArrayUsingTransform;
 use Algolia\ScoutExtended\Transformers\ConvertNumericStringsToNumbers;
+use Tests\Features\Fixtures\ThreadWithSearchableArrayUsingDateTransform;
 
 final class TransformersTest extends TestCase
 {
@@ -76,4 +77,22 @@ public function testTransformMethod(): void
 
         dispatch(new UpdateJob(collect([$threadWithSearchableArrayUsingTransform])));
     }
+
+  public function testTransformsAreRespectedOnSingleModel(): void
+  {
+    $thread = factory(Thread::class)->create();
+
+    $extendedThread = new ThreadWithSearchableArrayUsingDateTransform($thread->toArray());
+
+    $threadsIndexMock = $this->mockIndex($extendedThread->searchableAs());
+
+    $threadsIndexMock->shouldReceive('saveObjects')->once()->with(\Mockery::on(function ($argument) {
+        // Assert subscriber count has not been transformed
+        return $argument[0]['subscriber_count'] === '100';
+    }));
+
+    $extendedThread->created_at = now();
+
+    dispatch(new UpdateJob(collect([$extendedThread])));
+  }
 }