Implement basic tests for avatars

FAForever · May 23, 2017 · 4e7dcbf · 4e7dcbf
1 parent e916bcf
commit 4e7dcbf
Show file tree

Hide file tree

Showing 12 changed files with 214 additions and 22 deletions.
diff --git a/src/main/java/com/faforever/api/avatar/AvatarAssignmentRepository.java b/src/main/java/com/faforever/api/avatar/AvatarAssignmentRepository.java
@@ -0,0 +1,9 @@
+package com.faforever.api.avatar;
+
+import com.faforever.api.data.domain.AvatarAssignment;
+import org.springframework.data.jpa.repository.JpaRepository;
+import org.springframework.stereotype.Repository;
+
+@Repository
+public interface AvatarAssignmentRepository extends JpaRepository<AvatarAssignment, Integer> {
+}
diff --git a/src/main/java/com/faforever/api/avatar/AvatarRepository.java b/src/main/java/com/faforever/api/avatar/AvatarRepository.java
@@ -0,0 +1,9 @@
+package com.faforever.api.avatar;
+
+import com.faforever.api.data.domain.Avatar;
+import org.springframework.data.jpa.repository.JpaRepository;
+import org.springframework.stereotype.Repository;
+
+@Repository
+public interface AvatarRepository extends JpaRepository<Avatar, Integer> {
+}
diff --git a/src/main/java/com/faforever/api/data/domain/Avatar.java b/src/main/java/com/faforever/api/data/domain/Avatar.java
@@ -9,6 +9,7 @@
 import javax.persistence.Entity;
 import javax.persistence.OneToMany;
 import javax.persistence.Table;
+import javax.validation.constraints.NotNull;
 import java.util.List;
 
 @Entity
@@ -22,6 +23,7 @@ public class Avatar extends AbstractEntity {
   private List<AvatarAssignment> assignments;
 
   @Column(name = "url")
+  @NotNull
   public String getUrl() {
     return url;
   }

diff --git a/src/main/java/com/faforever/api/data/domain/AvatarAssignment.java b/src/main/java/com/faforever/api/data/domain/AvatarAssignment.java
@@ -9,6 +9,7 @@
 import javax.persistence.JoinColumn;
 import javax.persistence.ManyToOne;
 import javax.persistence.Table;
+import javax.validation.constraints.NotNull;
 import java.time.OffsetDateTime;
 
 @Entity
@@ -33,12 +34,14 @@ public OffsetDateTime getExpiresAt() {
 
   @ManyToOne(fetch = FetchType.LAZY)
   @JoinColumn(name = "idAvatar")
+  @NotNull
   public Avatar getAvatar() {
     return avatar;
   }
 
   @ManyToOne(fetch = FetchType.LAZY)
   @JoinColumn(name = "idUser")
+  @NotNull
   public Player getPlayer() {
     return player;
   }

diff --git a/src/test/java/com/faforever/api/clan/ClanControllerIntegrationTest.java b/src/test/java/com/faforever/api/clan/ClanControllerIntegrationTest.java
@@ -161,7 +161,7 @@ public void createSecondClan() throws Exception {
 
   @Test
   public void createClanWithSameName() throws Exception {
-    Player otherLeader = PlayerFactory.createPlayer("AnotherJunitUser", database);
+    Player otherLeader = PlayerFactory.builder().database(database).build();
     Session session = SessionFactory.createUserAndGetAccessToken(
       database, mvc);
     String clanName = "My Cool ClanName";
@@ -194,7 +194,7 @@ public void createClanWithSameName() throws Exception {
 
   @Test
   public void createClanWithSameTag() throws Exception {
-    Player otherLeader = PlayerFactory.createPlayer("AnotherJunitUser", database);
+    Player otherLeader = PlayerFactory.builder().database(database).build();
     Session session = SessionFactory.createUserAndGetAccessToken(
       database, mvc);
     String clanName = "My Cool ClanName";

diff --git a/src/test/java/com/faforever/api/data/JsonApiAvatarIntegrationTest.java b/src/test/java/com/faforever/api/data/JsonApiAvatarIntegrationTest.java
@@ -0,0 +1,102 @@
+package com.faforever.api.data;
+
+import com.faforever.api.data.domain.Avatar;
+import com.faforever.api.data.domain.Player;
+import com.faforever.integration.TestDatabase;
+import com.faforever.integration.factories.AvatarFactory;
+import com.faforever.integration.factories.PlayerFactory;
+import com.faforever.integration.utils.MockMvcHelper;
+import org.junit.After;
+import org.junit.Before;
+import org.junit.Test;
+import org.junit.runner.RunWith;
+import org.springframework.boot.test.context.SpringBootTest;
+import org.springframework.context.annotation.Import;
+import org.springframework.test.context.junit4.SpringRunner;
+import org.springframework.test.web.servlet.MockMvc;
+import org.springframework.test.web.servlet.setup.MockMvcBuilders;
+import org.springframework.web.context.WebApplicationContext;
+
+import javax.inject.Inject;
+import javax.servlet.Filter;
+
+import static org.hamcrest.Matchers.hasSize;
+import static org.hamcrest.Matchers.is;
+import static org.junit.Assert.assertEquals;
+import static org.springframework.test.web.servlet.request.MockMvcRequestBuilders.get;
+import static org.springframework.test.web.servlet.result.MockMvcResultMatchers.jsonPath;
+import static org.springframework.test.web.servlet.result.MockMvcResultMatchers.status;
+
+@RunWith(SpringRunner.class)
+@SpringBootTest
+@Import(TestDatabase.class)
+public class JsonApiAvatarIntegrationTest {
+
+  private MockMvc mvc;
+  private WebApplicationContext context;
+  private Filter springSecurityFilterChain;
+
+  private TestDatabase database;
+
+  @Inject
+  public void init(TestDatabase database, WebApplicationContext context, Filter springSecurityFilterChain) {
+    this.context = context;
+    this.springSecurityFilterChain = springSecurityFilterChain;
+    this.database = database;
+  }
+
+  @Before
+  public void setUp() {
+    mvc = MockMvcBuilders
+      .webAppContextSetup(context)
+      .addFilter(springSecurityFilterChain)
+      .build();
+    database.assertEmptyDatabase();
+  }
+
+  @After
+  public void tearDown() {
+    // TODO: This is needed, because Elide has some problems with @Transactional annotation #71
+    database.tearDown();
+  }
+
+
+  @Test
+  public void getSingleAvatar() throws Exception {
+    Avatar avatar = AvatarFactory.builder().database(database).build();
+
+    assertEquals(1, database.getAvatarRepository().count());
+
+    MockMvcHelper.of(this.mvc).perform(get("/data/avatar/"))
+      .andExpect(status().isOk())
+      .andExpect(jsonPath("$.data", hasSize(1)))
+      .andExpect(jsonPath("$.data[0].id", is(String.valueOf(avatar.getId()))))
+      .andExpect(jsonPath("$.data[0].type", is("avatar")))
+      .andExpect(jsonPath("$.data[0].attributes.tooltip", is(avatar.getTooltip())))
+      .andExpect(jsonPath("$.data[0].attributes.url", is(avatar.getUrl())))
+      .andExpect(jsonPath("$.data[0].relationships.assignments.data", hasSize(0)));
+    assertEquals(1, database.getAvatarRepository().count());
+  }
+
+  @Test
+  public void getSingleAvatarWithPlayer() throws Exception {
+    Player player = PlayerFactory.builder().database(database).build();
+    Avatar avatar = AvatarFactory.builder().player(player).database(database).build();
+
+    assertEquals(1, database.getAvatarRepository().count());
+
+    MockMvcHelper.of(this.mvc).perform(get("/data/avatar/"))
+      .andExpect(status().isOk())
+      .andExpect(jsonPath("$.data", hasSize(1)))
+      .andExpect(jsonPath("$.data[0].id", is(String.valueOf(avatar.getId()))))
+      .andExpect(jsonPath("$.data[0].type", is("avatar")))
+      .andExpect(jsonPath("$.data[0].attributes.tooltip", is(avatar.getTooltip())))
+      .andExpect(jsonPath("$.data[0].attributes.url", is(avatar.getUrl())))
+      .andExpect(jsonPath("$.data[0].relationships.assignments.data", hasSize(1)))
+      .andExpect(jsonPath("$.data[0].relationships.assignments.data[0].id",
+        is(String.valueOf(avatar.getAssignments().get(0).getId()))))
+      .andExpect(jsonPath("$.data[0].relationships.assignments.data[0].type", is("avatarAssignment")));
+
+    assertEquals(1, database.getAvatarRepository().count());
+  }
+}
diff --git a/src/test/java/com/faforever/api/data/JsonApiClanIntegrationTest.java b/src/test/java/com/faforever/api/data/JsonApiClanIntegrationTest.java
@@ -101,7 +101,7 @@ public void cannotKickLeaderFromClan() throws Exception {
   public void cannotKickAsMember() throws Exception {
     Session session = SessionFactory.createUserAndGetAccessToken(database, mvc);
 
-    Player bob = PlayerFactory.createPlayer("Bob", database);
+    Player bob = PlayerFactory.builder().login("Bob").database(database).build();
     Clan clan = new Clan().setLeader(bob).setTag("123").setName("abcClanName");
     ClanMembership myMembership = new ClanMembership().setPlayer(session.getPlayer()).setClan(clan);
     ClanMembership bobsMembership = new ClanMembership().setPlayer(bob).setClan(clan);
@@ -121,7 +121,7 @@ public void canKickMember() throws Exception {
     Session session = SessionFactory.createUserAndGetAccessToken(database, mvc);
     Player player = session.getPlayer();
 
-    Player bob = PlayerFactory.createPlayer("Bob", database);
+    Player bob = PlayerFactory.builder().login("Bob").database(database).build();
     Clan clan = new Clan().setLeader(player).setTag("123").setName("abcClanName");
     ClanMembership myMembership = new ClanMembership().setPlayer(player).setClan(clan);
     ClanMembership bobsMembership = new ClanMembership().setPlayer(bob).setClan(clan);
@@ -141,7 +141,7 @@ public void canKickMember() throws Exception {
   public void canLeaveClan() throws Exception {
     Session session = SessionFactory.createUserAndGetAccessToken(database, mvc);
 
-    Player bob = PlayerFactory.createPlayer("Bob", database);
+    Player bob = PlayerFactory.builder().login("Bob").database(database).build();
     Player player = session.getPlayer();
 
     Clan clan = new Clan().setLeader(bob).setTag("123").setName("abcClanName");
@@ -161,7 +161,7 @@ public void canLeaveClan() throws Exception {
   @Test
   public void getFilteredPlayerForClanInvite() throws Exception {
     String[] players = new String[]{"Dragonfire", "DRAGON", "Fire of Dragon", "d r a g o n", "firedragon"};
-    Arrays.stream(players).forEach(name -> noCatch(() -> PlayerFactory.createPlayer(name, database)));
+    Arrays.stream(players).forEach(name -> noCatch(() -> PlayerFactory.builder().login(name).database(database).build()));
     assertEquals(players.length, database.getPlayerRepository().count());
     ResultActions action = MockMvcHelper.of(this.mvc).perform(get("/data/player?filter=login==dragon*&sort=login"));
     action.andExpect(status().isOk())
@@ -195,7 +195,7 @@ public void transferLeadership() throws Exception {
     Session session = SessionFactory.createUserAndGetAccessToken("Leader", "foo", database, mvc);
     Player player = session.getPlayer();
 
-    Player bob = PlayerFactory.createPlayer("Bob", database);
+    Player bob = PlayerFactory.builder().login("Bob").database(database).build();
     Clan clan = new Clan().setLeader(player).setTag("123").setName("abcClanName");
     ClanMembership myMembership = new ClanMembership().setPlayer(player).setClan(clan);
     ClanMembership bobsMembership = new ClanMembership().setPlayer(bob).setClan(clan);
@@ -246,7 +246,7 @@ public void transferLeadershipToNonClanMember() throws Exception {
     Session session = SessionFactory.createUserAndGetAccessToken("Leader", "foo", database, mvc);
     Player player = session.getPlayer();
 
-    Player bob = PlayerFactory.createPlayer("Bob", database);
+    Player bob = PlayerFactory.builder().login("Bob").database(database).build();
 
     Clan clan = new Clan().setLeader(player).setTag("123").setName("abcClanName");
     ClanMembership myMembership = new ClanMembership().setPlayer(player).setClan(clan);
@@ -273,8 +273,8 @@ public void transferLeadershipAsNonLeader() throws Exception {
     Session session = SessionFactory.createUserAndGetAccessToken("Leader", "foo", database, mvc);
     Player player = session.getPlayer();
 
-    Player bob = PlayerFactory.createPlayer("Bob", database);
-    Player charlie = PlayerFactory.createPlayer("Charlie", database);
+    Player bob = PlayerFactory.builder().login("Bob").database(database).build();
+    Player charlie = PlayerFactory.builder().login("Charlie").database(database).build();
 
     Clan clan = new Clan().setLeader(bob).setTag("123").setName("abcClanName");
     ClanMembership myMembership = new ClanMembership().setPlayer(player).setClan(clan);

diff --git a/src/test/java/com/faforever/integration/TestDatabase.java b/src/test/java/com/faforever/integration/TestDatabase.java
@@ -1,5 +1,7 @@
 package com.faforever.integration;
 
+import com.faforever.api.avatar.AvatarAssignmentRepository;
+import com.faforever.api.avatar.AvatarRepository;
 import com.faforever.api.ban.BanRepository;
 import com.faforever.api.clan.ClanMembershipRepository;
 import com.faforever.api.clan.ClanRepository;
@@ -24,6 +26,8 @@ public class TestDatabase {
   private OAuthClientRepository oAuthClientRepository;
   private ObjectMapper objectMapper;
   private BanRepository banRepository;
+  private AvatarRepository avatarRepository;
+  private AvatarAssignmentRepository avatarAssignmentRepository;
 
   @Inject
   public void init(ClanRepository clanRepository,
@@ -32,18 +36,24 @@ public void init(ClanRepository clanRepository,
                    OAuthClientRepository oAuthClientRepository,
                    ClanMembershipRepository clanMembershipRepository,
                    BanRepository banRepository,
+                   AvatarRepository avatarRepository,
+                   AvatarAssignmentRepository avatarAssignmentRepository,
                    ObjectMapper objectMapper) {
     this.clanRepository = clanRepository;
     this.userRepository = userRepository;
     this.playerRepository = playerRepository;
     this.oAuthClientRepository = oAuthClientRepository;
     this.clanMembershipRepository = clanMembershipRepository;
     this.banRepository = banRepository;
+    this.avatarRepository = avatarRepository;
+    this.avatarAssignmentRepository = avatarAssignmentRepository;
     this.objectMapper = objectMapper;
   }
 
   public void assertEmptyDatabase() {
     assertEquals(0, banRepository.count());
+    assertEquals(0, avatarAssignmentRepository.count());
+    assertEquals(0, avatarRepository.count());
     assertEquals(0, clanRepository.count());
     assertEquals(0, userRepository.count());
     assertEquals(0, playerRepository.count());
@@ -53,6 +63,8 @@ public void assertEmptyDatabase() {
 
   public void tearDown() {
     banRepository.deleteAll();
+    avatarAssignmentRepository.deleteAll();
+    avatarRepository.deleteAll();
     clanMembershipRepository.deleteAll();
     clanRepository.deleteAll();
     userRepository.deleteAll();

diff --git a/src/test/java/com/faforever/integration/factories/AvatarFactory.java b/src/test/java/com/faforever/integration/factories/AvatarFactory.java
@@ -0,0 +1,43 @@
+package com.faforever.integration.factories;
+
+import com.faforever.api.data.domain.Avatar;
+import com.faforever.api.data.domain.AvatarAssignment;
+import com.faforever.api.data.domain.Player;
+import com.faforever.integration.TestDatabase;
+import lombok.Builder;
+import org.springframework.util.Assert;
+
+import java.util.Collections;
+
+import static org.junit.Assert.assertEquals;
+
+public class AvatarFactory {
+  public static final String DEFAULT_TOOLTIP = "Cool Test Avatar Tooltip";
+  public static final String DEFAULT_URL = "https://www.faf-test.org/myavatar.png";
+
+  @Builder
+  public static Avatar createAvatar(String tooltip,
+                                    String url,
+                                    Player player,
+                                    TestDatabase database) {
+    Assert.notNull(database, "'database' must not be null");
+    Avatar avatar = new Avatar()
+      .setTooltip(tooltip != null ? tooltip : DEFAULT_TOOLTIP)
+      .setUrl(url != null ? url : DEFAULT_URL);
+
+    long assignmentCount = database.getAvatarAssignmentRepository().count();
+    if (player != null) {
+      AvatarAssignment assignment = new AvatarAssignment()
+        .setAvatar(avatar)
+        .setPlayer(player);
+      avatar.setAssignments(Collections.singletonList(assignment));
+      assignmentCount++;
+    }
+
+    long count = database.getAvatarRepository().count();
+    database.getAvatarRepository().save(avatar);
+    assertEquals(count + 1, database.getAvatarRepository().count());
+    assertEquals(assignmentCount, database.getAvatarAssignmentRepository().count());
+    return avatar;
+  }
+}
diff --git a/src/test/java/com/faforever/integration/factories/BanFactory.java b/src/test/java/com/faforever/integration/factories/BanFactory.java
@@ -18,12 +18,15 @@ public class BanFactory {
 
   @Builder
   @SneakyThrows
-  public static BanInfo createBan(Player player, Player author, String reason,
-                                  BanLevel level, OffsetDateTime expiresAt, TestDatabase database) {
+  private static BanInfo createBan(Player player, Player author, String reason,
+                                   BanLevel level, OffsetDateTime expiresAt, TestDatabase database) {
     Assert.notNull(database, "'database' must not be null");
     BanInfo ban = new BanInfo()
-      .setPlayer(player != null ? player : PlayerFactory.createPlayer(PLAYER_NAME_BANNED, database))
-      .setAuthor(author != null ? author : PlayerFactory.createPlayer(PLAYER_NAME_AUTHOR, database))
+      .setPlayer(player != null
+        ? player
+        : PlayerFactory.builder().login(PLAYER_NAME_BANNED).database(database).build())
+      .setAuthor(author != null
+        ? author : PlayerFactory.builder().login(PLAYER_NAME_AUTHOR).database(database).build())
       .setExpiresAt(expiresAt)
       .setLevel(level)
       .setReason(reason != null ? reason : DEFAULT_REASON);

diff --git a/src/test/java/com/faforever/integration/factories/PlayerFactory.java b/src/test/java/com/faforever/integration/factories/PlayerFactory.java
@@ -3,16 +3,25 @@
 import com.faforever.api.data.domain.Player;
 import com.faforever.api.data.domain.User;
 import com.faforever.integration.TestDatabase;
-import lombok.SneakyThrows;
+import lombok.Builder;
+import org.springframework.util.Assert;
+
+import static org.junit.Assert.assertEquals;
 
 public class PlayerFactory {
-  @SneakyThrows
-  public static Player createPlayer(String login, TestDatabase database) {
+  private static final String DEFAULT_USERNAME = "JUnitTestUser_PlayerFactory";
+  private static final String DEFAULT_PASSWORD = "testPassword";
+
+  @Builder
+  private static Player create(String login, String password, TestDatabase database) throws Exception {
+    Assert.notNull(database, "'database' must not be null");
     User user = (User) new User()
-      .setPassword("foo")
-      .setLogin(login)
-      .setEmail(login + "@faforever.com");
+      .setPassword(password != null ? password : DEFAULT_PASSWORD)
+      .setLogin(login != null ? login : DEFAULT_USERNAME)
+      .setEmail(login + "@junit-test.org");
+    long count = database.getUserRepository().count();
     database.getUserRepository().save(user);
+    assertEquals(count + 1, database.getUserRepository().count());
     return database.getPlayerRepository().findOne(user.getId());
   }
 }