From a5e86ef98425aa9b69e2bd5f08d427cf9d5ba6d2 Mon Sep 17 00:00:00 2001 From: Tim Elliott Date: Mon, 4 Feb 2019 10:52:09 -0800 Subject: [PATCH] producer: Add host='' to connect_s3 for SigV4 --- muskrat/producer.py | 2 +- muskrat/tests/test_producer.py | 4 +++- 2 files changed, 4 insertions(+), 2 deletions(-) diff --git a/muskrat/producer.py b/muskrat/producer.py index 4270f9f..8d382e0 100644 --- a/muskrat/producer.py +++ b/muskrat/producer.py @@ -88,7 +88,7 @@ def __init__(self, **kwargs): @property def s3conn(self): if self._s3conn is None: - self._s3conn = boto.connect_s3( self.config.s3_key, self.config.s3_secret ) + self._s3conn = boto.connect_s3( self.config.s3_key, self.config.s3_secret, host=self.config.s3_host ) return self._s3conn @property diff --git a/muskrat/tests/test_producer.py b/muskrat/tests/test_producer.py index cd1ef09..14fb612 100644 --- a/muskrat/tests/test_producer.py +++ b/muskrat/tests/test_producer.py @@ -23,6 +23,8 @@ TEST_KEY_PREFIX = 'Muskrat.Producer' +boto.config.add_section('s3') +boto.config.set('s3', 'use-sigv4', 'True') class TestProducerBase( unittest.TestCase ): @@ -70,7 +72,7 @@ class TestS3Producer( TestProducerBase ): def setUp(self): super( TestS3Producer, self ).setUp() - self.conn = boto.connect_s3( config.s3_key, config.s3_secret ) + self.conn = boto.connect_s3( config.s3_key, config.s3_secret, host=config.s3_host ) self.bucket = self.conn.get_bucket( config.s3_bucket ) self.key += '.Test.S3Producer'