Skip to content

Commit d2d3264

Browse files
committed
rename messages fixture to aleph_messages
1 parent 8635114 commit d2d3264

File tree

2 files changed

+28
-28
lines changed

2 files changed

+28
-28
lines changed

tests/unit/conftest.py

+5-5
Original file line numberDiff line numberDiff line change
@@ -55,7 +55,7 @@ def json_messages():
5555

5656

5757
@pytest.fixture
58-
def messages() -> List[AlephMessage]:
58+
def aleph_messages() -> List[AlephMessage]:
5959
return [
6060
AggregateMessage.parse_obj(
6161
{
@@ -112,11 +112,11 @@ def messages() -> List[AlephMessage]:
112112

113113

114114
@pytest.fixture
115-
def raw_messages_response(messages):
115+
def raw_messages_response(aleph_messages):
116116
return lambda page: {
117-
"messages": [message.dict() for message in messages] if page == 1 else [],
117+
"messages": [message.dict() for message in aleph_messages] if page == 1 else [],
118118
"pagination_item": "messages",
119119
"pagination_page": page,
120-
"pagination_per_page": max(len(messages), 20),
121-
"pagination_total": len(messages) if page == 1 else 0,
120+
"pagination_per_page": max(len(aleph_messages), 20),
121+
"pagination_total": len(aleph_messages) if page == 1 else 0,
122122
}

tests/unit/test_node_get.py

+23-23
Original file line numberDiff line numberDiff line change
@@ -18,20 +18,20 @@
1818

1919

2020
@pytest.mark.asyncio
21-
async def test_base(messages):
21+
async def test_base(aleph_messages):
2222
# test add_many
2323
cache = MessageCache()
24-
cache.add(messages)
25-
assert len(cache) == len(messages)
24+
cache.add(aleph_messages)
25+
assert len(cache) == len(aleph_messages)
2626

27-
item_hashes = [message.item_hash for message in messages]
27+
item_hashes = [message.item_hash for message in aleph_messages]
2828
cached_messages = cache.get(item_hashes)
29-
assert len(cached_messages) == len(messages)
29+
assert len(cached_messages) == len(aleph_messages)
3030

31-
for message in messages:
31+
for message in aleph_messages:
3232
assert cache[message.item_hash] == message
3333

34-
for message in messages:
34+
for message in aleph_messages:
3535
assert message.item_hash in cache
3636

3737
for message in cache:
@@ -47,8 +47,8 @@ class TestMessageQueries:
4747
cache: MessageCache
4848

4949
@pytest.fixture(autouse=True)
50-
def class_setup(self, messages):
51-
self.messages = messages
50+
def class_setup(self, aleph_messages):
51+
self.messages = aleph_messages
5252
self.cache = MessageCache()
5353
self.cache.add(self.messages)
5454

@@ -127,8 +127,8 @@ class TestPostQueries:
127127
cache: MessageCache
128128

129129
@pytest.fixture(autouse=True)
130-
def class_setup(self, messages):
131-
self.messages = messages
130+
def class_setup(self, aleph_messages):
131+
self.messages = aleph_messages
132132
self.cache = MessageCache()
133133
self.cache.add(self.messages)
134134

@@ -196,33 +196,33 @@ async def mock_message_stream():
196196

197197

198198
@pytest.mark.asyncio
199-
async def test_fetch_aggregate(messages):
199+
async def test_fetch_aggregate(aleph_messages):
200200
cache = MessageCache()
201-
cache.add(messages)
201+
cache.add(aleph_messages)
202202

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

205-
assert aggregate == messages[0].content.content
205+
assert aggregate == aleph_messages[0].content.content
206206

207207

208208
@pytest.mark.asyncio
209-
async def test_fetch_aggregates(messages):
209+
async def test_fetch_aggregates(aleph_messages):
210210
cache = MessageCache()
211-
cache.add(messages)
211+
cache.add(aleph_messages)
212212

213-
aggregates = await cache.fetch_aggregates(messages[0].sender)
213+
aggregates = await cache.fetch_aggregates(aleph_messages[0].sender)
214214

215-
assert aggregates == {messages[0].content.key: messages[0].content.content}
215+
assert aggregates == {aleph_messages[0].content.key: aleph_messages[0].content.content}
216216

217217

218218
@pytest.mark.asyncio
219-
async def test_get_message(messages):
219+
async def test_get_message(aleph_messages):
220220
cache = MessageCache()
221-
cache.add(messages)
221+
cache.add(aleph_messages)
222222

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

225-
assert message == messages[0]
225+
assert message == aleph_messages[0]
226226

227227

228228
@pytest.mark.asyncio

0 commit comments

Comments
 (0)