71 lines
2.8 KiB
PHP
71 lines
2.8 KiB
PHP
|
<?php
|
||
|
|
|
||
|
|
namespace Tests\Unit;
|
||
|
|
|
||
|
|
use App\Jobs\RunFinding;
|
||
|
|
use App\Models\Investigation;
|
||
|
|
use App\Models\Source;
|
||
|
|
use App\Models\User;
|
||
|
|
use App\Osint\FindingDispatcher;
|
||
|
|
use Illuminate\Foundation\Testing\RefreshDatabase;
|
||
|
|
use Illuminate\Support\Facades\Queue;
|
||
|
|
use Tests\TestCase;
|
||
|
|
|
||
|
|
class FindingDispatcherTest extends TestCase
|
||
|
|
{
|
||
|
|
use RefreshDatabase;
|
||
|
|
|
||
|
|
public function test_queues_only_applicable_registered_sources(): void
|
||
|
|
{
|
||
|
|
Queue::fake();
|
||
|
|
|
||
|
|
$user = User::factory()->create();
|
||
|
|
$investigation = Investigation::factory()->for($user, 'owner')->create();
|
||
|
|
$target = $investigation->targets()->create(['type' => 'username', 'value' => 'someone', 'created_by' => $user->id]);
|
||
|
|
|
||
|
|
Source::create(['key' => 'sherlock', 'name' => 'Sherlock', 'target_types' => ['username'], 'enabled' => true]);
|
||
|
|
Source::create(['key' => 'crtsh', 'name' => 'crt.sh', 'target_types' => ['domain'], 'enabled' => true]);
|
||
|
|
Source::create(['key' => 'unregistered', 'name' => 'Unregistered', 'target_types' => ['username'], 'enabled' => true]);
|
||
|
|
|
||
|
|
$queued = app(FindingDispatcher::class)->dispatchAllApplicable($target, $user);
|
||
|
|
|
||
|
|
$this->assertSame(1, $queued);
|
||
|
|
$this->assertDatabaseCount('findings', 1);
|
||
|
|
Queue::assertPushed(RunFinding::class, 1);
|
||
|
|
}
|
||
|
|
|
||
|
|
public function test_skips_sources_already_pending_or_running(): void
|
||
|
|
{
|
||
|
|
Queue::fake();
|
||
|
|
|
||
|
|
$user = User::factory()->create();
|
||
|
|
$investigation = Investigation::factory()->for($user, 'owner')->create();
|
||
|
|
$target = $investigation->targets()->create(['type' => 'username', 'value' => 'someone', 'created_by' => $user->id]);
|
||
|
|
|
||
|
|
$sherlock = Source::create(['key' => 'sherlock', 'name' => 'Sherlock', 'target_types' => ['username'], 'enabled' => true]);
|
||
|
|
$target->findings()->create(['source_id' => $sherlock->id, 'status' => 'pending']);
|
||
|
|
|
||
|
|
$queued = app(FindingDispatcher::class)->dispatchAllApplicable($target, $user);
|
||
|
|
|
||
|
|
$this->assertSame(0, $queued);
|
||
|
|
$this->assertDatabaseCount('findings', 1);
|
||
|
|
}
|
||
|
|
|
||
|
|
public function test_does_not_skip_a_source_whose_last_finding_already_completed(): void
|
||
|
|
{
|
||
|
|
Queue::fake();
|
||
|
|
|
||
|
|
$user = User::factory()->create();
|
||
|
|
$investigation = Investigation::factory()->for($user, 'owner')->create();
|
||
|
|
$target = $investigation->targets()->create(['type' => 'username', 'value' => 'someone', 'created_by' => $user->id]);
|
||
|
|
|
||
|
|
$sherlock = Source::create(['key' => 'sherlock', 'name' => 'Sherlock', 'target_types' => ['username'], 'enabled' => true]);
|
||
|
|
$target->findings()->create(['source_id' => $sherlock->id, 'status' => 'completed']);
|
||
|
|
|
||
|
|
$queued = app(FindingDispatcher::class)->dispatchAllApplicable($target, $user);
|
||
|
|
|
||
|
|
$this->assertSame(1, $queued);
|
||
|
|
$this->assertDatabaseCount('findings', 2);
|
||
|
|
}
|
||
|
|
}
|