def setUp(self):
#Config
config = TestConfig()
#NoSuchKey Error
error = {
'Error': {
'Code': 'NoSuchKey',
'Message': 'The specified key does not exist.',
'Key': 'd20b1c38-2f5f-4b48-b604-eb90f82ff800'
},
'ResponseMetadata': {
'HTTPStatusCode': 404,
'RetryAttempts': 0
}
}
#Mock
mock = MagicMock()
mock.get_object(Bucket=config.BUCKET_NAME,
Key='d20b1c38-2f5f-4b48-b604-eb90f82ff800')
mock.get_object.side_effect = ClientError(error, 'GetObject')
config.S3 = mock
#Storage
self.storage = AWSStorage(config)
self.config = config
def setUp(self):
#Config
config = TestConfig()
#Mock
mock = MagicMock()
mock.delete_object(Bucket=config.BUCKET_NAME,
Key='d20b1c38-2f5f-4b48-b604-eb90f82ff800')
config.S3 = mock
#Storage
self.storage = AWSStorage(config)
self.config = config
def setUp(self):
#Temp File
self.file = tempfile.TemporaryFile(mode='w+b')
self.file.write(b'It is a file!')
self.file.seek(0)
self.raw_file = self.file.read()
#Config
config = TestConfig()
#Mock
mock = MagicMock()
mock.put_object(Bucket=config.BUCKET_NAME,
Key='d20b1c38-2f5f-4b48-b604-eb90f82ff800',
Body=b'It is a file!')
config.S3 = mock
#Storage
self.storage = AWSStorage(config)
self.config = config
