Skip to content

Commit c2ce808

Browse files
committed
Return expected keys from database summary
1 parent 848aaa7 commit c2ce808

File tree

2 files changed

+29
-1
lines changed

2 files changed

+29
-1
lines changed

src/mock_vws/_mock.py

Lines changed: 13 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -178,8 +178,20 @@ def database_summary(self,
178178

179179
context.status_code = codes.OK # pylint: disable=no-member
180180
body = {
181-
'transaction_id': uuid.uuid4().hex,
182181
'result_code': ResultCodes.SUCCESS.value,
182+
'transaction_id': uuid.uuid4().hex,
183+
'name': '',
184+
'active_images': '',
185+
'inactive_images': '',
186+
'failed_images': '',
187+
'target_quota': '',
188+
'total_recos': '',
189+
'current_month_recos': '',
190+
'previous_month_recos': '',
191+
'processing_images': '',
192+
'reco_threshold': '',
193+
'request_quota': '',
194+
'request_usage': '',
183195
}
184196
return json.dumps(body)
185197

tests/mock_vws/test_database_summary.py

Lines changed: 16 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -45,3 +45,19 @@ def test_success(self,
4545
data=b'',
4646
)
4747
assert response.status_code == codes.OK
48+
assert response.json().keys() == {
49+
'active_images',
50+
'current_month_recos',
51+
'failed_images',
52+
'inactive_images',
53+
'name',
54+
'previous_month_recos',
55+
'processing_images',
56+
'reco_threshold',
57+
'request_quota',
58+
'request_usage',
59+
'result_code',
60+
'target_quota',
61+
'total_recos',
62+
'transaction_id',
63+
}

0 commit comments

Comments
 (0)