# 安装必要的依赖库
pip install requests
# 配置认证服务器地址
AUTH_SERVER = "https://auth.example.com"
# 用户登录接口
def authenticate_user(username, password):
response = requests.post(f"{AUTH_SERVER}/login", data={
"username": username,
"password": password
})
if response.status_code == 200:
return response.json()["token"]
else:
raise Exception("Authentication failed.")
]]>
def test_sso(token):
headers = {"Authorization": f"Bearer {token}"}
response = requests.get(f"{AUTH_SERVER}/profile", headers=headers)
assert response.status_code == 200
print("SSO test passed.")
]]>