Skip to content

Commit e6564dc

Browse files
committed
Add test for recognition counts [skip ci]
1 parent 88926f8 commit e6564dc

File tree

2 files changed

+34
-3
lines changed

2 files changed

+34
-3
lines changed

src/mock_vws/_mock_web_services_api.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -536,9 +536,9 @@ def database_summary(
536536
'inactive_images': inactive_images,
537537
'failed_images': failed_images,
538538
'target_quota': 1000,
539-
'total_recos': '',
540-
'current_month_recos': '',
541-
'previous_month_recos': '',
539+
'total_recos': 0,
540+
'current_month_recos': 0,
541+
'previous_month_recos': 0,
542542
'processing_images': processing_images,
543543
'reco_threshold': 1000,
544544
'request_quota': 100000,

tests/mock_vws/test_database_summary.py

Lines changed: 31 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -380,6 +380,37 @@ def test_quotas(self, vuforia_database_keys: VuforiaDatabaseKeys) -> None:
380380
assert response.json()['reco_threshold'] == 1000
381381

382382

383+
@pytest.mark.usefixtures('verify_mock_vuforia')
384+
class TestRecos:
385+
"""
386+
Tests for the recognition count fields.
387+
"""
388+
389+
def test_query_request(
390+
self,
391+
vuforia_database_keys: VuforiaDatabaseKeys,
392+
high_quality_image: io.BytesIO,
393+
) -> None:
394+
"""
395+
The ``*_recos`` counts are always 0.
396+
"""
397+
image_content = high_quality_image.getvalue()
398+
body = {'image': ('image.jpeg', image_content, 'image/jpeg')}
399+
query_resp = query(
400+
vuforia_database_keys=vuforia_database_keys,
401+
body=body,
402+
)
403+
404+
assert query_resp.status_code == codes.OK
405+
406+
response = database_summary(
407+
vuforia_database_keys=vuforia_database_keys,
408+
)
409+
assert response.json()['total_recos'] == 0
410+
assert response.json()['current_month_recos'] == 0
411+
assert response.json()['previous_month_recos'] == 0
412+
413+
383414
@pytest.mark.usefixtures('verify_mock_vuforia')
384415
class TestRequestUsage:
385416
"""

0 commit comments

Comments
 (0)