ExternalVendorCode/tahoe-lafs
1
0
Fork 0
mirror of https://github.com/tahoe-lafs/tahoe-lafs.git synced 2025-04-14 14:16:39 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Make it easier to read.

Browse Source
This commit is contained in:
Itamar Turner-Trauring 2021-04-02 12:49:58 -04:00
parent dd6e0d5467
commit c21288b3dd
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
src/allmydata/test/test_consumer.py
View File

@ -28,8 +28,8 @@ class Producer(object):
If used as streaming, the test should call iterate() manually.
"""
def __init__(self, consumer):
self.data = [b"abc", b"def", b"ghi"]
def __init__(self, consumer, data):
self.data = data
self.consumer = consumer
self.done = False
@ -58,7 +58,7 @@ class MemoryConsumerTests(TestCase):
A MemoryConsumer accumulates all data sent by a streaming producer.
"""
consumer = MemoryConsumer()
producer = Producer(consumer)
producer = Producer(consumer, [b"abc", b"def", b"ghi"])
consumer.registerProducer(producer, True)
self.assertEqual(consumer.chunks, [b"abc"])
producer.iterate()
@ -74,7 +74,7 @@ class MemoryConsumerTests(TestCase):
A MemoryConsumer accumulates all data sent by a non-streaming producer.
"""
consumer = MemoryConsumer()
producer = Producer(consumer)
producer = Producer(consumer, [b"abc", b"def", b"ghi"])
consumer.registerProducer(producer, False)
self.assertEqual(consumer.chunks, [b"abc", b"def", b"ghi"])
self.assertEqual(consumer.done, True)