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