def status_proto(self): return status_pb2.Status(status_code=self.proto_status_code, error_message=str(self))
@patch( "bentoml.yatai.deployment.aws_ec2.operator.ensure_sam_available_or_raise", MagicMock(), ) @patch( "bentoml.yatai.deployment.aws_ec2.operator.ensure_docker_available_or_raise", MagicMock(), ) @patch( "bentoml.yatai.deployment.aws_ec2.operator.AwsEc2DeploymentOperator.deploy_service", MagicMock(), ) @patch( "bentoml.yatai.deployment.aws_ec2.operator.AwsEc2DeploymentOperator.describe", MagicMock(return_value=DescribeDeploymentResponse(status=status_pb2.Status( status_code=status_pb2.Status.INTERNAL, error_message="failed"))), ) def test_ec2_update_describe_failure(): yatai_service_mock = create_yatai_service_mock() test_deployment_pb = generate_ec2_deployment_pb() operator = AwsEc2DeploymentOperator(yatai_service_mock) result_pb = operator.update(test_deployment_pb, test_deployment_pb) assert result_pb.status.status_code == status_pb2.Status.INTERNAL assert result_pb.deployment.state.state == DeploymentState.ERROR @patch( "bentoml.yatai.deployment.aws_ec2.operator.ensure_sam_available_or_raise", MagicMock(), )