From 80b6cbfc9c861469146318d0b3dd5f8b8b525b8a Mon Sep 17 00:00:00 2001 From: xiejun <xiejun@vci-tech.com> Date: 星期五, 01 十一月 2024 15:11:19 +0800 Subject: [PATCH] Revert "集成获取mdm分发通用数据格式接口集成" --- Source/BladeX-Tool/blade-core-test/src/main/java/org/springblade/core/test/BladeSpringExtension.java | 88 ++++++++++++++++++++++++++++++++++++++++++++ 1 files changed, 88 insertions(+), 0 deletions(-) diff --git a/Source/BladeX-Tool/blade-core-test/src/main/java/org/springblade/core/test/BladeSpringExtension.java b/Source/BladeX-Tool/blade-core-test/src/main/java/org/springblade/core/test/BladeSpringExtension.java new file mode 100644 index 0000000..c57f457 --- /dev/null +++ b/Source/BladeX-Tool/blade-core-test/src/main/java/org/springblade/core/test/BladeSpringExtension.java @@ -0,0 +1,88 @@ +/* + * Copyright (c) 2018-2028, DreamLu All rights reserved. + * + * Redistribution and use in source and binary forms, with or without + * modification, are permitted provided that the following conditions are met: + * + * Redistributions of source code must retain the above copyright notice, + * this list of conditions and the following disclaimer. + * Redistributions in binary form must reproduce the above copyright + * notice, this list of conditions and the following disclaimer in the + * documentation and/or other materials provided with the distribution. + * Neither the name of the dreamlu.net developer nor the names of its + * contributors may be used to endorse or promote products derived from + * this software without specific prior written permission. + * Author: DreamLu 鍗㈡槬姊� (596392912@qq.com) + */ + +package org.springblade.core.test; + + +import org.junit.jupiter.api.extension.ExtensionContext; +import org.springblade.core.launch.BladeApplication; +import org.springblade.core.launch.constant.AppConstant; +import org.springblade.core.launch.constant.NacosConstant; +import org.springblade.core.launch.constant.SentinelConstant; +import org.springblade.core.launch.service.LauncherService; +import org.springframework.boot.builder.SpringApplicationBuilder; +import org.springframework.core.annotation.AnnotationUtils; +import org.springframework.lang.NonNull; +import org.springframework.test.context.junit.jupiter.SpringExtension; + +import java.util.*; +import java.util.stream.Collectors; + +/** + * 璁剧疆鍚姩鍙傛暟 + * + * @author L.cm + */ +public class BladeSpringExtension extends SpringExtension { + + @Override + public void beforeAll(@NonNull ExtensionContext context) throws Exception { + super.beforeAll(context); + setUpTestClass(context); + } + + private void setUpTestClass(ExtensionContext context) { + Class<?> clazz = context.getRequiredTestClass(); + BladeBootTest bladeBootTest = AnnotationUtils.getAnnotation(clazz, BladeBootTest.class); + if (bladeBootTest == null) { + throw new BladeBootTestException(String.format("%s must be @BladeBootTest .", clazz)); + } + String appName = bladeBootTest.appName(); + String profile = bladeBootTest.profile(); + boolean isLocalDev = BladeApplication.isLocalDev(); + Properties props = System.getProperties(); + props.setProperty("blade.env", profile); + props.setProperty("blade.name", appName); + props.setProperty("blade.is-local", String.valueOf(isLocalDev)); + props.setProperty("blade.dev-mode", profile.equals(AppConstant.PROD_CODE) ? "false" : "true"); + props.setProperty("blade.service.version", AppConstant.APPLICATION_VERSION); + props.setProperty("spring.application.name", appName); + props.setProperty("spring.profiles.active", profile); + props.setProperty("info.version", AppConstant.APPLICATION_VERSION); + props.setProperty("info.desc", appName); + props.setProperty("loadbalancer.client.name", appName); + props.setProperty("spring.cloud.sentinel.transport.dashboard", SentinelConstant.SENTINEL_ADDR); + props.setProperty("spring.main.allow-bean-definition-overriding", "true"); + props.setProperty("spring.cloud.nacos.config.shared-configs[0].data-id", NacosConstant.sharedDataId()); + props.setProperty("spring.cloud.nacos.config.shared-configs[0].group", NacosConstant.NACOS_CONFIG_GROUP); + props.setProperty("spring.cloud.nacos.config.shared-configs[0].refresh", NacosConstant.NACOS_CONFIG_REFRESH); + props.setProperty("spring.cloud.nacos.config.file-extension", NacosConstant.NACOS_CONFIG_FORMAT); + props.setProperty("spring.cloud.nacos.config.shared-configs[1].data-id", NacosConstant.sharedDataId(profile)); + props.setProperty("spring.cloud.nacos.config.shared-configs[1].group", NacosConstant.NACOS_CONFIG_GROUP); + props.setProperty("spring.cloud.nacos.config.shared-configs[1].refresh", NacosConstant.NACOS_CONFIG_REFRESH); + // 鍔犺浇鑷畾涔夌粍浠� + if (bladeBootTest.enableLoader()) { + SpringApplicationBuilder builder = new SpringApplicationBuilder(clazz); + List<LauncherService> launcherList = new ArrayList<>(); + ServiceLoader.load(LauncherService.class).forEach(launcherList::add); + launcherList.stream().sorted(Comparator.comparing(LauncherService::getOrder)).collect(Collectors.toList()) + .forEach(launcherService -> launcherService.launcher(builder, appName, profile, isLocalDev)); + } + System.err.printf("---[junit.test]:[%s]---鍚姩涓紝璇诲彇鍒扮殑鐜鍙橀噺:[%s]%n", appName, profile); + } + +} -- Gitblit v1.9.3