moto/tests/test_sts/test_server.py

40 lines
1.1 KiB
Python
Raw Normal View History

from __future__ import unicode_literals
2013-08-03 21:21:25 +00:00
import sure # noqa
2013-05-24 21:22:34 +00:00
import moto.server as server
2019-10-31 15:44:26 +00:00
"""
2013-05-24 21:22:34 +00:00
Test the different server responses
2019-10-31 15:44:26 +00:00
"""
2013-05-24 21:22:34 +00:00
def test_sts_get_session_token():
2013-12-29 01:15:37 +00:00
backend = server.create_backend_app("sts")
test_client = backend.test_client()
2019-10-31 15:44:26 +00:00
res = test_client.get("/?Action=GetSessionToken")
2013-05-24 21:22:34 +00:00
res.status_code.should.equal(200)
2014-08-26 17:25:50 +00:00
res.data.should.contain(b"SessionToken")
res.data.should.contain(b"AccessKeyId")
def test_sts_get_federation_token():
backend = server.create_backend_app("sts")
test_client = backend.test_client()
2019-10-31 15:44:26 +00:00
res = test_client.get("/?Action=GetFederationToken&Name=Bob")
res.status_code.should.equal(200)
2014-08-26 17:25:50 +00:00
res.data.should.contain(b"SessionToken")
res.data.should.contain(b"AccessKeyId")
def test_sts_get_caller_identity():
backend = server.create_backend_app("sts")
test_client = backend.test_client()
2019-10-31 15:44:26 +00:00
res = test_client.get("/?Action=GetCallerIdentity")
res.status_code.should.equal(200)
res.data.should.contain(b"Arn")
res.data.should.contain(b"UserId")
res.data.should.contain(b"Account")