BJ_ZTEITS-Developers / parking-cloud-yudao

Commit cd656fad4b55c14ce5d49f2fc821f27883d3a102

Authored by puhui999
1 parent 2073ecb2

【代码优化】IoT: 数据桥梁的执行器单元测试优化

Inline Side-by-side
Showing 1 changed file with 85 additions and 74 deletions
yudao-module-iot/yudao-module-iot-biz/src/test/java/cn/iocoder/yudao/module/iot/service/rule/action/databridge/IotDataBridgeExecuteTest.java
  View file @cd656fa
@@ -22,7 +22,7 @@ import static org.mockito.ArgumentMatchers.anyString; @@ -22,7 +22,7 @@ import static org.mockito.ArgumentMatchers.anyString;
22 import static org.mockito.Mockito.when; 22 import static org.mockito.Mockito.when;
23 23
24 /** 24 /**
25 - * {@link IotDataBridgeExecute} 实现类的测试 25 + * {@link IotDataBridgeExecute} 实现类的单元测试
26 * 26 *
27 * @author HUIHUI 27 * @author HUIHUI
28 */ 28 */
@@ -41,114 +41,125 @@ public class IotDataBridgeExecuteTest extends BaseMockitoUnitTest { @@ -41,114 +41,125 @@ public class IotDataBridgeExecuteTest extends BaseMockitoUnitTest {
41 @BeforeEach 41 @BeforeEach
42 public void setUp() { 42 public void setUp() {
43 // 创建共享的测试消息 43 // 创建共享的测试消息
44 - message = IotDeviceMessage.builder().requestId("TEST-001").reportTime(LocalDateTime.now()).tenantId(1L)  
45 - .productKey("testProduct").deviceName("testDevice").deviceKey("testDeviceKey")  
46 - .type("property").identifier("temperature").data("{\"value\": 60}") 44 + message = IotDeviceMessage.builder()
  45 + .requestId("TEST-001")
  46 + .reportTime(LocalDateTime.now())
  47 + .tenantId(1L)
  48 + .productKey("testProduct")
  49 + .deviceName("testDevice")
  50 + .deviceKey("testDeviceKey")
  51 + .type("property")
  52 + .identifier("temperature")
  53 + .data("{\"value\": 60}")
47 .build(); 54 .build();
48 -  
49 - // 配置 RestTemplate mock 返回成功响应  
50 - // TODO @puhui999：这个应该放到 testHttpDataBridge 里  
51 - when(restTemplate.exchange(anyString(), any(HttpMethod.class), any(), any(Class.class)))  
52 - .thenReturn(new ResponseEntity<>("Success", HttpStatus.OK));  
53 } 55 }
54 56
55 @Test 57 @Test
56 - public void testKafkaMQDataBridge() { 58 + public void testKafkaMQDataBridge() throws Exception {
57 // 1. 创建执行器实例 59 // 1. 创建执行器实例
58 IotKafkaMQDataBridgeExecute action = new IotKafkaMQDataBridgeExecute(); 60 IotKafkaMQDataBridgeExecute action = new IotKafkaMQDataBridgeExecute();
59 61
60 // 2. 创建配置 62 // 2. 创建配置
61 - // TODO @puhui999：可以改成链式哈。  
62 - IotDataBridgeKafkaMQConfig config = new IotDataBridgeKafkaMQConfig();  
63 - config.setBootstrapServers("127.0.0.1:9092");  
64 - config.setTopic("test-topic");  
65 - config.setSsl(false);  
66 - config.setUsername(null);  
67 - config.setPassword(null);  
68 -  
69 - // 3. 执行两次测试，验证缓存  
70 - log.info("[testKafkaMQDataBridge][第一次执行，应该会创建新的 producer]");  
71 - action.execute(message, new IotDataBridgeDO().setType(action.getType()).setConfig(config));  
72 -  
73 - log.info("[testKafkaMQDataBridge][第二次执行，应该会复用缓存的 producer]");  
74 - action.execute(message, new IotDataBridgeDO().setType(action.getType()).setConfig(config)); 63 + IotDataBridgeKafkaMQConfig config = new IotDataBridgeKafkaMQConfig()
  64 + .setBootstrapServers("127.0.0.1:9092")
  65 + .setTopic("test-topic")
  66 + .setSsl(false)
  67 + .setUsername(null)
  68 + .setPassword(null);
  69 +
  70 + // 3. 执行测试并验证缓存
  71 + executeAndVerifyCache(action, config, "KafkaMQ");
75 } 72 }
76 73
77 @Test 74 @Test
78 - public void testRabbitMQDataBridge() { 75 + public void testRabbitMQDataBridge() throws Exception {
79 // 1. 创建执行器实例 76 // 1. 创建执行器实例
80 IotRabbitMQDataBridgeExecute action = new IotRabbitMQDataBridgeExecute(); 77 IotRabbitMQDataBridgeExecute action = new IotRabbitMQDataBridgeExecute();
81 78
82 // 2. 创建配置 79 // 2. 创建配置
83 - IotDataBridgeRabbitMQConfig config = new IotDataBridgeRabbitMQConfig();  
84 - config.setHost("localhost");  
85 - config.setPort(5672);  
86 - config.setVirtualHost("/");  
87 - config.setUsername("admin");  
88 - config.setPassword("123456");  
89 - config.setExchange("test-exchange");  
90 - config.setRoutingKey("test-key");  
91 - config.setQueue("test-queue");  
92 -  
93 - // 3. 执行两次测试，验证缓存  
94 - log.info("[testRabbitMQDataBridge][第一次执行，应该会创建新的 producer]");  
95 - action.execute(message, new IotDataBridgeDO().setType(action.getType()).setConfig(config));  
96 -  
97 - log.info("[testRabbitMQDataBridge][第二次执行，应该会复用缓存的 producer]");  
98 - action.execute(message, new IotDataBridgeDO().setType(action.getType()).setConfig(config)); 80 + IotDataBridgeRabbitMQConfig config = new IotDataBridgeRabbitMQConfig()
  81 + .setHost("localhost")
  82 + .setPort(5672)
  83 + .setVirtualHost("/")
  84 + .setUsername("admin")
  85 + .setPassword("123456")
  86 + .setExchange("test-exchange")
  87 + .setRoutingKey("test-key")
  88 + .setQueue("test-queue");
  89 +
  90 + // 3. 执行测试并验证缓存
  91 + executeAndVerifyCache(action, config, "RabbitMQ");
99 } 92 }
100 93
101 @Test 94 @Test
102 - public void testRedisStreamMQDataBridge() { 95 + public void testRedisStreamMQDataBridge() throws Exception {
103 // 1. 创建执行器实例 96 // 1. 创建执行器实例
104 IotRedisStreamMQDataBridgeExecute action = new IotRedisStreamMQDataBridgeExecute(); 97 IotRedisStreamMQDataBridgeExecute action = new IotRedisStreamMQDataBridgeExecute();
105 98
106 // 2. 创建配置 99 // 2. 创建配置
107 - IotDataBridgeRedisStreamMQConfig config = new IotDataBridgeRedisStreamMQConfig();  
108 - config.setHost("127.0.0.1");  
109 - config.setPort(6379);  
110 - config.setDatabase(0);  
111 - config.setPassword("123456");  
112 - config.setTopic("test-stream");  
113 -  
114 - // 3. 执行两次测试，验证缓存  
115 - log.info("[testRedisStreamMQDataBridge][第一次执行，应该会创建新的 producer]");  
116 - action.execute(message, new IotDataBridgeDO().setType(action.getType()).setConfig(config));  
117 -  
118 - log.info("[testRedisStreamMQDataBridge][第二次执行，应该会复用缓存的 producer]");  
119 - action.execute(message, new IotDataBridgeDO().setType(action.getType()).setConfig(config)); 100 + IotDataBridgeRedisMQConfig config = new IotDataBridgeRedisMQConfig()
  101 + .setHost("127.0.0.1")
  102 + .setPort(6379)
  103 + .setDatabase(0)
  104 + .setPassword("123456")
  105 + .setTopic("test-stream");
  106 +
  107 + // 3. 执行测试并验证缓存
  108 + executeAndVerifyCache(action, config, "RedisStreamMQ");
120 } 109 }
121 110
122 @Test 111 @Test
123 - public void testRocketMQDataBridge() { 112 + public void testRocketMQDataBridge() throws Exception {
124 // 1. 创建执行器实例 113 // 1. 创建执行器实例
125 IotRocketMQDataBridgeExecute action = new IotRocketMQDataBridgeExecute(); 114 IotRocketMQDataBridgeExecute action = new IotRocketMQDataBridgeExecute();
126 115
127 // 2. 创建配置 116 // 2. 创建配置
128 - IotDataBridgeRocketMQConfig config = new IotDataBridgeRocketMQConfig();  
129 - config.setNameServer("127.0.0.1:9876");  
130 - config.setGroup("test-group");  
131 - config.setTopic("test-topic");  
132 - config.setTags("test-tag");  
133 -  
134 - // 3. 执行两次测试，验证缓存  
135 - log.info("[testRocketMQDataBridge][第一次执行，应该会创建新的 producer]");  
136 - action.execute(message, new IotDataBridgeDO().setType(action.getType()).setConfig(config));  
137 -  
138 - log.info("[testRocketMQDataBridge][第二次执行，应该会复用缓存的 producer]");  
139 - action.execute(message, new IotDataBridgeDO().setType(action.getType()).setConfig(config)); 117 + IotDataBridgeRocketMQConfig config = new IotDataBridgeRocketMQConfig()
  118 + .setNameServer("127.0.0.1:9876")
  119 + .setGroup("test-group")
  120 + .setTopic("test-topic")
  121 + .setTags("test-tag");
  122 +
  123 + // 3. 执行测试并验证缓存
  124 + executeAndVerifyCache(action, config, "RocketMQ");
140 } 125 }
141 126
142 @Test 127 @Test
143 public void testHttpDataBridge() throws Exception { 128 public void testHttpDataBridge() throws Exception {
144 - // 创建配置  
145 - IotDataBridgeHttpConfig config = new IotDataBridgeHttpConfig();  
146 - config.setUrl("https://doc.iocoder.cn/");  
147 - config.setMethod(HttpMethod.GET.name()); 129 + // 1. 配置 RestTemplate mock 返回成功响应
  130 + when(restTemplate.exchange(anyString(), any(HttpMethod.class), any(), any(Class.class)))
  131 + .thenReturn(new ResponseEntity<>("Success", HttpStatus.OK));
148 132
149 - // 执行测试 133 + // 2. 创建配置
  134 + IotDataBridgeHttpConfig config = new IotDataBridgeHttpConfig()
  135 + .setUrl("https://doc.iocoder.cn/")
  136 + .setMethod(HttpMethod.GET.name());
  137 +
  138 + // 3. 执行测试
150 log.info("[testHttpDataBridge][执行HTTP数据桥接测试]"); 139 log.info("[testHttpDataBridge][执行HTTP数据桥接测试]");
151 - httpDataBridgeExecute.execute(message, new IotDataBridgeDO().setType(httpDataBridgeExecute.getType()).setConfig(config)); 140 + httpDataBridgeExecute.execute(message, new IotDataBridgeDO()
  141 + .setType(httpDataBridgeExecute.getType())
  142 + .setConfig(config));
  143 + }
  144 +
  145 + /**
  146 + * 执行测试并验证缓存的通用方法
  147 + *
  148 + * @param action 执行器实例
  149 + * @param config 配置对象
  150 + * @param mqType MQ类型
  151 + * @throws Exception 如果执行过程中发生异常
  152 + */
  153 + private void executeAndVerifyCache(IotDataBridgeExecute<?> action, IotDataBridgeAbstractConfig config, String mqType) throws Exception {
  154 + log.info("[test{}DataBridge][第一次执行，应该会创建新的 producer]", mqType);
  155 + action.execute(message, new IotDataBridgeDO()
  156 + .setType(action.getType())
  157 + .setConfig(config));
  158 +
  159 + log.info("[test{}DataBridge][第二次执行，应该会复用缓存的 producer]", mqType);
  160 + action.execute(message, new IotDataBridgeDO()
  161 + .setType(action.getType())
  162 + .setConfig(config));
152 } 163 }
153 164
154 } 165 }