diff --git a/src/test/java/org/ntlab/tampoposerver/repositories/FriendRepositoryTest.java b/src/test/java/org/ntlab/tampoposerver/repositories/FriendRepositoryTest.java new file mode 100644 index 0000000..dc74bd1 --- /dev/null +++ b/src/test/java/org/ntlab/tampoposerver/repositories/FriendRepositoryTest.java @@ -0,0 +1,49 @@ +package org.ntlab.tampoposerver.repositories; + +import org.junit.jupiter.api.Test; +import org.ntlab.tampoposerver.models.FriendPair; +import org.springframework.beans.factory.annotation.Autowired; +import org.springframework.boot.test.context.SpringBootTest; + +import static org.assertj.core.api.Assertions.assertThat; +import static org.junit.jupiter.api.Assertions.assertEquals; +import static org.junit.jupiter.api.Assertions.assertNotEquals; + +@SpringBootTest +public class FriendRepositoryTest { + @Autowired + private FriendRepository friendRepository; + + @Test + public void contextLoads() { + assertThat(friendRepository).isNotNull(); + } + + @Test + public void test() { + // Adding friend pair + FriendPair pair = new FriendPair("user-0", "user-1"); + FriendPair addedPair = friendRepository.add(pair); + assertEquals(0, addedPair.getId()); + assertEquals("user-0", addedPair.getUser0Id()); + assertEquals("user-1", addedPair.getUser1Id()); + + FriendPair anotherPair = new FriendPair("another-user-0", "another-user-1"); + FriendPair anotherAddedPair = friendRepository.add(anotherPair); + assertEquals(1, anotherAddedPair.getId()); + assertEquals("another-user-0", anotherAddedPair.getUser0Id()); + assertEquals("another-user-1", anotherAddedPair.getUser1Id()); + + FriendPair wrongIdPair = new FriendPair("wrong-user-0", "wrong-user-1"); + FriendPair addedWrongIdPair = friendRepository.add(wrongIdPair); + assertNotEquals(10, addedWrongIdPair.getId()); + assertNotEquals("wrong-user-10", addedWrongIdPair.getUser0Id()); + assertNotEquals("wrong-user-11", addedWrongIdPair.getUser1Id()); + + // Get friends + FriendPair foundPair = friendRepository.find(0); + assertEquals(0, foundPair.getId()); + assertEquals("user-0", foundPair.getUser0Id()); + assertEquals("user-1", foundPair.getUser1Id()); + } +}