Unit test for NTLM headers

1 files changed, 28 insertions, 13 deletions

diff --git a/library/SSRS/Soap/NTLM.php b/library/SSRS/Soap/NTLM.php
index cc5b0fd..7f75db7 100755
--- a/library/SSRS/Soap/NTLM.php
+++ b/library/SSRS/Soap/NTLM.php
@@ -131,24 +131,13 @@ class NTLM extends \SoapClient {
         curl_setopt($handle, CURLOPT_FOLLOWLOCATION, true);
         curl_setopt($handle, CURLOPT_HTTPAUTH, CURLAUTH_NTLM);
 
-        $headers = array(
-            'Method: ' . (($data === null) ? 'GET' : 'POST'),
-            'Connection: Keep-Alive',
-            'User-Agent: PHP-SOAP-CURL',
-        );
+        $headers = $this->generateHeaders($url, $data, $action);
+        curl_setopt($handle, CURLOPT_HTTPHEADER, $headers);
 
         if ($data !== null) {
-            $headers[] = 'Content-Type: text/xml; charset=utf-8';
-            $headers[] = 'Content-Length: ' . strlen($data);
             curl_setopt($handle, CURLOPT_POSTFIELDS, $data);
         }
 
-        if ($action !== null) {
-            $headers[] = 'SOAPAction: "' . $action . '"';
-        }
-
-        curl_setopt($handle, CURLOPT_HTTPHEADER, $headers);
-
         $response = curl_exec($handle);
         if ($response === false) {
             throw new SSRS_Soap_Exception('CURL error: ' . curl_error($handle), curl_errno($handle));
@@ -174,4 +163,30 @@ class NTLM extends \SoapClient {
         return $this->_lastResponse;
     }
 
+    /**
+     * 
+     * @param string $url
+     * @param mixed $data
+     * @param string $action
+     * @return array
+     */
+    public function generateHeaders($url, $data = null, $action = null) {
+        $headers = array(
+            'Method: ' . (($data === null) ? 'GET' : 'POST'),
+            'Connection: Keep-Alive',
+            'User-Agent: PHP-SOAP-CURL',
+        );
+
+        if ($data !== null) {
+            $headers[] = 'Content-Type: text/xml; charset=utf-8';
+            $headers[] = 'Content-Length: ' . strlen($data);
+        }
+
+        if ($action !== null) {
+            $headers[] = 'SOAPAction: "' . $action . '"';
+        }
+
+        return $headers;
+    }
+
 }