PutKinesisStream.py

# Licensed to the Apache Software Foundation (ASF) under one or more
# contributor license agreements.  See the NOTICE file distributed with
# this work for additional information regarding copyright ownership.
# The ASF licenses this file to You under the Apache License, Version 2.0
# (the "License"); you may not use this file except in compliance with
# the License.  You may obtain a copy of the License at
#
#     http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.


from ..core.Processor import Processor


class PutKinesisStream(Processor):
    def __init__(
            self,
            context,
            proxy_host='',
            proxy_port='',
            proxy_username='',
            proxy_password=''):
        super(PutKinesisStream, self).__init__(
            context=context,
            clazz='PutKinesisStream',
            properties={
                'Amazon Kinesis Stream Name': 'test_stream',
                'Access Key': 'test_access_key',
                'Secret Key': 'test_secret',
                'Endpoint Override URL': f"http://kinesis-server-{context.feature_id}:4568",
                'Proxy Host': proxy_host,
                'Proxy Port': proxy_port,
                'Proxy Username': proxy_username,
                'Proxy Password': proxy_password,
                'Region': 'us-east-1'
            },
            auto_terminate=["success", "failure"])