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
91
92
93
94
95
96
97
98
99
100
|
######################################################################
#
# File: test/unit/fixtures/folder.py
#
# Copyright 2022 Backblaze Inc. All Rights Reserved.
#
# License https://www.backblaze.com/using_b2_code.html
#
######################################################################
from __future__ import annotations
from unittest import mock
import apiver_deps
import pytest
from apiver_deps import DEFAULT_SCAN_MANAGER, B2Folder, LocalFolder, LocalPath
if apiver_deps.V <= 1:
from apiver_deps import FileVersionInfo as VFileVersion
else:
from apiver_deps import FileVersion as VFileVersion
class FakeB2Folder(B2Folder):
def __init__(self, test_files):
self.file_versions = []
for test_file in test_files:
self.file_versions.extend(self._file_versions(*test_file))
super().__init__('test-bucket', 'folder', mock.MagicMock())
def get_file_versions(self):
yield from sorted(self.file_versions, key=lambda x: x.file_name)
def _file_versions(self, name, mod_times, size=10):
"""
Makes FileVersion objects.
Positive modification times are uploads, and negative modification
times are hides. It's a hack, but it works.
"""
if apiver_deps.V <= 1:
mandatory_kwargs = {}
else:
mandatory_kwargs = {
'api': None,
'account_id': 'account-id',
'bucket_id': 'bucket-id',
'content_md5': 'content_md5',
'server_side_encryption': None,
}
return [
VFileVersion(
id_='id_%s_%d' % (name[0], abs(mod_time)),
file_name='folder/' + name,
upload_timestamp=abs(mod_time),
action='upload' if 0 < mod_time else 'hide',
size=size,
file_info={'in_b2': 'yes'},
content_type='text/plain',
content_sha1='content_sha1',
**mandatory_kwargs,
)
for mod_time in mod_times
]
class FakeLocalFolder(LocalFolder):
def __init__(self, test_files):
super().__init__('folder')
self.local_paths = [self._local_path(*test_file) for test_file in test_files]
def all_files(self, reporter, policies_manager=DEFAULT_SCAN_MANAGER):
for single_path in sorted(self.local_paths, key=lambda x: x.relative_path):
if single_path.relative_path.endswith('/'):
if policies_manager.should_exclude_b2_directory(single_path.relative_path):
continue
else:
if policies_manager.should_exclude_local_path(single_path):
continue
yield single_path
def make_full_path(self, name):
return '/dir/' + name
def _local_path(self, name, mod_times, size=10):
"""
Makes a LocalPath object for a local file.
"""
return LocalPath(name, name, mod_times[0], size)
@pytest.fixture(scope='session')
def folder_factory():
def get_folder(f_type, *files):
if f_type == 'b2':
return FakeB2Folder(files)
return FakeLocalFolder(files)
return get_folder
|
