|
11 | 11 | import requests |
12 | 12 | from freezegun import freeze_time |
13 | 13 | from mock_vws import MockVWS |
14 | | -from mock_vws.database import VuforiaDatabase |
| 14 | +from mock_vws.database import CloudDatabase |
15 | 15 |
|
16 | 16 | from vws import VWS, CloudRecoService |
17 | 17 | from vws.exceptions.custom_exceptions import TargetProcessingTimeoutError |
@@ -122,8 +122,8 @@ def test_default_timeout( |
122 | 122 | )[1], |
123 | 123 | ) as mock, |
124 | 124 | ): |
125 | | - database = VuforiaDatabase() |
126 | | - mock.add_database(database=database) |
| 125 | + database = CloudDatabase() |
| 126 | + mock.add_cloud_database(cloud_database=database) |
127 | 127 | vws_client = VWS( |
128 | 128 | server_access_key=database.server_access_key, |
129 | 129 | server_secret_key=database.server_secret_key, |
@@ -187,8 +187,8 @@ def test_custom_timeout( |
187 | 187 | )[1], |
188 | 188 | ) as mock, |
189 | 189 | ): |
190 | | - database = VuforiaDatabase() |
191 | | - mock.add_database(database=database) |
| 190 | + database = CloudDatabase() |
| 191 | + mock.add_cloud_database(cloud_database=database) |
192 | 192 | vws_client = VWS( |
193 | 193 | server_access_key=database.server_access_key, |
194 | 194 | server_secret_key=database.server_secret_key, |
@@ -228,8 +228,8 @@ def test_custom_base_url(image: io.BytesIO | BinaryIO) -> None: |
228 | 228 | """ |
229 | 229 | base_vws_url = "http://example.com" |
230 | 230 | with MockVWS(base_vws_url=base_vws_url) as mock: |
231 | | - database = VuforiaDatabase() |
232 | | - mock.add_database(database=database) |
| 231 | + database = CloudDatabase() |
| 232 | + mock.add_cloud_database(cloud_database=database) |
233 | 233 | vws_client = VWS( |
234 | 234 | server_access_key=database.server_access_key, |
235 | 235 | server_secret_key=database.server_secret_key, |
@@ -489,8 +489,8 @@ def test_default_seconds_between_requests( |
489 | 489 | ) -> None: |
490 | 490 | """By default, 0.2 seconds are waited between polling requests.""" |
491 | 491 | with MockVWS(processing_time_seconds=0.5) as mock: |
492 | | - database = VuforiaDatabase() |
493 | | - mock.add_database(database=database) |
| 492 | + database = CloudDatabase() |
| 493 | + mock.add_cloud_database(cloud_database=database) |
494 | 494 | vws_client = VWS( |
495 | 495 | server_access_key=database.server_access_key, |
496 | 496 | server_secret_key=database.server_secret_key, |
@@ -541,8 +541,8 @@ def test_custom_seconds_between_requests( |
541 | 541 | requests. |
542 | 542 | """ |
543 | 543 | with MockVWS(processing_time_seconds=0.5) as mock: |
544 | | - database = VuforiaDatabase() |
545 | | - mock.add_database(database=database) |
| 544 | + database = CloudDatabase() |
| 545 | + mock.add_cloud_database(cloud_database=database) |
546 | 546 | vws_client = VWS( |
547 | 547 | server_access_key=database.server_access_key, |
548 | 548 | server_secret_key=database.server_secret_key, |
@@ -589,8 +589,8 @@ def test_custom_seconds_between_requests( |
589 | 589 | def test_custom_timeout(image: io.BytesIO | BinaryIO) -> None: |
590 | 590 | """It is possible to set a maximum timeout.""" |
591 | 591 | with MockVWS(processing_time_seconds=0.5) as mock: |
592 | | - database = VuforiaDatabase() |
593 | | - mock.add_database(database=database) |
| 592 | + database = CloudDatabase() |
| 593 | + mock.add_cloud_database(cloud_database=database) |
594 | 594 | vws_client = VWS( |
595 | 595 | server_access_key=database.server_access_key, |
596 | 596 | server_secret_key=database.server_secret_key, |
|
0 commit comments