oncall-engine/engine/apps/auth_token/tests/test_crypto.py
Michael Derynck 6b40f95033 World, meet OnCall!
Co-authored-by: Eve832 <eve.meelan@grafana.com>
    Co-authored-by: Francisco Montes de Oca <nevermind89x@gmail.com>
    Co-authored-by: Ildar Iskhakov <ildar.iskhakov@grafana.com>
    Co-authored-by: Innokentii Konstantinov <innokenty.konstantinov@grafana.com>
    Co-authored-by: Julia <ferril.darkdiver@gmail.com>
    Co-authored-by: maskin25 <kengurek@gmail.com>
    Co-authored-by: Matias Bordese <mbordese@gmail.com>
    Co-authored-by: Matvey Kukuy <motakuk@gmail.com>
    Co-authored-by: Michael Derynck <michael.derynck@grafana.com>
    Co-authored-by: Richard Hartmann <richih@richih.org>
    Co-authored-by: Robby Milo <robbymilo@fastmail.com>
    Co-authored-by: Timur Olzhabayev <timur.olzhabayev@grafana.com>
    Co-authored-by: Vadim Stepanov <vadimkerr@gmail.com>
    Co-authored-by: Yulia Shanyrova <yulia.shanyrova@grafana.com>
2022-06-03 08:09:47 -06:00

23 lines
702 B
Python

import binascii
from hmac import compare_digest
from apps.auth_token.crypto import (
generate_plugin_token_string,
generate_plugin_token_string_and_salt,
hash_token_string,
)
def test_plugin_token_round_trip():
stack_id = 100
org_id = 100
hex_token, hex_salt = generate_plugin_token_string_and_salt(stack_id, org_id)
hex_signature = hash_token_string(hex_token)
raw_salt = binascii.unhexlify(hex_salt)
hex_recreated_token = generate_plugin_token_string(raw_salt, stack_id, org_id)
hex_recreated_signature = hash_token_string(hex_recreated_token)
assert hex_token == hex_recreated_token
assert compare_digest(hex_signature, hex_recreated_signature)