1 files changed, 3 insertions, 14 deletions
diff --git a/site/app/tests/services/update.test.php b/site/app/tests/services/update.test.php
index 5209233..edd3957 100644
--- a/site/app/tests/services/update.test.php
+++ b/site/app/tests/services/update.test.php
@@ -60,7 +60,7 @@
 // Require update function file.
 require_once(APP.'webroot/services/functions.php');
 
-class UpdateServiceTest extends WebTestCase {
+class UpdateServiceTest extends WebTestHelper {
 	
     var $_args;
     var $_noUpdatesXml;
@@ -105,24 +105,13 @@ NoUpdateXml;
 	}
 
     /**
-     * Build an update URI based on _args.
-     * @param array $args update URI arguments
-     * @return string resulting URI
-     */
-    function _buildUpdateUri($args=array()) {
-        $_buf = array();
-        foreach ($args as $key=>$val) {
-            $_buf[] = $key.'='.$val;
-        }
-        return SERVICE_URL.'/update.php?test=1&'.implode('&',$_buf);
-    }
-
-    /**
      * Retrieve XML document based on _args.
      * @param array $args update URI arguments
      * @return string resulting XML document as a string
      */
     function _getXml($args=array()) {
+        // Use the test db.
+        $args['test'] = 1;
         return $this->get(SERVICE_URL.'/update.php',$args);
     }