27 lines
991 B
Python
27 lines
991 B
Python
import struct
|
|
import asyncio
|
|
from unittest.mock import AsyncMock
|
|
from pm3py.protocol import Cmd
|
|
from pm3py.transport import PM3Response
|
|
from pm3py.hf_mf import HFMFCommands
|
|
|
|
def test_mf_rdbl():
|
|
t = AsyncMock()
|
|
mf = HFMFCommands(t)
|
|
block_data = bytes(range(16))
|
|
t.send_ng.return_value = PM3Response(cmd=Cmd.HF_MIFARE_READBL, status=0,
|
|
reason=0, ng=True, data=block_data)
|
|
result = asyncio.get_event_loop().run_until_complete(
|
|
mf.rdbl(block=0, key="FFFFFFFFFFFF"))
|
|
assert result["data"] == block_data.hex()
|
|
assert len(result["raw"]) == 16
|
|
|
|
def test_mf_wrbl():
|
|
t = AsyncMock()
|
|
mf = HFMFCommands(t)
|
|
t.send_mix.return_value = PM3Response(cmd=Cmd.ACK, status=0, reason=0,
|
|
ng=False, data=b"", oldarg=[1, 0, 0])
|
|
result = asyncio.get_event_loop().run_until_complete(
|
|
mf.wrbl(block=4, key="FFFFFFFFFFFF", data="00" * 16))
|
|
assert result["success"] is True
|