<artifactId>awaitility</artifactId>

本文探讨了awaitility库在Java测试中的使用,介绍了如何设置超时时间和间隔,如atMost()与pollInterval()。重点展示了如何确保异步任务完成并在特定时间内进行判断。
摘要由CSDN通过智能技术生成

<dependencies>
  <dependency>
    <groupId>org.awaitility</groupId>
    <artifactId>awaitility</artifactId>
    <version>4.0.2</version>
    <scope>test</scope>
  </dependency>
</dependencies>

import org.awaitility.Awaitility;

// ...

Awaitility.await().atMost(10, TimeUnit.SECONDS).until(() -> asyncTask.isDone());

import org.awaitility.Awaitility;

// ...

int value = 0;

Awaitility.await().atMost(10, TimeUnit.SECONDS).until(() -> value > 0);

import static java.util.concurrent.TimeUnit.SECONDS;

import org.awaitility.Awaitility;

// ...

Awaitility.await().atMost(10, SECONDS).pollInterval(1, SECONDS).until(() -> asyncTask.isDone());

Awaitility.await().atMost(10, SECONDS).pollInterval(1, SECONDS).pollDelay(2, SECONDS).until(() -> asyncTask.isDone());

package com.xymzsfxy.backend.service; import com.xymzsfxy.backend.entity.PriceHistory; import com.xymzsfxy.backend.entity.Product; import com.xymzsfxy.backend.repository.PriceHistoryRepository; import com.xymzsfxy.backend.repository.ProductRepository; import org.jsoup.Jsoup; import org.junit.jupiter.api.Test; import org.springframework.beans.factory.annotation.Autowired; import org.springframework.boot.test.context.SpringBootTest; import org.springframework.boot.test.context.TestConfiguration; import org.springframework.boot.test.mock.mockito.MockBean; import org.springframework.boot.test.mock.mockito.SpyBean; import org.springframework.context.annotation.Bean; import org.springframework.scheduling.annotation.EnableAsync; import org.springframework.test.context.TestPropertySource; import java.math.BigDecimal; import java.util.Optional; import java.util.concurrent.TimeUnit; import static org.awaitility.Awaitility.await; import static org.mockito.ArgumentMatchers.any; import static org.mockito.Mockito.*; // 测试类注解 @SpringBootTest @EnableAsync @TestPropertySource(properties = { "crawler.sources.source1=https://example.com/price?id=%s", "crawler.selectors.source1=div.price" }) class PriceCrawlerServiceTest { @Autowired private PriceCrawlerService priceCrawlerService; @MockBean private ProductRepository productRepository; @MockBean private PriceHistoryRepository priceHistoryRepository; @SpyBean private PriceCrawlerService spyPriceCrawlerService; @TestConfiguration static class TestConfig { @Bean public JsoupWrapper jsoupWrapper() { return new JsoupWrapper(); } } @Test void shouldSavePriceWhenCrawlSuccess() throws Exception { // 模拟测试数据 Product mockProduct = new Product(); mockProduct.setId(1L); mockProduct.setExternalId("P123"); // 配置Mock行为 when(productRepository.findById(1L)).thenReturn(Optional.of(mockProduct)); doReturn(new BigDecimal("299.00")).when(spyPriceCrawlerService).extractPrice(any(), anyString()); // 执行测试方法 priceCrawlerService.crawlPrices(1L); // 验证异步执行结果 await().atMost(5, TimeUnit.SECONDS).untilAsserted(() -> { verify(priceHistoryRepository, times(1)).save(any(PriceHistory.class)); verify(productRepository, times(1)).save(argThat(p -> p.getLatestPrice().equals(new BigDecimal("299.00")) )); }); } @Test void shouldHandleMissingProduct() { when(productRepository.findById(999L)).thenReturn(Optional.empty()); // 直接调用方法,若抛出异常测试会自动失败 priceCrawlerService.crawlPrices(999L); verify(priceHistoryRepository, never()).save(any()); } } // 辅助类用于模拟网页请求 class JsoupWrapper { Document fetchDocument(String url) throws IOException { // 模拟返回包含价格的HTML文档 String html = "<html><div class='price'>¥299.00</div></html>"; return Jsoup.parse(html); } }无法解析符号 'Document'
最新发布
03-09
package com.xymzsfxy.backend.service; import com.xymzsfxy.backend.entity.PriceHistory; import com.xymzsfxy.backend.entity.Product; import com.xymzsfxy.backend.repository.PriceHistoryRepository; import com.xymzsfxy.backend.repository.ProductRepository; import lombok.RequiredArgsConstructor; import lombok.extern.slf4j.Slf4j; import org.jsoup.Jsoup; import org.jsoup.nodes.Document; import org.springframework.beans.factory.annotation.Value; import org.springframework.scheduling.annotation.Async; import org.springframework.stereotype.Service; import org.springframework.transaction.annotation.Transactional; import java.io.IOException; import java.math.BigDecimal; import java.time.LocalDateTime; import java.util.Map; @Slf4j @Service @RequiredArgsConstructor public class PriceCrawlerService { private final ProductRepository productRepository; private final PriceHistoryRepository priceHistoryRepository; @Value(“#{${crawler.sources}}”) private Map<String, String> crawlerSources; @Value(“#{${crawler.selectors}}”) private Map<String, String> crawlerSelectors; private static final int TIMEOUT_MS = 10_000; private static final String USER_AGENT = “Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36”; @Async(“crawlerTaskExecutor”) public void crawlPrices(Long productId) { productRepository.findById(productId).ifPresentOrElse( product -> crawlerSources.forEach((source, url) -> processSource(product, source, url)), () -> log.warn(“商品不存在: ID={}”, productId) ); } private void processSource(Product product, String sourceName, String urlTemplate) { try { final String targetUrl = buildTargetUrl(product, urlTemplate); final Document document = fetchDocument(targetUrl); final BigDecimal price = extractPrice(document, sourceName); savePriceData(product, sourceName, price); log.debug("价格抓取成功: {} | {} | {}", sourceName, product.getId(), price); } catch (IOException e) { log.error("[{}] 网络请求异常: {}", sourceName, e.getMessage()); } catch (Exception e) { log.error("[{}] 数据处理异常: {}", sourceName, e.getMessage(), e); } } private String buildTargetUrl(Product product, String urlTemplate) { return String.format(urlTemplate, product.getExternalId()); } private Document fetchDocument(String url) throws IOException { return Jsoup.connect(url) .timeout(TIMEOUT_MS) .userAgent(USER_AGENT) .get(); } private BigDecimal extractPrice(Document document, String sourceName) { String selector = crawlerSelectors.get(sourceName); String priceText = document.selectFirst(selector).text(); return parsePrice(priceText); } private BigDecimal parsePrice(String priceText) { String numericString = priceText.replaceAll(“\d.,”, “”); return new BigDecimal(numericString.replace(“,”, “”)); } @Transactional protected void savePriceData(Product product, String source, BigDecimal price) { persistPriceHistory(product, source, price); updateProductPrice(product, price); } private void persistPriceHistory(Product product, String source, BigDecimal price) { PriceHistory history = PriceHistory.builder() .productId(product.getId()) .source(source) .price(price) .crawlTime(LocalDateTime.now()) .build(); priceHistoryRepository.save(history); } private void updateProductPrice(Product product, BigDecimal price) { product.setLatestPrice(price); product.setUpdatedTime(LocalDateTime.now()); productRepository.save(product); } }在SpringBootTest测试是否能自动获取的功能给出完整代码
03-09
评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

童心同萌

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值