public VmRngDevice generateRngDevice() { VmRngDevice dev = new VmRngDevice(); dev.setBytes(rngBytes.getEntity()); dev.setPeriod(rngPeriod.getEntity()); dev.setSource(Boolean.TRUE.equals(rngSourceUrandom.getEntity()) ? getBehavior().getUrandomOrRandomRngSource() : VmRngDevice.Source.HWRNG); return dev; }
@Mapping(from = RngDevice.class, to = VmRngDevice.class) public static VmRngDevice map(RngDevice model, VmRngDevice template) { if (model != null && model.isSetSource()) { VmRngDevice dev = new VmRngDevice(); if (model.isSetRate()) { dev.setBytes(model.getRate().getBytes()); if (model.getRate().isSetPeriod()) { dev.setPeriod(model.getRate().getPeriod()); } } VmRngDevice.Source source = VmRngDevice.Source.valueOf(model.getSource().name()); if (source != null) { dev.setSource(source); return dev; } } return null; }
@Test public void testMapFromRestToBackend() { RngDevice model = new RngDevice(); model.setSource(RngSource.HWRNG); model.setRate(new Rate()); model.getRate().setBytes(10); model.getRate().setPeriod(11); VmRngDevice expected = new VmRngDevice(); expected.setBytes(10); expected.setPeriod(11); expected.setSource(VmRngDevice.Source.HWRNG); assertEquals(expected, RngDeviceMapper.map(model, null)); } }
@Test public void testMapFromBackendToRest() { VmRngDevice entity = new VmRngDevice(); entity.setBytes(11); entity.setPeriod(10); entity.setSource(VmRngDevice.Source.RANDOM); RngDevice expected = new RngDevice(); expected.setRate(new Rate()); expected.getRate().setBytes(11); expected.getRate().setPeriod(10); expected.setSource(RngSource.RANDOM); assertEquals(expected.getRate().getBytes(), RngDeviceMapper.map(entity, null).getRate().getBytes()); assertEquals(expected.getRate().getPeriod(), RngDeviceMapper.map(entity, null).getRate().getPeriod()); assertEquals(expected.getSource(), RngDeviceMapper.map(entity, null).getSource()); }