package in.partake.controller.api.user;
import static org.hamcrest.Matchers.is;
import static org.junit.Assert.assertThat;
import in.partake.base.DateTime;
import in.partake.base.Pair;
import in.partake.base.TimeUtil;
import in.partake.controller.api.APIControllerTest;
import in.partake.model.dto.Event;
import in.partake.model.dto.EventTicket;
import in.partake.model.dto.UserTicket;
import in.partake.model.dto.auxiliary.AttendanceStatus;
import in.partake.model.dto.auxiliary.EventCategory;
import in.partake.model.dto.auxiliary.ModificationStatus;
import in.partake.model.dto.auxiliary.ParticipationStatus;
import java.util.ArrayList;
import java.util.Collections;
import java.util.List;
import java.util.UUID;
import org.codehaus.jackson.JsonNode;
import org.codehaus.jackson.node.ObjectNode;
import org.junit.Test;
import in.partake.controller.ActionProxy;
public class GetTicketsAPITest extends APIControllerTest {
@Test
public void testGetEnrollments() throws Exception {
List<Pair<UUID, String>> ids = prepareEvents(20);
prepareEnrollment(DEFAULT_USER_ID, ids);
ActionProxy proxy = getActionProxy(GET, "/api/user/tickets?userId=" + DEFAULT_USER_ID + "&limit=10");
proxy.execute();
assertResultOK(proxy);
ObjectNode obj = getJSON(proxy);
assertThat(obj.get("totalTicketCount").asInt(), is(20));
JsonNode array = obj.get("ticketStatuses");
assertThat(array.size(), is(5));
for (int i = 0; i < 5; ++i) {
assertThat(array.get(i).get("ticket").get("id").asText(), is(ids.get(i * 2).getFirst().toString()));
assertThat(array.get(i).get("status").asText(), is("enrolled"));
}
}
private List<Pair<UUID, String>> prepareEvents(int n) throws Exception {
List<Pair<UUID, String>> ids = new ArrayList<Pair<UUID, String>>();
DateTime now = TimeUtil.getCurrentDateTime();
DateTime late = new DateTime(now.getTime() + 1000 * 3600);
String category = EventCategory.getCategories().get(0).getKey();
for (int i = 0; i < n; ++i) {
boolean isPrivate = i % 2 == 1;
Event event = new Event(null, "title", "summary", category,
late, late, "url", "place",
"address", "description", "#hashTag", EVENT_OWNER_ID,
EVENT_FOREIMAGE_ID, EVENT_BACKIMAGE_ID, isPrivate ? "passcode" : null, false,
Collections.singletonList(EVENT_EDITOR_TWITTER_SCREENNAME), new ArrayList<String>(), null,
now, now, -1);
String eventId = storeEvent(event);
UUID uuid = UUID.randomUUID();
EventTicket ticket = EventTicket.createDefaultTicket(uuid, eventId);
storeEventTicket(ticket);
ids.add(new Pair<UUID, String>(uuid, eventId));
}
return ids;
}
private List<String> prepareEnrollment(String userId, List<Pair<UUID, String>> ids) throws Exception {
List<String> userTicketIds = new ArrayList<String>();
for (int i = 0; i < ids.size(); ++i) {
UUID ticketId = ids.get(i).getFirst();
String eventId = ids.get(i).getSecond();
ParticipationStatus status = ParticipationStatus.ENROLLED;
ModificationStatus modificationStatus = ModificationStatus.CHANGED;
AttendanceStatus attendanceStatus = AttendanceStatus.UNKNOWN;
DateTime enrolleAt = new DateTime(TimeUtil.getCurrentTime() + (ids.size() - i) * 1000);
UserTicket enrollment = new UserTicket(null, userId, ticketId, eventId, "comment", status, modificationStatus, attendanceStatus, null, enrolleAt, enrolleAt, enrolleAt);
userTicketIds.add(storeEnrollment(enrollment));
}
return userTicketIds;
}
}