@Test public void testFireAudit() throws Exception { AuditSupport auditSupport = new AuditSupport(this); Field mapField = AuditSupport.class.getDeclaredField("map"); mapField.setAccessible(true); AuditListenerMap map = (AuditListenerMap)mapField.get(auditSupport); AuditListener auditListener1 = new AuditListenerTestHelper(); AuditListener auditListener2 = new AuditListenerTestHelper(); AuditListener[] listeners = {auditListener1, auditListener2}; map.set("testServiceName", listeners); map.set(null, listeners); List<GenericAuditEntity> auditEntities = new ArrayList<>(); GenericAuditEntity genericAuditEntity = new GenericAuditEntity(); auditEntities.add(genericAuditEntity); auditSupport.fireAudit("testServiceName", auditEntities); } }
@Test public void addAuditProxyHBase() throws Exception { AuditSupport auditSupport = new AuditSupport(this); Field mapField = AuditSupport.class.getDeclaredField("map"); mapField.setAccessible(true); AuditListenerMap map = (AuditListenerMap)mapField.get(auditSupport); AuditListener auditListener = new AuditListenerTestHelper(); auditSupport.addAuditListener(auditListener); Assert.assertNull(map.get("test")); }
@Test public void testRemoveAuditProxy() { AuditSupport auditSupport = new AuditSupport(this); AuditListener auditListener = new AuditListenerTestHelper(); auditSupport.addAuditListener(auditListener); auditSupport.removeAuditListener(auditListener); }