async def test_producer_producer_limit(self): # Expect some throughput errors async with Producer( stream_name=self.STREAM_NAME_SINGLE_SHARD, processor=StringProcessor(), put_bandwidth_limit_per_shard=1500, ) as producer: async with Consumer( stream_name=self.STREAM_NAME_SINGLE_SHARD, processor=StringProcessor(), iterator_type="LATEST", ) as consumer: await consumer.start_consumer() # Wait a bit just to be sure iterator is gonna get late await asyncio.sleep(3) for x in range(20): await producer.put(self.random_string(1024 * 250)) # todo: async timeout output = [] while len(output) < 20: async for item in consumer: output.append(item) self.assertEquals(len(output), 20) self.assertTrue(producer.throughput_exceeded_count > 0)
async def test_producer_and_consumer_consume_from_start_after(self): # Don't flush, close producer immediately to test all data is written to stream on exit. async with Producer( stream_name=self.stream_name, endpoint_url=ENDPOINT_URL, processor=StringProcessor(), ) as producer: # Put enough data to ensure it will require more than one put # ie test overflow behaviour for _ in range(15): await producer.put(self.random_string(100 * 1024)) results = [] async with Consumer( stream_name=self.stream_name, endpoint_url=ENDPOINT_URL, processor=StringProcessor(), ) as consumer: async for item in consumer: results.append(item) # Expect to have consumed from start as default iterator_type=TRIM_HORIZON self.assertEquals(len(results), 15)
async def test_producer_put_below_limit(self): async with Producer( stream_name=self.stream_name, processor=StringProcessor(), endpoint_url=ENDPOINT_URL, ) as producer: # The maximum size of the data payload of a record before base64-encoding is up to 1 MiB. # Limit is set in aggregators.BaseAggregator (few bytes short of 1MiB) await producer.put(self.random_string(40 * 25 * 1024))
async def test_consumer_checkpoint(self): checkpointer = MemoryCheckPointer(name="test") results = [] async with Producer( stream_name=self.STREAM_NAME_SINGLE_SHARD, processor=StringProcessor(), ) as producer: async with Consumer( stream_name=self.STREAM_NAME_SINGLE_SHARD, checkpointer=checkpointer, processor=StringProcessor(), iterator_type="LATEST", ) as consumer: # Manually start await consumer.start_consumer() await producer.put("test") await producer.flush() for i in range(3): async for item in consumer: results.append(item) checkpoints = checkpointer.get_all_checkpoints() # Expect 1 as only 1 shard self.assertEquals(1, len(checkpoints)) self.assertIsNotNone(checkpoints[list( checkpoints.keys())[0]]["sequence"]) self.assertListEqual(results, ["test"])
def test_string_processor(self): processor = StringProcessor() self.assertEquals(processor.max_bytes, 1024 * 25 * 40) output = list(processor.add_item("test")) self.assertEqual(len(output), 1) self.assertIsInstance(output[0], OutputItem) self.assertEqual(output[0].size, len("test")) self.assertEqual(output[0].n, 1) self.assertEqual(output[0].data, b"test") self.assertFalse(processor.has_items())
def test_processor_exceed_put_limit(self): processor = StringProcessor() with self.assertRaises(exceptions.ExceededPutLimit): list(processor.add_item(self.random_string(1024 * 1024 + 1)))