def setUp(self):
self.service1 = LoginService("MusterService", ["fileStorage"])
self.service2 = LoginService("BetonService", ["fileStorage"])
self.oauthservice1 = OAuth2Service.from_service(
self.service1,
"http://localhost:5000/oauth/authorize",
"http://localhost:5000/oauth/refresh",
"ABC",
"XYZ",
)
self.oauthservice2 = OAuth2Service.from_service(
self.service2,
"http://localhost:5000/oauth/authorize",
"http://localhost:5000/oauth/refresh",
"DEF",
"MNO",
)
self.user1 = User("Max Mustermann")
self.user2 = User("Mimi Mimikri")
self.token1 = Token(self.user1, self.service1, "ABC")
self.token2 = Token(self.user1, self.service2, "DEF")
self.oauthtoken1 = OAuth2Token(self.user1, self.oauthservice1, "ABC",
"XYZ")
self.oauthtoken2 = OAuth2Token(self.user1, self.oauthservice2, "DEF",
"UVW")
def setUp(self):
Util.monkeypatch()
self.empty_storage = Storage(**get_opts())
self.user1 = User("Max Mustermann")
self.user2 = User("Mimi Mimikri")
self.service1 = LoginService(servicename="MusterService",
implements=["metadata"])
self.service2 = LoginService(servicename="FahrService",
implements=["metadata"])
self.oauthservice1 = OAuth2Service(
servicename="BetonService",
implements=["metadata"],
authorize_url="http://localhost/oauth/authorize",
refresh_url="http://localhost/oauth/token",
client_id="MNO",
client_secret="UVW",
)
self.oauthservice2 = OAuth2Service(
servicename="FlugService",
implements=["metadata"],
authorize_url="http://localhost21/oauth/authorize",
refresh_url="http://localhost21/oauth/token",
client_id="XCA",
client_secret="BCXY",
)
self.empty_storage.addService(self.service1)
self.empty_storage.addService(self.oauthservice1)
self.empty_storage.addService(self.oauthservice2)
self.token1 = Token(self.user1, self.service1, "ABC")
self.token_like_token1 = Token(self.user1, self.service1, "DEF")
self.token2 = Token(self.user1, self.oauthservice1, "XYZ")
self.token3 = Token(self.user2, self.service2, "XASD")
self.token4 = Token(self.user2, self.service1, "IOAJSD")
self.oauthtoken1 = OAuth2Token(self.user1, self.oauthservice1,
"ABC", "X_ABC")
self.oauthtoken_like_token1 = OAuth2Token(self.user1,
self.oauthservice1,
"ABC", "X_DEF")
self.oauthtoken2 = OAuth2Token(self.user1, self.oauthservice1,
"XYZ", "X_XYZ")
self.oauthtoken3 = OAuth2Token(self.user1, self.oauthservice2,
"XYZ", "X_XYZ")
def test_parseToken(self):
user1 = User("MaxMustermann")
service1 = LoginService("MusterService", ["fileStorage"])
token1 = Token(user1, service1, "ABC")
serviceport = "{}".format(token1.service.servicename)
data = {
"userId":
"port-{}://{}:{}".format("musterservice", "MaxMustermann", "ABC")
}
self.assertEqual(Util.parseToken(token1), data)
def test_token_json(self):
dump = json.dumps(self.token1)
# self.assertEqual(dump, json.dumps(expected))
self.assertEqual(Token.from_json(dump), self.token1)
expected = {
"type": "OAuth2Token",
"data": {
"service": self.service1,
"access_token": self.oauthtoken1.access_token,
"refresh_token": self.oauthtoken1.refresh_token,
"expiration_date": str(self.oauthtoken1.expiration_date),
},
}
dump = json.dumps(self.oauthtoken1)
# self.assertEqual(dump, json.dumps(expected))
self.assertEqual(OAuth2Token.from_json(dump),
self.oauthtoken1,
msg=dump)
def test_tokenstorage_service_implementstype(self):
empty_storage = Storage(**get_opts())
service = LoginService(servicename="longname",
implements=["fileStorage", "metadata"])
empty_storage.addUser(self.user1)
token1 = Token(self.user1, service, "ISADF")
# test the exception raise
with self.assertRaises(ServiceNotExistsError):
empty_storage.addTokenToUser(token1, self.user1)
# now should work
self.assertTrue(empty_storage.addService(service))
self.assertTrue(empty_storage.addTokenToUser(token1, self.user1))
self.assertEqual(empty_storage.getTokens(self.user1), [token1])
with self.assertRaises(ServiceExistsAlreadyError):
empty_storage.addService(service)
def test_tokenstorage_tokens_under_user(self):
oauthtoken1 = OAuth2Token(self.user1, self.oauthservice1, "ABC",
"X_ABC")
self.empty_storage.addTokenToUser(oauthtoken1,
self.user1,
Force=True)
oauthtoken2 = OAuth2Token(self.user1, self.oauthservice2, "XYZ",
"X_XYZ")
self.empty_storage.addTokenToUser(oauthtoken2,
self.user1,
Force=True)
token1 = Token(self.user1, self.service2, "ISADF")
with self.assertRaises(ServiceNotExistsError):
self.empty_storage.addTokenToUser(token1,
self.user1,
Force=True)
self.empty_storage.addTokenToUser(self.token1,
self.user1,
Force=True)
