Ejemplo n.º 1
0
class TestDictRefreshDB(object):
    @pytest.fixture(autouse=True)
    def create_rdb(self):
        self.rdb = DictRefreshDB()

    def test_verify_token(self):
        token = self.rdb.create_token('client1', 'uid', 'openid', 'sub1',
                                      'authzreq', 'sid')
        assert self.rdb.verify_token('client1', token)
        assert self.rdb.verify_token('client2', token) is False

    def test_revoke_token(self):
        token = self.rdb.create_token('client1', 'uid', 'openid', 'sub1',
                                      'authzreq', 'sid')
        self.rdb.remove(token)
        assert self.rdb.verify_token('client1', token) is False
        assert pytest.raises(KeyError, 'self.rdb.get(token)')

    def test_get_token(self):
        assert pytest.raises(KeyError, 'self.rdb.get("token")')
        token = self.rdb.create_token('client1', 'uid', ['openid'], 'sub1',
                                      'authzreq', 'sid')
        assert self.rdb.get(token) == {
            'client_id': 'client1',
            'sub': 'sub1',
            'scope': ['openid'],
            'uid': 'uid',
            'authzreq': 'authzreq',
            'sid': 'sid'
        }
Ejemplo n.º 2
0
class TestDictRefreshDB(object):
    @pytest.fixture(autouse=True)
    def create_rdb(self):
        self.rdb = DictRefreshDB()

    def test_verify_token(self):
        token = self.rdb.create_token('client1', 'uid', 'openid', 'sub1',
                                      'authzreq', 'sid')
        assert self.rdb.verify_token('client1', token)
        assert self.rdb.verify_token('client2', token) is False

    def test_revoke_token(self):
        token = self.rdb.create_token('client1', 'uid', 'openid', 'sub1',
                                      'authzreq', 'sid')
        self.rdb.remove(token)
        assert self.rdb.verify_token('client1', token) is False
        assert pytest.raises(KeyError, 'self.rdb.get(token)')

    def test_get_token(self):
        assert pytest.raises(KeyError, 'self.rdb.get("token")')
        token = self.rdb.create_token('client1', 'uid', ['openid'], 'sub1',
                                      'authzreq', 'sid')
        assert self.rdb.get(token) == {'client_id': 'client1', 'sub': 'sub1',
                                       'scope': ['openid'], 'uid': 'uid',
                                       'authzreq': 'authzreq', 'sid': 'sid'}
Ejemplo n.º 3
0
 def create_rdb(self):
     self.rdb = DictRefreshDB()
Ejemplo n.º 4
0
 def create_rdb(self):
     self.rdb = DictRefreshDB()
Ejemplo n.º 5
0
class TestDictRefreshDB(object):
    @pytest.fixture(autouse=True)
    def create_rdb(self):
        self.rdb = DictRefreshDB()

    def test_verify_token(self):
        token = self.rdb.create_token("client1", "uid", "openid", "sub1",
                                      "authzreq", "sid")
        assert self.rdb.verify_token("client1", token)
        assert self.rdb.verify_token("client2", token) is False

    def test_revoke_token(self):
        token = self.rdb.create_token("client1", "uid", "openid", "sub1",
                                      "authzreq", "sid")
        self.rdb.remove(token)
        assert self.rdb.verify_token("client1", token) is False
        with pytest.raises(KeyError):
            self.rdb.get(token)

    def test_get_token(self):
        with pytest.raises(KeyError):
            self.rdb.get("token")
        token = self.rdb.create_token("client1", "uid", ["openid"], "sub1",
                                      "authzreq", "sid")
        assert self.rdb.get(token) == {
            "client_id": "client1",
            "sub": "sub1",
            "scope": ["openid"],
            "uid": "uid",
            "authzreq": "authzreq",
            "sid": "sid",
        }