File: _decorators_async.py

package info (click to toggle)
python-azure 20250603%2Bgit-1
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid, trixie
  • size: 851,724 kB
  • sloc: python: 7,362,925; ansic: 804; javascript: 287; makefile: 195; sh: 145; xml: 109
file content (45 lines) | stat: -rw-r--r-- 1,829 bytes parent folder | download
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
# -------------------------------------------------------------------------
# Copyright (c) Microsoft Corporation. All rights reserved.
# Licensed under the MIT License. See License.txt in the project root for
# license information.
# --------------------------------------------------------------------------

import os
from typing import Callable, Any

from devtools_testutils import is_live, is_live_and_not_recording, trim_kwargs_from_test_function
from azure.communication.messages._shared.utils import parse_connection_str


class MessagesPreparersAsync(object):

    @staticmethod
    def messages_test_decorator_async(func: Callable[[], object], **kwargs: Any):
        async def wrapper(self, *args, **kwargs):
            if is_live() or is_live_and_not_recording():
                self.connection_string = os.getenv("COMMUNICATION_SAMPLES_CONNECTION_STRING")
                endpoint, _ = parse_connection_str(self.connection_string)
                self.resource_name = endpoint.split(".")[0]

            else:
                self.connection_string = (
                    "endpoint=https://sanitized.unitedstates.ppe.communication.azure.net/;accesskey=fake==="
                )
                self.resource_name = "sanitized"

            return await func(self, *args, **kwargs)

        return wrapper

    @staticmethod
    def messages_test_decorator_for_token_async(func: Callable[[], object], **kwargs: Any):
        async def wrapper(self, *args, **kwargs):
            breakpoint
            if is_live() or is_live_and_not_recording():
                self.endpoint_str = os.getenv("COMMUNICATION_SAMPLES_CONNECTION_STRING")
            else:
                self.endpoint = "sanitized.unitedstates.communication.azure.com"

            return await func(self, *args, **kwargs)

        return wrapper