gergely/PySecretHandshake
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

test: Add tests for util.inc_nonce

Browse Source
This commit is contained in:
Gergely Polonkai 2023-11-03 08:57:57 +01:00
parent 88516b230b
commit 995f0dabed
1 changed files with 11 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
tests/test_util.py
View File

@ -28,11 +28,21 @@ from typing import List, Sequence, TypeVar
import pytest import pytest
from secret_handshake.util import bytes_to_long, long_to_bytes, split_chunks from secret_handshake.util import bytes_to_long, inc_nonce, long_to_bytes, split_chunks
T = TypeVar("T") T = TypeVar("T")
@pytest.mark.parametrize("in_,out", ((b"\x00\x00\x00\x00", b"\x00" * 23 + b"\x01"),))
def test_inc_nonce(in_: bytes, out: bytes) -> None:
"""Test the inc_nonce function"""
result = inc_nonce(in_)
assert len(result) == 24
assert result == out
def test_split_chunks_is_generator() -> None: def test_split_chunks_is_generator() -> None:
"""Test if split_chunks returns a generator""" """Test if split_chunks returns a generator"""