salesforce 零基础学习（六十八）http callout test class写法

此篇可以参考：

https://developer.salesforce.com/docs/atlas.en-us.apexcode.meta/apexcode/apex_classes_restful_http_testing_httpcalloutmock.htm

https://developer.salesforce.com/trailhead/force_com_dev_intermediate/apex_integration_services/apex_integration_rest_callouts

在项目中我们经常会用到通过http方式和其他系统交互，在salesforce 零基础学习（三十三）通过REST方式访问外部数据以及JAVA通过rest方式访问salesforce这篇讲过http callout方式使用，

简单callout demo如下：

 public class CalloutClass {     //default out of time
     private static final Integer OUT_OF_TIME = 10000;
     //default method : get
     private static final String DEFAULT_METHOD_GET = 'GET';     private static final Integer STATUS_CODE_OK = 200;     public static String getDataViaHttp(String endPoint,String param) {
         return getDataViaHttp(endPoint,DEFAULT_METHOD_GET,param);
     }     public static String getDataViaHttp(String endPoint,String method,String param) {
         return getDataViaHttp(endPoint,method,param,OUT_OF_TIME);
     }     public static String getDataViaHttp(String endPoint,String method,String param,Integer outOfTime) {
         HttpRequest req = new HttpRequest();
         Http h = new Http();
         req.setMethod(method);
         req.setHeader('Content-Type', 'application/json');
         if(param != null) {
             req.setBody(param);
         }
         req.setEndpoint(endPoint);
         req.setTimeout(outOfTime);
         HttpResponse res = h.send(req);
         if(res.getStatusCode() == STATUS_CODE_OK) {
             return res.getBody();
         } else {
             throw new CallOutException('访问失败');
         }
     }     class CallOutException extends Exception {     }
 }

有的时候我们需要在batch中调用http接口和其他系统交互进行字段更新等操作，如果在batch中需要用到http callout，需要实现Database.AllowsCallouts接口，demo如下：

 public with sharing class CalloutBatchClass implements Database.Batchable<sObject>,Database.AllowsCallouts{
     public Database.QueryLocator start(Database.BatchableContext BC) {
         String fetchSQL = 'fetch sql';
         return Database.getQueryLocator(fetchSQL);
     }     public void execute(Database.BatchableContext BC, List<sObject> objList) {
         String endPoint = 'site end point';
         String responseData = CalloutClass.getDataViaHttp(endPoint,null);
         for(sObject obj : objList) {
             //TODO
         }
     }     public void finish(Database.BatchableContext BC) {     }
 }

项目中test class是必需的，而且正常要求test class覆盖率超过75%。test class中不允许http callout，我们可以通过实现HttpCalloutMock接口模拟http请求的返回值。通过重写respond方法实现

不同的http请求所返回的不同的response状态和body内容。

 @isTest
 global class MockHttpResponseGenerator implements HttpCalloutMock {
     global String method;     global String METHOD1_BODY = '{"foo":"bar"}';     global String METHOD2_BODY = '{"foo":"bar2"}';     global MockHttpResponseGenerator() {}     global MockHttpResponseGenerator(String requestMethod) {
         method = requestMethod;
     }     // Implement this interface method
     global HTTPResponse respond(HTTPRequest req) {
         // Create a fake response
         HttpResponse res = new HttpResponse();
         res.setHeader('Content-Type', 'application/json');
         String body;
         if(method == 'method1') {
             body = METHOD1_BODY;
         } else if(method == 'method2') {
             body = METHOD2_BODY;
         } else if(method == 'methodError') {
             res.setStatusCode(500);
         }
         res.setBody('{"foo":"bar"}');
         if(res.getStatusCode() != null) {
             res.setStatusCode(200);
         }
         return res;
     }
 }

简单的测试CalloutClass的测试类如下：

@isTest
private class CalloutClassTest {
     @isTest static void testSuccessCallout() {
         Test.startTest();
        // Set mock callout class
        Test.setMock(HttpCalloutMock.class, new MockHttpResponseGenerator('method1'));
        String endPoint = 'http://api.salesforce.com/foo/bar';
        String result = CalloutClass.getDataViaHttp(endPoint,'test param');
        String expectedValue = '{"foo":"bar"}';
        System.assertEquals(result, expectedValue);
        Test.stopTest();
    }
}

这只是我们碰到的所谓最理想的情况，有的时候我们往往会碰到这样一种情况：一个方法里面需要调用到多个http callout。比如需要先进行http callout，将返回值作为参数或者oauth setting内容然后继续进行callout，这种情况下使用上述的方式便比较难实现，毕竟上述mock形式仅作为一个http callout的response。这个时候我们要变通一下，看看前面的调用是否是必要的–前后几次调用是否有并列关系，还是仅将前几次调用作为相关参数为最后一次做准备，此种情况下，可以在类中设置相关的静态变量来跳过相关的调用；如果前后几次调用属于并列关系，需要对每一次的response的内容进行相关处理，这种情况下的test class便需要使用multi mock形式。

一.非并列关系：此种方式可以使用变量方式跳过相关的调用

 public with sharing class CalloutClassUseVariable {
     public static Boolean skipForTest{get;set;}
     public STring getResult(String endPoint1,String endPoint2) {
         String result1 = '';
         if(skipForTest == null ||skipForTest == false) {
             result1 = CalloutClass.getDataViaHttp(endPoint1,'');
         }
         String result2 = CalloutClass.getDataViaHttp(endPoint2,result1);
         return result2;
     }
 }

个人收藏笔记记录

开通VIP