def generate_kinesis_events()

in lib/integration-test/lambda/generate_kinesis_events.py [0:0]


def generate_kinesis_events(event, context):
    try:
        LOGGER.info("Received event: %s", event)
        d=event
        for i in range(0, event['record_count']):
            data = json.dumps(getReferrer())
            kinesis.put_record(
                StreamName=d['KinesisInputStreamName'],
                Data=data,
                PartitionKey='partitionKey')
        return event
    except Exception as e:
        LOGGER.exception("Error while writing to Kinesis stream.")
        return {"status": "FAILED", "error_message": str(e)}