Index: modules/simpletest/drupal_web_test_case.php
===================================================================
RCS file: /cvs/drupal/drupal/modules/simpletest/drupal_web_test_case.php,v
retrieving revision 1.27
diff -u -r1.27 drupal_web_test_case.php
--- modules/simpletest/drupal_web_test_case.php	18 Jul 2008 07:30:34 -0000	1.27
+++ modules/simpletest/drupal_web_test_case.php	26 Jul 2008 01:43:26 -0000
@@ -562,6 +562,40 @@
   }
 
   /**
+   * Set the current user stored in $GLOBALS['user'].
+   *
+   * @param $new_user User to switch to, either UID or user object.
+   * @return Current user.
+   */
+  function drupalSetUser($new_user = NULL) {
+    global $user;
+    static $user_original;
+
+    if (!$new_user) {
+      $user = $user_original;
+      return $user;
+    }
+
+    $user_original = $user;
+    if (is_int($user)) {
+      $user = user_load(array('uid' => $new_user));
+    }
+    else {
+      $user = $new_user;
+    }
+    return $user;
+  }
+
+  /**
+   * Restore the user that was originally loaded.
+   *
+   * @return Current user.
+   */
+  function drupalRestoreUser() {
+    return $this->drupalSetUser();
+  }
+
+  /**
    * Logs in a user with the internal browser. If already logged in then logs the current
    * user out before logging in the specified user. If no user is specified then a new
    * user will be created and logged in.