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
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
|
######################################################################
#
# File: test/unit/test_session.py
#
# Copyright 2021 Backblaze Inc. All Rights Reserved.
#
# License https://www.backblaze.com/using_b2_code.html
#
######################################################################
from __future__ import annotations
from unittest import mock
from apiver_deps import AuthInfoCache, B2Session, DummyCache, InMemoryAccountInfo
from .account_info.fixtures import * # noqa
from .fixtures import * # noqa
class TestAuthorizeAccount:
@pytest.fixture(autouse=True)
def setup(self, b2_session):
self.b2_session = b2_session
@pytest.mark.parametrize(
'authorize_call_kwargs',
[
pytest.param(
dict(
account_id=mock.ANY,
auth_token=mock.ANY,
api_url=mock.ANY,
download_url=mock.ANY,
recommended_part_size=mock.ANY,
absolute_minimum_part_size=mock.ANY,
application_key='456',
realm='dev',
s3_api_url=mock.ANY,
allowed=mock.ANY,
application_key_id='123',
),
marks=pytest.mark.apiver(from_ver=2),
),
pytest.param(
dict(
account_id=mock.ANY,
auth_token=mock.ANY,
api_url=mock.ANY,
download_url=mock.ANY,
minimum_part_size=mock.ANY,
application_key='456',
realm='dev',
s3_api_url=mock.ANY,
allowed=mock.ANY,
application_key_id='123',
),
marks=pytest.mark.apiver(to_ver=1),
),
],
)
def test_simple_authorization(self, authorize_call_kwargs):
self.b2_session.authorize_account('dev', '123', '456')
self.b2_session.raw_api.authorize_account.assert_called_once_with(
'http://api.backblazeb2.xyz:8180', '123', '456'
)
assert self.b2_session.cache.clear.called is False
self.b2_session.account_info.set_auth_data.assert_called_once_with(**authorize_call_kwargs)
def test_clear_cache(self):
self.b2_session.account_info.is_same_account.return_value = False
self.b2_session.authorize_account('dev', '123', '456')
assert self.b2_session.cache.clear.called is True
def test_session__with_in_memory_account_info(apiver_int):
memory_info = InMemoryAccountInfo()
b2_session = B2Session(
account_info=memory_info,
)
assert b2_session.account_info is memory_info
if apiver_int < 3:
assert isinstance(b2_session.cache, DummyCache)
else:
assert isinstance(b2_session.cache, AuthInfoCache)
assert b2_session.cache.info is memory_info
|
