18
18
19
19
20
20
@pytest .mark .asyncio
21
- async def test_base (messages ):
21
+ async def test_base (aleph_messages ):
22
22
# test add_many
23
23
cache = MessageCache ()
24
- cache .add (messages )
25
- assert len (cache ) == len (messages )
24
+ cache .add (aleph_messages )
25
+ assert len (cache ) == len (aleph_messages )
26
26
27
- item_hashes = [message .item_hash for message in messages ]
27
+ item_hashes = [message .item_hash for message in aleph_messages ]
28
28
cached_messages = cache .get (item_hashes )
29
- assert len (cached_messages ) == len (messages )
29
+ assert len (cached_messages ) == len (aleph_messages )
30
30
31
- for message in messages :
31
+ for message in aleph_messages :
32
32
assert cache [message .item_hash ] == message
33
33
34
- for message in messages :
34
+ for message in aleph_messages :
35
35
assert message .item_hash in cache
36
36
37
37
for message in cache :
@@ -47,8 +47,8 @@ class TestMessageQueries:
47
47
cache : MessageCache
48
48
49
49
@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
52
52
self .cache = MessageCache ()
53
53
self .cache .add (self .messages )
54
54
@@ -127,8 +127,8 @@ class TestPostQueries:
127
127
cache : MessageCache
128
128
129
129
@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
132
132
self .cache = MessageCache ()
133
133
self .cache .add (self .messages )
134
134
@@ -196,33 +196,33 @@ async def mock_message_stream():
196
196
197
197
198
198
@pytest .mark .asyncio
199
- async def test_fetch_aggregate (messages ):
199
+ async def test_fetch_aggregate (aleph_messages ):
200
200
cache = MessageCache ()
201
- cache .add (messages )
201
+ cache .add (aleph_messages )
202
202
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 )
204
204
205
- assert aggregate == messages [0 ].content .content
205
+ assert aggregate == aleph_messages [0 ].content .content
206
206
207
207
208
208
@pytest .mark .asyncio
209
- async def test_fetch_aggregates (messages ):
209
+ async def test_fetch_aggregates (aleph_messages ):
210
210
cache = MessageCache ()
211
- cache .add (messages )
211
+ cache .add (aleph_messages )
212
212
213
- aggregates = await cache .fetch_aggregates (messages [0 ].sender )
213
+ aggregates = await cache .fetch_aggregates (aleph_messages [0 ].sender )
214
214
215
- assert aggregates == {messages [0 ].content .key : messages [0 ].content .content }
215
+ assert aggregates == {aleph_messages [0 ].content .key : aleph_messages [0 ].content .content }
216
216
217
217
218
218
@pytest .mark .asyncio
219
- async def test_get_message (messages ):
219
+ async def test_get_message (aleph_messages ):
220
220
cache = MessageCache ()
221
- cache .add (messages )
221
+ cache .add (aleph_messages )
222
222
223
- message : AlephMessage = await cache .get_message (messages [0 ].item_hash )
223
+ message : AlephMessage = await cache .get_message (aleph_messages [0 ].item_hash )
224
224
225
- assert message == messages [0 ]
225
+ assert message == aleph_messages [0 ]
226
226
227
227
228
228
@pytest .mark .asyncio
0 commit comments