Skip to content

Commit c345c39

Browse files
committed
Put query in test
1 parent 16d2011 commit c345c39

File tree

2 files changed

+11
-11
lines changed

2 files changed

+11
-11
lines changed

src/mock_vws/_mock_web_services_api.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -510,7 +510,7 @@ def database_summary(
510510
'inactive_images': inactive_images,
511511
'failed_images': failed_images,
512512
'target_quota': 1000,
513-
'total_recos': '',
513+
'total_recos': self.num_requests,
514514
'current_month_recos': '',
515515
'previous_month_recos': '',
516516
'processing_images': processing_images,

tests/mock_vws/test_database_summary.py

Lines changed: 10 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -17,6 +17,7 @@
1717
add_target_to_vws,
1818
database_summary,
1919
delete_target,
20+
query,
2021
wait_for_target_processed,
2122
)
2223
from tests.mock_vws.utils.assertions import assert_vws_response
@@ -396,26 +397,25 @@ def test_request_usage(
396397
vuforia_database_keys=vuforia_database_keys,
397398
)
398399

399-
previous_request_usage = response.json()['request_usage']
400+
original_request_usage = response.json()['request_usage']
400401

401402
response = database_summary(
402403
vuforia_database_keys=vuforia_database_keys,
403404
)
404405

405406
new_request_usage = response.json()['request_usage']
406-
assert new_request_usage == previous_request_usage + 1
407+
assert new_request_usage == original_request_usage + 1
407408

408-
def test_reco_counts(
409-
self,
410-
vuforia_database_keys: VuforiaDatabaseKeys,
411-
) -> None:
412-
response = database_summary(
409+
query(
413410
vuforia_database_keys=vuforia_database_keys,
411+
body={},
414412
)
415413

416-
assert response.json()['total_recos'] == 0
417-
assert response.json()['current_month_recos'] == 0
418-
assert response.json()['previous_month_recos'] == 0
414+
response = database_summary(
415+
vuforia_database_keys=vuforia_database_keys,
416+
)
417+
new_request_usage = response.json()['request_usage']
418+
assert new_request_usage == original_request_usage + 2
419419

420420

421421
@pytest.mark.usefixtures('verify_mock_vuforia_inactive')

0 commit comments

Comments
 (0)