commit b5f39b2fccdedced0f3755bc2c7a28f6b98ff0df
author Karthik Ramasamy <karthik@streaml.io> Tue Jan 30 00:02:54 2018 -0800
committer Karthik Ramasamy <karthik@streaml.io> Tue Jan 30 00:02:54 2018 -0800
tree 59437039743bd048c15de82d1275976f31404b55
parent 64b3cbd0ca49c187853f00c575a50fa98623d1fa

fix the AuroraSchedulerTest

heron/schedulers/tests/java/com/twitter/heron/scheduler/aurora/AuroraSchedulerTest.java

diff --git a/heron/schedulers/tests/java/com/twitter/heron/scheduler/aurora/AuroraSchedulerTest.java b/heron/schedulers/tests/java/com/twitter/heron/scheduler/aurora/AuroraSchedulerTest.java
index 2a8a015..490bcd0 100644
--- a/heron/schedulers/tests/java/com/twitter/heron/scheduler/aurora/AuroraSchedulerTest.java
+++ b/heron/schedulers/tests/java/com/twitter/heron/scheduler/aurora/AuroraSchedulerTest.java
@@ -213,6 +213,8 @@
     PowerMockito.mockStatic(Config.class);
     when(Config.toClusterMode(mockConfig)).thenReturn(mockConfig);
 
+    AuroraController controller = Mockito.mock(AuroraController.class);
+    doReturn(controller).when(scheduler).getController();
     scheduler.initialize(mockConfig, Mockito.mock(Config.class));
 
     PowerMockito.spy(TokenSub.class);