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
101
102
103
104
105
106
107
108
|
"""
Copyright (c) 2017-2018 Nitrokey UG
This file is part of libnitrokey.
libnitrokey is free software: you can redistribute it and/or modify
it under the terms of the GNU Lesser General Public License as published by
the Free Software Foundation, either version 3 of the License, or
any later version.
libnitrokey is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
GNU General Public License for more details.
You should have received a copy of the GNU Lesser General Public License
along with libnitrokey. If not, see <http://www.gnu.org/licenses/>.
SPDX-License-Identifier: LGPL-3.0
"""
import pprint
from time import sleep
import pytest
from collections import defaultdict
from tqdm import tqdm
from conftest import skip_if_device_version_lower_than
from constants import DefaultPasswords, DeviceErrorCode
from misc import gs, wait, ffi, bb
pprint = pprint.PrettyPrinter(indent=4).pprint
@pytest.mark.other
@pytest.mark.info
def test_list_devices(C):
infos = C.NK_list_devices()
assert infos != ffi.NULL
C.NK_free_device_info(infos)
@pytest.mark.other
@pytest.mark.info
def test_connect_with_path(C):
ids = gs(C.NK_list_devices_by_cpuID())
# NK_list_devices_by_cpuID already opened the devices, so we have to close
# them before trying to reconnect
assert C.NK_logout() == 0
devices_list = ids.split(b';')
for value in devices_list:
parts = value.split(b'_p_')
assert len(parts) < 3
if len(parts) == 2:
path = parts[1]
else:
path = parts[0]
assert C.NK_connect_with_path(path) == 1
@pytest.mark.other
@pytest.mark.info
def test_get_status_storage_multiple(C):
ids = gs(C.NK_list_devices_by_cpuID())
print(ids)
devices_list = ids.split(b';')
#
# for s in devices_list:
# res = C.NK_connect_with_ID(s)
# assert res == 1
# # st = gs(C.NK_get_status_storage_as_string())
# # print(len(st))
# C.NK_lock_device()
#
for s in devices_list:
res = C.NK_connect_with_ID(s)
assert res == 1
v = C.NK_fill_SD_card_with_random_data(b'12345678')
# print(v)
devices_count = len(devices_list)
assert devices_count != 0
b = 0
last_b = 0
with tqdm(total=devices_count*100) as pbar:
while b/devices_count < 100:
pbar.update(b - last_b)
b = defaultdict (lambda: 0)
ids = gs(C.NK_list_devices_by_cpuID())
devices_list = ids.split(b';')
devices_count = len(devices_list)
for s in devices_list:
res = C.NK_connect_with_ID(s)
if res != 1: continue
b[s] += C.NK_get_progress_bar_value()
print(b)
b = sum(b.values())
print('{}: {}'.format(b, int(b/devices_count) * '='))
sleep(5)
|
