add mock authentication

datahub-project · Jul 7, 2022 · 412c22e · 412c22e
1 parent 04f739d
commit 412c22e
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/metadata-service/openapi-servlet/src/test/java/entities/EntitiesControllerTest.java b/metadata-service/openapi-servlet/src/test/java/entities/EntitiesControllerTest.java
@@ -1,5 +1,9 @@
 package entities;
 
+import com.datahub.authentication.Actor;
+import com.datahub.authentication.ActorType;
+import com.datahub.authentication.Authentication;
+import com.datahub.authentication.AuthenticationContext;
 import com.fasterxml.jackson.databind.ObjectMapper;
 import com.linkedin.metadata.entity.AspectDao;
 import com.linkedin.metadata.event.EventProducer;
@@ -34,6 +38,7 @@
 import org.testng.annotations.Test;
 
 import static com.linkedin.metadata.Constants.*;
+import static org.mockito.Mockito.when;
 
 
 public class EntitiesControllerTest {
@@ -53,6 +58,9 @@ public void setup()
     EventProducer mockEntityEventProducer = Mockito.mock(EventProducer.class);
     MockEntityService mockEntityService = new MockEntityService(aspectDao, mockEntityEventProducer, mockEntityRegistry);
     _entitiesController = new EntitiesController(mockEntityService, new ObjectMapper());
+    Authentication authentication = Mockito.mock(Authentication.class);
+    when(authentication.getActor()).thenReturn(new Actor(ActorType.USER, "datahub"));
+    AuthenticationContext.setAuthentication(authentication);
   }
 
   EntitiesController _entitiesController;