mirror of
https://github.com/aljazceru/python-nostr.git
synced 2025-12-18 14:54:23 +01:00
94 lines
3.2 KiB
Python
94 lines
3.2 KiB
Python
import pytest
|
|
import time
|
|
from nostr.event import Event, EncryptedDirectMessage
|
|
from nostr.key import PrivateKey
|
|
|
|
|
|
|
|
class TestEvent:
|
|
def test_event_default_time(self):
|
|
"""
|
|
ensure created_at default value reflects the time at Event object instantiation
|
|
see: https://github.com/jeffthibault/python-nostr/issues/23
|
|
"""
|
|
event1 = Event(content='test event')
|
|
time.sleep(1.5)
|
|
event2 = Event(content='test event')
|
|
assert event1.created_at < event2.created_at
|
|
|
|
|
|
def test_content_only_instantiation(self):
|
|
""" should be able to create an Event by only specifying content without kwarg """
|
|
event = Event("Hello, world!")
|
|
assert event.content is not None
|
|
|
|
|
|
def test_event_id_recomputes(self):
|
|
""" should recompute the Event.id to reflect the current Event attrs """
|
|
event = Event(content="some event")
|
|
|
|
# id should be computed on the fly
|
|
event_id = event.id
|
|
|
|
event.created_at += 10
|
|
|
|
# Recomputed id should now be different
|
|
assert event.id != event_id
|
|
|
|
|
|
def test_add_event_ref(self):
|
|
""" should add an 'e' tag for each event_ref added """
|
|
some_event_id = "some_event_id"
|
|
event = Event(content="Adding an 'e' tag")
|
|
event.add_event_ref(some_event_id)
|
|
assert ['e', some_event_id] in event.tags
|
|
|
|
|
|
def test_add_pubkey_ref(self):
|
|
""" should add a 'p' tag for each pubkey_ref added """
|
|
some_pubkey = "some_pubkey"
|
|
event = Event(content="Adding a 'p' tag")
|
|
event.add_pubkey_ref(some_pubkey)
|
|
assert ['p', some_pubkey] in event.tags
|
|
|
|
|
|
|
|
class TestEncryptedDirectMessage:
|
|
def setup_class(self):
|
|
self.sender_pk = PrivateKey()
|
|
self.sender_pubkey = self.sender_pk.public_key.hex()
|
|
self.recipient_pk = PrivateKey()
|
|
self.recipient_pubkey = self.recipient_pk.public_key.hex()
|
|
|
|
|
|
def test_content_field_moved_to_cleartext_content(self):
|
|
""" Should transfer `content` field data to `cleartext_content` """
|
|
dm = EncryptedDirectMessage(content="My message!", recipient_pubkey=self.recipient_pubkey)
|
|
assert dm.content is None
|
|
assert dm.cleartext_content is not None
|
|
|
|
|
|
def test_nokwarg_content_allowed(self):
|
|
""" Should allow creating a new DM w/no `content` nor `cleartext_content` kwarg """
|
|
dm = EncryptedDirectMessage("My message!", recipient_pubkey=self.recipient_pubkey)
|
|
assert dm.cleartext_content is not None
|
|
|
|
|
|
def test_recipient_p_tag(self):
|
|
""" Should generate recipient 'p' tag """
|
|
dm = EncryptedDirectMessage(cleartext_content="Secret message!", recipient_pubkey=self.recipient_pubkey)
|
|
assert ['p', self.recipient_pubkey] in dm.tags
|
|
|
|
|
|
def test_unencrypted_dm_has_undefined_id(self):
|
|
""" Should raise Exception if `id` is requested before DM is encrypted """
|
|
dm = EncryptedDirectMessage(cleartext_content="My message!", recipient_pubkey=self.recipient_pubkey)
|
|
|
|
with pytest.raises(Exception) as e:
|
|
dm.id
|
|
assert "undefined" in str(e)
|
|
|
|
# But once we encrypt it, we can request its id
|
|
self.sender_pk.encrypt_dm(dm)
|
|
assert dm.id is not None
|