@fleetbase/fleetops-engine 0.6.67 → 0.6.69

package/server/tests/Unit/Support/Ai/Tools/FleetOpsAiToolsTest.php

@@ -0,0 +1,246 @@
+<?php
+
+use Fleetbase\Ai\Models\AiTask;
+use Fleetbase\Ai\Support\AiAudience;
+use Fleetbase\Ai\Support\AiToolContext;
+use Fleetbase\FleetOps\Models\OrderConfig;
+use Fleetbase\FleetOps\Support\Ai\Tools\CreateOrderTool;
+use Fleetbase\FleetOps\Support\Ai\Tools\SearchResourcesTool;
+
+class FleetOpsCreateOrderToolProbe extends CreateOrderTool
+{
+    public array $resolvedPlaces = [];
+
+    public function exposeBuildDraft(AiTask $task, array $input = []): array
+    {
+        return $this->buildDraft($task, $input);
+    }
+
+    protected function can(string $permission): bool
+    {
+        return true;
+    }
+
+    protected function draftFromPrompt(string $prompt): array
+    {
+        return array_filter([
+            'order_config_uuid' => 'order-config-uuid',
+            'type'              => 'transport',
+            'payload'           => $prompt !== '' ? ['pickup_query' => 'Parsed from prompt'] : [],
+            'dispatched'        => false,
+        ], fn ($value) => $value !== []) + ['payload' => []];
+    }
+
+    protected function resolvePlace(?string $query): ?array
+    {
+        return $this->resolvedPlaces[$query] ?? null;
+    }
+
+    protected function resolveOrderConfig(array $draft): ?OrderConfig
+    {
+        $config = new OrderConfig();
+        $config->setRawAttributes(['uuid' => 'order-config-uuid', 'key' => 'transport', 'name' => 'Transport'], true);
+
+        return $config;
+    }
+
+    protected function resolveDriver(array $draft): ?Fleetbase\FleetOps\Models\Driver
+    {
+        return null;
+    }
+
+    protected function resolveVehicle(array $draft): ?Fleetbase\FleetOps\Models\Vehicle
+    {
+        return null;
+    }
+}
+
+class FleetOpsSearchResourcesToolProbe extends SearchResourcesTool
+{
+    public array $searched = [];
+
+    protected function searchAll(array $terms, ?array $types = null): array
+    {
+        $this->searched[] = [$terms, $types];
+
+        return ['query_terms' => $terms, 'results' => []];
+    }
+}
+
+test('create order tool prepares one draft preview per call from structured fields without parsing the prompt', function () {
+    $tool    = new FleetOpsCreateOrderToolProbe();
+    $task    = new AiTask(['prompt' => 'Create some dummy orders with made up data', 'metadata' => []]);
+    $context = new AiToolContext($task, new AiAudience(false, ['fleet-ops create order']));
+
+    $first = $tool->invoke($task, [
+        'pickup'       => 'Kabelweg 57, Amsterdam',
+        'dropoff'      => 'Danzigerkade 15, Amsterdam',
+        'scheduled_at' => '2026-09-16T09:30:00+02:00',
+        'driver'       => '  ',
+        'notes'        => 'Test order',
+    ], $context);
+    $second = $tool->invoke($task, ['pickup' => 'Diemerhof 32, Diemen', 'dropoff' => 'Herikerbergweg 238, Amsterdam', 'dispatch' => true], $context);
+
+    expect($first['preview_id'])->toBe('preview-1')
+        ->and($second['preview_id'])->toBe('preview-2')
+        ->and($first['status'])->toBe('draft_shown_to_user')
+        ->and($first['message'])->toContain('does not exist until the user reviews the card')
+        ->and($context->actionPreviews)->toHaveCount(2)
+        ->and($context->actionPreviews[0]['draft']['payload']['pickup_query'])->toBe('Kabelweg 57, Amsterdam')
+        ->and($context->actionPreviews[0]['draft']['scheduled_at'])->toBe('2026-09-16T09:30:00+02:00')
+        ->and($context->actionPreviews[0]['draft']['notes'])->toBe('Test order')
+        ->and($context->actionPreviews[0]['draft'])->not->toHaveKey('driver_query')
+        ->and($context->actionPreviews[0]['draft']['dispatched'])->toBeFalse()
+        ->and($context->actionPreviews[1]['draft']['payload']['pickup_query'])->toBe('Diemerhof 32, Diemen')
+        ->and($context->actionPreviews[1]['draft']['dispatched'])->toBeTrue()
+        ->and(json_encode($context->actionPreviews))->not->toContain('Parsed from prompt')
+        ->and($tool->toolName())->toBe('propose_create_order')
+        ->and($tool->key())->toBe('fleet-ops.create_order')
+        ->and($tool->toolDescription())->toContain('NOT created by this tool')
+        ->and($tool->toolParameters()['required'])->toBe(['pickup', 'dropoff'])
+        ->and($tool->availableFor($context))->toBeTrue()
+        ->and($tool->availableFor(new AiToolContext($task, new AiAudience(false, []))))->toBeFalse();
+});
+
+test('create order drafts refresh from the edited preview and still parse prompts in keyword mode', function () {
+    $tool = new FleetOpsCreateOrderToolProbe();
+    $task = new AiTask(['prompt' => 'create order', 'metadata' => ['action_previews' => [['draft' => ['notes' => 'first preview']]]]]);
+
+    $refreshed = $tool->exposeBuildDraft($task, ['existing_draft' => ['notes' => 'second preview'], 'draft' => ['dispatched' => true]]);
+    $keyword   = $tool->exposeBuildDraft($task, ['dispatched' => false]);
+
+    expect($refreshed['notes'])->toBe('second preview')
+        ->and($refreshed['dispatched'])->toBeTrue()
+        ->and($refreshed['payload']['pickup_query'])->toBe('Parsed from prompt')
+        ->and($keyword['notes'])->toBe('first preview')
+        ->and($keyword)->not->toHaveKey('existing_draft');
+});
+
+test('search tool searches the requested phrase and identifiers for the chosen record types', function () {
+    $tool    = new FleetOpsSearchResourcesToolProbe();
+    $task    = new AiTask(['prompt' => 'where is Jane']);
+    $context = new AiToolContext($task, new AiAudience(false, ['fleet-ops see driver']));
+
+    $tool->invoke($task, ['query' => 'Jane Doe', 'types' => ['drivers', 'pets']], $context);
+    $tool->invoke($task, ['query' => 'order order_yhkejdnzgz'], $context);
+
+    expect($tool->searched[0])->toBe([['Jane Doe', 'Doe'], ['drivers']])
+        ->and($tool->searched[1])->toBe([['order order_yhkejdnzgz', 'order_yhkejdnzgz'], null])
+        ->and($tool->invoke($task, ['query' => ' x '], $context))->toBe(['error' => 'Provide an identifier or name to search for.'])
+        ->and($tool->toolName())->toBe('fleetops_search')
+        ->and($tool->key())->toBe('fleet-ops.search_resources')
+        ->and($tool->toolDescription())->toContain('particular record')
+        ->and($tool->toolParameters()['properties']['types']['items']['enum'])->toBe(SearchResourcesTool::TYPES)
+        ->and($tool->availableFor($context))->toBeTrue()
+        ->and($tool->availableFor(new AiToolContext($task, new AiAudience(false, []))))->toBeFalse();
+});
+
+test('search tool narrows the real search to the requested types', function () {
+    $tool = new class extends SearchResourcesTool {
+        public array $called = [];
+
+        protected function orders(array $terms): array
+        {
+            $this->called[] = 'orders';
+
+            return [['id' => 'order_1']];
+        }
+
+        protected function drivers(array $terms): array
+        {
+            $this->called[] = 'drivers';
+
+            return [];
+        }
+    };
+
+    $result = $tool->invoke(new AiTask(['prompt' => '']), ['query' => 'ORDER-1', 'types' => ['orders']], new AiToolContext(new AiTask(), new AiAudience(true)));
+
+    expect($tool->called)->toBe(['orders'])
+        ->and($result['results'])->toBe(['orders' => [['id' => 'order_1']]]);
+});
+
+class FleetOpsOptimizeOrderRouteToolProbe extends Fleetbase\FleetOps\Support\Ai\Tools\OptimizeOrderRouteTool
+{
+    public array $previewedPrompts = [];
+
+    public function preview(AiTask $task, array $input = []): array
+    {
+        $this->previewedPrompts[] = $task->prompt;
+
+        return [
+            'action'         => $this->key(),
+            'ready'          => true,
+            'message'        => 'Fleetbase AI prepared an optimized waypoint sequence. Review it before applying.',
+            'missing_fields' => [],
+            'fields'         => [['label' => 'Order', 'value' => 'order_abc']],
+        ];
+    }
+}
+
+test('optimize route tool proposes a reviewable preview for the order the model named', function () {
+    $tool    = new FleetOpsOptimizeOrderRouteToolProbe();
+    $task    = new AiTask(['prompt' => 'can you optimise the route please', 'metadata' => []]);
+    $context = new AiToolContext($task, new AiAudience(false, ['fleet-ops optimize order', 'fleet-ops update-route-for order']));
+
+    $result = $tool->invoke($task, ['order' => 'order_abc'], $context);
+
+    expect($result['preview_id'])->toBe('preview-1')
+        ->and($result['ready'])->toBeTrue()
+        ->and($result['status'])->toBe('proposal_shown_to_user')
+        ->and($result['fields'])->toBe([['label' => 'Order', 'value' => 'order_abc']])
+        ->and($result['missing_fields'])->toBe([])
+        // The capability finds the order in the prompt text, so the tool hands it the id the model gave.
+        ->and($tool->previewedPrompts)->toBe(['order_abc'])
+        ->and($task->prompt)->toBe('can you optimise the route please')
+        ->and($context->actionPreviews)->toHaveCount(1)
+        ->and($tool->toolName())->toBe('propose_optimize_order_route')
+        ->and($tool->key())->toBe('fleet-ops.optimize_order_route')
+        ->and($tool->toolDescription())->toContain('NOT changed by this tool')
+        ->and($tool->toolParameters()['required'])->toBe(['order'])
+        ->and($tool->availableFor($context))->toBeTrue()
+        ->and($tool->availableFor(new AiToolContext($task, new AiAudience(false, ['fleet-ops optimize order']))))->toBeFalse();
+});
+
+test('optimize route tool refuses to build a preview without an order', function () {
+    $tool    = new FleetOpsOptimizeOrderRouteToolProbe();
+    $task    = new AiTask(['prompt' => 'optimise a route', 'metadata' => []]);
+    $context = new AiToolContext($task, new AiAudience(false, ['fleet-ops optimize order', 'fleet-ops update-route-for order']));
+
+    expect($tool->invoke($task, ['order' => '   '], $context))->toBe([
+        'error'   => 'invalid_arguments',
+        'message' => 'An order id is required to optimize a route.',
+    ])
+        ->and($tool->invoke($task, [], $context)['error'])->toBe('invalid_arguments')
+        ->and($context->actionPreviews)->toBe([])
+        ->and($tool->previewedPrompts)->toBe([]);
+});
+
+class FleetOpsImportOrdersToolProbe extends Fleetbase\FleetOps\Support\Ai\Tools\ImportOrdersTool
+{
+    protected function can(string $permission): bool
+    {
+        return true;
+    }
+}
+
+test('import tool reports what an import needs and never claims to have imported anything', function () {
+    $tool    = new FleetOpsImportOrdersToolProbe();
+    $task    = new AiTask(['prompt' => 'can I import orders from a spreadsheet', 'metadata' => []]);
+    $context = new AiToolContext($task, new AiAudience(false, ['fleet-ops import order']));
+
+    $result = $tool->invoke($task, [], $context);
+
+    expect($result['accepted_sources'])->toBe(['xlsx', 'csv'])
+        ->and($result['minimum_columns'])->toContain('pickup address or pickup place')
+        ->and($result['can_import_here'])->toBeFalse()
+        ->and($result['message'])->toContain('will not process uploaded spreadsheets')
+        ->and($context->actionPreviews)->toBe([])
+        ->and($tool->toolName())->toBe('fleetops_import_requirements')
+        ->and($tool->key())->toBe('fleet-ops.import_orders_preview')
+        ->and($tool->toolDescription())->toContain('does NOT import anything')
+        ->and($tool->toolParameters()['properties'])->toBeInstanceOf(stdClass::class)
+        ->and(json_encode($tool->toolParameters()['properties']))->toBe('{}')
+        ->and($tool->availableFor($context))->toBeTrue()
+        ->and($tool->availableFor(new AiToolContext($task, new AiAudience(false, []))))->toBeFalse();
+});

package/server/tests/Unit/Support/AiCapabilityPermissionsTest.php

@@ -96,7 +96,7 @@ test('permission checks for non admin users delegate to the gate', function () {
         ->and(fn () => fleetopsAiPermissionsHelper('canAll')->invoke($capability, ['fleet-ops list order']))->toThrow(TypeError::class);
 });
 
-test('search terms extract identifiers and fall back to the raw prompt', function () {
+test('search terms extract identifiers and never fall back to the raw prompt', function () {
     fleetopsAiPermissionsBoot();
     $capability  = new OperationalQueryCapability();
     $searchTerms = fleetopsAiPermissionsHelper('searchTerms');
@@ -105,8 +105,8 @@ test('search terms extract identifiers and fall back to the raw prompt', functio
     expect($terms)->toContain('TRK-12345', 'atlas99')
         ->and($terms)->not->toContain('find', 'order');
 
-    // Stop-word-only prompts fall back to the trimmed prompt
-    expect($searchTerms->invoke($capability, 'find order'))->toBe(['find order']);
+    // Prompts without a record reference produce no search terms, so nothing is searched
+    expect($searchTerms->invoke($capability, 'find order'))->toBe([]);
 });
 
 test('where like any matches across columns and terms', function () {