package org.jboss.aerogear.unifiedpush.rest.registry.applications;
import org.jboss.resteasy.specimpl.UriInfoImpl;
import org.jboss.resteasy.spi.Link;
import org.jboss.resteasy.spi.LinkHeader;
import org.junit.Test;
import javax.ws.rs.core.PathSegment;
import java.net.URI;
import java.net.URISyntaxException;
import java.util.Collections;
import static org.assertj.core.api.Assertions.assertThat;
/**
*/
public class InstallationManagementEndpointTest {
private final InstallationManagementEndpoint endpoint = new InstallationManagementEndpoint();
@Test
public void shouldGenerateHeaderLinksFirstPage() throws URISyntaxException {
//given
final UriInfoImpl uriInfo = getUriInfo();
//when
final LinkHeader linkHeader = endpoint.getLinkHeader(0, 3, uriInfo);
//then
assertThat(findLinkByRel(linkHeader, "prev")).isNull();
final Link next = findLinkByRel(linkHeader, "next");
assertThat(next).isNotNull();
assertThat(next.getHref()).isEqualTo("/?page=1");
assertThat(findLinkByRel(linkHeader, "first")).isNull();
}
@Test
public void shouldGenerateHeaderLinksNormalPage() throws URISyntaxException {
//given
final UriInfoImpl uriInfo = getUriInfo();
//when
final LinkHeader linkHeader = endpoint.getLinkHeader(2, 3, uriInfo);
final Link prev = findLinkByRel(linkHeader, "prev");
assertThat(prev).isNotNull();
assertThat(prev.getHref()).isEqualTo("/?page=1");
final Link next = findLinkByRel(linkHeader, "next");
assertThat(next).isNotNull();
assertThat(next.getHref()).isEqualTo("/?page=3");
}
@Test
public void shouldGenerateHeaderLinksLastPage() throws URISyntaxException {
//given
final UriInfoImpl uriInfo = getUriInfo();
//when
final LinkHeader linkHeader = endpoint.getLinkHeader(3, 3, uriInfo);
final Link prev = findLinkByRel(linkHeader, "prev");
assertThat(prev).isNotNull();
assertThat(prev.getHref()).isEqualTo("/?page=2");
final Link first = findLinkByRel(linkHeader, "first");
assertThat(first).isNotNull();
assertThat(first.getHref()).isEqualTo("/?page=0");
assertThat(findLinkByRel(linkHeader, "last")).isNull();
}
private Link findLinkByRel(LinkHeader linkHeader, String rel) {
for (Link link : linkHeader.getLinks()) {
if (link.getRelationship().equals(rel)) {
return link;
}
}
return null;
}
private UriInfoImpl getUriInfo() throws URISyntaxException {
return new UriInfoImpl(new URI("/"), new URI("http://localhost"), "/", "", Collections.<PathSegment>emptyList());
}
}