diff --git a/squeaknode/core/crypto.py b/squeaknode/core/crypto.py index 50fc50cd..c2573b01 100644 --- a/squeaknode/core/crypto.py +++ b/squeaknode/core/crypto.py @@ -22,13 +22,6 @@ import random -DATA_KEY_LENGTH = 32 -VERSION_NONCE_LENGTH = 8 - -HASH_LENGTH = 32 -EMPTY_HASH = b'\x00' * HASH_LENGTH - - def generate_version_nonce() -> int: return random.SystemRandom().getrandbits(64) diff --git a/tests/core/test_crypto.py b/tests/core/test_crypto.py new file mode 100644 index 00000000..200240c7 --- /dev/null +++ b/tests/core/test_crypto.py @@ -0,0 +1,35 @@ +# MIT License +# +# Copyright (c) 2020 Jonathan Zernik +# +# Permission is hereby granted, free of charge, to any person obtaining a copy +# of this software and associated documentation files (the "Software"), to deal +# in the Software without restriction, including without limitation the rights +# to use, copy, modify, merge, publish, distribute, sublicense, and/or sell +# copies of the Software, and to permit persons to whom the Software is +# furnished to do so, subject to the following conditions: +# +# The above copyright notice and this permission notice shall be included in all +# copies or substantial portions of the Software. +# +# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR +# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, +# FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE +# AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER +# LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, +# OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE +# SOFTWARE. +from squeaknode.core.crypto import generate_ping_nonce +from squeaknode.core.crypto import generate_version_nonce + + +def test_gen_version_nonce(): + generated_nonce = generate_version_nonce() + + assert type(generated_nonce) is int + + +def test_gen_ping_nonce(): + generated_nonce = generate_ping_nonce() + + assert type(generated_nonce) is int