diff --git a/superset/results_backends.py b/superset/results_backends.py index 614d28c2f6b09..ccf755fc7d146 100644 --- a/superset/results_backends.py +++ b/superset/results_backends.py @@ -35,6 +35,7 @@ def __init__(self, default_timeout=300): self.default_timeout = default_timeout self.s3_client = boto3.client('s3') + self.bucket = config.get('S3_CACHE_BUCKET') self.key_prefix = config.get('S3_CACHE_KEY_PREFIX') diff --git a/tests/results_backends_tests.py b/tests/results_backends_tests.py index 8ebbb53a77e4c..f628ebdbf975a 100644 --- a/tests/results_backends_tests.py +++ b/tests/results_backends_tests.py @@ -1,4 +1,5 @@ import cPickle + import mock from superset import app, results_backends @@ -44,7 +45,7 @@ def test_s3_cache_set(self): self.assertEquals(call_args[1], 'test-bucket') self.assertEquals(call_args[2], 'test-prefix/test-key') - def test_s3_cache_exception(self): + def test_s3_cache_set_exception(self): self.mock_s3_client.upload_fileobj.side_effect = Exception('Something bad happened!') result = self.s3_cache.set('test-key', 'test-value')