JUnit测试异步方法问题

爱被打了一巴掌 2021-12-21 11:11 345阅读 0赞

参考案例：[https://stackoverflow.com/questions/631598/how-to-use-junit-to-test-asynchronous-processes][https_stackoverflow.com_questions_631598_how-to-use-junit-to-test-asynchronous-processes]

通过注解@Async的异步方法，在Junit5测试时，数据一直保存不到mongodb数据库中

@Async
    	public CompletableFuture<Integer> addUsers(List<User> users) {
    		Collection<User> result = mongoTemplate.insertAll(users);
    		return CompletableFuture.completedFuture(result.size());
    	}
    
    	@Async
    	public CompletableFuture<Boolean> saveUser(User user) {
    		User result = mongoTemplate.save(user);
    		return CompletableFuture.completedFuture(result != null);
    	}

原测试方法

@Test
        void testAddUsers() {
            List<User> users = new ArrayList<>();
            users.add(new User(UUID.randomUUID().toString(), "55", "zhangsan", "1234"));
            users.add(new User(UUID.randomUUID().toString(), "77", "zhangsan", "1234"));
            mongoRepository.addUsers(users);
        }

原测试结果

![watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3pzeDE4MjczMTE3MDAz_size_16_color_FFFFFF_t_70][]

数据库空空如也

![20190627171245177.png][]

修改后方法

@Test
        void testAddUsers() throws ExecutionException, InterruptedException {
            List<User> users = new ArrayList<>();
            users.add(new User(UUID.randomUUID().toString(), "55", "zhangsan", "1234"));
            users.add(new User(UUID.randomUUID().toString(), "77", "zhangsan", "1234"));
            CompletableFuture<Integer> future = mongoRepository.addUsers(users);
            assertEquals("2", String.valueOf(future.get()));
        }

再进行测试结果

![watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3pzeDE4MjczMTE3MDAz_size_16_color_FFFFFF_t_70 1][]

数据库结果

![20190627171443487.png][]

[https_stackoverflow.com_questions_631598_how-to-use-junit-to-test-asynchronous-processes]: https://stackoverflow.com/questions/631598/how-to-use-junit-to-test-asynchronous-processes
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3pzeDE4MjczMTE3MDAz_size_16_color_FFFFFF_t_70]: /images/20211221/b250a5ed867e400997f0de09b21b3517.png
[20190627171245177.png]: /images/20211221/449acca672fb45e5bb24137b69b226c4.png
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3pzeDE4MjczMTE3MDAz_size_16_color_FFFFFF_t_70 1]: /images/20211221/b9d9d16a741146abb25733f1302f3e0f.png
[20190627171443487.png]: /images/20211221/2f76e5621b204f8ca9c8515584e347f2.png