From ccd0bf2b454d58405612d3ff582107c5c73702df Mon Sep 17 00:00:00 2001
From: Brooke Bryan <brooke@bajb.net>
Date: Wed, 12 Sep 2018 13:39:06 +0100
Subject: [PATCH] Allow overriding of response type in call data

---
 src/Response/ApiCallData.php | 28 +++++++++++++++++++---------
 1 file changed, 19 insertions(+), 9 deletions(-)

diff --git a/src/Response/ApiCallData.php b/src/Response/ApiCallData.php
index eaed62b..ddcf25e 100644
--- a/src/Response/ApiCallData.php
+++ b/src/Response/ApiCallData.php
@@ -10,18 +10,15 @@ class ApiCallData
   protected $_type;
   protected $_result;
 
-  public static function create(
-    $type, $result, $statusCode, $statusMessage, $totalTime, $executionTime,
-    $callTime
-  )
+  public static function create($type, $result, $statusCode, $statusMessage, $totalTime, $executionTime, $callTime)
   {
-    $callData = new static;
+    $callData = new static();
     $callData->_setStatus($statusCode, $statusMessage);
     $callData->_setRawResult($result);
     $callData->_setResponseType($type);
     $callData->_totalTime = $totalTime;
-    $callData->_execTime  = $executionTime;
-    $callData->_callTime  = $callTime;
+    $callData->_execTime = $executionTime;
+    $callData->_callTime = $callTime;
     return $callData;
   }
 
@@ -38,8 +35,8 @@ public function getRawResult()
 
   protected function _setStatus($code, $message = '')
   {
-    $this->_status          = new \stdClass();
-    $this->_status->code    = $code;
+    $this->_status = new \stdClass();
+    $this->_status->code = $code;
     $this->_status->message = $message;
     return $this;
   }
@@ -55,6 +52,19 @@ public function getResponseType()
     return $this->_type;
   }
 
+  /**
+   * Override response type
+   *
+   * @param $type
+   *
+   * @return $this
+   */
+  public function setResponseType($type)
+  {
+    $this->_type = $type;
+    return $this;
+  }
+
   /**
    * Get the time taken to process the call internally on the server
    *