aws-cli/tests/functional/test_streaming_output.py at develop · lovejavaee/aws-cli · GitHub

45 lines (40 loc) · 1.64 KB

#!/usr/bin/env python

# Licensed under the Apache License, Version 2.0 (the "License"). You

# may not use this file except in compliance with the License. A copy of

# the License is located at

# http://aws.amazon.com/apache2.0/

# or in the "license" file accompanying this file. This file 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 awscli.compat import six

from awscli.testutils import BaseAWSCommandParamsTest

from awscli.testutils import FileCreator

class TestStreamingOutput(BaseAWSCommandParamsTest):

def setUp(self):

super(TestStreamingOutput, self).setUp()

self.files = FileCreator()

def tearDown(self):

super(TestStreamingOutput, self).tearDown()

self.files.remove_all()

def test_get_media_streaming_output(self):

cmdline = (

'kinesis-video-media get-media --stream-name test-stream '

'--start-selector StartSelectorType=EARLIEST %s'

)

self.parsed_response = {

'ContentType': 'video/webm',

'Payload': six.BytesIO(b'testbody')

}

outpath = self.files.full_path('outfile')

params = {

'StartSelector': {'StartSelectorType': 'EARLIEST'},

'StreamName': 'test-stream'

}

self.assert_params_for_cmd(cmdline % outpath, params)

with open(outpath, 'rb') as outfile:

self.assertEqual(outfile.read(), b'testbody')

Expand file tree