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
|
######################################################################
#
# File: b2sdk/_internal/testing/helpers/api.py
#
# Copyright 2022 Backblaze Inc. All Rights Reserved.
#
# License https://www.backblaze.com/using_b2_code.html
#
######################################################################
from __future__ import annotations
import os
from b2sdk._internal.account_info.in_memory import InMemoryAccountInfo
from b2sdk._internal.api import B2Api
from b2sdk._internal.api_config import DEFAULT_HTTP_API_CONFIG
def get_realm() -> str:
return os.environ.get('B2_TEST_ENVIRONMENT', 'production')
def get_b2_auth_data():
application_key_id = os.environ.get('B2_TEST_APPLICATION_KEY_ID')
if application_key_id is None:
raise ValueError('B2_TEST_APPLICATION_KEY_ID is not set.')
application_key = os.environ.get('B2_TEST_APPLICATION_KEY')
if application_key is None:
raise ValueError('B2_TEST_APPLICATION_KEY is not set.')
return application_key_id, application_key
def authorize(b2_auth_data, realm, api_config=DEFAULT_HTTP_API_CONFIG):
info = InMemoryAccountInfo()
b2_api = B2Api(info, api_config=api_config)
b2_api.authorize_account(*b2_auth_data, realm=realm)
return b2_api, info
|
