Revize f8581e41
Přidáno uživatelem Stanislav Král před asi 4 roky(ů)
tests/integration_tests/dao/certificate_repository_test.py | ||
---|---|---|
4 | 4 |
from src.model.private_key import PrivateKey |
5 | 5 |
|
6 | 6 |
|
7 |
def test_set_certificate_revoked(certificate_repository, private_key_repository): |
|
7 |
def test_set_certificate_revoked(certificate_repository, private_key_repository_unique):
|
|
8 | 8 |
private_key = PrivateKey(private_key_id=-1, private_key="test_pk", password="test_password") |
9 |
private_key_id = private_key_repository.create(private_key) |
|
9 |
private_key_id = private_key_repository_unique.create(private_key)
|
|
10 | 10 |
|
11 | 11 |
assert type(private_key_id) is int |
12 | 12 |
|
... | ... | |
45 | 45 |
assert certificate.revocation_reason == REV_REASON_UNSPECIFIED |
46 | 46 |
|
47 | 47 |
|
48 |
def test_clear_certificate_revocation(certificate_repository, private_key_repository): |
|
48 |
def test_clear_certificate_revocation(certificate_repository, private_key_repository_unique):
|
|
49 | 49 |
private_key = PrivateKey(private_key_id=-1, private_key="test_pk", password="test_password") |
50 |
private_key_id = private_key_repository.create(private_key=private_key) |
|
50 |
private_key_id = private_key_repository_unique.create(private_key=private_key)
|
|
51 | 51 |
|
52 | 52 |
assert type(private_key_id) is int |
53 | 53 |
|
... | ... | |
84 | 84 |
assert certificate.revocation_reason == "" |
85 | 85 |
|
86 | 86 |
|
87 |
def test_get_all_revoked_by(certificate_repository, private_key_repository): |
|
87 |
def test_get_all_revoked_by(certificate_repository, private_key_repository_unique):
|
|
88 | 88 |
private_key = PrivateKey(private_key_id=-1, private_key="test_pk", password="test_password") |
89 |
private_key_id = private_key_repository.create(private_key=private_key) |
|
89 |
private_key_id = private_key_repository_unique.create(private_key=private_key)
|
|
90 | 90 |
|
91 | 91 |
assert type(private_key_id) is int |
92 | 92 |
|
... | ... | |
139 | 139 |
assert certificates[0].certificate_id == child_1_id and certificates[1].certificate_id == child_3_id |
140 | 140 |
|
141 | 141 |
|
142 |
def test_get_all_issued_by(certificate_repository, private_key_repository): |
|
142 |
def test_get_all_issued_by(certificate_repository, private_key_repository_unique):
|
|
143 | 143 |
private_key = PrivateKey(private_key_id=-1, private_key="test_pk", password="test_password") |
144 |
private_key_id = private_key_repository.create(private_key=private_key) |
|
144 |
private_key_id = private_key_repository_unique.create(private_key=private_key)
|
|
145 | 145 |
|
146 | 146 |
assert type(private_key_id) is int |
147 | 147 |
|
tests/integration_tests/dao/conftest.py | ||
---|---|---|
2 | 2 |
import sqlite3 |
3 | 3 |
from sqlite3 import Connection, Cursor |
4 | 4 |
|
5 |
from src.config.configuration import test_configuration |
|
6 |
from src.config.connection_provider import ConnectionProvider |
|
5 | 7 |
from src.dao.certificate_repository import CertificateRepository |
6 | 8 |
from src.dao.private_key_repository import PrivateKeyRepository |
7 | 9 |
from src.db.init_queries import SCHEMA_SQL, DEFAULT_VALUES_SQL |
... | ... | |
10 | 12 |
# scope="module" means that this fixture is run once per module |
11 | 13 |
@pytest.fixture(scope="module") |
12 | 14 |
def connection(): |
13 |
print("Creating a new SQLITE connection to the test DB") |
|
14 |
connection: Connection = sqlite3.connect(':memory:') |
|
15 |
|
|
16 |
# yield the created connection |
|
17 |
yield connection |
|
15 |
return ConnectionProvider().connect(test_configuration()) |
|
18 | 16 |
|
19 | 17 |
|
20 | 18 |
@pytest.fixture(scope="module") |
21 | 19 |
def cursor(connection): |
22 |
cursor = connection.cursor() |
|
23 |
|
|
24 |
# execute db initialisation script |
|
25 |
cursor.executescript(SCHEMA_SQL) |
|
20 |
return connection.cursor() |
|
26 | 21 |
|
27 |
# insert default values |
|
28 |
cursor.executescript(DEFAULT_VALUES_SQL) |
|
29 |
|
|
30 |
return cursor |
|
31 | 22 |
|
32 | 23 |
@pytest.fixture |
33 | 24 |
def connection_unique(): |
34 |
print("Creating a new SQLITE connection to the test DB") |
|
35 |
connection: Connection = sqlite3.connect(':memory:') |
|
36 |
|
|
37 |
# yield the created connection |
|
38 |
yield connection |
|
25 |
return ConnectionProvider().connect(test_configuration()) |
|
39 | 26 |
|
40 | 27 |
|
41 | 28 |
@pytest.fixture |
42 | 29 |
def cursor_unique(connection): |
43 |
cursor = connection.cursor()
|
|
30 |
return connection.cursor()
|
|
44 | 31 |
|
45 |
# execute db initialisation script |
|
46 |
cursor.executescript(SCHEMA_SQL) |
|
47 |
|
|
48 |
# insert default values |
|
49 |
cursor.executescript(DEFAULT_VALUES_SQL) |
|
50 |
|
|
51 |
return cursor |
|
52 | 32 |
|
53 | 33 |
@pytest.fixture |
54 |
def certificate_repository(connection, cursor): |
|
55 |
return CertificateRepository(connection, cursor) |
|
34 |
def certificate_repository(connection): |
|
35 |
return CertificateRepository(connection) |
|
36 |
|
|
56 | 37 |
|
57 | 38 |
@pytest.fixture |
58 |
def private_key_repository(connection_unique, cursor_unique): |
|
59 |
return PrivateKeyRepository(connection_unique, cursor_unique) |
|
39 |
def private_key_repository_unique(connection_unique, cursor_unique): |
|
40 |
return PrivateKeyRepository(connection_unique) |
tests/integration_tests/dao/private_key_repository_test.py | ||
---|---|---|
4 | 4 |
from src.model.private_key import PrivateKey |
5 | 5 |
|
6 | 6 |
|
7 |
def test_create(private_key_repository): |
|
7 |
def test_create(private_key_repository_unique):
|
|
8 | 8 |
private_key_1 = PrivateKey(private_key_id=-1, private_key="test_private_key_1", password="test_password_1") |
9 | 9 |
private_key_2 = PrivateKey(private_key_id=-1, private_key="test_private_key_2") |
10 | 10 |
|
... | ... | |
16 | 16 |
assert private_key_2.private_key == "test_private_key_2" |
17 | 17 |
assert private_key_2.password is None |
18 | 18 |
|
19 |
private_key_1_id = private_key_repository.create(private_key=private_key_1) |
|
20 |
private_key_2_id = private_key_repository.create(private_key=private_key_2) |
|
19 |
private_key_1_id = private_key_repository_unique.create(private_key=private_key_1)
|
|
20 |
private_key_2_id = private_key_repository_unique.create(private_key=private_key_2)
|
|
21 | 21 |
|
22 | 22 |
assert type(private_key_1_id) is int |
23 | 23 |
assert type(private_key_2_id) is int |
24 | 24 |
|
25 |
private_key_1 = private_key_repository.read(private_key_id=private_key_1_id) |
|
26 |
private_key_2 = private_key_repository.read(private_key_id=private_key_2_id) |
|
25 |
private_key_1 = private_key_repository_unique.read(private_key_id=private_key_1_id)
|
|
26 |
private_key_2 = private_key_repository_unique.read(private_key_id=private_key_2_id)
|
|
27 | 27 |
|
28 | 28 |
assert private_key_1.private_key_id == private_key_1_id |
29 | 29 |
assert private_key_1.private_key == "test_private_key_1" |
... | ... | |
34 | 34 |
assert private_key_2.password is None |
35 | 35 |
|
36 | 36 |
|
37 |
def test_read(private_key_repository): |
|
37 |
def test_read(private_key_repository_unique):
|
|
38 | 38 |
private_key = PrivateKey(private_key_id=-1, private_key="test_private_key", password="test_password") |
39 | 39 |
|
40 |
private_key_id = private_key_repository.create(private_key=private_key) |
|
40 |
private_key_id = private_key_repository_unique.create(private_key=private_key)
|
|
41 | 41 |
|
42 | 42 |
assert type(private_key_id) is int |
43 | 43 |
|
44 |
private_key = private_key_repository.read(private_key_id=private_key_id) |
|
44 |
private_key = private_key_repository_unique.read(private_key_id=private_key_id)
|
|
45 | 45 |
|
46 | 46 |
assert private_key.private_key_id == private_key_id |
47 | 47 |
assert private_key.private_key == "test_private_key" |
48 | 48 |
assert private_key.password == "test_password" |
49 | 49 |
|
50 | 50 |
|
51 |
def test_read_invalid_id(private_key_repository): |
|
51 |
def test_read_invalid_id(private_key_repository_unique):
|
|
52 | 52 |
none_id = 100 |
53 |
private_key = private_key_repository.read(private_key_id=none_id) |
|
53 |
private_key = private_key_repository_unique.read(private_key_id=none_id)
|
|
54 | 54 |
|
55 | 55 |
assert private_key is None |
56 | 56 |
|
57 | 57 |
|
58 |
def test_read_all(private_key_repository): |
|
59 |
private_keys = private_key_repository.read_all() |
|
58 |
def test_read_all(private_key_repository_unique):
|
|
59 |
private_keys = private_key_repository_unique.read_all()
|
|
60 | 60 |
|
61 | 61 |
assert len(private_keys) == 0 |
62 | 62 |
|
63 | 63 |
private_key_1 = PrivateKey(private_key_id=-1, private_key="test_private_key_1", password="test_password_1") |
64 | 64 |
private_key_2 = PrivateKey(private_key_id=-1, private_key="test_private_key_2") |
65 | 65 |
|
66 |
private_key_1_id = private_key_repository.create(private_key=private_key_1) |
|
67 |
private_key_2_id = private_key_repository.create(private_key=private_key_2) |
|
66 |
private_key_1_id = private_key_repository_unique.create(private_key=private_key_1)
|
|
67 |
private_key_2_id = private_key_repository_unique.create(private_key=private_key_2)
|
|
68 | 68 |
|
69 | 69 |
assert type(private_key_1_id) is int |
70 | 70 |
assert type(private_key_2_id) is int |
71 | 71 |
|
72 |
private_keys = private_key_repository.read_all() |
|
72 |
private_keys = private_key_repository_unique.read_all()
|
|
73 | 73 |
|
74 | 74 |
assert len(private_keys) == 2 |
75 | 75 |
|
... | ... | |
82 | 82 |
assert private_keys[1].password is None |
83 | 83 |
|
84 | 84 |
|
85 |
def test_update(private_key_repository): |
|
85 |
def test_update(private_key_repository_unique):
|
|
86 | 86 |
private_key = PrivateKey(private_key_id=-1, private_key="test_private_key") |
87 | 87 |
|
88 |
private_key_id = private_key_repository.create(private_key=private_key) |
|
88 |
private_key_id = private_key_repository_unique.create(private_key=private_key)
|
|
89 | 89 |
|
90 | 90 |
assert type(private_key_id) is int |
91 | 91 |
|
92 |
private_key = private_key_repository.read(private_key_id=private_key_id) |
|
92 |
private_key = private_key_repository_unique.read(private_key_id=private_key_id)
|
|
93 | 93 |
|
94 | 94 |
assert private_key.private_key_id == private_key_id |
95 | 95 |
assert private_key.private_key == "test_private_key" |
... | ... | |
97 | 97 |
|
98 | 98 |
private_key.password = "test_password" |
99 | 99 |
|
100 |
check_updated = private_key_repository.update(private_key_id=private_key_id, private_key=private_key) |
|
100 |
check_updated = private_key_repository_unique.update(private_key_id=private_key_id, private_key=private_key)
|
|
101 | 101 |
|
102 | 102 |
assert check_updated is True |
103 | 103 |
|
104 |
private_key = private_key_repository.read(private_key_id=private_key_id) |
|
104 |
private_key = private_key_repository_unique.read(private_key_id=private_key_id)
|
|
105 | 105 |
|
106 | 106 |
assert private_key.private_key_id == private_key_id |
107 | 107 |
assert private_key.private_key == "test_private_key" |
108 | 108 |
assert private_key.password == "test_password" |
109 | 109 |
|
110 | 110 |
|
111 |
def test_update_invalid_id(private_key_repository): |
|
111 |
def test_update_invalid_id(private_key_repository_unique):
|
|
112 | 112 |
private_key = PrivateKey(private_key_id=-1, private_key="test_private_key") |
113 | 113 |
|
114 |
private_key_id = private_key_repository.create(private_key=private_key) |
|
114 |
private_key_id = private_key_repository_unique.create(private_key=private_key)
|
|
115 | 115 |
|
116 | 116 |
assert type(private_key_id) is int |
117 | 117 |
|
118 |
private_key = private_key_repository.read(private_key_id=private_key_id) |
|
118 |
private_key = private_key_repository_unique.read(private_key_id=private_key_id)
|
|
119 | 119 |
|
120 | 120 |
assert private_key.private_key_id == private_key_id |
121 | 121 |
assert private_key.private_key == "test_private_key" |
... | ... | |
124 | 124 |
private_key.password = "test_password" |
125 | 125 |
|
126 | 126 |
none_id = 100 |
127 |
check_updated = private_key_repository.update(private_key_id=none_id, private_key=private_key) |
|
127 |
check_updated = private_key_repository_unique.update(private_key_id=none_id, private_key=private_key)
|
|
128 | 128 |
|
129 | 129 |
assert check_updated is False |
130 | 130 |
|
131 |
def test_delete(private_key_repository): |
|
131 |
def test_delete(private_key_repository_unique):
|
|
132 | 132 |
private_key_1 = PrivateKey(private_key_id=-1, private_key="test_private_key_1", password="test_password_1") |
133 | 133 |
private_key_2 = PrivateKey(private_key_id=-1, private_key="test_private_key_2") |
134 | 134 |
|
135 |
private_key_1_id = private_key_repository.create(private_key=private_key_1) |
|
136 |
private_key_2_id = private_key_repository.create(private_key=private_key_2) |
|
135 |
private_key_1_id = private_key_repository_unique.create(private_key=private_key_1)
|
|
136 |
private_key_2_id = private_key_repository_unique.create(private_key=private_key_2)
|
|
137 | 137 |
|
138 | 138 |
assert type(private_key_1_id) is int |
139 | 139 |
assert type(private_key_2_id) is int |
140 | 140 |
|
141 |
private_keys = private_key_repository.read_all() |
|
141 |
private_keys = private_key_repository_unique.read_all()
|
|
142 | 142 |
|
143 | 143 |
assert len(private_keys) == 2 |
144 | 144 |
|
... | ... | |
150 | 150 |
assert private_keys[1].private_key == "test_private_key_2" |
151 | 151 |
assert private_keys[1].password is None |
152 | 152 |
|
153 |
check_deleted = private_key_repository.delete(private_key_id=private_key_1_id) |
|
153 |
check_deleted = private_key_repository_unique.delete(private_key_id=private_key_1_id)
|
|
154 | 154 |
|
155 |
private_keys = private_key_repository.read_all() |
|
155 |
private_keys = private_key_repository_unique.read_all()
|
|
156 | 156 |
|
157 | 157 |
assert len(private_keys) == 1 |
158 | 158 |
|
Také k dispozici: Unified diff
Fixed failing tests after performing a merge.