/**
* Copyright (C) 2012 - present by OpenGamma Inc. and the OpenGamma group of companies
*
* Please see distribution for license.
*/
package com.opengamma.batch.rest;
import static com.google.common.collect.Lists.newArrayList;
import static com.google.common.collect.Sets.newHashSet;
import static org.mockito.Mockito.doNothing;
import static org.mockito.Mockito.mock;
import static org.mockito.Mockito.verify;
import static org.mockito.Mockito.when;
import static org.testng.AssertJUnit.assertEquals;
import static org.testng.AssertJUnit.assertSame;
import java.util.List;
import javax.ws.rs.core.Response;
import org.testng.annotations.BeforeMethod;
import org.testng.annotations.Test;
import org.threeten.bp.Instant;
import com.opengamma.batch.BatchMaster;
import com.opengamma.batch.domain.CalculationConfiguration;
import com.opengamma.batch.domain.MarketData;
import com.opengamma.batch.domain.RiskRun;
import com.opengamma.batch.domain.RiskRunProperty;
import com.opengamma.engine.view.ViewResultEntry;
import com.opengamma.id.ObjectId;
import com.opengamma.id.UniqueId;
import com.opengamma.id.VersionCorrection;
import com.opengamma.transport.jaxrs.FudgeResponse;
import com.opengamma.util.paging.Paging;
import com.opengamma.util.paging.PagingRequest;
import com.opengamma.util.test.TestGroup;
import com.opengamma.util.tuple.Pair;
import com.sun.jersey.api.client.ClientResponse.Status;
/**
* Tests BatchRunResource.
*/
@Test(groups = TestGroup.UNIT)
public class DataBatchRunResourceTest {
private RiskRun _riskRun;
private BatchMaster _underlying;
private DataBatchRunResource _resource;
private static final ObjectId _riskRunId = ObjectId.of("Test", "RiskRun");
@BeforeMethod
public void setUp() {
_riskRun = new RiskRun(
new MarketData(UniqueId.of(BatchMaster.BATCH_IDENTIFIER_SCHEME, "market-data")),
Instant.now(),
Instant.now(),
0,
newHashSet(new CalculationConfiguration("calc-config")),
newHashSet(new RiskRunProperty()),
false,
VersionCorrection.LATEST,
UniqueId.of("Scheme", "view-def")
);
_underlying = mock(BatchMaster.class);
_resource = new DataBatchRunResource(_riskRunId, _underlying);
when(_underlying.getRiskRun(_riskRunId)).thenReturn(_riskRun);
}
//-------------------------------------------------------------------------
@Test
public void testGet() {
Response test = _resource.get();
assertEquals(Status.OK.getStatusCode(), test.getStatus());
assertSame(_riskRun, test.getEntity());
}
@Test
public void testDelete() {
doNothing().when(_underlying).deleteRiskRun(_riskRunId.getObjectId());
_resource.deleteBatchRun();
verify(_underlying).deleteRiskRun(_riskRunId.getObjectId());
}
@SuppressWarnings("unchecked")
@Test
public void testGetBatchValues() {
PagingRequest pagingRequest = PagingRequest.FIRST_PAGE;
ViewResultEntry mockViewResultEntry = mock(ViewResultEntry.class);
List<ViewResultEntry> viewResultEntries = newArrayList(mockViewResultEntry);
Paging paging = Paging.of(pagingRequest, viewResultEntries);
when(_underlying.getBatchValues(_riskRunId, pagingRequest)).thenReturn(Pair.of(viewResultEntries, paging));
Response response = _resource.getBatchValues(pagingRequest);
FudgeResponse entity = (FudgeResponse) response.getEntity();
Pair<List<ViewResultEntry>, Paging> result = (Pair<List<ViewResultEntry>, Paging>) entity.getValue();
assertEquals(Status.OK.getStatusCode(), response.getStatus());
assertSame(result.getFirst().size(), 1);
assertSame(result.getFirst().get(0), mockViewResultEntry);
assertSame(result.getSecond(), paging);
}
}