Files
hate_crack/tests/test_attacks_pcfg.py

23 lines
628 B
Python

from unittest.mock import MagicMock
from hate_crack.attacks import pcfg_attack, prince_ling_attack
def _make_ctx(hash_type: str = "1000", hash_file: str = "/tmp/hashes.txt") -> MagicMock:
ctx = MagicMock()
ctx.hcatHashType = hash_type
ctx.hcatHashFile = hash_file
return ctx
def test_pcfg_attack_invokes_hcatPCFG():
ctx = _make_ctx()
pcfg_attack(ctx)
ctx.hcatPCFG.assert_called_once_with("1000", "/tmp/hashes.txt")
def test_prince_ling_attack_invokes_hcatPrinceLing():
ctx = _make_ctx()
prince_ling_attack(ctx)
ctx.hcatPrinceLing.assert_called_once_with("1000", "/tmp/hashes.txt")