Skip to content

Commit

Permalink
rename messages fixture to aleph_messages
Browse files Browse the repository at this point in the history
  • Loading branch information
MHHukiewitz committed Sep 27, 2023
1 parent 8635114 commit d2d3264
Show file tree
Hide file tree
Showing 2 changed files with 28 additions and 28 deletions.
10 changes: 5 additions & 5 deletions tests/unit/conftest.py
Original file line number Diff line number Diff line change
Expand Up @@ -55,7 +55,7 @@ def json_messages():


@pytest.fixture
def messages() -> List[AlephMessage]:
def aleph_messages() -> List[AlephMessage]:
return [
AggregateMessage.parse_obj(
{
Expand Down Expand Up @@ -112,11 +112,11 @@ def messages() -> List[AlephMessage]:


@pytest.fixture
def raw_messages_response(messages):
def raw_messages_response(aleph_messages):
return lambda page: {
"messages": [message.dict() for message in messages] if page == 1 else [],
"messages": [message.dict() for message in aleph_messages] if page == 1 else [],
"pagination_item": "messages",
"pagination_page": page,
"pagination_per_page": max(len(messages), 20),
"pagination_total": len(messages) if page == 1 else 0,
"pagination_per_page": max(len(aleph_messages), 20),
"pagination_total": len(aleph_messages) if page == 1 else 0,
}
46 changes: 23 additions & 23 deletions tests/unit/test_node_get.py
Original file line number Diff line number Diff line change
Expand Up @@ -18,20 +18,20 @@


@pytest.mark.asyncio
async def test_base(messages):
async def test_base(aleph_messages):
# test add_many
cache = MessageCache()
cache.add(messages)
assert len(cache) == len(messages)
cache.add(aleph_messages)
assert len(cache) == len(aleph_messages)

item_hashes = [message.item_hash for message in messages]
item_hashes = [message.item_hash for message in aleph_messages]
cached_messages = cache.get(item_hashes)
assert len(cached_messages) == len(messages)
assert len(cached_messages) == len(aleph_messages)

for message in messages:
for message in aleph_messages:
assert cache[message.item_hash] == message

for message in messages:
for message in aleph_messages:
assert message.item_hash in cache

for message in cache:
Expand All @@ -47,8 +47,8 @@ class TestMessageQueries:
cache: MessageCache

@pytest.fixture(autouse=True)
def class_setup(self, messages):
self.messages = messages
def class_setup(self, aleph_messages):
self.messages = aleph_messages
self.cache = MessageCache()
self.cache.add(self.messages)

Expand Down Expand Up @@ -127,8 +127,8 @@ class TestPostQueries:
cache: MessageCache

@pytest.fixture(autouse=True)
def class_setup(self, messages):
self.messages = messages
def class_setup(self, aleph_messages):
self.messages = aleph_messages
self.cache = MessageCache()
self.cache.add(self.messages)

Expand Down Expand Up @@ -196,33 +196,33 @@ async def mock_message_stream():


@pytest.mark.asyncio
async def test_fetch_aggregate(messages):
async def test_fetch_aggregate(aleph_messages):
cache = MessageCache()
cache.add(messages)
cache.add(aleph_messages)

aggregate = await cache.fetch_aggregate(messages[0].sender, messages[0].content.key)
aggregate = await cache.fetch_aggregate(aleph_messages[0].sender, aleph_messages[0].content.key)

assert aggregate == messages[0].content.content
assert aggregate == aleph_messages[0].content.content


@pytest.mark.asyncio
async def test_fetch_aggregates(messages):
async def test_fetch_aggregates(aleph_messages):
cache = MessageCache()
cache.add(messages)
cache.add(aleph_messages)

aggregates = await cache.fetch_aggregates(messages[0].sender)
aggregates = await cache.fetch_aggregates(aleph_messages[0].sender)

assert aggregates == {messages[0].content.key: messages[0].content.content}
assert aggregates == {aleph_messages[0].content.key: aleph_messages[0].content.content}


@pytest.mark.asyncio
async def test_get_message(messages):
async def test_get_message(aleph_messages):
cache = MessageCache()
cache.add(messages)
cache.add(aleph_messages)

message: AlephMessage = await cache.get_message(messages[0].item_hash)
message: AlephMessage = await cache.get_message(aleph_messages[0].item_hash)

assert message == messages[0]
assert message == aleph_messages[0]


@pytest.mark.asyncio
Expand Down

0 comments on commit d2d3264

Please sign in to comment.