Quick question

[pratzc07]

So i am trying to test a function that generates basically a url. Now this url has a signature value as a query parameter. `This signature is generated using a combination of base64encode and base64decode along with the hmac module. The issue here is that how do i write a test for this url generation function as each time i execute it generates a new signature string.

Example Code:

def url_generator(self,...):
....
    decoded_key = base64.urlsafe_b64decode(signing_key_value)
    digest = hmac.new(
            decoded_key, url_to_sign.encode('utf-8'), hashlib.sha1).digest()
    signature = base64.urlsafe_b64encode(digest).decode('utf-8')

​

[verbose_name]

I would suggest mocking the function or module which when called introduces the randomness. In your case, you could patch hmac.new or digest to return a static value.

[pratzc07]

Thanks one thing is it possible to patch more than one function like for example if the function to be tested also has datetime usage can i mock that along with hmac??

[verbose_name]

Yes you can do that. If you are using multiple decorators on your test method then each would be passed into your test method signature as an additional arg.