Skip to content

Commit 7091ea3

Browse files
committed
Add configuration as public class variable
1 parent a12de92 commit 7091ea3

File tree

1 file changed

+7
-3
lines changed

1 file changed

+7
-3
lines changed

src/mock_vws/_mock_web_query_api.py

Lines changed: 7 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -462,11 +462,15 @@ def __init__(
462462
secret_key (str): A VWS client secret key.
463463
mock_web_services_api (MockVuforiaWebServicesAPI): An instance of a
464464
mock web services API.
465+
query_recognizes_deletion_seconds (int): The number of seconds
466+
after a target has been deleted that the query endpoint will
467+
return a 500 response for on a match.
465468
"""
466469
self.routes: Set[Route] = ROUTES
467470
self.access_key: str = client_access_key
468471
self.secret_key: str = client_secret_key
469472
self.mock_web_services_api = mock_web_services_api
473+
self.query_recognizes_deletion_seconds = 3
470474

471475
@route(path_pattern='/v1/query', http_methods=[POST])
472476
def query(
@@ -497,9 +501,9 @@ def query(
497501
gmt = pytz.timezone('GMT')
498502
now = datetime.datetime.now(tz=gmt)
499503

500-
# See https://github.com/adamtheturtle/vws-python/issues/623 for making
501-
# this customizable.
502-
minimum_time_since_delete = datetime.timedelta(seconds=3)
504+
minimum_time_since_delete = datetime.timedelta(
505+
seconds=self.query_recognizes_deletion_seconds,
506+
)
503507

504508
for target in self.mock_web_services_api.targets:
505509
delete_processing = bool(

0 commit comments

Comments
 (0)