From 235b7348ae936be848163ff327f6e2d5aaea605c Mon Sep 17 00:00:00 2001
From: Calum Lind <calumlind+deluge@gmail.com>
Date: Thu, 16 May 2013 12:54:33 +0100
Subject: [PATCH] Fix test_client test

---
 deluge/tests/test_client.py | 12 +++++-------
 1 file changed, 5 insertions(+), 7 deletions(-)

diff --git a/deluge/tests/test_client.py b/deluge/tests/test_client.py
index 0e9bf5c1b..735803ac2 100644
--- a/deluge/tests/test_client.py
+++ b/deluge/tests/test_client.py
@@ -71,16 +71,14 @@ class ClientTestCase(unittest.TestCase):
 
     def test_connect_no_credentials(self):
 
-        d = client.connect("localhost", 58846)
+        d = client.connect()
 
-        def on_failure(failure):
-            self.assertEqual(
-                failure.trap(error.AuthenticationRequired),
-                error.AuthenticationRequired
-            )
+        def on_connect(result):
+            self.assertEqual(client.get_auth_level(), AUTH_LEVEL_ADMIN)
             self.addCleanup(client.disconnect)
+            return result
 
-        d.addErrback(on_failure)
+        d.addCallback(on_connect)
         return d
 
     def test_connect_localclient(self):