WebFlux 集成测试及部署

Author: liqiangqiang

springboot-webflux-9-test/pom.xml

@@ -0,0 +1,53 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<project xmlns="http://maven.apache.org/POM/4.0.0" xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance"
+         xsi:schemaLocation="http://maven.apache.org/POM/4.0.0 http://maven.apache.org/xsd/maven-4.0.0.xsd">
+  <modelVersion>4.0.0</modelVersion>
+
+  <groupId>springboot</groupId>
+  <artifactId>springboot-webflux-9-test</artifactId>
+  <version>0.0.1-SNAPSHOT</version>
+  <name>springboot-webflux-9-test :: Spring Boot WebFlux 集成测试及部署</name>
+
+  <!-- Spring Boot 启动父依赖 -->
+  <parent>
+    <groupId>org.springframework.boot</groupId>
+    <artifactId>spring-boot-starter-parent</artifactId>
+    <version>2.0.1.RELEASE</version>
+  </parent>
+
+  <dependencies>
+
+    <!-- Spring Boot Web Flux 依赖 -->
+    <dependency>
+      <groupId>org.springframework.boot</groupId>
+      <artifactId>spring-boot-starter-webflux</artifactId>
+    </dependency>
+
+    <!-- ❤️Spring Boot 响应式 MongoDB 支持 -->
+    <dependency>
+      <groupId>org.springframework.boot</groupId>
+      <artifactId>spring-boot-starter-data-mongodb-reactive</artifactId>
+    </dependency>
+
+    <!-- Spring Boot Test 依赖 -->
+    <dependency>
+      <groupId>org.springframework.boot</groupId>
+      <artifactId>spring-boot-starter-test</artifactId>
+      <scope>test</scope>
+    </dependency>
+
+    <dependency>
+      <groupId>io.projectreactor</groupId>
+      <artifactId>reactor-test</artifactId>
+      <scope>test</scope>
+    </dependency>
+
+    <!-- Junit -->
+    <dependency>
+      <groupId>junit</groupId>
+      <artifactId>junit</artifactId>
+      <version>4.12</version>
+    </dependency>
+  </dependencies>
+
+</project>

springboot-webflux-9-test/src/main/java/org/spring/springboot/Application.java

@@ -0,0 +1,20 @@
+package org.spring.springboot;
+
+import org.springframework.boot.SpringApplication;
+import org.springframework.boot.autoconfigure.SpringBootApplication;
+
+/**
+ * Spring Boot 应用启动类
+ *
+ * Created by bysocket on 09/29/2017.
+ */
+// Spring Boot 应用的标识
+@SpringBootApplication
+public class Application {
+
+    public static void main(String[] args) {
+        // 程序启动入口
+        // 启动嵌入式的 Tomcat 并初始化 Spring 环境及其各 Spring 组件
+        SpringApplication.run(Application.class,args);
+    }
+}

springboot-webflux-9-test/src/main/java/org/spring/springboot/dao/CityRepository.java

@@ -0,0 +1,10 @@
+package org.spring.springboot.dao;
+
+import org.spring.springboot.domain.City;
+import org.springframework.data.mongodb.repository.ReactiveMongoRepository;
+import org.springframework.stereotype.Repository;
+
+@Repository
+public interface CityRepository extends ReactiveMongoRepository<City, Long> {
+
+}

springboot-webflux-9-test/src/main/java/org/spring/springboot/domain/City.java

@@ -0,0 +1,63 @@
+package org.spring.springboot.domain;
+
+import org.springframework.data.annotation.Id;
+
+/**
+ * 城市实体类
+ *
+ */
+public class City {
+
+    /**
+     * 城市编号
+     */
+    @Id
+    private Long id;
+
+    /**
+     * 省份编号
+     */
+    private Long provinceId;
+
+    /**
+     * 城市名称
+     */
+    private String cityName;
+
+    /**
+     * 描述
+     */
+    private String description;
+
+    public Long getId() {
+        return id;
+    }
+
+    public void setId(Long id) {
+        this.id = id;
+    }
+
+    public Long getProvinceId() {
+        return provinceId;
+    }
+
+    public void setProvinceId(Long provinceId) {
+        this.provinceId = provinceId;
+    }
+
+    public String getCityName() {
+        return cityName;
+    }
+
+    public void setCityName(String cityName) {
+        this.cityName = cityName;
+    }
+
+    public String getDescription() {
+        return description;
+    }
+
+    public void setDescription(String description) {
+        this.description = description;
+    }
+}

springboot-webflux-9-test/src/main/java/org/spring/springboot/handler/CityHandler.java

@@ -0,0 +1,43 @@
+package org.spring.springboot.handler;
+
+import org.spring.springboot.dao.CityRepository;
+import org.spring.springboot.domain.City;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.stereotype.Component;
+import reactor.core.publisher.Flux;
+import reactor.core.publisher.Mono;
+
+@Component
+public class CityHandler {
+
+    private final CityRepository cityRepository;
+
+    @Autowired
+    public CityHandler(CityRepository cityRepository) {
+        this.cityRepository = cityRepository;
+    }
+
+    public Mono<City> save(City city) {
+        return cityRepository.save(city);
+    }
+
+    public Mono<City> findCityById(Long id) {
+
+        return cityRepository.findById(id);
+    }
+
+    public Flux<City> findAllCity() {
+
+        return cityRepository.findAll();
+    }
+
+    public Mono<City> modifyCity(City city) {
+
+        return cityRepository.save(city);
+    }
+
+    public Mono<Long> deleteCity(Long id) {
+        cityRepository.deleteById(id);
+        return Mono.create(cityMonoSink -> cityMonoSink.success(id));
+    }
+}

springboot-webflux-9-test/src/main/java/org/spring/springboot/webflux/controller/CityWebFluxController.java

@@ -0,0 +1,41 @@
+package org.spring.springboot.webflux.controller;
+
+import org.spring.springboot.domain.City;
+import org.spring.springboot.handler.CityHandler;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.web.bind.annotation.*;
+import reactor.core.publisher.Flux;
+import reactor.core.publisher.Mono;
+
+@RestController
+@RequestMapping(value = "/city")
+public class CityWebFluxController {
+
+    @Autowired
+    private CityHandler cityHandler;
+
+    @GetMapping(value = "/{id}")
+    public Mono<City> findCityById(@PathVariable("id") Long id) {
+        return cityHandler.findCityById(id);
+    }
+
+    @GetMapping()
+    public Flux<City> findAllCity() {
+        return cityHandler.findAllCity();
+    }
+
+    @PostMapping()
+    public Mono<City> saveCity(@RequestBody City city) {
+        return cityHandler.save(city);
+    }
+
+    @PutMapping()
+    public Mono<City> modifyCity(@RequestBody City city) {
+        return cityHandler.modifyCity(city);
+    }
+
+    @DeleteMapping(value = "/{id}")
+    public Mono<Long> deleteCity(@PathVariable("id") Long id) {
+        return cityHandler.deleteCity(id);
+    }
+}

springboot-webflux-9-test/src/main/resources/application.properties

@@ -0,0 +1,5 @@
+spring.data.mongodb.host=localhost
+spring.data.mongodb.database=admin
+spring.data.mongodb.port=27017
+spring.data.mongodb.username=admin
+spring.data.mongodb.password=admin

springboot-webflux-9-test/src/test/java/org/spring/springboot/handler/CityHandlerTest.java

@@ -0,0 +1,52 @@
+package org.spring.springboot.handler;
+
+import org.junit.Assert;
+import org.junit.BeforeClass;
+import org.junit.Test;
+import org.junit.runner.RunWith;
+import org.spring.springboot.domain.City;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.boot.test.context.SpringBootTest;
+import org.springframework.http.MediaType;
+import org.springframework.test.context.junit4.SpringRunner;
+import org.springframework.test.web.reactive.server.WebTestClient;
+import org.springframework.web.reactive.function.BodyInserters;
+
+import java.util.HashMap;
+import java.util.Map;
+
+@RunWith(SpringRunner.class)
+@SpringBootTest(webEnvironment = SpringBootTest.WebEnvironment.RANDOM_PORT)
+public class CityHandlerTest {
+
+    @Autowired
+    private WebTestClient webClient;
+
+    private static Map<String, City> cityMap = new HashMap<>();
+
+    @BeforeClass
+    public static void setup() throws Exception {
+        City wl = new City();
+        wl.setId(1L);
+        wl.setProvinceId(2L);
+        wl.setCityName("WL");
+        wl.setDescription("WL IS GOOD");
+        cityMap.put("WL", wl);
+    }
+
+    @Test
+    public void testSave() throws Exception {
+
+        City expectCity = webClient.post().uri("/city")
+                .contentType(MediaType.APPLICATION_JSON)
+                .body(BodyInserters.fromObject(cityMap.get("WL")))
+                .exchange()
+                .expectStatus().isOk()
+                .expectBody(City.class).returnResult().getResponseBody();
+
+        Assert.assertNotNull(expectCity);
+        Assert.assertEquals(expectCity.getId(), cityMap.get("WL").getId());
+        Assert.assertEquals(expectCity.getCityName(), cityMap.get("WL").getCityName());
+    }
+
+}