|
| 1 | +package com.didispace; |
| 2 | + |
| 3 | +import com.didispace.domain.User; |
| 4 | +import com.didispace.domain.UserRepository; |
| 5 | +import org.junit.Assert; |
| 6 | +import org.junit.Before; |
| 7 | +import org.junit.Test; |
| 8 | +import org.junit.runner.RunWith; |
| 9 | +import org.springframework.beans.factory.annotation.Autowired; |
| 10 | +import org.springframework.boot.test.SpringApplicationConfiguration; |
| 11 | +import org.springframework.test.context.junit4.SpringJUnit4ClassRunner; |
| 12 | + |
| 13 | + |
| 14 | +@RunWith(SpringJUnit4ClassRunner.class) |
| 15 | +@SpringApplicationConfiguration(Application.class) |
| 16 | +public class ApplicationTests { |
| 17 | + |
| 18 | + @Autowired |
| 19 | + private UserRepository userRepository; |
| 20 | + |
| 21 | + @Before |
| 22 | + public void setUp() { |
| 23 | + userRepository.deleteAll(); |
| 24 | + } |
| 25 | + |
| 26 | + @Test |
| 27 | + public void test() throws Exception { |
| 28 | + |
| 29 | + // 创建三个User,并验证User总数 |
| 30 | + userRepository.save(new User(1L, "didi", 30)); |
| 31 | + userRepository.save(new User(2L, "mama", 40)); |
| 32 | + userRepository.save(new User(3L, "kaka", 50)); |
| 33 | + Assert.assertEquals(3, userRepository.findAll().size()); |
| 34 | + |
| 35 | + // 删除一个User,再验证User总数 |
| 36 | + User u = userRepository.findOne(1L); |
| 37 | + userRepository.delete(u); |
| 38 | + Assert.assertEquals(2, userRepository.findAll().size()); |
| 39 | + |
| 40 | + // 删除一个User,再验证User总数 |
| 41 | + u = userRepository.findByUsername("mama"); |
| 42 | + userRepository.delete(u); |
| 43 | + Assert.assertEquals(1, userRepository.findAll().size()); |
| 44 | + |
| 45 | + } |
| 46 | + |
| 47 | +} |
0 commit comments