|
1 | 1 | from unittest.mock import patch |
2 | 2 |
|
3 | 3 | from app.api.auth import (ApiKeyError, ApiKeyErrorCode, authenticate, |
4 | | - blacklist_key, find_key_by_apikey_or_email, |
| 4 | + deny_key, find_key_by_apikey_or_email, |
5 | 5 | rotate_key) |
6 | 6 | from app.models import Key |
7 | 7 | from flask import g |
@@ -56,9 +56,9 @@ def callback(*args, **kwargs): |
56 | 56 | assert g.auth_key == key |
57 | 57 |
|
58 | 58 |
|
59 | | -def test_authenticate_blacklisted(module_client, function_empty_db): |
| 59 | +def test_authenticate_denied(module_client, function_empty_db): |
60 | 60 | # Arrange |
61 | | - create_fake_key(function_empty_db.session, blacklisted=True) |
| 61 | + create_fake_key(function_empty_db.session, denied=True) |
62 | 62 |
|
63 | 63 | def callback(*args, **kwargs): |
64 | 64 | return 1 |
@@ -88,75 +88,75 @@ def test_find_key_by_apikey_or_email(module_client, function_empty_db): |
88 | 88 | assert key == key2 |
89 | 89 |
|
90 | 90 |
|
91 | | -def test_blacklist_key_not_found(module_client, function_empty_db): |
| 91 | +def test_denied_key_not_found(module_client, function_empty_db): |
92 | 92 | # Arrange |
93 | 93 | error = None |
94 | 94 |
|
95 | 95 | # Act |
96 | 96 | try: |
97 | | - blacklist_key(FAKE_APIKEY + 'b', True, function_empty_db.session) |
| 97 | + deny_key(FAKE_APIKEY + 'b', True, function_empty_db.session) |
98 | 98 | except ApiKeyError as e: |
99 | 99 | error = e |
100 | 100 |
|
101 | 101 | # Assert |
102 | 102 | assert error.error_code == ApiKeyErrorCode.NOT_FOUND |
103 | 103 |
|
104 | 104 |
|
105 | | -def test_blacklist_key_already_blacklisted(module_client, function_empty_db): |
| 105 | +def test_deny_key_already_denied(module_client, function_empty_db): |
106 | 106 | # Arrange |
107 | 107 | error = None |
108 | 108 | key1 = None |
109 | | - create_fake_key(function_empty_db.session, blacklisted=True) |
| 109 | + create_fake_key(function_empty_db.session, denied=True) |
110 | 110 |
|
111 | 111 | # Act |
112 | 112 | try: |
113 | | - key1 = blacklist_key(FAKE_APIKEY, True, function_empty_db.session) |
| 113 | + key1 = deny_key(FAKE_APIKEY, True, function_empty_db.session) |
114 | 114 | except ApiKeyError as e: |
115 | 115 | error = e |
116 | 116 |
|
117 | 117 | # Assert |
118 | | - assert error.error_code == ApiKeyErrorCode.ALREADY_BLACKLISTED |
| 118 | + assert error.error_code == ApiKeyErrorCode.ALREADY_DENIED |
119 | 119 | assert key1 is None |
120 | 120 |
|
121 | 121 |
|
122 | | -def test_blacklist_key_not_blacklisted(module_client, function_empty_db): |
| 122 | +def test_deny_key_not_denied(module_client, function_empty_db): |
123 | 123 | # Arrange |
124 | 124 | error = None |
125 | 125 | key1 = None |
126 | 126 | create_fake_key(function_empty_db.session) |
127 | 127 |
|
128 | 128 | # Act |
129 | 129 | try: |
130 | | - key1 = blacklist_key(FAKE_APIKEY, False, function_empty_db.session) |
| 130 | + key1 = deny_key(FAKE_APIKEY, False, function_empty_db.session) |
131 | 131 | except ApiKeyError as e: |
132 | 132 | error = e |
133 | 133 |
|
134 | 134 | # Assert |
135 | | - assert error.error_code == ApiKeyErrorCode.NOT_BLACKLISTED |
| 135 | + assert error.error_code == ApiKeyErrorCode.NOT_DENIED |
136 | 136 | assert key1 is None |
137 | 137 |
|
138 | 138 |
|
139 | | -def test_blacklist_key_set_blacklisted_on(module_client, function_empty_db): |
| 139 | +def test_deny_key_set_denied_on(module_client, function_empty_db): |
140 | 140 | # Arrange |
141 | 141 | key = create_fake_key(function_empty_db.session) |
142 | 142 |
|
143 | 143 | # Act |
144 | | - key1 = blacklist_key(FAKE_APIKEY, True, function_empty_db.session) |
| 144 | + key1 = deny_key(FAKE_APIKEY, True, function_empty_db.session) |
145 | 145 |
|
146 | 146 | # Assert |
147 | | - assert key.blacklisted |
| 147 | + assert key.denied |
148 | 148 | assert key == key1 |
149 | 149 |
|
150 | 150 |
|
151 | | -def test_blacklist_key_set_blacklisted_off(module_client, function_empty_db): |
| 151 | +def test_deny_key_set_denied_off(module_client, function_empty_db): |
152 | 152 | # Arrange |
153 | | - key = create_fake_key(function_empty_db.session, blacklisted=True) |
| 153 | + key = create_fake_key(function_empty_db.session, denied=True) |
154 | 154 |
|
155 | 155 | # Act |
156 | | - key1 = blacklist_key(FAKE_APIKEY, False, function_empty_db.session) |
| 156 | + key1 = deny_key(FAKE_APIKEY, False, function_empty_db.session) |
157 | 157 |
|
158 | 158 | # Assert |
159 | | - assert not key.blacklisted |
| 159 | + assert not key.denied |
160 | 160 | assert key == key1 |
161 | 161 |
|
162 | 162 |
|
|
0 commit comments