HarmonizedDatasetServiceTest.java 文件源码

@Test
public void testPopulateHarmonizedVariablesMap() {
  List<DatasetVariable> l = new ArrayList<DatasetVariable>() {
    {
      add(new DatasetVariable(dataset, Variable.Builder.newVariable("v1", BooleanType.get(), "test").build(), st));
      add(new DatasetVariable(dataset, Variable.Builder.newVariable("v2", BooleanType.get(), "test").build(), st2));
    }};

  doReturn(dataset).when(datasetService).findById(anyString());
  when(helper.asyncGetDatasetVariables(any(Supplier.class))).thenReturn(new AsyncResult<>(l));
  doReturn(l).when(datasetService).getDatasetVariables(any(HarmonizationDataset.class));
  doReturn(l).when(datasetService).getDatasetVariables(any(HarmonizationDataset.class), any(StudyTable.class));

  Map<String, List<DatasetVariable>> res = datasetService.populateHarmonizedVariablesMap(dataset);

  assertEquals(2, res.keySet().size());
  assertEquals(2, res.get("testds:v1:Dataschema").size());
  assertEquals(2, res.get("testds:v2:Dataschema").size());
}