1 files changed, 21 insertions, 25 deletions

diff --git a/deluge/tests/test_rpcserver.py b/deluge/tests/test_rpcserver.py
index 02f9af023..982d1d5f1 100644
--- a/deluge/tests/test_rpcserver.py
+++ b/deluge/tests/test_rpcserver.py
@@ -1,4 +1,3 @@
-# -*- coding: utf-8 -*-
 #
 # Copyright (C) 2013 Bro <bro.development@gmail.com>
 #
@@ -7,18 +6,15 @@
 # See LICENSE for more details.
 #
 
-from __future__ import unicode_literals
-
 import deluge.component as component
 import deluge.error
 from deluge.common import get_localhost_auth
+from deluge.conftest import BaseTestCase
 from deluge.core import rpcserver
 from deluge.core.authmanager import AuthManager
 from deluge.core.rpcserver import DelugeRPCProtocol, RPCServer
 from deluge.log import setup_logger
 
-from .basetest import BaseTestCase
-
 setup_logger('none')
 
 
@@ -30,7 +26,7 @@ class DelugeRPCProtocolTester(DelugeRPCProtocol):
         self.messages.append(data)
 
 
-class RPCServerTestCase(BaseTestCase):
+class TestRPCServer(BaseTestCase):
     def set_up(self):
         self.rpcserver = RPCServer(listen=False)
         self.rpcserver.factory.protocol = DelugeRPCProtocolTester
@@ -60,15 +56,15 @@ class RPCServerTestCase(BaseTestCase):
         e = TorrentFolderRenamedEvent(*data)
         self.rpcserver.emit_event_for_session_id(self.session_id, e)
         msg = self.protocol.messages.pop()
-        self.assertEqual(msg[0], rpcserver.RPC_EVENT, str(msg))
-        self.assertEqual(msg[1], 'TorrentFolderRenamedEvent', str(msg))
-        self.assertEqual(msg[2], data, str(msg))
+        assert msg[0] == rpcserver.RPC_EVENT, str(msg)
+        assert msg[1] == 'TorrentFolderRenamedEvent', str(msg)
+        assert msg[2] == data, str(msg)
 
     def test_invalid_client_login(self):
         self.protocol.dispatch(self.request_id, 'daemon.login', [1], {})
         msg = self.protocol.messages.pop()
-        self.assertEqual(msg[0], rpcserver.RPC_ERROR)
-        self.assertEqual(msg[1], self.request_id)
+        assert msg[0] == rpcserver.RPC_ERROR
+        assert msg[1] == self.request_id
 
     def test_valid_client_login(self):
         self.authmanager = AuthManager()
@@ -77,9 +73,9 @@ class RPCServerTestCase(BaseTestCase):
             self.request_id, 'daemon.login', auth, {'client_version': 'Test'}
         )
         msg = self.protocol.messages.pop()
-        self.assertEqual(msg[0], rpcserver.RPC_RESPONSE, str(msg))
-        self.assertEqual(msg[1], self.request_id, str(msg))
-        self.assertEqual(msg[2], rpcserver.AUTH_LEVEL_ADMIN, str(msg))
+        assert msg[0] == rpcserver.RPC_RESPONSE, str(msg)
+        assert msg[1] == self.request_id, str(msg)
+        assert msg[2] == rpcserver.AUTH_LEVEL_ADMIN, str(msg)
 
     def test_client_login_error(self):
         # This test causes error log prints while running the test...
@@ -90,24 +86,24 @@ class RPCServerTestCase(BaseTestCase):
             self.request_id, 'daemon.login', auth, {'client_version': 'Test'}
         )
         msg = self.protocol.messages.pop()
-        self.assertEqual(msg[0], rpcserver.RPC_ERROR)
-        self.assertEqual(msg[1], self.request_id)
-        self.assertEqual(msg[2], 'WrappedException')
-        self.assertEqual(msg[3][1], 'AttributeError')
+        assert msg[0] == rpcserver.RPC_ERROR
+        assert msg[1] == self.request_id
+        assert msg[2] == 'WrappedException'
+        assert msg[3][1] == 'AttributeError'
 
     def test_client_invalid_method_call(self):
         self.authmanager = AuthManager()
         auth = get_localhost_auth()
         self.protocol.dispatch(self.request_id, 'invalid_function', auth, {})
         msg = self.protocol.messages.pop()
-        self.assertEqual(msg[0], rpcserver.RPC_ERROR)
-        self.assertEqual(msg[1], self.request_id)
-        self.assertEqual(msg[2], 'WrappedException')
-        self.assertEqual(msg[3][1], 'AttributeError')
+        assert msg[0] == rpcserver.RPC_ERROR
+        assert msg[1] == self.request_id
+        assert msg[2] == 'WrappedException'
+        assert msg[3][1] == 'AttributeError'
 
     def test_daemon_info(self):
         self.protocol.dispatch(self.request_id, 'daemon.info', [], {})
         msg = self.protocol.messages.pop()
-        self.assertEqual(msg[0], rpcserver.RPC_RESPONSE, str(msg))
-        self.assertEqual(msg[1], self.request_id, str(msg))
-        self.assertEqual(msg[2], deluge.common.get_version(), str(msg))
+        assert msg[0] == rpcserver.RPC_RESPONSE, str(msg)
+        assert msg[1] == self.request_id, str(msg)
+        assert msg[2] == deluge.common.get_version(), str(msg)