def test_s3_get_exists(self, mock_session): aws_manager = AwsManager('test-profile', 'us-east-1') mock_session.return_value.client.return_value.get_object.return_value = 'response' response = aws_manager.s3_get('some-test-bucket', 'key') mock_session.return_value.client.return_value.get_object.assert_called_with( Bucket='some-test-bucket', Key='key') self.assertEqual(response, 'response')
def test_s3_head(self, mock_session): aws_manager = AwsManager('test-profile', 'us-east-1') mock_session.return_value.client.return_value.get_object.side_effect = ClientError( {'Error': {}}, 'test-op') response = aws_manager.s3_get('some-test-bucket', 'key') mock_session.return_value.client.return_value.get_object.assert_called_with( Bucket='some-test-bucket', Key='key') self.assertEqual(response, None)
def test_s3_head(self, mock_session): aws_manager = AwsManager('test-profile', 'us-east-1') mock_session.return_value.client.return_value.get_object.side_effect = ClientError({'Error': {}}, 'test-op') response = aws_manager.s3_get('some-test-bucket', 'key') mock_session.return_value.client.return_value.get_object.assert_called_with(Bucket='some-test-bucket', Key='key') self.assertEqual(response, None)
def test_s3_get_exists(self, mock_session): aws_manager = AwsManager('test-profile', 'us-east-1') mock_session.return_value.client.return_value.get_object.return_value = 'response' response = aws_manager.s3_get('some-test-bucket', 'key') mock_session.return_value.client.return_value.get_object.assert_called_with(Bucket='some-test-bucket', Key='key') self.assertEqual(response, 'response')