o tBh@s,ddlZddlZddlZGdddeZdS)Nc@sDeZdZdZiZe d ddZe   d ddZedd Z d S) TestHelperaA Helper Class for various functions associated with API testing. This class contains static and class methods for operations that need to be performed during API testing. All of the methods inside this class are static or class methods, there is no need to ever initialise an instance of this class. Attributes: cache (Set): Class variable which stores hashes of file URLs so we don't download the same file twice in a test session. TcCs|t|t|ks|dkrt|t|krdSdd|D}|D]}||vr*dS||dkr;||||kr;dSq!dS)aStatic method to compare the received headers with the expected headers. Args: expected_headers (dict): A dictionary of expected headers (keys in lower case). received_headers (dict): A dictionary of headers received. allow_extra (Boolean, optional): A flag which determines if we allow extra headers. Returns: Boolean: True if headers match, False otherwise. FcSsi|] \}}||qS)lower).0kvrrh/var/www/html/riverr-enterprise-integrations-main/venv/lib/python3.10/site-packages/tests/test_helper.py )sz,TestHelper.match_headers..NT)lenitems)expected_headersreceived_headers allow_extrae_keyrrr match_headerss zTestHelper.match_headersFcs0t|tkr8t|tkrdS|D]%}||vrdSs#t||tkr5t||||dkr5dSqdSt|tkrt|tkrFdSdkrTt|t|krTdSgt|D]3\}fddt|D}t|dkrudSdkr|dkrtfdd|DrdS|qZdS||krdSdS)aStatic method to compare the received body with the expected body. Args: expected_body (dynamic): The expected body. received_body (dynamic): The received body. check_values (Boolean, optional): A flag which determines if we check values in dictionaries. check_order (Boolean, optional): A flag which determines if we check the order of array elements. check_count (Boolean, optional): A flag which determines if we check the count of array elements. Returns: Boolean: True if bodies match, False otherwise. FTc s&g|]\}}t|r|qSr)r match_body)rjreceived_element) check_count check_order check_valuesexpected_elementrr Zs  z)TestHelper.match_body..rcs"g|] tfddDqS)c3s|]}|kVqdS)Nr)ryxrr asz3TestHelper.match_body...)all)r)previous_matchesrrras")typedictrrlistr enumerater) expected_body received_bodyrrrkeyimatchesr)rrrrrrr3sH    zTestHelper.match_bodycCsJ||jvrt|j|<|j|t|j|j|d|j|S)anClass method which takes a URL, downloads the file (if not already downloaded for this test session) and returns a file object for the file in read-binary mode. Args: url (string): The URL of the required file. Returns: FileObject: The file object of the required file (opened with "rb"). r)cachetempfileNamedTemporaryFilewriterequestsgetcontentseek)clsurlrrrget_filehs  zTestHelper.get_fileN)T)FFF) __name__ __module__ __qualname____doc__r( staticmethodrr classmethodr2rrrrrs   4r)osr)r,objectrrrrrs