From c21288b3ddf40a338f53293bd7810813b3ad0f78 Mon Sep 17 00:00:00 2001 From: Itamar Turner-Trauring Date: Fri, 2 Apr 2021 12:49:58 -0400 Subject: [PATCH] Make it easier to read. --- src/allmydata/test/test_consumer.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/src/allmydata/test/test_consumer.py b/src/allmydata/test/test_consumer.py index 46b87e477..73c99d8a3 100644 --- a/src/allmydata/test/test_consumer.py +++ b/src/allmydata/test/test_consumer.py @@ -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)