Browse Source

YARN-1760. TestRMAdminService assumes CapacityScheduler. (kasha)

git-svn-id: https://svn.apache.org/repos/asf/hadoop/common/branches/branch-2@1571778 13f79535-47bb-0310-9956-ffa450edef68
Karthik Kambatla 11 years ago
parent
commit
d4b882d09f

+ 2 - 0
hadoop-yarn-project/CHANGES.txt

@@ -337,6 +337,8 @@ Release 2.4.0 - UNRELEASED
     transits from standby to active mode so as to assimilate any changes that
     transits from standby to active mode so as to assimilate any changes that
     happened while it was in standby mode. (Xuan Gong via vinodkv)
     happened while it was in standby mode. (Xuan Gong via vinodkv)
 
 
+    YARN-1760. TestRMAdminService assumes CapacityScheduler. (kasha)
+
 Release 2.3.1 - UNRELEASED
 Release 2.3.1 - UNRELEASED
 
 
   INCOMPATIBLE CHANGES
   INCOMPATIBLE CHANGES

+ 10 - 2
hadoop-yarn-project/hadoop-yarn/hadoop-yarn-server/hadoop-yarn-server-resourcemanager/src/test/java/org/apache/hadoop/yarn/server/resourcemanager/TestRMAdminService.java

@@ -62,7 +62,7 @@ import org.junit.Test;
 
 
 public class TestRMAdminService {
 public class TestRMAdminService {
 
 
-  private final Configuration configuration = new YarnConfiguration();
+  private Configuration configuration;;
   private MockRM rm = null;
   private MockRM rm = null;
   private FileSystem fs;
   private FileSystem fs;
   private Path workingPath;
   private Path workingPath;
@@ -70,7 +70,7 @@ public class TestRMAdminService {
 
 
   @Before
   @Before
   public void setup() throws IOException {
   public void setup() throws IOException {
-    Configuration.addDefaultResource(YarnConfiguration.CS_CONFIGURATION_FILE);
+    configuration = new YarnConfiguration();
     fs = FileSystem.get(configuration);
     fs = FileSystem.get(configuration);
     workingPath =
     workingPath =
         new Path(new File("target", this.getClass().getSimpleName()
         new Path(new File("target", this.getClass().getSimpleName()
@@ -94,9 +94,16 @@ public class TestRMAdminService {
     fs.delete(tmpDir, true);
     fs.delete(tmpDir, true);
   }
   }
 
 
+  private void useCapacityScheduler() {
+    configuration.set(YarnConfiguration.RM_SCHEDULER,
+        CapacityScheduler.class.getCanonicalName());
+    configuration.addResource(YarnConfiguration.CS_CONFIGURATION_FILE);
+  }
+
   @Test
   @Test
   public void testAdminRefreshQueuesWithLocalConfigurationProvider()
   public void testAdminRefreshQueuesWithLocalConfigurationProvider()
       throws IOException, YarnException {
       throws IOException, YarnException {
+    useCapacityScheduler();
     rm = new MockRM(configuration);
     rm = new MockRM(configuration);
     rm.init(configuration);
     rm.init(configuration);
     rm.start();
     rm.start();
@@ -119,6 +126,7 @@ public class TestRMAdminService {
       throws IOException, YarnException {
       throws IOException, YarnException {
     configuration.set(YarnConfiguration.RM_CONFIGURATION_PROVIDER_CLASS,
     configuration.set(YarnConfiguration.RM_CONFIGURATION_PROVIDER_CLASS,
         "org.apache.hadoop.yarn.FileSystemBasedConfigurationProvider");
         "org.apache.hadoop.yarn.FileSystemBasedConfigurationProvider");
+    useCapacityScheduler();
     try {
     try {
       rm = new MockRM(configuration);
       rm = new MockRM(configuration);
       rm.init(configuration);
       rm.init(configuration);