Handle Elastica Transport errors

composer.json

@@ -19,7 +19,7 @@
         "symfony/property-access": "^3.2|^4",
         "pagerfanta/pagerfanta": "^1.0.5|^2.0",
         "psr/log": "^1.0",
-        "ruflin/elastica": "^5.2.1|^6.0"
+        "ruflin/elastica": "dev-master"
     },
     "require-dev": {
         "doctrine/orm": "^2.5",

src/Elastica/Client.php

@@ -24,6 +24,8 @@
  */
 class Client extends BaseClient
 {
+    private $forbiddenHttpCodes = [403, 404, 400];
+
     /**
      * Stores created indexes to avoid recreation.
      *
@@ -48,8 +50,15 @@ public function request($path, $method = Request::GET, $data = [], array $query
         }
 
         $response = parent::request($path, $method, $data, $query, $contentType);
-        $responseData = $response->getData();
 
+        $transportInfo = $response->getTransferInfo();
+        if (isset($transportInfo['http_code']) && in_array($transportInfo['http_code'], $this->forbiddenHttpCodes)) {
+            $body = isset($transportInfo['body']) ? $transportInfo['body'] : 'blank';
+            $message = sprintf('Error in transportInfo: response code is %s, response body is %s', $transportInfo['http_code'], $body);
+            throw new \Exception($message);
+        }
+
+        $responseData = $response->getData();
         if (isset($responseData['took']) && isset($responseData['hits'])) {
             $this->logQuery($path, $method, $data, $query, $response->getQueryTime(), $response->getEngineTime(), $responseData['hits']['total']);
         } else {

tests/Unit/Elastica/ClientTest.php

@@ -11,7 +11,9 @@
 
 namespace FOS\ElasticaBundle\Tests\Unit\Client;
 
+use Elastica\Client as BaseClient;
 use Elastica\Connection;
+use Elastica\JSON;
 use Elastica\Request;
 use Elastica\Response;
 use Elastica\Transport\NullTransport;
@@ -21,14 +23,28 @@
 
 class ClientTest extends TestCase
 {
-    public function testRequestsAreLogged()
+    private function getClientMock(Response $response = null)
     {
         $transport = new NullTransport();
+        if ($response) {
+            $transport->setResponse($response);
+        }
 
         $connection = $this->createMock(Connection::class);
         $connection->expects($this->any())->method('getTransportObject')->will($this->returnValue($transport));
         $connection->expects($this->any())->method('toArray')->will($this->returnValue([]));
 
+        $client = $this->getMockBuilder(Client::class)
+            ->setMethods(['getConnection'])
+            ->getMock();
+
+        $client->expects($this->any())->method('getConnection')->will($this->returnValue($connection));
+        return $client;
+    }
+
+    public function testRequestsAreLogged()
+    {
+        $client = $this->getClientMock();
         $logger = $this->createMock(ElasticaLogger::class);
         $logger
             ->expects($this->once())
@@ -44,17 +60,30 @@ public function testRequestsAreLogged()
                 $this->isType('array'),
                 $this->isType('array')
             );
-
-        $client = $this->getMockBuilder(Client::class)
-            ->setMethods(['getConnection'])
-            ->getMock();
-
-        $client->expects($this->any())->method('getConnection')->will($this->returnValue($connection));
-
         $client->setLogger($logger);
 
         $response = $client->request('foo');
 
         $this->assertInstanceOf(Response::class, $response);
     }
+
+    public function testRequestsWithTransportInfoErrorsRaiseExceptions()
+    {
+        $httpCode = 403;
+        $responseString = JSON::stringify(['message' => 'some AWS error']);
+        $transferInfo = [
+            'request_header' => 'bar',
+            'http_code' => $httpCode,
+            'body' => $responseString,
+        ];
+        $response = new Response($responseString);
+        $response->setTransferInfo($transferInfo);
+
+        $client = $this->getClientMock($response);
+
+        $desiredMessage = sprintf('Error in transportInfo: response code is %d, response body is %s', $httpCode, $responseString);
+        $this->expectException(\Exception::class);
+        $this->expectExceptionMessage($desiredMessage);
+        $response = $client->request('foo');
+    }
 }