[开发测试]一个典型spock UT demo

一个完整的功能完备的spock demo

class LarkServiceTest extends Specification {
    @Spy
    LarkAdapter lAdapter;

    @Spy
    LarkService lService;

    @Mock
    AopContext aopContext;

    @Mock
    CustomDataRepositoryV2 dataRepository

	// 将bean的定义直接copy过来，这里用来处理Spring bean注入
    RestTemplate restTemplate = restTemplate()

    // 需要注入的数据
    private String appId = "cli_a06";

    private String domain = "https://open.feishu.cn/open-apis/";

    private String batchUserIdUri = "user/v1/batch_get_id";

    void setup() {
    	// RequestContext.getProperty()静态方法中需要查询的数据，在cache中预设数据
        ThreadLocalCache.set(ContextConstant.TENANT_CODE, '1000')
        ThreadLocalCache.set(ContextConstant.EMPLOYEE_ID, 2012300012)

        MockitoAnnotations.initMocks(this)
        // 返回mock的复杂数据结构，每次调用dataRepository.load方法时，就会按照mockEntityData()返回mock数据
        Mockito.when(dataRepository.load(Matchers.anyString(), Matchers.anyString(), Matchers.anyString()))
                .thenAnswer({ args -> mockEntityData(args) })
        // 植入mock好的各种属性字段        
        Whitebox.setInternalState(lAdapter, "restTemplate", restTemplate)
        Whitebox.setInternalState(lAdapter, "appId", appId)
        Whitebox.setInternalState(lAdapter, "appSecret", appSecret)
        Whitebox.setInternalState(lAdapter, "domain", domain)
        Whitebox.setInternalState(lAdapter, "batchUserIdUri", batchUserIdUri)
        // 因此lService中每次调用load方法时就会走上述mock
        Whitebox.setInternalState(larkService, "dataRepository", dataRepository)
        Whitebox.setInternalState(larkService, "larkAdapter", larkAdapter)
    }

    void cleanup() {
    }

    def "exchangeEmailForIDTest"() {
        when:
        Map map = lAdapter.exchangeEmailsForOpenIds(Arrays.asList("test@mail.com"))
        then:
        map != null && !map.isEmpty()
    }

    def "CreateGroupWithWrongEmployee"() {
        when:
        // 临时修改cache来模拟负向case
        ThreadLocalCache.set(ContextConstant.EMPLOYEE_ID, 1000000L)
        String str = lService.createGroup("test1", "111")
        then:
        def e = thrown(Exception)
        e.message != ""
    }

    def "CreateGroupTest"() {
        when:
        Pair pair = Pair.of("2012300012", Arrays.asList("2012300012"))
        // 当需要某个方法返回预设值而不是走实际方法时通过下述方法执行
        Mockito.doReturn(pair).when(lService).getLarkMemberPair(Mockito.any())
        Mockito.doReturn("test").when(lAdapter).createLarkGroup(Mockito.any(), Mockito.any(), Mockito.any())
        String str = larkService.createGroup("test0", "111")
        then:
        def e = thrown(Exception)
        e.message != ""
    }

    // 将bean的定义直接copy过来，这里用来处理Spring bean注入
    private RestTemplate restTemplate() {
        OkHttp3ClientHttpRequestFactory okHttpFactory = new OkHttp3ClientHttpRequestFactory();
        okHttpFactory.setConnectTimeout(10000);
        okHttpFactory.setReadTimeout(10000);
        RestTemplate restTemplate = new RestTemplate(okHttpFactory);
        MappingJackson2HttpMessageConverter converter = new MappingJackson2HttpMessageConverter();
        List<MediaType> mediaTypes = new ArrayList<>();
        mediaTypes.add(MediaType.TEXT_PLAIN);
        mediaTypes.add(MediaType.TEXT_HTML);
        converter.setSupportedMediaTypes(mediaTypes);
        restTemplate.getMessageConverters().add(converter);
        return restTemplate;
    }

	// mock复杂对象
	// 使用方法：Mockito.when(customDataRepository.load(Matchers.anyString(), Matchers.anyString(), Matchers.anyString()))
                .thenAnswer({ args -> mockLoadEntityData(args) })
    // 这样在调用customDataRepository.load（）时候就会根据传入的entityName决定用哪一个实体
    EntityData mockLoadEntityData(InvocationOnMock args) {
        args.getArgumentAt(0, String.class)
        def entityName = args.getArgumentAt(1, String.class)
        EntityData result = null
        if (entityName == 'test0') {
            result = [
                    primaryKey  : '111',
                    tenantCode  : RequestContext.tenantCode,
                    sysName     : 'test',
                    recordType  : 'rec',
                    entityName  : entityName,
                    fieldList   : [
                            [
                                    fieldName : 'primaryName',
                                    fieldValue: 'primaryName',
                                    action    : ActionEnum.NONE,
                                    fieldMeta : [
                                            fieldName   : 'primaryName',
                                            type        : FieldTemplateTypeEnum.TYPE_STRING,
									 // 注意：以，结尾，否则将会导致map映射不上的问题。这是因为这里不是该结果的全量数据
                                            valueType   : FieldValueTypeEnum.TYPE_STRING,
                                    ] as EntityField,
                            ] as OthersField,
                            [
                                    fieldName : 'ownerId',
                                    fieldMeta : [
                                            fieldName   : 'Lark_Group',
                                            valueType   : FieldValueTypeEnum.TYPE_STRING,
                                            type        : FieldTemplateTypeEnum.TYPE_STRING,
                                            relationName: []
                                    ] as EntityField,
                                    fieldValue: '8011869',
                                    action    : ActionEnum.NONE,
                            ] as OthersField,
                    ] as List<CustomField>,
                    entityAction: ActionEnum.NONE,
                      // ， 结尾，否则会映射不上
            ] as EntityData
        } else if (entityName == 'test3') {
            result = [
                    primaryKey  : '111',
                    tenantCode  : RequestContext.tenantCode,
                    sysName     : 'test',
                    recordType  : 'rec',
                    entityName  : entityName,
                    fieldList   : [
                            [
                                    fieldName : 'ownerId',
                                    fieldMeta : [
                                            fieldName   : 'Lark_Group',
                                            valueType   : FieldValueTypeEnum.TYPE_STRING,
                                            type        : FieldTemplateTypeEnum.TYPE_STRING,
                                            relationName: []
                                    ] as EntityField,
                                    fieldValue: '',
                                    action    : ActionEnum.NONE,
                            ] as OthersField
                    ] as List<CustomField>,
                    entityAction: ActionEnum.NONE,
            ] as EntityData
        }
        return result
    }
}

adapter源码：

public class LarkAdapter {
    @Value("${open.app_id:cli_9f28c167c635900c}")
    private String appId;

    @Value("${open.app_secret:jDdumhMXPh9FInntZ2hb4bPGLOEiyED4}")
    private String appSecret;

    @Value("${open.domain:https:/……}")
    private String domain;

    private RestTemplate restTemplate = restTemplate();
	
	……
}

关注博主即可阅读全文