python-crypto/tests/utils/test_message.py at b15b3872f5eab2c602e87af2b490dabc0c077e95 · ArkEcosystem/python-crypto · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
import json

from crypto.utils.message import Message

def test_verify_with_publicKey(load_transaction_fixture):
    fixture = load_transaction_fixture('message-sign')

    result = Message(
        message=fixture['message'],
        signature=fixture['signature'][2:],
        public_key=fixture['publicKey'],
    )

    isVerified = result.verify()

    assert isVerified is True

def test_message_sign_verification(load_transaction_fixture, passphrase):
    fixture = load_transaction_fixture('message-sign')

    message: Message = Message.sign(fixture['message'], passphrase)

    assert message.signature.decode() == fixture['signature'][2:]
    assert message.public_key.decode() == fixture['publicKey']
    assert message.message.decode() == fixture['message']

    isVerified = message.verify()

    assert isVerified is True

def test_to_dict(load_transaction_fixture):
    fixture = load_transaction_fixture('message-sign')

    result = Message(
        message=fixture['message'],
        signature=fixture['signature'],
        public_key=fixture['publicKey'],
    )

    data = result.to_dict()

    assert data['signature'] == fixture['signature']
    assert data['public_key'] == fixture['publicKey']
    assert data['message'] == fixture['message']

def test_to_json(load_transaction_fixture):
    fixture = load_transaction_fixture('message-sign')

    result = Message(
        message=fixture['message'],
        signature=fixture['signature'],
        public_key=fixture['publicKey'],
    )

    json_data = result.to_json()
    data = json.loads(json_data)

    assert data['signature'] == fixture['signature']
    assert data['public_key'] == fixture['publicKey']
    assert data['message'] == fixture['message']