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
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
|
import dataclasses
import unittest
import os
import time
from electrum_aionostr.event import Event, InvalidEvent
from electrum_aionostr.key import PrivateKey
class TestEvent(unittest.TestCase):
def test_verify(self):
privkey1 = PrivateKey(os.urandom(32))
privkey2 = PrivateKey(os.urandom(32))
unsigned_event = Event(
pubkey=privkey1.public_key.hex(),
content="test"
)
# verify event without signature
self.assertFalse(unsigned_event.verify())
# verify event with correct signature
event = unsigned_event.sign(privkey1.hex())
self.assertTrue(event.verify())
# Event with incorrect signature cannot even be created:
with self.assertRaises(InvalidEvent):
event = unsigned_event.sign(privkey2.hex())
def test_expiration(self):
privkey1 = PrivateKey(os.urandom(32))
event = Event(
pubkey=privkey1.public_key.hex(),
content="test"
)
# Test event with no expiration tag
self.assertFalse(event.is_expired())
# Test event with expiration tag set in the future
future_time = int(time.time()) + 3600
assert event.tags == []
event = event.add_expiration_tag(future_time)
self.assertFalse(event.is_expired())
# Test event with expiration tag set in the past
event = dataclasses.replace(event, tags=[["expiration", str(int(time.time()) - 3600)]])
self.assertTrue(event.is_expired())
# Test event with expiration tag set during initialization
future_time = int(time.time()) + 999999
event_with_expiration = Event(
pubkey=privkey1.public_key.hex(),
content="test",
tags=[["expiration", str(future_time)]],
)
self.assertFalse(event_with_expiration.is_expired())
self.assertEqual(future_time, event_with_expiration.expires_at())
# test expired event with multiple tags
expiration_time = int(time.time())
tags = []
tags.append(["test", "test"])
tags.append(["test"])
tags.append(["test", "21312", "test"])
tags.append(["expiration", str(expiration_time)])
tags.append(["test", "test"])
tags.append(["test"])
tags.append(["test", "21312", "test"])
event = Event(
pubkey=privkey1.public_key.hex(),
content="test",
tags=tags,
)
self.assertTrue(event.is_expired())
self.assertEqual(event.expires_at(), expiration_time)
|
