spring MockMvc集成测试-模拟Azure ShareFileClient和MockFileInputStream无限期挂起

30byixjq  于 2024-01-05  发布在  Spring
关注(0)|答案(1)|浏览(136)

我在这个测试中遇到了问题，这个端点会挂起（也尝试了WebTestClient，它会抛出超时，所以这个问题与mock mvc无关），在返回被调用后，改变了一些配置，现在我得到了HttpMessageNotableException。
我几乎可以肯定它是处理模拟InputStreamResource类型的东西。
我已经创建了最小的设置来重现此错误。
下面是测试类：

  1. import static org.mockito.Mockito.anyString;
  2. import static org.mockito.Mockito.when;
  3. import static org.springframework.test.web.servlet.result.MockMvcResultHandlers.print;
  4. import static org.springframework.test.web.servlet.result.MockMvcResultMatchers.status;
  6. import java.io.File;
  7. import java.io.FileInputStream;
  8. import org.junit.jupiter.api.BeforeEach;
  9. import org.junit.jupiter.api.DisplayName;
  10. import org.junit.jupiter.api.Test;
  11. import org.junit.jupiter.api.extension.ExtendWith;
  12. import org.mockito.junit.jupiter.MockitoExtension;
  13. import org.msf.sits.health.mail.MailService;
  14. import org.msf.sits.health.share.FileShareService;
  15. import org.msf.sits.health.share.FileDownloadInfo;
  16. import org.springframework.beans.factory.annotation.Autowired;
  17. import org.springframework.boot.test.autoconfigure.web.servlet.AutoConfigureMockMvc;
  18. import org.springframework.boot.test.autoconfigure.web.servlet.AutoConfigureWebMvc;
  19. import org.springframework.boot.test.context.SpringBootTest;
  20. import org.springframework.boot.test.mock.mockito.MockBean;
  21. import org.springframework.core.io.InputStreamResource;
  22. import org.springframework.http.MediaType;
  23. import org.springframework.http.converter.HttpMessageConverter;
  24. import org.springframework.test.web.servlet.MockMvc;
  25. import org.springframework.test.web.servlet.MvcResult;
  26. import org.springframework.test.web.servlet.request.MockMvcRequestBuilders;
  27. import org.springframework.test.web.servlet.setup.MockMvcBuilders;
  29. @SpringBootTest(webEnvironment = SpringBootTest.WebEnvironment.RANDOM_PORT)
  30. @AutoConfigureMockMvc
  31. @AutoConfigureWebMvc
  32. @ExtendWith(MockitoExtension.class)
  33. public class FileDownloadControllerTest extends TestcontainersInitializer{
  35.   @MockBean
  36.   private MailService mailService;
  37.   
  38.   @MockBean
  39.   private FileShareService fileShareService;
  41.   @SuppressWarnings("SpringJavaInjectionPointsAutowiringInspection")
  42.   @Autowired
  43.   private MockMvc mockMvc;
  45.   @Autowired
  46.   private HttpMessageConverter[] httpMessageConverters;
  48.   @Autowired
  49.   private FilesDownloadController controller;
  51.   @BeforeEach
  52.   void setUp() {
  53.     this.mockMvc = MockMvcBuilders.standaloneSetup(controller)
  54.         .setMessageConverters(httpMessageConverters).build();
  55.   }
  57.   @Test
  58.   @DisplayName("200 - GET /downloadz/{token}")
  59.   void downloadzFileTest() throws Exception {
  60.     var token = "GbdARtD594xVpQQqOHAg0cOQSvx5LKGD";
  62.     File testFile = new File("src/test/resources", "integration/mocked-file.txt");
  63.     InputStreamResource testInputStream = new InputStreamResource(new FileInputStream(
  64.         testFile));
  65.     when(fileShareService.getFileDownloadInfo(anyString(), anyString())).thenReturn(
  66.         new FileDownloadInfo("mocked-file.txt", testInputStream.contentLength(),
  67.             testInputStream)
  68.     );
  70.     MvcResult result = null;
  71.     try {
  72.       result = mockMvc.perform(MockMvcRequestBuilders
  73.               .get("/downloadz/{token}", token)
  74.               .contentType(MediaType.APPLICATION_OCTET_STREAM)
  75.               .accept(MediaType.APPLICATION_OCTET_STREAM))
  76.           .andDo(print())
  77.           .andExpect(status().isOk()).andReturn();
  78.     } catch (Exception e) {
  79.       System.out.println("Exception: " + e.getMessage());
  80.     }
  81.   }
  82. }

字符串
这里是控制器：

  1. import jakarta.servlet.http.HttpServletRequest;
  2. import lombok.RequiredArgsConstructor;
  3. import org.msf.sits.health.share.FileShareService;
  4. import org.msf.sits.health.share.MedicalFileDownloadInfo;
  5. import org.springframework.core.io.InputStreamResource;
  6. import org.springframework.http.HttpHeaders;
  7. import org.springframework.http.MediaType;
  8. import org.springframework.http.ResponseEntity;
  9. import org.springframework.web.bind.annotation.GetMapping;
  10. import org.springframework.web.bind.annotation.PathVariable;
  11. import org.springframework.web.bind.annotation.RestController;
  13. @RestController
  14. @RequiredArgsConstructor
  15. public class FilesDownloadController {
  17.   private final FileShareService fileShareService;
  19.   @GetMapping("/downloadz/{token}")
  20.   public ResponseEntity<InputStreamResource> downloadFile(@PathVariable String token,
  21.       HttpServletRequest request) {
  22.     MedicalFileDownloadInfo medicalFileDownloadInfo = fileShareService.getMedicalFileDownloadInfo(
  23.         token, request.getRemoteAddr());
  25.     HttpHeaders headers = new HttpHeaders();
  26.     headers.add(HttpHeaders.CONTENT_DISPOSITION,
  27.         "attachment; filename=" + medicalFileDownloadInfo.getFileName());
  29.     var response = ResponseEntity.ok()
  30.         .headers(headers)
  31.         .contentLength(medicalFileDownloadInfo.getContentLength())
  32.         .contentType(MediaType.APPLICATION_OCTET_STREAM)
  33.         .body(medicalFileDownloadInfo.getContent());
  35.     return response;
  36.   }
  37. }

型
POM.xml:

  1. <?xml version="1.0" encoding="UTF-8"?>
  2. <project xmlns="http://maven.apache.org/POM/4.0.0" xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance"
  3.          xsi:schemaLocation="http://maven.apache.org/POM/4.0.0 https://maven.apache.org/xsd/maven-4.0.0.xsd">
  4.     <modelVersion>4.0.0</modelVersion>
  5.     <parent>
  6.         <groupId>org.springframework.boot</groupId>
  7.         <artifactId>spring-boot-starter-parent</artifactId>
  8.         <version>3.1.6</version>
  9.         <relativePath/> <!-- lookup parent from repository -->
  10.     </parent>
  11.     <groupId>org.msf.sits</groupId>
  12.     <artifactId>health</artifactId>
  13.     <version>0.0.1-SNAPSHOT</version>
  14.     <packaging>war</packaging>
  15.     <name>health</name>
  16.     <description>health</description>
  17.     <properties>
  18.         <java.version>17</java.version>
  19.         <spring-cloud-azure.version>5.5.0</spring-cloud-azure.version>
  20.         <vaadin.version>24.1.7</vaadin.version>
  21.         <tikka.version>2.9.1</tikka.version>
  22.         <springdoc-openapi-starter-webmvc-ui.version>2.0.2</springdoc-openapi-starter-webmvc-ui.version>
  23.         <spring-boot.testcontainers.version>3.2.0-M3</spring-boot.testcontainers.version>
  24.         <testcontainers.version>1.19.3</testcontainers.version>
  25.     </properties>
  26.     <dependencies>
  27.         <dependency>
  28.             <groupId>org.springframework.boot</groupId>
  29.             <artifactId>spring-boot-starter-actuator</artifactId>
  30.         </dependency>
  31.         <dependency>
  32.             <groupId>com.azure.spring</groupId>
  33.             <artifactId>spring-cloud-azure-starter-actuator</artifactId>
  34.         </dependency>
  35.         <dependency>
  36.             <groupId>org.springframework.boot</groupId>
  37.             <artifactId>spring-boot-starter-oauth2-client</artifactId>
  38.         </dependency>
  39.         <dependency>
  40.             <groupId>org.springframework.boot</groupId>
  41.             <artifactId>spring-boot-starter-data-jpa</artifactId>
  42.         </dependency>
  43.         <dependency>
  44.             <groupId>org.flywaydb</groupId>
  45.             <artifactId>flyway-core</artifactId>
  46.         </dependency>
  47.         <dependency>
  48.             <groupId>org.postgresql</groupId>
  49.             <artifactId>postgresql</artifactId>
  50.             <scope>runtime</scope>
  51.         </dependency>
  52.         <dependency>
  53.             <groupId>org.springframework.boot</groupId>
  54.             <artifactId>spring-boot-starter-oauth2-resource-server</artifactId>
  55.         </dependency>
  56.         <dependency>
  57.             <groupId>org.springframework.boot</groupId>
  58.             <artifactId>spring-boot-starter-security</artifactId>
  59.         </dependency>
  60.         <dependency>
  61.             <groupId>com.azure.spring</groupId>
  62.             <artifactId>spring-cloud-azure-starter-active-directory</artifactId>
  63.         </dependency>
  64.         <dependency>
  65.             <groupId>com.azure.spring</groupId>
  66.             <artifactId>spring-cloud-azure-starter-keyvault-secrets</artifactId>
  67.         </dependency>
  68.         <dependency>
  69.             <groupId>com.azure.spring</groupId>
  70.             <artifactId>spring-cloud-azure-starter-storage-file-share</artifactId>
  71.         </dependency>
  72.         <dependency>
  73.             <groupId>com.vaadin</groupId>
  74.             <artifactId>vaadin-spring-boot-starter</artifactId>
  75.         </dependency>
  76.         <dependency>
  77.             <groupId>com.fasterxml.jackson.core</groupId>
  78.             <artifactId>jackson-databind</artifactId>
  79.         </dependency>
  80.         <dependency>
  81.             <groupId>com.sendgrid</groupId>
  82.             <artifactId>sendgrid-java</artifactId>
  83.             <version>4.9.3</version>
  84.         </dependency>
  85.         <dependency>
  86.             <groupId>org.springframework.boot</groupId>
  87.             <artifactId>spring-boot-devtools</artifactId>
  88.             <scope>runtime</scope>
  89.             <optional>true</optional>
  90.         </dependency>
  91.         <dependency>
  92.             <groupId>org.springdoc</groupId>
  93.             <artifactId>springdoc-openapi-starter-webmvc-ui</artifactId>
  94.             <version>${springdoc-openapi-starter-webmvc-ui.version}</version>
  95.         </dependency>
  96.         <dependency>
  97.             <groupId>org.apache.tika</groupId>
  98.             <artifactId>tika-core</artifactId>
  99.             <version>${tikka.version}</version>
  100.         </dependency>
  101.         <dependency>
  102.             <groupId>org.projectlombok</groupId>
  103.             <artifactId>lombok</artifactId>
  104.             <optional>true</optional>
  105.         </dependency>
  106.         <dependency>
  107.             <groupId>org.springframework.boot</groupId>
  108.             <artifactId>spring-boot-starter-tomcat</artifactId>
  109.             <scope>provided</scope>
  110.         </dependency>
  112.         <!-- test dependencies -->
  113.         <dependency>
  114.             <groupId>org.springframework.boot</groupId>
  115.             <artifactId>spring-boot-starter-test</artifactId>
  116.             <scope>test</scope>
  117.         </dependency>
  118.         <dependency>
  119.             <groupId>org.mockito</groupId>
  120.             <artifactId>mockito-inline</artifactId>
  121.             <version>5.2.0</version>
  122.             <scope>test</scope>
  123.         </dependency>
  124.         <dependency>
  125.             <groupId>org.springframework.boot</groupId>
  126.             <artifactId>spring-boot-testcontainers</artifactId>
  127.             <version>${spring-boot.testcontainers.version}</version>
  128.             <scope>test</scope>
  129.         </dependency>
  130.         <dependency>
  131.             <groupId>org.junit.jupiter</groupId>
  132.             <artifactId>junit-jupiter</artifactId>
  133.             <version>5.9.3</version>
  134.             <scope>test</scope>
  135.         </dependency>
  136.         <dependency>
  137.             <groupId>org.testcontainers</groupId>
  138.             <artifactId>postgresql</artifactId>
  139.             <version>${testcontainers.version}</version>
  140.             <scope>test</scope>
  141.         </dependency>
  142.         <dependency>
  143.             <groupId>org.testcontainers</groupId>
  144.             <artifactId>testcontainers</artifactId>
  145.             <version>1.19.3</version>
  146.             <scope>test</scope>
  147.         </dependency>
  148.       <dependency>
  149.         <groupId>org.mockito</groupId>
  150.         <artifactId>mockito-core</artifactId>
  151.         <version>5.8.0</version>
  152.       </dependency>
  153.       <dependency>
  154.         <groupId>org.springframework.boot</groupId>
  155.         <artifactId>spring-boot-starter-webflux</artifactId>
  156.         <scope>test</scope>
  157.       </dependency>
  158.       <!--  test dependencies -->
  159.     </dependencies>
  160.     <dependencyManagement>
  161.         <dependencies>
  162.             <dependency>
  163.                 <groupId>com.azure.spring</groupId>
  164.                 <artifactId>spring-cloud-azure-dependencies</artifactId>
  165.                 <version>${spring-cloud-azure.version}</version>
  166.                 <type>pom</type>
  167.                 <scope>import</scope>
  168.             </dependency>
  169.             <dependency>
  170.                 <groupId>com.vaadin</groupId>
  171.                 <artifactId>vaadin-bom</artifactId>
  172.                 <version>${vaadin.version}</version>
  173.                 <type>pom</type>
  174.                 <scope>import</scope>
  175.             </dependency>
  176.         </dependencies>
  177.     </dependencyManagement>
  179.     <build>
  180.         <plugins>
  181.             <plugin>
  182.                 <groupId>org.springframework.boot</groupId>
  183.                 <artifactId>spring-boot-maven-plugin</artifactId>
  184.                 <configuration>
  185.                     <excludes>
  186.                         <exclude>
  187.                             <groupId>org.projectlombok</groupId>
  188.                             <artifactId>lombok</artifactId>
  189.                         </exclude>
  190.                     </excludes>
  191.                 </configuration>
  192.             </plugin>
  193.             <plugin>
  194.                 <groupId>com.microsoft.azure</groupId>
  195.                 <artifactId>azure-webapp-maven-plugin</artifactId>
  196.                 <version>2.12.0</version>
  197.                 <configuration>
  198.                     <schemaVersion>v2</schemaVersion>
  199.                     <subscriptionId>${env.SUBSCRIPTION_ID}</subscriptionId>
  200.                     <resourceGroup>${env.RESOURCE_GROUP}</resourceGroup>
  201.                     <appName>${env.APP_NAME}</appName>
  202.                     <pricingTier>${env.TIER}</pricingTier>
  203.                     <region>${env.REGION}</region>
  204.                     <runtime>
  205.                         <os>Linux</os>
  206.                         <javaVersion>Java 17</javaVersion>
  207.                         <webContainer>Tomcat 10.0</webContainer>
  208.                     </runtime>
  209.                     <deployment>
  210.                         <resources>
  211.                             <resource>
  212.                                 <directory>${project.basedir}/target</directory>
  213.                                 <includes>
  214.                                     <include>*.war</include>
  215.                                 </includes>
  216.                             </resource>
  217.                         </resources>
  218.                     </deployment>
  219.                 </configuration>
  220.             </plugin>
  221.         </plugins>
  222.     </build>
  224.     <reporting>
  225.         <plugins>
  226.             <plugin>
  227.                 <groupId>org.apache.maven.plugins</groupId>
  228.                 <artifactId>maven-jxr-plugin</artifactId>
  229.                 <version>3.3.0</version>
  230.             </plugin>
  231.             <plugin>
  232.                 <groupId>org.apache.maven.plugins</groupId>
  233.                 <artifactId>maven-checkstyle-plugin</artifactId>
  234.                 <version>3.3.0</version>
  235.                 <configuration>
  236.                     <configLocation>sun_checks.xml</configLocation>
  237.                 </configuration>
  238.                 <reportSets>
  239.                     <reportSet>
  240.                         <reports>
  241.                             <report>checkstyle</report>
  242.                         </reports>
  243.                     </reportSet>
  244.                 </reportSets>
  245.             </plugin>
  246.             <plugin>
  247.                 <groupId>org.apache.maven.plugins</groupId>
  248.                 <artifactId>maven-pmd-plugin</artifactId>
  249.                 <version>3.21.0</version>
  250.             </plugin>
  251.         </plugins>
  252.     </reporting>
  254.     <profiles>
  255.         <profile>
  256.             <id>production</id>
  257.             <build>
  258.                 <plugins>
  259.                     <plugin>
  260.                         <groupId>com.vaadin</groupId>
  261.                         <artifactId>vaadin-maven-plugin</artifactId>
  262.                         <version>${vaadin.version}</version>
  263.                         <executions>
  264.                             <execution>
  265.                                 <id>frontend</id>
  266.                                 <phase>compile</phase>
  267.                                 <goals>
  268.                                     <goal>prepare-frontend</goal>
  269.                                     <goal>build-frontend</goal>
  270.                                 </goals>
  271.                             </execution>
  272.                         </executions>
  273.                     </plugin>
  274.                 </plugins>
  275.             </build>
  276.         </profile>
  277.     </profiles>
  279.     <repositories>
  280.         <repository>
  281.             <id>spring-milestones</id>
  282.             <name>Spring Milestones</name>
  283.             <url>https://repo.spring.io/milestone</url>
  284.         </repository>
  285.     </repositories>
  287. </project>

型

spring

1条答案

按热度按时间
v09wglhw

v09wglhw1#

为了解决由流mock引起的无限挂起的问题，我必须以下面的方式模拟storageFileInputStream及其读取方法：
设置模拟的代码段：

  1. ShareFileClient shareFileClient = Mockito.mock(ShareFileClient.class,
  2.         Mockito.RETURNS_DEEP_STUBS);
  3.     StorageFileInputStream storageFileInputStream = Mockito.mock(StorageFileInputStream.class);
  5.     when(shareClient.getFileClient(anyString())).thenReturn(shareFileClient);
  6.     when(shareFileClient.openInputStream()).thenReturn(storageFileInputStream);
  8.     byte[] mockData = "Mocked File data!".getBytes();
  9.     mockStorageFileInputStreamRead(storageFileInputStream, mockData);

字符串
使用read() mock实现的方法：

  1. private static void mockStorageFileInputStreamRead(StorageFileInputStream storageFileInputStream,
  2.       byte[] mockData)
  3.       throws IOException {
  4.     AtomicInteger bytesRead = new AtomicInteger(0);
  6.     when(storageFileInputStream.read(any(byte[].class), anyInt(), anyInt())).thenAnswer(
  7.         invocation -> {
  8.           Object[] args = invocation.getArguments();
  9.           byte[] output = (byte[]) args[0];
  10.           int offset = (int) args[1];
  11.           int len = (int) args[2];
  13.           if (bytesRead.get() >= mockData.length) {
  14.             return -1;
  15.           }
  17.           int bytesToRead = Math.min(mockData.length - bytesRead.get(), len);
  18.           System.arraycopy(mockData, bytesRead.get(), output, offset, bytesToRead);
  19.           bytesRead.addAndGet(bytesToRead);
  21.           return bytesToRead;
  22.         });
  23.   }

型

展开查看全部
赞(0）分享  回复(0）举报  2024-01-05
我来回答

相关问题

    查看更多

    热门标签

    更多
    JavaquerypythonNode开发语言requestUtil数据库Table后端算法LoggerMessageElementParser

    最新问答

    更多
    • 蜀ICP备13028337号-1 大数据知识库 https://www.saoniuhuo.com © All rights reserved
    • 本站内容来源互联网,如果侵犯您的权益请联系我们删除, 联系方式：448109455@qq.com