Re-push remove from Scout queue fix

laravel#479

src/Jobs/RemoveFromSearch.php

@@ -2,6 +2,7 @@
 
 namespace Laravel\Scout\Jobs;
 
+use Illuminate\Support\Str;
 use Illuminate\Bus\Queueable;
 use Illuminate\Contracts\Queue\ShouldQueue;
 use Illuminate\Database\Eloquent\Collection as EloquentCollection;
@@ -14,7 +15,7 @@ class RemoveFromSearch implements ShouldQueue
     /**
      * The models to be removed from the search index.
      *
-     * @var \Illuminate\Database\Eloquent\Collection
+     * @var RemoveableScoutCollection
      */
     public $models;
 
@@ -26,7 +27,7 @@ class RemoveFromSearch implements ShouldQueue
      */
     public function __construct($models)
     {
-        $this->models = $models;
+        $this->models = RemoveableScoutCollection::make($models);
     }
 
     /**
@@ -56,9 +57,25 @@ protected function restoreCollection($value)
         return new EloquentCollection(
             collect($value->id)->map(function ($id) use ($value) {
                 return tap(new $value->class, function ($model) use ($id) {
-                    $model->forceFill([$model->getKeyName() => $id]);
+                    $keyName = $this->getUnqualifiedScoutKeyName(
+                        $model->getScoutKeyName()
+                    );
+
+                    $model->forceFill([$keyName => $id]);
                 });
             })
         );
     }
+
+    /**
+     * Get the unqualified Scout key name.
+     *
+     * @param string $keyName
+     *
+     * @return string
+     */
+    protected function getUnqualifiedScoutKeyName($keyName)
+    {
+        return Str::afterLast($keyName, '.');
+    }
 }

src/Jobs/RemoveableScoutCollection.php

@@ -0,0 +1,25 @@
+<?php
+
+namespace Laravel\Scout\Jobs;
+
+use Laravel\Scout\Searchable;
+use Illuminate\Database\Eloquent\Collection;
+
+class RemoveableScoutCollection extends Collection
+{
+    /**
+     * Get the Scout identifiers for all of the entities.
+     *
+     * @return array
+     */
+    public function getQueueableIds()
+    {
+        if ($this->isEmpty()) {
+            return [];
+        }
+
+        return in_array(Searchable::class, class_uses_recursive($this->first()))
+                    ? $this->map->getScoutKey()->all()
+                    : parent::getQueueableIds();
+    }
+}

tests/Fixtures/SearchableModelWithCustomKey.php

@@ -0,0 +1,28 @@
+<?php
+
+namespace Laravel\Scout\Tests\Fixtures;
+
+use Illuminate\Database\Eloquent\Model;
+use Laravel\Scout\Searchable;
+
+class SearchableModelWithCustomKey extends Model
+{
+    use Searchable;
+
+    /**
+     * The attributes that are mass assignable.
+     *
+     * @var array
+     */
+    protected $fillable = ['other_id'];
+
+    public function getScoutKey()
+    {
+        return $this->other_id;
+    }
+
+    public function getScoutKeyName()
+    {
+        return $this->qualifyColumn('other_id');
+    }
+}

tests/Unit/RemoveFromSearchTest.php

@@ -4,8 +4,10 @@
 
 use Illuminate\Database\Eloquent\Collection;
 use Illuminate\Support\Facades\Config;
+use Laravel\Scout\Jobs\RemoveableScoutCollection;
 use Laravel\Scout\Jobs\RemoveFromSearch;
 use Laravel\Scout\Tests\Fixtures\SearchableModel;
+use Laravel\Scout\Tests\Fixtures\SearchableModelWithCustomKey;
 use Mockery as m;
 use PHPUnit\Framework\TestCase;
 
@@ -34,7 +36,7 @@ public function test_handle_passes_the_collection_to_engine()
 
     public function test_models_are_deserialized_without_the_database()
     {
-        $job = new RemoveFromSearch($collection = Collection::make([
+        $job = new RemoveFromSearch(Collection::make([
             $model = new SearchableModel(['id' => 1234]),
         ]));
 
@@ -46,4 +48,37 @@ public function test_models_are_deserialized_without_the_database()
         $this->assertTrue($model->is($job->models->first()));
         $this->assertEquals(1234, $job->models->first()->getScoutKey());
     }
+
+    public function test_models_are_deserialized_without_the_database_using_custom_scout_key()
+    {
+        $job = new RemoveFromSearch(Collection::make([
+            $model = new SearchableModelWithCustomKey(['other_id' => 1234]),
+        ]));
+
+        $job = unserialize(serialize($job));
+
+        $this->assertInstanceOf(Collection::class, $job->models);
+        $this->assertCount(1, $job->models);
+        $this->assertInstanceOf(SearchableModelWithCustomKey::class, $job->models->first());
+        $this->assertTrue($model->is($job->models->first()));
+        $this->assertEquals(1234, $job->models->first()->getScoutKey());
+        $this->assertEquals('searchable_model_with_custom_keys.other_id', $job->models->first()->getScoutKeyName());
+    }
+
+    public function test_removeable_scout_collection_returns_scout_keys()
+    {
+        $collection = RemoveableScoutCollection::make([
+            new SearchableModelWithCustomKey(['other_id' => 1234]),
+            new SearchableModelWithCustomKey(['other_id' => 2345]),
+            new SearchableModel(['id' => 3456]),
+            new SearchableModel(['id' => 7891]),
+        ]);
+
+        $this->assertEquals([
+            1234,
+            2345,
+            3456,
+            7891,
+        ], $collection->getQueueableIds());
+    }
 }