From a10e162270e40d818c519bd579409cf64bbb890a Mon Sep 17 00:00:00 2001
From: Angie Byron <webchick@24967.no-reply.drupal.org>
Date: Thu, 20 Aug 2009 02:40:06 +0000
Subject: #554106 by dropcube, boombatower, deekayen, and sun: Let DWTC handle
 HTTP location header to ensure drupal_generate_test_ua() is called. Fix for
 testing bot.

---
 modules/simpletest/drupal_web_test_case.php | 19 +++++++++++++------
 1 file changed, 13 insertions(+), 6 deletions(-)

(limited to 'modules/simpletest/drupal_web_test_case.php')

diff --git a/modules/simpletest/drupal_web_test_case.php b/modules/simpletest/drupal_web_test_case.php
index 51b983a34..1755d0bc7 100644
--- a/modules/simpletest/drupal_web_test_case.php
+++ b/modules/simpletest/drupal_web_test_case.php
@@ -1237,8 +1237,7 @@ class DrupalWebTestCase extends DrupalTestCase {
       $curl_options = $this->additionalCurlOptions + array(
         CURLOPT_COOKIEJAR => $this->cookieFile,
         CURLOPT_URL => $base_url,
-        CURLOPT_FOLLOWLOCATION => TRUE,
-        CURLOPT_MAXREDIRS => 5,
+        CURLOPT_FOLLOWLOCATION => FALSE,
         CURLOPT_RETURNTRANSFER => TRUE,
         CURLOPT_SSL_VERIFYPEER => FALSE, // Required to make the tests run on https.
         CURLOPT_SSL_VERIFYHOST => FALSE, // Required to make the tests run on https.
@@ -1389,7 +1388,7 @@ class DrupalWebTestCase extends DrupalTestCase {
     $this->refreshVariables(); // Ensure that any changes to variables in the other thread are picked up.
 
     // Replace original page output with new output from redirected page(s).
-    if (($new = $this->checkForMetaRefresh())) {
+    if (($new = $this->checkForRedirect())) {
       $out = $new;
     }
     $this->verbose('GET request to: ' . $path .
@@ -1479,7 +1478,7 @@ class DrupalWebTestCase extends DrupalTestCase {
           $this->refreshVariables();
 
           // Replace original page output with new output from redirected page(s).
-          if (($new = $this->checkForMetaRefresh())) {
+          if (($new = $this->checkForRedirect())) {
             $out = $new;
           }
           $this->verbose('POST request to: ' . $path .
@@ -1499,14 +1498,22 @@ class DrupalWebTestCase extends DrupalTestCase {
   }
 
   /**
-   * Check for meta refresh tag and if found call drupalGet() recursively. This
+   * Check for meta refresh tag or HTTP location header.
+   *
+   * If either redirect indicators are found call drupalGet() recursively. This
    * function looks for the http-equiv attribute to be set to "Refresh"
    * and is case-sensitive.
    *
    * @return
    *   Either the new page content or FALSE.
    */
-  protected function checkForMetaRefresh() {
+  protected function checkForRedirect() {
+    foreach ($this->headers as $header) {
+      if (strpos($header, 'Location') !== FALSE) {
+        return $this->drupalGet(trim(str_replace('Location:', '', $header)));
+      }
+    }
+
     if ($this->drupalGetContent() != '' && $this->parse()) {
       $refresh = $this->xpath('//meta[@http-equiv="Refresh"]');
       if (!empty($refresh)) {
-- 
cgit v1.2.3