MgmtVmCfgEao { public Map<String, String> findAll() { List<MgmtVmCfgItemEntity> cfgItemList = mgmtVmCfgDao.findAll(); Map<String, String> cfgMap = null; if (cfgItemList != null && !cfgItemList.isEmpty()) { cfgMap = new HashMap<>(); for (MgmtVmCfgItemEntity mgmtVmCfgItemEntity : cfgItemList) { cfgMap.put(mgmtVmCfgItemEntity.getName(), mgmtVmCfgItemEntity.getValue()); } } else { throw new UserMgmtPersistException("MGMT_VM_CFG.CFG_NOT_FOUND", null); } return cfgMap; }  Map<String, String> findAll(); void update(Map<String, String> newConfig);  }
@Test(groups = { "TestMgmtVmCfgEao" }) public void testFindAll() { mgmtVmCfgDao = new MockUp<IBaseDAO<MgmtVmCfgItemEntity>>() { @Mock List<MgmtVmCfgItemEntity> findAll() { List<MgmtVmCfgItemEntity> mgmtVmCfgItemEntitys = new ArrayList<MgmtVmCfgItemEntity>(); MgmtVmCfgItemEntity mgmtVmCfgItemEntity = new MgmtVmCfgItemEntity(); mgmtVmCfgItemEntity.setName(ngmtVmCfgItemName); mgmtVmCfgItemEntity.setValue(ngmtVmCfgItemValue); mgmtVmCfgItemEntitys.add(mgmtVmCfgItemEntity); return mgmtVmCfgItemEntitys; } }.getMockInstance(); TestUtils.setPrivateField(mgmtVmCfgEao, "mgmtVmCfgDao", mgmtVmCfgDao); Map<String, String> mgmtVmCfgItemEntityMap= mgmtVmCfgEao.findAll(); assertEquals(mgmtVmCfgItemEntityMap.size(), 1); assertTrue(mgmtVmCfgItemEntityMap.containsKey(ngmtVmCfgItemName)); assertEquals(mgmtVmCfgItemEntityMap.get(ngmtVmCfgItemName), ngmtVmCfgItemValue); }