/**
* Copyright (C) 2012 - present by OpenGamma Inc. and the OpenGamma group of companies
*
* Please see distribution for license.
*/
package com.opengamma.financial.temptarget.rest;
import static org.testng.Assert.assertEquals;
import static org.testng.Assert.assertNull;
import static org.testng.Assert.assertTrue;
import java.net.URI;
import javax.ws.rs.WebApplicationException;
import org.fudgemsg.FudgeMsg;
import org.mockito.Mockito;
import org.mockito.invocation.InvocationOnMock;
import org.mockito.stubbing.Answer;
import org.testng.annotations.Test;
import com.opengamma.financial.temptarget.InMemoryTempTargetRepository;
import com.opengamma.financial.temptarget.MockTempTarget;
import com.opengamma.financial.temptarget.TempTargetRepository;
import com.opengamma.id.UniqueId;
import com.opengamma.util.rest.UniformInterfaceException404NotFound;
import com.opengamma.util.test.TestGroup;
import com.sun.jersey.api.client.ClientResponse;
import com.sun.jersey.api.client.UniformInterface;
/**
* Tests the {@link RemoteTempTargetRepository} and {@link DataTempTargetRepositoryResource} classes.
*/
@Test(groups = TestGroup.UNIT)
public class RemoteTempTargetRepositoryTest {
public void testGet() {
final InMemoryTempTargetRepository underlying = new InMemoryTempTargetRepository();
final UniqueId uid = underlying.locateOrStore(new MockTempTarget("Test"));
assertEquals(uid, UniqueId.of("TmpMem", "1"));
final DataTempTargetRepositoryResource server = new DataTempTargetRepositoryResource(underlying);
final TempTargetRepository client = new RemoteTempTargetRepository(URI.create("http://localhost/")) {
@Override
protected UniformInterface accessRemote(final URI uri) {
assertTrue(uri.getPath().startsWith("/target/"));
final UniformInterface builder = Mockito.mock(UniformInterface.class);
Mockito.when(builder.get(FudgeMsg.class)).thenAnswer(new Answer<FudgeMsg>() {
@Override
public FudgeMsg answer(final InvocationOnMock invocation) throws Throwable {
try {
return (FudgeMsg) server.get(uri.getPath().substring(8)).getEntity();
} catch (final WebApplicationException e) {
assertEquals(e.getResponse().getStatus(), 404);
throw new UniformInterfaceException404NotFound(new ClientResponse(404, null, null, null), false);
}
}
});
return builder;
}
};
assertNull(client.get(UniqueId.of("Invalid", "Foo")));
assertEquals(client.get(uid), new MockTempTarget("Test"));
}
public void testLocateOrStore() {
final InMemoryTempTargetRepository underlying = new InMemoryTempTargetRepository();
final DataTempTargetRepositoryResource server = new DataTempTargetRepositoryResource(underlying);
final TempTargetRepository client = new RemoteTempTargetRepository(URI.create("http://localhost/")) {
@SuppressWarnings("unchecked")
@Override
protected UniformInterface accessRemote(final URI uri) {
assertTrue(uri.getPath().equals("/target"));
final UniformInterface builder = Mockito.mock(UniformInterface.class);
Mockito.doAnswer(new Answer<FudgeMsg>() {
@Override
public FudgeMsg answer(final InvocationOnMock invocation) throws Throwable {
return (FudgeMsg) server.locateOrStore((FudgeMsg) invocation.getArguments()[1]).getEntity();
}
}).when(builder).post(Mockito.any(Class.class), Mockito.anyObject());
return builder;
}
};
assertEquals(client.locateOrStore(new MockTempTarget("Test")), UniqueId.of("TmpMem", "1"));
assertEquals(client.locateOrStore(new MockTempTarget("Test")), UniqueId.of("TmpMem", "1"));
assertEquals(client.locateOrStore(new MockTempTarget("Foo")), UniqueId.of("TmpMem", "2"));
}
}