[internal-api-client] Use specific elastic mapping for search endpoints

Author: janvanicek

src/Client.php

@@ -354,7 +354,7 @@ public function searchJobs(
         $request = new Request('GET', 'search/jobs?' . http_build_query($query));
         $response = $this->sendRequest($request);
 
-        return $this->mapJobsFromResponse($response);
+        return $this->mapJobsFromSearchResponse($response);
     }
 
     /**
@@ -418,7 +418,16 @@ public function searchJobsGrouped(
         $request = new Request('GET', 'search/grouped-jobs?' . http_build_query($query));
         $response = $this->sendRequest($request);
 
-        return $this->mapJobsFromResponse($response);
+        return $this->mapJobsFromSearchResponse($response);
+    }
+
+    /** @return JobInterface[] */
+    private function mapJobsFromSearchResponse(array $responseBody): array
+    {
+        return array_map(
+            fn(array $jobData): JobInterface => $this->existingJobFactory->loadFromElasticJobData($jobData),
+            $responseBody,
+        );
     }
 
     /**

src/ExistingJobFactory.php

@@ -4,6 +4,7 @@
 
 namespace Keboola\JobQueueInternalClient;
 
+use Keboola\JobQueueInternalClient\JobFactory\ElasticJobDefinition;
 use Keboola\JobQueueInternalClient\JobFactory\FullJobDefinition;
 use Keboola\JobQueueInternalClient\JobFactory\Job;
 use Keboola\JobQueueInternalClient\JobFactory\JobInterface;
@@ -24,4 +25,11 @@ public function loadFromExistingJobData(array $data): JobInterface
 
         return new Job($this->objectEncryptor, $this->storageClientFactory, $data);
     }
+
+    public function loadFromElasticJobData(array $data): JobInterface
+    {
+        $data = $this->validateJobData($data, ElasticJobDefinition::class);
+
+        return new Job($this->objectEncryptor, $this->storageClientFactory, $data);
+    }
 }

tests/ClientSearchTest.php

@@ -311,7 +311,7 @@ private function createClient(array &$requests, array $responses): Client
         $handlerStack->push(Middleware::history($requests));
 
         $existingJobFactory = $this->createMock(ExistingJobFactory::class);
-        $existingJobFactory->method('loadFromExistingJobData')
+        $existingJobFactory->method('loadFromElasticJobData')
             ->willReturnCallback(function (array $data) {
                 $job = $this->createMock(Job::class);
                 $job->method('getId')->willReturn((string) $data['id']);