81 lines
3.0 KiB
Java
81 lines
3.0 KiB
Java
package io.spring.infrastructure.user;
|
|
|
|
import io.spring.core.user.FollowRelation;
|
|
import io.spring.core.user.User;
|
|
import io.spring.core.user.UserRepository;
|
|
import io.spring.infrastructure.repository.MyBatisUserRepository;
|
|
import org.junit.Before;
|
|
import org.junit.Test;
|
|
import org.junit.runner.RunWith;
|
|
import org.mybatis.spring.boot.test.autoconfigure.MybatisTest;
|
|
import org.springframework.beans.factory.annotation.Autowired;
|
|
import org.springframework.context.annotation.Import;
|
|
import org.springframework.test.context.junit4.SpringRunner;
|
|
|
|
import java.util.Optional;
|
|
|
|
import static org.hamcrest.CoreMatchers.is;
|
|
import static org.junit.Assert.*;
|
|
|
|
@RunWith(SpringRunner.class)
|
|
@MybatisTest
|
|
@Import(MyBatisUserRepository.class)
|
|
public class MyBatisUserRepositoryTest {
|
|
@Autowired
|
|
private UserRepository userRepository;
|
|
private User user;
|
|
|
|
@Before
|
|
public void setUp() throws Exception {
|
|
user = new User("aisensiy@163.com", "aisensiy", "123", "", "default");
|
|
}
|
|
|
|
@Test
|
|
public void should_save_and_fetch_user_success() throws Exception {
|
|
userRepository.save(user);
|
|
Optional<User> userOptional = userRepository.findByUsername("aisensiy");
|
|
assertThat(userOptional.get(), is(user));
|
|
Optional<User> userOptional2 = userRepository.findByEmail("aisensiy@163.com");
|
|
assertThat(userOptional2.get(), is(user));
|
|
}
|
|
|
|
@Test
|
|
public void should_update_user_success() throws Exception {
|
|
String newEmail = "newemail@email.com";
|
|
user.update(newEmail, "", "", "", "");
|
|
userRepository.save(user);
|
|
Optional<User> optional = userRepository.findByUsername(user.getUsername());
|
|
assertThat(optional.isPresent(), is(true));
|
|
assertThat(optional.get().getEmail(), is(newEmail));
|
|
|
|
String newUsername = "newUsername";
|
|
user.update("", newUsername, "", "", "");
|
|
userRepository.save(user);
|
|
optional = userRepository.findByEmail(user.getEmail());
|
|
assertThat(optional.isPresent(), is(true));
|
|
assertThat(optional.get().getUsername(), is(newUsername));
|
|
assertThat(optional.get().getImage(), is(user.getImage()));
|
|
}
|
|
|
|
@Test
|
|
public void should_create_new_user_follow_success() throws Exception {
|
|
User other = new User("other@example.com", "other", "123", "", "");
|
|
userRepository.save(other);
|
|
|
|
FollowRelation followRelation = new FollowRelation(user.getId(), other.getId());
|
|
userRepository.saveRelation(followRelation);
|
|
assertThat(userRepository.findRelation(user.getId(), other.getId()).isPresent(), is(true));
|
|
}
|
|
|
|
@Test
|
|
public void should_unfollow_user_success() throws Exception {
|
|
User other = new User("other@example.com", "other", "123", "", "");
|
|
userRepository.save(other);
|
|
|
|
FollowRelation followRelation = new FollowRelation(user.getId(), other.getId());
|
|
userRepository.saveRelation(followRelation);
|
|
|
|
userRepository.removeRelation(followRelation);
|
|
assertThat(userRepository.findRelation(user.getId(), other.getId()).isPresent(), is(false));
|
|
}
|
|
} |