Skip to content

Commit 9072640

Browse files
Merge pull request #627 from adamtheturtle/set-time-from-mock
Set query recognises deletion time from mock.__init__
2 parents ca76b7d + 89692d1 commit 9072640

File tree

2 files changed

+8
-1
lines changed

2 files changed

+8
-1
lines changed

src/mock_vws/__init__.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -115,6 +115,7 @@ def __enter__(self) -> 'MockVWS':
115115
client_access_key=self.client_access_key,
116116
client_secret_key=self.client_secret_key,
117117
mock_web_services_api=mock_vws_api,
118+
query_recognizes_deletion_seconds=3,
118119
)
119120

120121
date = email.utils.formatdate(None, localtime=False, usegmt=True)

src/mock_vws/_mock_web_query_api.py

Lines changed: 7 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -449,12 +449,16 @@ def __init__(
449449
client_access_key: str,
450450
client_secret_key: str,
451451
mock_web_services_api: MockVuforiaWebServicesAPI,
452+
query_recognizes_deletion_seconds: int,
452453
) -> None:
453454
"""
454455
Args:
455456
client_access_key: A VWS client access key.
456457
client_secret_key: A VWS client secret key.
457458
mock_web_services_api: An instance of a mock web services API.
459+
query_recognizes_deletion_seconds: The number of seconds after a
460+
target has been deleted that the query endpoint will return a
461+
500 response for on a match.
458462
459463
Attributes:
460464
routes: The `Route`s to be used in the mock.
@@ -470,7 +474,9 @@ def __init__(
470474
self.access_key: str = client_access_key
471475
self.secret_key: str = client_secret_key
472476
self.mock_web_services_api = mock_web_services_api
473-
self.query_recognizes_deletion_seconds = 3
477+
self.query_recognizes_deletion_seconds = (
478+
query_recognizes_deletion_seconds
479+
)
474480

475481
@route(path_pattern='/v1/query', http_methods=[POST])
476482
def query(

0 commit comments

Comments
 (0)