def test_endpoint()

in load-testing/locustfile.py [0:0]

• 25 lines of code
• 1 McCabe index (conditional complexity)

    def test_endpoint(self):
        data, target_model = sample_data(x_test)
        start_time = time.time()
        try:
            sagemaker_runtime.invoke_endpoint(
                            EndpointName=endpoint_name, 
                            ContentType='text/csv',
                            TargetModel=target_model,
                            Body=data)
            total_time = int((time.time() - start_time) * 1000)
            events.request_success.fire(
                request_type="sagemaker-mme",
                name=endpoint_name,
                response_time=total_time,
                response_length=0,
            )

        except:
            total_time = int((time.time() - start_time) * 1000)
            events.request_failure.fire(
                request_type="sagemaker-mme",
                name=endpoint_name,
                response_time=total_time,
                response_length=0,
                exception=sys.exc_info(),
            )