use of org.summerb.easycrud.api.relations.ReferencesRegistry in project summerb by skarpushin.
the class EasyCrudRestControllerBase method resolveReferences.
protected void resolveReferences(List<String> referencesToResolve, CrudQueryResult<TId, TDto> ret, List<TDto> items) throws EntityNotFoundException, NotAuthorizedException {
Preconditions.checkState(dataSetLoader != null, "DataSetLoader is required to resolve references");
Preconditions.checkState(referencesRegistry != null, "referencesRegistry is required to resolve references");
DataSet ds = new DataSet();
DataTable<TId, TDto> table = new DataTable<>(service.getEntityTypeMessageCode());
table.putAll(items);
ds.getTables().put(table.getName(), table);
List<Ref> references = referencesToResolve.stream().map(name -> referencesRegistry.getRefByName(name)).collect(Collectors.toList());
Ref[] refsArr = (Ref[]) references.toArray(new Ref[references.size()]);
dataSetLoader.resolveReferencedObjects(ds, refsArr);
// now remove initial table from dataset because we don't want to
// duplicate this. It's already populated to rows
ds.getTables().remove(table.getName());
// x. ret
ret.setRefsResolved(references.stream().collect(Collectors.toMap(Ref::getName, Function.identity())));
ret.setRefs(ds);
}
use of org.summerb.easycrud.api.relations.ReferencesRegistry in project summerb by skarpushin.
the class DomLoaderDeviceGatewayTest method testMapDtoToDom_expectCorrectFieldTypeResolution.
@Test
public void testMapDtoToDom_expectCorrectFieldTypeResolution() throws Exception {
// Deps and fixture
ReferencesRegistry referencesRegistry = new Refs();
EasyCrudServiceResolver easyCrudServiceResolver = mock(EasyCrudServiceResolver.class);
DeviceService deviceService = mock(DeviceService.class);
when(deviceService.getEntityTypeMessageCode()).thenReturn(DeviceService.ENTITY_TYPE_MESSAGE_CODE);
when(easyCrudServiceResolver.resolveByDtoClass(DeviceRow.class)).thenReturn(deviceService);
when(easyCrudServiceResolver.resolveByEntityType(DeviceService.ENTITY_TYPE_MESSAGE_CODE)).thenReturn(deviceService);
EnvService envService = mock(EnvService.class);
when(envService.getEntityTypeMessageCode()).thenReturn(EnvService.ENTITY_TYPE_MESSAGE_CODE);
when(easyCrudServiceResolver.resolveByDtoClass(EnvironmentRow.class)).thenReturn(envService);
when(easyCrudServiceResolver.resolveByEntityType(EnvService.ENTITY_TYPE_MESSAGE_CODE)).thenReturn(envService);
DataSetLoaderImpl dataSetLoader = new DataSetLoaderImpl();
dataSetLoader.setEasyCrudServiceResolver(easyCrudServiceResolver);
dataSetLoader.setReferencesRegistry(referencesRegistry);
DomLoaderImpl f = new DomLoaderImpl(dataSetLoader, easyCrudServiceResolver);
// Setup DataSet
// DataSet ds = new DataSet();
EnvironmentRow envRow = new EnvironmentRow();
envRow.setId(1L);
envRow.setName("Hurray");
when(envService.query(any(), any(), any())).thenReturn(new PaginatedList<>(PagerParams.ALL, Arrays.asList(envRow), 1));
when(envService.findById(1L)).thenReturn(envRow);
// ds.get(EnvService.ENTITY_TYPE_MESSAGE_CODE).put(envRow);
DeviceRow deviceRow = new DeviceRow();
deviceRow.setId(2L);
deviceRow.setEnvId(1);
deviceRow.setName("Yes it is");
when(deviceService.query(any(), any())).thenReturn(new PaginatedList<>(PagerParams.ALL, Arrays.asList(deviceRow), 1));
when(deviceService.findById(2L)).thenReturn(deviceRow);
// ds.get(DeviceService.ENTITY_TYPE_MESSAGE_CODE).put(deviceRow);
// when(dataSetLoader.loadObjectsByIds(any(),
// anyString())).thenReturn(Arrays.asList(envRow));
// Now let's invoke it
Env env = f.load(Env.class, 1L, Refs.envDevices, Refs.deviceEnv);
// verify(deviceService, times(1)).query(any(), any(), any());
// verify(envService, times(1)).query(any(), any(), any());
// Verify
assertNotNull(env);
assertEquals("Hurray", env.getName());
assertNotNull(env.getDevices());
assertEquals(1, env.getDevices().size());
Device device = env.getDevices().get(0);
assertEquals("Yes it is", device.getName());
assertNotNull(device.getEnv());
assertEquals(env, device.getEnv());
}
use of org.summerb.easycrud.api.relations.ReferencesRegistry in project summerb by skarpushin.
the class DataSetLoaderImplTest method buildMockedInstance.
private DataSetLoaderImpl buildMockedInstance() {
DataSetLoaderImpl ret = new DataSetLoaderImpl();
EasyCrudServiceResolver easyCrudServiceResolver = Mockito.mock(EasyCrudServiceResolver.class);
ret.setEasyCrudServiceResolver(easyCrudServiceResolver);
ReferencesRegistry referencesRegistry = Mockito.mock(ReferencesRegistry.class);
ret.setReferencesRegistry(referencesRegistry);
return ret;
}
Aggregations