|
1 | | -from collections import namedtuple |
| 1 | +from typing import Any |
2 | 2 |
|
3 | | -from azure.core.credentials import TokenCredential |
4 | | - |
5 | | -MockToken = namedtuple("MockToken", ["token", "expires_on"]) |
| 3 | +from azure.core.credentials import AccessToken, TokenCredential |
6 | 4 |
|
7 | 5 |
|
8 | 6 | class MockAzureCredential(TokenCredential): |
9 | | - def get_token(self, uri): |
10 | | - return MockToken("", 9999999999) |
| 7 | + def get_token( |
| 8 | + self, |
| 9 | + *scopes: str, |
| 10 | + claims: str | None = None, |
| 11 | + tenant_id: str | None = None, |
| 12 | + enable_cae: bool = False, |
| 13 | + **kwargs: Any, |
| 14 | + ) -> AccessToken: |
| 15 | + return AccessToken("", 9999999999) |
11 | 16 |
|
12 | 17 |
|
13 | 18 | class MockAzureCredentialExpired(TokenCredential): |
14 | 19 | def __init__(self): |
15 | 20 | self.access_number = 0 |
16 | 21 |
|
17 | | - async def get_token(self, uri): |
| 22 | + def get_token( |
| 23 | + self, |
| 24 | + *scopes: str, |
| 25 | + claims: str | None = None, |
| 26 | + tenant_id: str | None = None, |
| 27 | + enable_cae: bool = False, |
| 28 | + **kwargs: Any, |
| 29 | + ) -> AccessToken: |
18 | 30 | self.access_number += 1 |
19 | 31 | if self.access_number == 1: |
20 | | - return MockToken("", 0) |
| 32 | + return AccessToken("", 0) |
21 | 33 | else: |
22 | | - return MockToken("", 9999999999) |
| 34 | + return AccessToken("", 9999999999) |
0 commit comments