2525 database_summary ,
2626 delete_target ,
2727 get_vws_target ,
28+ query ,
2829 rfc_1123_date ,
2930 wait_for_target_processed ,
3031)
32+ from tests .mock_vws .utils .assertions import assert_query_success
3133
3234
3335def request_unmocked_address () -> None :
@@ -431,7 +433,7 @@ class TestCustomQueryRecognizesDeletionSeconds:
431433 until it is not recognized by the query endpoint.
432434 """
433435
434- def _time_to_recognize_deletion (
436+ def _seconds_to_recognize_deletion (
435437 self ,
436438 high_quality_image : io .BytesIO ,
437439 vuforia_database_keys : VuforiaDatabaseKeys ,
@@ -485,7 +487,8 @@ def _time_to_recognize_deletion(
485487 continue
486488
487489 assert response .json ()['results' ] == []
488- return datetime .datetime .now () - time_after_delete
490+ time_difference = datetime .datetime .now () - time_after_delete
491+ return time_difference .total_seconds ()
489492
490493 def test_default (
491494 self ,
@@ -496,19 +499,16 @@ def test_default(
496499 XXX
497500 """
498501 with MockVWS (
499- base_vwq_url = 'https://vuforia.vwq.example.com' ,
500502 real_http = False ,
501503 client_access_key = vuforia_database_keys .client_access_key .decode (),
502504 client_secret_key = vuforia_database_keys .client_secret_key .decode (),
503505 server_access_key = vuforia_database_keys .server_access_key .decode (),
504506 server_secret_key = vuforia_database_keys .server_secret_key .decode (),
505507 ) as mock :
506- time_to_recognize_deletion = self ._time_to_recognize_deletion (
508+ seconds_to_recognize_deletion = self ._seconds_to_recognize_deletion (
507509 high_quality_image = high_quality_image ,
508510 vuforia_database_keys = vuforia_database_keys ,
509511 )
510512
511- import pdb
512- pdb .set_trace ()
513- expected == 3
514- assert abs (expected - 1 ) == 3
513+ expected = 3
514+ assert abs (expected - seconds_to_recognize_deletion ) < 0.2
0 commit comments