예제 #1
0
class InMemoryProviderTest(unittest.TestCase):

    def setUp(self):
        self.provider = InMemoryProvider([
            {'username': '******', 'password': '******', 'roles': []},
            {'username': '******'}
        ])

    def test_init_valid_users(self):
        self.assertEquals(2, len(self.provider.users))
        self.assertEquals('foo', self.provider.users['foo'].username)
        self.assertEquals('bar', self.provider.users['foo'].password)

        self.assertEquals('bar', self.provider.users['bar'].username)
        self.assertEquals(None, self.provider.users['bar'].password)

    def test_find_by_username(self):

        with self.assertRaises(UsernameNotFoundException):
            self.provider.loadUserByUsername('fake')

        user = self.provider.loadUserByUsername('bar')

        self.assertEquals('bar', user.username)

    def test_refresh_user(self):

        user = self.provider.loadUserByUsername('bar')

        user = self.provider.refreshUser(user)

        self.assertEquals('bar', user.username)

    def test_support_class(self):
        self.assertTrue(self.provider.supportClass(User))
예제 #2
0
 def setUp(self):
     self.provider = InMemoryProvider([
         {'username': '******', 'password': '******', 'roles': []},
         {'username': '******'}
     ])