Fixes tests.

I also split Athenticaton class into two. So I keep the same API for old code
and have different things in two different classes. #795

1 files changed, 8 insertions, 10 deletions

diff --git a/service/test/unit/test_authentication.py b/service/test/unit/test_authentication.py
index 2fb97d69..34138b5e 100644
--- a/service/test/unit/test_authentication.py
+++ b/service/test/unit/test_authentication.py
@@ -1,34 +1,32 @@
 from twisted.trial import unittest
 
-from leap.bitmask.bonafide._srp import SRPAuthError
-from pixelated.authentication import Authentication
+from pixelated.authentication import Authenticator
 
 
-class AuthenticationTest(unittest.TestCase):
-
+class AuthenticatorTest(unittest.TestCase):
     def test_authenticates_with_username_and_password(self):
         self.fail()
 
     def test_validate_username_accepts_username(self):
-        auth = Authentication('domain.org')
+        auth = Authenticator('domain.org')
         self.assertTrue(auth.validate_username('username'))
 
     def test_validate_username_accepts_email_address(self):
-        auth = Authentication('domain.org')
+        auth = Authenticator('domain.org')
         self.assertTrue(auth.validate_username('username@domain.org'))
 
     def test_validate_username_denies_other_domains(self):
-        auth = Authentication('domain.org')
+        auth = Authenticator('domain.org')
         self.assertFalse(auth.validate_username('username@wrongdomain.org'))
 
     def test_username_with_domain(self):
-        auth = Authentication('domain.org')
+        auth = Authenticator('domain.org')
         self.assertEqual('user@domain.org', auth.username_with_domain('user'))
 
     def test_extract_username_extracts_from_plain_username(self):
-        auth = Authentication('domain.org')
+        auth = Authenticator('domain.org')
         self.assertEqual(auth.extract_username('user'), 'user')
 
     def test_extract_username_extracts_from_email_address(self):
-        auth = Authentication('domain.org')
+        auth = Authenticator('domain.org')
         self.assertEqual(auth.extract_username('user@domain.org'), 'user')