Use action=tokens for api test tokens

[lhc/web/wiklou.git] / tests / phpunit / includes / api / ApiTestCase.php

diff --git a/tests/phpunit/includes/api/ApiTestCase.php b/tests/phpunit/includes/api/ApiTestCase.php
index 1559bef..374eded 100644 (file)
--- a/tests/phpunit/includes/api/ApiTestCase.php
+++ b/tests/phpunit/includes/api/ApiTestCase.php
@@ -139,11 +139,15 @@ abstract class ApiTestCase extends MediaWikiLangTestCase {
                }
        }
 
-       protected function doLogin() {
+       protected function doLogin( $user = 'sysop' ) {
+               if( !array_key_exists( $user, self::$users ) ){
+                       throw new MWException( "Can not log in to undefined user $user" );
+               }
+
                $data = $this->doApiRequest( array(
                        'action' => 'login',
-                       'lgname' => self::$users['sysop']->username,
-                       'lgpassword' => self::$users['sysop']->password ) );
+                       'lgname' => self::$users[ $user ]->username,
+                       'lgpassword' => self::$users[ $user ]->password ) );
 
                $token = $data[0]['login']['token'];
 
@@ -151,8 +155,8 @@ abstract class ApiTestCase extends MediaWikiLangTestCase {
                        array(
                                'action' => 'login',
                                'lgtoken' => $token,
-                               'lgname' => self::$users['sysop']->username,
-                               'lgpassword' => self::$users['sysop']->password,
+                               'lgname' => self::$users[ $user ]->username,
+                               'lgpassword' => self::$users[ $user ]->password,
                        ),
                        $data[2]
                );
@@ -162,12 +166,15 @@ abstract class ApiTestCase extends MediaWikiLangTestCase {
 
        protected function getTokenList( $user, $session = null ) {
                $data = $this->doApiRequest( array(
-                       'action' => 'query',
-                       'titles' => 'Main Page',
-                       'intoken' => 'edit|delete|protect|move|block|unblock|watch',
-                       'prop' => 'info' ), $session, false, $user->user );
+                       'action' => 'tokens',
+                       'type' => 'edit|delete|protect|move|block|unblock|watch'
+               ), $session, false, $user->user );
 
-               return $data;
+               if( !array_key_exists( 'tokens', $data[0] ) ){
+                       throw new MWException( 'Api failed to return a token list' );
+               }
+
+               return $data[0]['tokens'];
        }
 
        public function testApiTestGroup() {