diff --git a/repository/src/test/java/org/apache/atlas/repository/audit/AuditRepositoryTestBase.java b/repository/src/test/java/org/apache/atlas/repository/audit/AuditRepositoryTestBase.java index 7ea15cb..3d17ca9 100644 --- a/repository/src/test/java/org/apache/atlas/repository/audit/AuditRepositoryTestBase.java +++ b/repository/src/test/java/org/apache/atlas/repository/audit/AuditRepositoryTestBase.java @@ -21,6 +21,7 @@ package org.apache.atlas.repository.audit; import org.apache.atlas.EntityAuditEvent; import org.apache.atlas.typesystem.Referenceable; import org.apache.commons.lang.RandomStringUtils; +import org.testng.annotations.BeforeTest; import org.testng.annotations.Test; import java.util.ArrayList; @@ -36,6 +37,11 @@ public class AuditRepositoryTestBase { return RandomStringUtils.randomAlphanumeric(10); } + @BeforeTest + public void setUp() throws Exception{ + eventRepository = new InMemoryEntityAuditRepository(); + } + @Test public void testAddEvents() throws Exception { EntityAuditEvent event = new EntityAuditEvent(rand(), System.currentTimeMillis(), "u1",