Search in sources :

Example 11 with SnapshotRequestModel

use of bio.terra.model.SnapshotRequestModel in project jade-data-repo by DataBiosphere.

the class SnapshotConnectedTest method testMinimal.

@Test
public void testMinimal() throws Exception {
    DatasetSummaryModel datasetMinimalSummary = setupMinimalDataset();
    String datasetName = PDAO_PREFIX + datasetMinimalSummary.getName();
    BigQueryProject bigQueryProject = TestUtils.bigQueryProjectForDatasetName(datasetDao, dataLocationService, datasetMinimalSummary.getName());
    long datasetParticipants = queryForCount(datasetName, "participant", bigQueryProject);
    assertThat("dataset participants loaded properly", datasetParticipants, equalTo(2L));
    long datasetSamples = queryForCount(datasetName, "sample", bigQueryProject);
    assertThat("dataset samples loaded properly", datasetSamples, equalTo(5L));
    SnapshotRequestModel snapshotRequest = makeSnapshotTestRequest(datasetMinimalSummary, "dataset-minimal-snapshot.json");
    MockHttpServletResponse response = performCreateSnapshot(snapshotRequest, "");
    SnapshotSummaryModel summaryModel = validateSnapshotCreated(snapshotRequest, response);
    SnapshotModel snapshotModel = getTestSnapshot(summaryModel.getId(), snapshotRequest, datasetMinimalSummary);
    List<TableModel> tables = snapshotModel.getTables();
    Optional<TableModel> participantTable = tables.stream().filter(t -> t.getName().equals("participant")).findFirst();
    Optional<TableModel> sampleTable = tables.stream().filter(t -> t.getName().equals("sample")).findFirst();
    assertThat("participant table exists", participantTable.isPresent(), equalTo(true));
    assertThat("sample table exists", sampleTable.isPresent(), equalTo(true));
    long snapshotParticipants = queryForCount(summaryModel.getName(), "participant", bigQueryProject);
    assertThat("dataset participants loaded properly", snapshotParticipants, equalTo(1L));
    assertThat("participant row count matches expectation", participantTable.get().getRowCount(), equalTo(1));
    long snapshotSamples = queryForCount(summaryModel.getName(), "sample", bigQueryProject);
    assertThat("dataset samples loaded properly", snapshotSamples, equalTo(2L));
    assertThat("sample row count matches expectation", sampleTable.get().getRowCount(), equalTo(2));
    List<RelationshipModel> relationships = snapshotModel.getRelationships();
    assertThat("a relationship comes back", relationships.size(), equalTo(1));
    RelationshipModel relationshipModel = relationships.get(0);
    assertThat("relationship name is right", relationshipModel.getName(), equalTo("participant_sample"));
    assertThat("from table is right", relationshipModel.getFrom().getTable(), equalTo("participant"));
    assertThat("from column is right", relationshipModel.getFrom().getColumn(), equalTo("id"));
    assertThat("to table is right", relationshipModel.getTo().getTable(), equalTo("sample"));
    assertThat("to column is right", relationshipModel.getTo().getColumn(), equalTo("participant_id"));
}
Also used : BillingProfileModel(bio.terra.model.BillingProfileModel) FieldValue(com.google.cloud.bigquery.FieldValue) CoreMatchers(org.hamcrest.CoreMatchers) SnapshotModel(bio.terra.model.SnapshotModel) Autowired(org.springframework.beans.factory.annotation.Autowired) ActiveProfiles(org.springframework.test.context.ActiveProfiles) StorageOptions(com.google.cloud.storage.StorageOptions) StringUtils(org.apache.commons.lang3.StringUtils) BigQuery(com.google.cloud.bigquery.BigQuery) Assert.assertThat(org.junit.Assert.assertThat) RelationshipModel(bio.terra.model.RelationshipModel) FieldValueList(com.google.cloud.bigquery.FieldValueList) After(org.junit.After) DrsId(bio.terra.service.filedata.DrsId) TableResult(com.google.cloud.bigquery.TableResult) SpringRunner(org.springframework.test.context.junit4.SpringRunner) URI(java.net.URI) DataLocationService(bio.terra.service.resourcemanagement.DataLocationService) MockBean(org.springframework.boot.test.mock.mockito.MockBean) CoreMatchers.containsString(org.hamcrest.CoreMatchers.containsString) PDAO_PREFIX(bio.terra.common.PdaoConstant.PDAO_PREFIX) MediaType(org.springframework.http.MediaType) QueryJobConfiguration(com.google.cloud.bigquery.QueryJobConfiguration) EnumerateSnapshotModel(bio.terra.model.EnumerateSnapshotModel) UUID(java.util.UUID) Category(org.junit.experimental.categories.Category) SnapshotRequestContentsModel(bio.terra.model.SnapshotRequestContentsModel) Collectors(java.util.stream.Collectors) DatasetSummaryModel(bio.terra.model.DatasetSummaryModel) StandardCharsets(java.nio.charset.StandardCharsets) ConfigEnum(bio.terra.service.configuration.ConfigEnum) IOUtils(org.apache.commons.io.IOUtils) List(java.util.List) SpringBootTest(org.springframework.boot.test.context.SpringBootTest) ST(org.stringtemplate.v4.ST) Assert.assertFalse(org.junit.Assert.assertFalse) TestUtils(bio.terra.common.TestUtils) Matchers.equalTo(org.hamcrest.Matchers.equalTo) Optional(java.util.Optional) Storage(com.google.cloud.storage.Storage) ConfigurationService(bio.terra.service.configuration.ConfigurationService) ArgumentMatchers.any(org.mockito.ArgumentMatchers.any) GoogleResourceConfiguration(bio.terra.service.resourcemanagement.google.GoogleResourceConfiguration) ConnectedOperations(bio.terra.common.fixtures.ConnectedOperations) Names(bio.terra.common.fixtures.Names) DRSObject(bio.terra.model.DRSObject) RunWith(org.junit.runner.RunWith) BigQueryProject(bio.terra.service.tabulardata.google.BigQueryProject) DrsIdService(bio.terra.service.filedata.DrsIdService) TableModel(bio.terra.model.TableModel) MockMvcResultMatchers.content(org.springframework.test.web.servlet.result.MockMvcResultMatchers.content) MockHttpServletResponse(org.springframework.mock.web.MockHttpServletResponse) ArrayList(java.util.ArrayList) MockMvc(org.springframework.test.web.servlet.MockMvc) SnapshotRequestModel(bio.terra.model.SnapshotRequestModel) MockMvcRequestBuilders.delete(org.springframework.test.web.servlet.request.MockMvcRequestBuilders.delete) MockMvcRequestBuilders.post(org.springframework.test.web.servlet.request.MockMvcRequestBuilders.post) MvcResult(org.springframework.test.web.servlet.MvcResult) JsonLoader(bio.terra.common.fixtures.JsonLoader) ProfileDao(bio.terra.service.resourcemanagement.ProfileDao) ConnectedTestConfiguration(bio.terra.app.configuration.ConnectedTestConfiguration) StringStartsWith.startsWith(org.hamcrest.core.StringStartsWith.startsWith) Before(org.junit.Before) BlobInfo(com.google.cloud.storage.BlobInfo) FileLoadModel(bio.terra.model.FileLoadModel) FileModel(bio.terra.model.FileModel) Assert.assertNotNull(org.junit.Assert.assertNotNull) ObjectMapper(com.fasterxml.jackson.databind.ObjectMapper) Connected(bio.terra.common.category.Connected) Assert.assertTrue(org.junit.Assert.assertTrue) DeleteResponseModel(bio.terra.model.DeleteResponseModel) Test(org.junit.Test) IamProviderInterface(bio.terra.service.iam.IamProviderInterface) Mockito.when(org.mockito.Mockito.when) SnapshotSourceModel(bio.terra.model.SnapshotSourceModel) DatasetDao(bio.terra.service.dataset.DatasetDao) TimeUnit(java.util.concurrent.TimeUnit) ErrorModel(bio.terra.model.ErrorModel) HttpStatus(org.springframework.http.HttpStatus) IngestRequestModel(bio.terra.model.IngestRequestModel) AutoConfigureMockMvc(org.springframework.boot.test.autoconfigure.web.servlet.AutoConfigureMockMvc) Assert.assertNull(org.junit.Assert.assertNull) SnapshotSummaryModel(bio.terra.model.SnapshotSummaryModel) MockMvcRequestBuilders.get(org.springframework.test.web.servlet.request.MockMvcRequestBuilders.get) Assert.assertEquals(org.junit.Assert.assertEquals) SnapshotSummaryModel(bio.terra.model.SnapshotSummaryModel) RelationshipModel(bio.terra.model.RelationshipModel) CoreMatchers.containsString(org.hamcrest.CoreMatchers.containsString) SnapshotRequestModel(bio.terra.model.SnapshotRequestModel) SnapshotModel(bio.terra.model.SnapshotModel) EnumerateSnapshotModel(bio.terra.model.EnumerateSnapshotModel) BigQueryProject(bio.terra.service.tabulardata.google.BigQueryProject) DatasetSummaryModel(bio.terra.model.DatasetSummaryModel) MockHttpServletResponse(org.springframework.mock.web.MockHttpServletResponse) TableModel(bio.terra.model.TableModel) SpringBootTest(org.springframework.boot.test.context.SpringBootTest) Test(org.junit.Test)

Example 12 with SnapshotRequestModel

use of bio.terra.model.SnapshotRequestModel in project jade-data-repo by DataBiosphere.

the class SnapshotConnectedTest method testArrayStruct.

@Test
public void testArrayStruct() throws Exception {
    DatasetSummaryModel datasetArraySummary = setupArrayStructDataset();
    String datasetName = PDAO_PREFIX + datasetArraySummary.getName();
    BigQueryProject bigQueryProject = TestUtils.bigQueryProjectForDatasetName(datasetDao, dataLocationService, datasetArraySummary.getName());
    long datasetParticipants = queryForCount(datasetName, "participant", bigQueryProject);
    assertThat("dataset participants loaded properly", datasetParticipants, equalTo(2L));
    long datasetSamples = queryForCount(datasetName, "sample", bigQueryProject);
    assertThat("dataset samples loaded properly", datasetSamples, equalTo(5L));
    SnapshotRequestModel snapshotRequest = makeSnapshotTestRequest(datasetArraySummary, "snapshot-array-struct.json");
    MockHttpServletResponse response = performCreateSnapshot(snapshotRequest, "");
    SnapshotSummaryModel summaryModel = validateSnapshotCreated(snapshotRequest, response);
    getTestSnapshot(summaryModel.getId(), snapshotRequest, datasetArraySummary);
    long snapshotParticipants = queryForCount(summaryModel.getName(), "participant", bigQueryProject);
    assertThat("dataset participants loaded properly", snapshotParticipants, equalTo(2L));
    long snapshotSamples = queryForCount(summaryModel.getName(), "sample", bigQueryProject);
    assertThat("dataset samples loaded properly", snapshotSamples, equalTo(3L));
}
Also used : SnapshotSummaryModel(bio.terra.model.SnapshotSummaryModel) BigQueryProject(bio.terra.service.tabulardata.google.BigQueryProject) DatasetSummaryModel(bio.terra.model.DatasetSummaryModel) CoreMatchers.containsString(org.hamcrest.CoreMatchers.containsString) SnapshotRequestModel(bio.terra.model.SnapshotRequestModel) MockHttpServletResponse(org.springframework.mock.web.MockHttpServletResponse) SpringBootTest(org.springframework.boot.test.context.SpringBootTest) Test(org.junit.Test)

Example 13 with SnapshotRequestModel

use of bio.terra.model.SnapshotRequestModel in project jade-data-repo by DataBiosphere.

the class SnapshotConnectedTest method testDuplicateName.

@Test
public void testDuplicateName() throws Exception {
    // create a snapshot
    SnapshotRequestModel snapshotRequest = makeSnapshotTestRequest(datasetSummary, "snapshot-test-snapshot.json");
    MockHttpServletResponse response = performCreateSnapshot(snapshotRequest, "_dup_");
    SnapshotSummaryModel summaryModel = validateSnapshotCreated(snapshotRequest, response);
    // fetch the snapshot and confirm the metadata matches the request
    SnapshotModel snapshotModel = getTestSnapshot(summaryModel.getId(), snapshotRequest, datasetSummary);
    assertNotNull("fetched snapshot successfully after creation", snapshotModel);
    // check that the snapshot metadata row is unlocked
    String exclusiveLock = snapshotDao.getExclusiveLockState(UUID.fromString(snapshotModel.getId()));
    assertNull("snapshot row is unlocked", exclusiveLock);
    // try to create the same snapshot again and check that it fails
    snapshotRequest.setName(snapshotModel.getName());
    response = performCreateSnapshot(snapshotRequest, null);
    ErrorModel errorModel = handleCreateSnapshotFailureCase(response);
    assertThat(response.getStatus(), equalTo(HttpStatus.BAD_REQUEST.value()));
    assertThat("error message includes name conflict", errorModel.getMessage(), containsString("Snapshot name already exists"));
    // delete and confirm deleted
    connectedOperations.deleteTestSnapshot(snapshotModel.getId());
    connectedOperations.getSnapshotExpectError(snapshotModel.getId(), HttpStatus.NOT_FOUND);
}
Also used : SnapshotSummaryModel(bio.terra.model.SnapshotSummaryModel) ErrorModel(bio.terra.model.ErrorModel) CoreMatchers.containsString(org.hamcrest.CoreMatchers.containsString) SnapshotRequestModel(bio.terra.model.SnapshotRequestModel) SnapshotModel(bio.terra.model.SnapshotModel) EnumerateSnapshotModel(bio.terra.model.EnumerateSnapshotModel) MockHttpServletResponse(org.springframework.mock.web.MockHttpServletResponse) SpringBootTest(org.springframework.boot.test.context.SpringBootTest) Test(org.junit.Test)

Example 14 with SnapshotRequestModel

use of bio.terra.model.SnapshotRequestModel in project jade-data-repo by DataBiosphere.

the class SnapshotTest method snapshotByFullViewHappyPathTest.

@Test
public void snapshotByFullViewHappyPathTest() throws Exception {
    DatasetModel dataset = dataRepoFixtures.getDataset(steward(), datasetId);
    String datasetName = dataset.getName();
    SnapshotRequestModel requestModel = jsonLoader.loadObject("ingest-test-snapshot-fullviews.json", SnapshotRequestModel.class);
    // swap in the correct dataset name (with the id at the end)
    requestModel.getContents().get(0).setDatasetName(datasetName);
    SnapshotSummaryModel snapshotSummary = dataRepoFixtures.createSnapshotWithRequest(steward(), datasetName, requestModel);
    TimeUnit.SECONDS.sleep(10);
    createdSnapshotIds.add(snapshotSummary.getId());
    SnapshotModel snapshot = dataRepoFixtures.getSnapshot(steward(), snapshotSummary.getId());
    assertEquals("new snapshot has been created", snapshot.getName(), requestModel.getName());
    assertEquals("all 5 relationships come through", snapshot.getRelationships().size(), 5);
}
Also used : SnapshotSummaryModel(bio.terra.model.SnapshotSummaryModel) DatasetModel(bio.terra.model.DatasetModel) SnapshotRequestModel(bio.terra.model.SnapshotRequestModel) SnapshotModel(bio.terra.model.SnapshotModel) EnumerateSnapshotModel(bio.terra.model.EnumerateSnapshotModel) Test(org.junit.Test) SpringBootTest(org.springframework.boot.test.context.SpringBootTest)

Example 15 with SnapshotRequestModel

use of bio.terra.model.SnapshotRequestModel in project jade-data-repo by DataBiosphere.

the class SnapshotRequestValidator method validate.

@Override
public void validate(@NotNull Object target, Errors errors) {
    if (target != null && target instanceof SnapshotRequestModel) {
        SnapshotRequestModel snapshotRequestModel = (SnapshotRequestModel) target;
        validateSnapshotName(snapshotRequestModel.getName(), errors);
        validateSnapshotProfileId(snapshotRequestModel.getProfileId(), errors);
        validateSnapshotDescription(snapshotRequestModel.getDescription(), errors);
        validateSnapshotContents(snapshotRequestModel.getContents(), errors);
    }
}
Also used : SnapshotRequestModel(bio.terra.model.SnapshotRequestModel)

Aggregations

SnapshotRequestModel (bio.terra.model.SnapshotRequestModel)18 Test (org.junit.Test)11 SpringBootTest (org.springframework.boot.test.context.SpringBootTest)11 SnapshotSummaryModel (bio.terra.model.SnapshotSummaryModel)9 SnapshotModel (bio.terra.model.SnapshotModel)8 EnumerateSnapshotModel (bio.terra.model.EnumerateSnapshotModel)7 MockHttpServletResponse (org.springframework.mock.web.MockHttpServletResponse)7 SnapshotRequestContentsModel (bio.terra.model.SnapshotRequestContentsModel)6 DatasetSummaryModel (bio.terra.model.DatasetSummaryModel)5 CoreMatchers.containsString (org.hamcrest.CoreMatchers.containsString)5 DatasetModel (bio.terra.model.DatasetModel)4 ErrorModel (bio.terra.model.ErrorModel)4 BigQuery (com.google.cloud.bigquery.BigQuery)4 JsonLoader (bio.terra.common.fixtures.JsonLoader)3 IngestRequestModel (bio.terra.model.IngestRequestModel)3 BigQueryProject (bio.terra.service.tabulardata.google.BigQueryProject)3 ConnectedTestConfiguration (bio.terra.app.configuration.ConnectedTestConfiguration)2 PDAO_PREFIX (bio.terra.common.PdaoConstant.PDAO_PREFIX)2 TestUtils (bio.terra.common.TestUtils)2 Connected (bio.terra.common.category.Connected)2