|
@@ -39,6 +39,7 @@ import org.apache.hadoop.yarn.exceptions.YarnException;
|
|
|
import org.apache.hadoop.yarn.exceptions.YarnRuntimeException;
|
|
|
import org.apache.hadoop.yarn.server.resourcemanager.rmapp.attempt.RMAppAttemptState;
|
|
|
import org.apache.hadoop.yarn.server.resourcemanager.rmnode.RMNode;
|
|
|
+import org.apache.hadoop.yarn.server.resourcemanager.scheduler.capacity.CapacityScheduler;
|
|
|
import org.apache.hadoop.yarn.server.resourcemanager.scheduler.event.AppAttemptRemovedSchedulerEvent;
|
|
|
import org.apache.hadoop.yarn.server.resourcemanager.scheduler.event.NodeAddedSchedulerEvent;
|
|
|
import org.apache.hadoop.yarn.server.resourcemanager.scheduler.event.NodeUpdateSchedulerEvent;
|
|
@@ -57,6 +58,8 @@ public class TestResourceManager {
|
|
|
@Before
|
|
|
public void setUp() throws Exception {
|
|
|
Configuration conf = new YarnConfiguration();
|
|
|
+ conf.set(YarnConfiguration.RM_SCHEDULER,
|
|
|
+ CapacityScheduler.class.getCanonicalName());
|
|
|
UserGroupInformation.setConfiguration(conf);
|
|
|
resourceManager = new ResourceManager();
|
|
|
resourceManager.init(conf);
|
|
@@ -261,6 +264,8 @@ public class TestResourceManager {
|
|
|
}
|
|
|
};
|
|
|
Configuration conf = new YarnConfiguration();
|
|
|
+ conf.set(YarnConfiguration.RM_SCHEDULER,
|
|
|
+ CapacityScheduler.class.getCanonicalName());
|
|
|
conf.set(filterInitializerConfKey, filterInitializer);
|
|
|
conf.set("hadoop.security.authentication", "kerberos");
|
|
|
conf.set("hadoop.http.authentication.type", "kerberos");
|
|
@@ -295,6 +300,8 @@ public class TestResourceManager {
|
|
|
for (String filterInitializer : simpleFilterInitializers) {
|
|
|
resourceManager = new ResourceManager();
|
|
|
Configuration conf = new YarnConfiguration();
|
|
|
+ conf.set(YarnConfiguration.RM_SCHEDULER,
|
|
|
+ CapacityScheduler.class.getCanonicalName());
|
|
|
conf.set(filterInitializerConfKey, filterInitializer);
|
|
|
try {
|
|
|
UserGroupInformation.setConfiguration(conf);
|