| /** |
| * Licensed to the Apache Software Foundation (ASF) under one |
| * or more contributor license agreements. See the NOTICE file |
| * distributed with this work for additional information |
| * regarding copyright ownership. The ASF licenses this file |
| * to you under the Apache License, Version 2.0 (the |
| * "License"); you may not use this file except in compliance |
| * with the License. You may obtain a copy of the License at |
| * |
| * http://www.apache.org/licenses/LICENSE-2.0 |
| * |
| * Unless required by applicable law or agreed to in writing, software |
| * distributed under the License is distributed on an "AS IS" BASIS, |
| * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| * See the License for the specific language governing permissions and |
| * limitations under the License. |
| */ |
| |
| package org.apache.hadoop.yarn.server.resourcemanager; |
| |
| |
| import static org.mockito.Matchers.any; |
| import static org.mockito.Matchers.anyLong; |
| import static org.mockito.Matchers.isA; |
| import static org.mockito.Mockito.doAnswer; |
| import static org.mockito.Mockito.mock; |
| import static org.mockito.Mockito.never; |
| import static org.mockito.Mockito.times; |
| import static org.mockito.Mockito.verify; |
| import static org.mockito.Mockito.when; |
| |
| import java.io.IOException; |
| import java.nio.ByteBuffer; |
| import java.util.HashMap; |
| import java.util.List; |
| import java.util.concurrent.ConcurrentMap; |
| |
| import org.apache.commons.logging.Log; |
| import org.apache.commons.logging.LogFactory; |
| import org.apache.hadoop.conf.Configuration; |
| import org.apache.hadoop.io.DataOutputBuffer; |
| import org.apache.hadoop.security.AccessControlException; |
| import org.apache.hadoop.security.Credentials; |
| import org.apache.hadoop.service.Service; |
| import org.apache.hadoop.yarn.MockApps; |
| import org.apache.hadoop.yarn.api.protocolrecords.SubmitApplicationRequest; |
| import org.apache.hadoop.yarn.api.records.ApplicationAccessType; |
| import org.apache.hadoop.yarn.api.records.ApplicationId; |
| import org.apache.hadoop.yarn.api.records.ApplicationSubmissionContext; |
| import org.apache.hadoop.yarn.api.records.ContainerLaunchContext; |
| import org.apache.hadoop.yarn.api.records.Priority; |
| import org.apache.hadoop.yarn.api.records.Resource; |
| import org.apache.hadoop.yarn.api.records.ResourceRequest; |
| import org.apache.hadoop.yarn.conf.YarnConfiguration; |
| import org.apache.hadoop.yarn.event.AsyncDispatcher; |
| import org.apache.hadoop.yarn.event.Dispatcher; |
| import org.apache.hadoop.yarn.event.EventHandler; |
| import org.apache.hadoop.yarn.exceptions.YarnException; |
| import org.apache.hadoop.yarn.factories.RecordFactory; |
| import org.apache.hadoop.yarn.factory.providers.RecordFactoryProvider; |
| import org.apache.hadoop.yarn.server.resourcemanager.ahs.RMApplicationHistoryWriter; |
| import org.apache.hadoop.yarn.server.resourcemanager.metrics.SystemMetricsPublisher; |
| import org.apache.hadoop.yarn.server.resourcemanager.placement.PlacementManager; |
| import org.apache.hadoop.yarn.server.resourcemanager.recovery.RMStateStore; |
| import org.apache.hadoop.yarn.server.resourcemanager.rmapp.MockRMApp; |
| import org.apache.hadoop.yarn.server.resourcemanager.rmapp.RMApp; |
| import org.apache.hadoop.yarn.server.resourcemanager.rmapp.RMAppEvent; |
| import org.apache.hadoop.yarn.server.resourcemanager.rmapp.RMAppEventType; |
| import org.apache.hadoop.yarn.server.resourcemanager.rmapp.RMAppImpl; |
| import org.apache.hadoop.yarn.server.resourcemanager.rmapp.RMAppMetrics; |
| import org.apache.hadoop.yarn.server.resourcemanager.rmapp.RMAppState; |
| import org.apache.hadoop.yarn.server.resourcemanager.rmapp.attempt.AMLivelinessMonitor; |
| import org.apache.hadoop.yarn.server.resourcemanager.rmcontainer.ContainerAllocationExpirer; |
| import org.apache.hadoop.yarn.server.resourcemanager.scheduler.ResourceScheduler; |
| import org.apache.hadoop.yarn.server.resourcemanager.scheduler.YarnScheduler; |
| import org.apache.hadoop.yarn.server.resourcemanager.scheduler.capacity.CapacityScheduler; |
| import org.apache.hadoop.yarn.server.resourcemanager.security.ClientToAMTokenSecretManagerInRM; |
| import org.apache.hadoop.yarn.server.resourcemanager.timelineservice.RMTimelineCollectorManager; |
| import org.apache.hadoop.yarn.server.security.ApplicationACLsManager; |
| import org.apache.hadoop.yarn.util.Records; |
| import org.apache.hadoop.yarn.util.resource.ResourceCalculator; |
| import org.apache.hadoop.yarn.util.resource.Resources; |
| import static org.apache.hadoop.yarn.server.resourcemanager.scheduler.capacity.CapacitySchedulerConfiguration.PREFIX; |
| import org.junit.After; |
| import org.junit.Assert; |
| import org.junit.Before; |
| import org.junit.Test; |
| import org.mockito.invocation.InvocationOnMock; |
| import org.mockito.stubbing.Answer; |
| |
| import com.google.common.collect.Lists; |
| import com.google.common.collect.Maps; |
| |
| /** |
| * Testing applications being retired from RM. |
| * |
| */ |
| |
| public class TestAppManager{ |
| private Log LOG = LogFactory.getLog(TestAppManager.class); |
| private static RMAppEventType appEventType = RMAppEventType.KILL; |
| |
| public synchronized RMAppEventType getAppEventType() { |
| return appEventType; |
| } |
| public synchronized void setAppEventType(RMAppEventType newType) { |
| appEventType = newType; |
| } |
| |
| |
| public static List<RMApp> newRMApps(int n, long time, RMAppState state) { |
| List<RMApp> list = Lists.newArrayList(); |
| for (int i = 0; i < n; ++i) { |
| list.add(new MockRMApp(i, time, state)); |
| } |
| return list; |
| } |
| |
| public RMContext mockRMContext(int n, long time) { |
| final List<RMApp> apps = newRMApps(n, time, RMAppState.FINISHED); |
| final ConcurrentMap<ApplicationId, RMApp> map = Maps.newConcurrentMap(); |
| for (RMApp app : apps) { |
| map.put(app.getApplicationId(), app); |
| } |
| Dispatcher rmDispatcher = new AsyncDispatcher(); |
| ContainerAllocationExpirer containerAllocationExpirer = new ContainerAllocationExpirer( |
| rmDispatcher); |
| AMLivelinessMonitor amLivelinessMonitor = new AMLivelinessMonitor( |
| rmDispatcher); |
| AMLivelinessMonitor amFinishingMonitor = new AMLivelinessMonitor( |
| rmDispatcher); |
| RMApplicationHistoryWriter writer = mock(RMApplicationHistoryWriter.class); |
| RMContext context = new RMContextImpl(rmDispatcher, |
| containerAllocationExpirer, amLivelinessMonitor, amFinishingMonitor, |
| null, null, null, null, null) { |
| @Override |
| public ConcurrentMap<ApplicationId, RMApp> getRMApps() { |
| return map; |
| } |
| }; |
| ((RMContextImpl)context).setStateStore(mock(RMStateStore.class)); |
| metricsPublisher = mock(SystemMetricsPublisher.class); |
| context.setSystemMetricsPublisher(metricsPublisher); |
| context.setRMApplicationHistoryWriter(writer); |
| return context; |
| } |
| |
| public class TestAppManagerDispatcher implements |
| EventHandler<RMAppManagerEvent> { |
| |
| |
| public TestAppManagerDispatcher() { |
| } |
| |
| @Override |
| public void handle(RMAppManagerEvent event) { |
| // do nothing |
| } |
| } |
| |
| public class TestDispatcher implements |
| EventHandler<RMAppEvent> { |
| |
| public TestDispatcher() { |
| } |
| |
| @Override |
| public void handle(RMAppEvent event) { |
| //RMApp rmApp = this.rmContext.getRMApps().get(appID); |
| setAppEventType(event.getType()); |
| System.out.println("in handle routine " + getAppEventType().toString()); |
| } |
| } |
| |
| |
| // Extend and make the functions we want to test public |
| public class TestRMAppManager extends RMAppManager { |
| |
| public TestRMAppManager(RMContext context, Configuration conf) { |
| super(context, null, null, new ApplicationACLsManager(conf), conf); |
| } |
| |
| public TestRMAppManager(RMContext context, |
| ClientToAMTokenSecretManagerInRM clientToAMSecretManager, |
| YarnScheduler scheduler, ApplicationMasterService masterService, |
| ApplicationACLsManager applicationACLsManager, Configuration conf) { |
| super(context, scheduler, masterService, applicationACLsManager, conf); |
| } |
| |
| public void checkAppNumCompletedLimit() { |
| super.checkAppNumCompletedLimit(); |
| } |
| |
| public void finishApplication(ApplicationId appId) { |
| super.finishApplication(appId); |
| } |
| |
| public int getCompletedAppsListSize() { |
| return super.getCompletedAppsListSize(); |
| } |
| |
| public int getCompletedAppsInStateStore() { |
| return this.completedAppsInStateStore; |
| } |
| |
| public void submitApplication( |
| ApplicationSubmissionContext submissionContext, String user) |
| throws YarnException, IOException { |
| super.submitApplication(submissionContext, System.currentTimeMillis(), |
| user); |
| } |
| } |
| |
| protected void addToCompletedApps(TestRMAppManager appMonitor, RMContext rmContext) { |
| for (RMApp app : rmContext.getRMApps().values()) { |
| if (app.getState() == RMAppState.FINISHED |
| || app.getState() == RMAppState.KILLED |
| || app.getState() == RMAppState.FAILED) { |
| appMonitor.finishApplication(app.getApplicationId()); |
| } |
| } |
| } |
| |
| private RMContext rmContext; |
| private SystemMetricsPublisher metricsPublisher; |
| private TestRMAppManager appMonitor; |
| private ApplicationSubmissionContext asContext; |
| private ApplicationId appId; |
| |
| @SuppressWarnings("deprecation") |
| @Before |
| public void setUp() { |
| long now = System.currentTimeMillis(); |
| |
| rmContext = mockRMContext(1, now - 10); |
| rmContext |
| .setRMTimelineCollectorManager(mock(RMTimelineCollectorManager.class)); |
| ResourceScheduler scheduler = mockResourceScheduler(); |
| ((RMContextImpl)rmContext).setScheduler(scheduler); |
| Configuration conf = new Configuration(); |
| ((RMContextImpl) rmContext).setYarnConfiguration(conf); |
| ApplicationMasterService masterService = |
| new ApplicationMasterService(rmContext, scheduler); |
| appMonitor = new TestRMAppManager(rmContext, |
| new ClientToAMTokenSecretManagerInRM(), scheduler, masterService, |
| new ApplicationACLsManager(conf), conf); |
| |
| appId = MockApps.newAppID(1); |
| RecordFactory recordFactory = RecordFactoryProvider.getRecordFactory(null); |
| asContext = |
| recordFactory.newRecordInstance(ApplicationSubmissionContext.class); |
| asContext.setApplicationId(appId); |
| asContext.setAMContainerSpec(mockContainerLaunchContext(recordFactory)); |
| asContext.setResource(mockResource()); |
| asContext.setPriority(Priority.newInstance(0)); |
| setupDispatcher(rmContext, conf); |
| } |
| |
| @After |
| public void tearDown() { |
| setAppEventType(RMAppEventType.KILL); |
| ((Service)rmContext.getDispatcher()).stop(); |
| } |
| |
| @Test |
| public void testRMAppRetireNone() throws Exception { |
| long now = System.currentTimeMillis(); |
| |
| // Create such that none of the applications will retire since |
| // haven't hit max # |
| RMContext rmContext = mockRMContext(10, now - 10); |
| Configuration conf = new YarnConfiguration(); |
| conf.setInt(YarnConfiguration.RM_MAX_COMPLETED_APPLICATIONS, 10); |
| TestRMAppManager appMonitor = new TestRMAppManager(rmContext,conf); |
| |
| Assert.assertEquals("Number of apps incorrect before checkAppTimeLimit", |
| 10, rmContext.getRMApps().size()); |
| |
| // add them to completed apps list |
| addToCompletedApps(appMonitor, rmContext); |
| |
| // shouldn't have to many apps |
| appMonitor.checkAppNumCompletedLimit(); |
| Assert.assertEquals("Number of apps incorrect after # completed check", 10, |
| rmContext.getRMApps().size()); |
| Assert.assertEquals("Number of completed apps incorrect after check", 10, |
| appMonitor.getCompletedAppsListSize()); |
| verify(rmContext.getStateStore(), never()).removeApplication( |
| isA(RMApp.class)); |
| } |
| |
| @Test |
| public void testQueueSubmitWithNoPermission() throws IOException { |
| YarnConfiguration conf = new YarnConfiguration(); |
| conf.set(YarnConfiguration.RM_SCHEDULER, |
| CapacityScheduler.class.getCanonicalName()); |
| conf.set(PREFIX + "root.acl_submit_applications", " "); |
| conf.set(PREFIX + "root.acl_administer_queue", " "); |
| |
| conf.set(PREFIX + "root.default.acl_submit_applications", " "); |
| conf.set(PREFIX + "root.default.acl_administer_queue", " "); |
| conf.set(YarnConfiguration.YARN_ACL_ENABLE, "true"); |
| MockRM mockRM = new MockRM(conf); |
| ClientRMService rmService = mockRM.getClientRMService(); |
| SubmitApplicationRequest req = |
| Records.newRecord(SubmitApplicationRequest.class); |
| ApplicationSubmissionContext sub = |
| Records.newRecord(ApplicationSubmissionContext.class); |
| sub.setApplicationId(appId); |
| ResourceRequest resReg = |
| ResourceRequest.newInstance(Priority.newInstance(0), |
| ResourceRequest.ANY, Resource.newInstance(1024, 1), 1); |
| sub.setAMContainerResourceRequest(resReg); |
| req.setApplicationSubmissionContext(sub); |
| try { |
| rmService.submitApplication(req); |
| } catch (Exception e) { |
| if (e instanceof YarnException) { |
| Assert.assertTrue(e.getCause() instanceof AccessControlException); |
| } else { |
| Assert.fail("Yarn exception is expected"); |
| } |
| } finally { |
| mockRM.close(); |
| } |
| } |
| |
| @Test |
| public void testRMAppRetireSome() throws Exception { |
| long now = System.currentTimeMillis(); |
| |
| RMContext rmContext = mockRMContext(10, now - 20000); |
| Configuration conf = new YarnConfiguration(); |
| conf.setInt(YarnConfiguration.RM_STATE_STORE_MAX_COMPLETED_APPLICATIONS, 3); |
| conf.setInt(YarnConfiguration.RM_MAX_COMPLETED_APPLICATIONS, 3); |
| TestRMAppManager appMonitor = new TestRMAppManager(rmContext, conf); |
| |
| Assert.assertEquals("Number of apps incorrect before", 10, rmContext |
| .getRMApps().size()); |
| |
| // add them to completed apps list |
| addToCompletedApps(appMonitor, rmContext); |
| |
| // shouldn't have to many apps |
| appMonitor.checkAppNumCompletedLimit(); |
| Assert.assertEquals("Number of apps incorrect after # completed check", 3, |
| rmContext.getRMApps().size()); |
| Assert.assertEquals("Number of completed apps incorrect after check", 3, |
| appMonitor.getCompletedAppsListSize()); |
| verify(rmContext.getStateStore(), times(7)).removeApplication( |
| isA(RMApp.class)); |
| } |
| |
| @Test |
| public void testRMAppRetireSomeDifferentStates() throws Exception { |
| long now = System.currentTimeMillis(); |
| |
| // these parameters don't matter, override applications below |
| RMContext rmContext = mockRMContext(10, now - 20000); |
| Configuration conf = new YarnConfiguration(); |
| conf.setInt(YarnConfiguration.RM_STATE_STORE_MAX_COMPLETED_APPLICATIONS, 2); |
| conf.setInt(YarnConfiguration.RM_MAX_COMPLETED_APPLICATIONS, 2); |
| |
| TestRMAppManager appMonitor = new TestRMAppManager(rmContext, conf); |
| |
| // clear out applications map |
| rmContext.getRMApps().clear(); |
| Assert.assertEquals("map isn't empty", 0, rmContext.getRMApps().size()); |
| |
| // 6 applications are in final state, 4 are not in final state. |
| // / set with various finished states |
| RMApp app = new MockRMApp(0, now - 20000, RMAppState.KILLED); |
| rmContext.getRMApps().put(app.getApplicationId(), app); |
| app = new MockRMApp(1, now - 200000, RMAppState.FAILED); |
| rmContext.getRMApps().put(app.getApplicationId(), app); |
| app = new MockRMApp(2, now - 30000, RMAppState.FINISHED); |
| rmContext.getRMApps().put(app.getApplicationId(), app); |
| app = new MockRMApp(3, now - 20000, RMAppState.RUNNING); |
| rmContext.getRMApps().put(app.getApplicationId(), app); |
| app = new MockRMApp(4, now - 20000, RMAppState.NEW); |
| rmContext.getRMApps().put(app.getApplicationId(), app); |
| |
| // make sure it doesn't expire these since still running |
| app = new MockRMApp(5, now - 10001, RMAppState.KILLED); |
| rmContext.getRMApps().put(app.getApplicationId(), app); |
| app = new MockRMApp(6, now - 30000, RMAppState.ACCEPTED); |
| rmContext.getRMApps().put(app.getApplicationId(), app); |
| app = new MockRMApp(7, now - 20000, RMAppState.SUBMITTED); |
| rmContext.getRMApps().put(app.getApplicationId(), app); |
| app = new MockRMApp(8, now - 10001, RMAppState.FAILED); |
| rmContext.getRMApps().put(app.getApplicationId(), app); |
| app = new MockRMApp(9, now - 20000, RMAppState.FAILED); |
| rmContext.getRMApps().put(app.getApplicationId(), app); |
| |
| Assert.assertEquals("Number of apps incorrect before", 10, rmContext |
| .getRMApps().size()); |
| |
| // add them to completed apps list |
| addToCompletedApps(appMonitor, rmContext); |
| |
| // shouldn't have to many apps |
| appMonitor.checkAppNumCompletedLimit(); |
| Assert.assertEquals("Number of apps incorrect after # completed check", 6, |
| rmContext.getRMApps().size()); |
| Assert.assertEquals("Number of completed apps incorrect after check", 2, |
| appMonitor.getCompletedAppsListSize()); |
| // 6 applications in final state, 4 of them are removed |
| verify(rmContext.getStateStore(), times(4)).removeApplication( |
| isA(RMApp.class)); |
| } |
| |
| @Test |
| public void testRMAppRetireNullApp() throws Exception { |
| long now = System.currentTimeMillis(); |
| |
| RMContext rmContext = mockRMContext(10, now - 20000); |
| TestRMAppManager appMonitor = new TestRMAppManager(rmContext, new Configuration()); |
| |
| Assert.assertEquals("Number of apps incorrect before", 10, rmContext |
| .getRMApps().size()); |
| |
| appMonitor.finishApplication(null); |
| |
| Assert.assertEquals("Number of completed apps incorrect after check", 0, |
| appMonitor.getCompletedAppsListSize()); |
| } |
| |
| @Test |
| public void testRMAppRetireZeroSetting() throws Exception { |
| long now = System.currentTimeMillis(); |
| |
| RMContext rmContext = mockRMContext(10, now - 20000); |
| Configuration conf = new YarnConfiguration(); |
| conf.setInt(YarnConfiguration.RM_STATE_STORE_MAX_COMPLETED_APPLICATIONS, 0); |
| conf.setInt(YarnConfiguration.RM_MAX_COMPLETED_APPLICATIONS, 0); |
| TestRMAppManager appMonitor = new TestRMAppManager(rmContext, conf); |
| Assert.assertEquals("Number of apps incorrect before", 10, rmContext |
| .getRMApps().size()); |
| |
| addToCompletedApps(appMonitor, rmContext); |
| Assert.assertEquals("Number of completed apps incorrect", 10, |
| appMonitor.getCompletedAppsListSize()); |
| |
| appMonitor.checkAppNumCompletedLimit(); |
| |
| Assert.assertEquals("Number of apps incorrect after # completed check", 0, |
| rmContext.getRMApps().size()); |
| Assert.assertEquals("Number of completed apps incorrect after check", 0, |
| appMonitor.getCompletedAppsListSize()); |
| verify(rmContext.getStateStore(), times(10)).removeApplication( |
| isA(RMApp.class)); |
| } |
| |
| @Test |
| public void testStateStoreAppLimitLessThanMemoryAppLimit() { |
| long now = System.currentTimeMillis(); |
| RMContext rmContext = mockRMContext(10, now - 20000); |
| Configuration conf = new YarnConfiguration(); |
| int maxAppsInMemory = 8; |
| int maxAppsInStateStore = 4; |
| conf.setInt(YarnConfiguration.RM_MAX_COMPLETED_APPLICATIONS, maxAppsInMemory); |
| conf.setInt(YarnConfiguration.RM_STATE_STORE_MAX_COMPLETED_APPLICATIONS, |
| maxAppsInStateStore); |
| TestRMAppManager appMonitor = new TestRMAppManager(rmContext, conf); |
| |
| addToCompletedApps(appMonitor, rmContext); |
| Assert.assertEquals("Number of completed apps incorrect", 10, |
| appMonitor.getCompletedAppsListSize()); |
| appMonitor.checkAppNumCompletedLimit(); |
| |
| Assert.assertEquals("Number of apps incorrect after # completed check", |
| maxAppsInMemory, rmContext.getRMApps().size()); |
| Assert.assertEquals("Number of completed apps incorrect after check", |
| maxAppsInMemory, appMonitor.getCompletedAppsListSize()); |
| |
| int numRemoveAppsFromStateStore = 10 - maxAppsInStateStore; |
| verify(rmContext.getStateStore(), times(numRemoveAppsFromStateStore)) |
| .removeApplication(isA(RMApp.class)); |
| Assert.assertEquals(maxAppsInStateStore, |
| appMonitor.getCompletedAppsInStateStore()); |
| } |
| |
| @Test |
| public void testStateStoreAppLimitLargerThanMemoryAppLimit() { |
| long now = System.currentTimeMillis(); |
| RMContext rmContext = mockRMContext(10, now - 20000); |
| Configuration conf = new YarnConfiguration(); |
| int maxAppsInMemory = 8; |
| conf.setInt(YarnConfiguration.RM_MAX_COMPLETED_APPLICATIONS, maxAppsInMemory); |
| // larger than maxCompletedAppsInMemory, reset to RM_MAX_COMPLETED_APPLICATIONS. |
| conf.setInt(YarnConfiguration.RM_STATE_STORE_MAX_COMPLETED_APPLICATIONS, 1000); |
| TestRMAppManager appMonitor = new TestRMAppManager(rmContext, conf); |
| |
| addToCompletedApps(appMonitor, rmContext); |
| Assert.assertEquals("Number of completed apps incorrect", 10, |
| appMonitor.getCompletedAppsListSize()); |
| appMonitor.checkAppNumCompletedLimit(); |
| |
| int numRemoveApps = 10 - maxAppsInMemory; |
| Assert.assertEquals("Number of apps incorrect after # completed check", |
| maxAppsInMemory, rmContext.getRMApps().size()); |
| Assert.assertEquals("Number of completed apps incorrect after check", |
| maxAppsInMemory, appMonitor.getCompletedAppsListSize()); |
| verify(rmContext.getStateStore(), times(numRemoveApps)).removeApplication( |
| isA(RMApp.class)); |
| Assert.assertEquals(maxAppsInMemory, |
| appMonitor.getCompletedAppsInStateStore()); |
| } |
| |
| protected void setupDispatcher(RMContext rmContext, Configuration conf) { |
| TestDispatcher testDispatcher = new TestDispatcher(); |
| TestAppManagerDispatcher testAppManagerDispatcher = |
| new TestAppManagerDispatcher(); |
| rmContext.getDispatcher().register(RMAppEventType.class, testDispatcher); |
| rmContext.getDispatcher().register(RMAppManagerEventType.class, testAppManagerDispatcher); |
| ((Service)rmContext.getDispatcher()).init(conf); |
| ((Service)rmContext.getDispatcher()).start(); |
| Assert.assertEquals("app event type is wrong before", RMAppEventType.KILL, appEventType); |
| } |
| |
| @Test |
| public void testRMAppSubmit() throws Exception { |
| appMonitor.submitApplication(asContext, "test"); |
| RMApp app = rmContext.getRMApps().get(appId); |
| Assert.assertNotNull("app is null", app); |
| Assert.assertEquals("app id doesn't match", appId, app.getApplicationId()); |
| Assert.assertEquals("app state doesn't match", RMAppState.NEW, app.getState()); |
| verify(metricsPublisher).appACLsUpdated( |
| any(RMApp.class), any(String.class), anyLong()); |
| |
| // wait for event to be processed |
| int timeoutSecs = 0; |
| while ((getAppEventType() == RMAppEventType.KILL) && |
| timeoutSecs++ < 20) { |
| Thread.sleep(1000); |
| } |
| Assert.assertEquals("app event type sent is wrong", RMAppEventType.START, |
| getAppEventType()); |
| } |
| |
| @Test |
| public void testRMAppSubmitWithInvalidTokens() throws Exception { |
| // Setup invalid security tokens |
| DataOutputBuffer dob = new DataOutputBuffer(); |
| ByteBuffer securityTokens = ByteBuffer.wrap(dob.getData(), 0, |
| dob.getLength()); |
| asContext.getAMContainerSpec().setTokens(securityTokens); |
| try { |
| appMonitor.submitApplication(asContext, "test"); |
| Assert.fail("Application submission should fail because" + |
| " Tokens are invalid."); |
| } catch (YarnException e) { |
| // Exception is expected |
| Assert.assertTrue("The thrown exception is not" + |
| " java.io.EOFException", |
| e.getMessage().contains("java.io.EOFException")); |
| } |
| int timeoutSecs = 0; |
| while ((getAppEventType() == RMAppEventType.KILL) && |
| timeoutSecs++ < 20) { |
| Thread.sleep(1000); |
| } |
| Assert.assertEquals("app event type sent is wrong", |
| RMAppEventType.APP_REJECTED, getAppEventType()); |
| asContext.getAMContainerSpec().setTokens(null); |
| } |
| |
| @Test |
| public void testRMAppSubmitWithValidTokens() throws Exception { |
| // Setup valid security tokens |
| DataOutputBuffer dob = new DataOutputBuffer(); |
| Credentials credentials = new Credentials(); |
| credentials.writeTokenStorageToStream(dob); |
| ByteBuffer securityTokens = ByteBuffer.wrap(dob.getData(), 0, |
| dob.getLength()); |
| asContext.getAMContainerSpec().setTokens(securityTokens); |
| appMonitor.submitApplication(asContext, "test"); |
| RMApp app = rmContext.getRMApps().get(appId); |
| Assert.assertNotNull("app is null", app); |
| Assert.assertEquals("app id doesn't match", appId, |
| app.getApplicationId()); |
| Assert.assertEquals("app state doesn't match", RMAppState.NEW, |
| app.getState()); |
| verify(metricsPublisher).appACLsUpdated( |
| any(RMApp.class), any(String.class), anyLong()); |
| |
| // wait for event to be processed |
| int timeoutSecs = 0; |
| while ((getAppEventType() == RMAppEventType.KILL) && |
| timeoutSecs++ < 20) { |
| Thread.sleep(1000); |
| } |
| Assert.assertEquals("app event type sent is wrong", RMAppEventType.START, |
| getAppEventType()); |
| asContext.getAMContainerSpec().setTokens(null); |
| } |
| |
| @Test (timeout = 30000) |
| public void testRMAppSubmitMaxAppAttempts() throws Exception { |
| int[] globalMaxAppAttempts = new int[] { 10, 1 }; |
| int[][] individualMaxAppAttempts = new int[][]{ |
| new int[]{ 9, 10, 11, 0 }, |
| new int[]{ 1, 10, 0, -1 }}; |
| int[][] expectedNums = new int[][]{ |
| new int[]{ 9, 10, 10, 10 }, |
| new int[]{ 1, 1, 1, 1 }}; |
| for (int i = 0; i < globalMaxAppAttempts.length; ++i) { |
| for (int j = 0; j < individualMaxAppAttempts.length; ++j) { |
| ResourceScheduler scheduler = mockResourceScheduler(); |
| Configuration conf = new Configuration(); |
| conf.setInt(YarnConfiguration.RM_AM_MAX_ATTEMPTS, globalMaxAppAttempts[i]); |
| ApplicationMasterService masterService = |
| new ApplicationMasterService(rmContext, scheduler); |
| TestRMAppManager appMonitor = new TestRMAppManager(rmContext, |
| new ClientToAMTokenSecretManagerInRM(), scheduler, masterService, |
| new ApplicationACLsManager(conf), conf); |
| |
| ApplicationId appID = MockApps.newAppID(i * 4 + j + 1); |
| asContext.setApplicationId(appID); |
| if (individualMaxAppAttempts[i][j] != 0) { |
| asContext.setMaxAppAttempts(individualMaxAppAttempts[i][j]); |
| } |
| appMonitor.submitApplication(asContext, "test"); |
| RMApp app = rmContext.getRMApps().get(appID); |
| Assert.assertEquals("max application attempts doesn't match", |
| expectedNums[i][j], app.getMaxAppAttempts()); |
| |
| // wait for event to be processed |
| int timeoutSecs = 0; |
| while ((getAppEventType() == RMAppEventType.KILL) && |
| timeoutSecs++ < 20) { |
| Thread.sleep(1000); |
| } |
| setAppEventType(RMAppEventType.KILL); |
| } |
| } |
| } |
| |
| @Test (timeout = 30000) |
| public void testRMAppSubmitDuplicateApplicationId() throws Exception { |
| ApplicationId appId = MockApps.newAppID(0); |
| asContext.setApplicationId(appId); |
| RMApp appOrig = rmContext.getRMApps().get(appId); |
| Assert.assertTrue("app name matches but shouldn't", "testApp1" != appOrig.getName()); |
| |
| // our testApp1 should be rejected and original app with same id should be left in place |
| try { |
| appMonitor.submitApplication(asContext, "test"); |
| Assert.fail("Exception is expected when applicationId is duplicate."); |
| } catch (YarnException e) { |
| Assert.assertTrue("The thrown exception is not the expectd one.", |
| e.getMessage().contains("Cannot add a duplicate!")); |
| } |
| |
| // make sure original app didn't get removed |
| RMApp app = rmContext.getRMApps().get(appId); |
| Assert.assertNotNull("app is null", app); |
| Assert.assertEquals("app id doesn't match", appId, app.getApplicationId()); |
| Assert.assertEquals("app state doesn't match", RMAppState.FINISHED, app.getState()); |
| } |
| |
| @SuppressWarnings("deprecation") |
| @Test (timeout = 30000) |
| public void testRMAppSubmitInvalidResourceRequest() throws Exception { |
| asContext.setResource(Resources.createResource( |
| YarnConfiguration.DEFAULT_RM_SCHEDULER_MAXIMUM_ALLOCATION_MB + 1)); |
| |
| // submit an app |
| try { |
| appMonitor.submitApplication(asContext, "test"); |
| Assert.fail("Application submission should fail because resource" + |
| " request is invalid."); |
| } catch (YarnException e) { |
| // Exception is expected |
| // TODO Change this to assert the expected exception type - post YARN-142 |
| // sub-task related to specialized exceptions. |
| Assert.assertTrue("The thrown exception is not" + |
| " InvalidResourceRequestException", |
| e.getMessage().contains("Invalid resource request")); |
| } |
| } |
| |
| @Test (timeout = 30000) |
| public void testEscapeApplicationSummary() { |
| RMApp app = mock(RMAppImpl.class); |
| when(app.getApplicationId()).thenReturn( |
| ApplicationId.newInstance(100L, 1)); |
| when(app.getName()).thenReturn("Multiline\n\n\r\rAppName"); |
| when(app.getUser()).thenReturn("Multiline\n\n\r\rUserName"); |
| when(app.getQueue()).thenReturn("Multiline\n\n\r\rQueueName"); |
| when(app.getState()).thenReturn(RMAppState.RUNNING); |
| when(app.getApplicationType()).thenReturn("MAPREDUCE"); |
| RMAppMetrics metrics = |
| new RMAppMetrics(Resource.newInstance(1234, 56), |
| 10, 1, 16384, 64, 0, 0); |
| when(app.getRMAppMetrics()).thenReturn(metrics); |
| |
| RMAppManager.ApplicationSummary.SummaryBuilder summary = |
| new RMAppManager.ApplicationSummary().createAppSummary(app); |
| String msg = summary.toString(); |
| LOG.info("summary: " + msg); |
| Assert.assertFalse(msg.contains("\n")); |
| Assert.assertFalse(msg.contains("\r")); |
| |
| String escaped = "\\n\\n\\r\\r"; |
| Assert.assertTrue(msg.contains("Multiline" + escaped +"AppName")); |
| Assert.assertTrue(msg.contains("Multiline" + escaped +"UserName")); |
| Assert.assertTrue(msg.contains("Multiline" + escaped +"QueueName")); |
| Assert.assertTrue(msg.contains("memorySeconds=16384")); |
| Assert.assertTrue(msg.contains("vcoreSeconds=64")); |
| Assert.assertTrue(msg.contains("preemptedAMContainers=1")); |
| Assert.assertTrue(msg.contains("preemptedNonAMContainers=10")); |
| Assert.assertTrue(msg.contains("preemptedResources=<memory:1234\\, vCores:56>")); |
| Assert.assertTrue(msg.contains("applicationType=MAPREDUCE")); |
| } |
| |
| @Test |
| public void testRMAppSubmitWithQueueChanged() throws Exception { |
| // Setup a PlacementManager returns a new queue |
| PlacementManager placementMgr = mock(PlacementManager.class); |
| doAnswer(new Answer<Void>() { |
| |
| @Override |
| public Void answer(InvocationOnMock invocation) throws Throwable { |
| ApplicationSubmissionContext ctx = |
| (ApplicationSubmissionContext) invocation.getArguments()[0]; |
| ctx.setQueue("newQueue"); |
| return null; |
| } |
| |
| }).when(placementMgr).placeApplication(any(ApplicationSubmissionContext.class), |
| any(String.class)); |
| rmContext.setQueuePlacementManager(placementMgr); |
| |
| asContext.setQueue("oldQueue"); |
| appMonitor.submitApplication(asContext, "test"); |
| RMApp app = rmContext.getRMApps().get(appId); |
| Assert.assertNotNull("app is null", app); |
| Assert.assertEquals("newQueue", asContext.getQueue()); |
| |
| // wait for event to be processed |
| int timeoutSecs = 0; |
| while ((getAppEventType() == RMAppEventType.KILL) && timeoutSecs++ < 20) { |
| Thread.sleep(1000); |
| } |
| Assert.assertEquals("app event type sent is wrong", RMAppEventType.START, |
| getAppEventType()); |
| } |
| |
| private static ResourceScheduler mockResourceScheduler() { |
| ResourceScheduler scheduler = mock(ResourceScheduler.class); |
| when(scheduler.getMinimumResourceCapability()).thenReturn( |
| Resources.createResource( |
| YarnConfiguration.DEFAULT_RM_SCHEDULER_MINIMUM_ALLOCATION_MB)); |
| when(scheduler.getMaximumResourceCapability()).thenReturn( |
| Resources.createResource( |
| YarnConfiguration.DEFAULT_RM_SCHEDULER_MAXIMUM_ALLOCATION_MB)); |
| |
| ResourceCalculator rs = mock(ResourceCalculator.class); |
| when(scheduler.getResourceCalculator()).thenReturn(rs); |
| |
| return scheduler; |
| } |
| |
| private static ContainerLaunchContext mockContainerLaunchContext( |
| RecordFactory recordFactory) { |
| ContainerLaunchContext amContainer = recordFactory.newRecordInstance( |
| ContainerLaunchContext.class); |
| amContainer.setApplicationACLs(new HashMap<ApplicationAccessType, String>());; |
| return amContainer; |
| } |
| |
| private static Resource mockResource() { |
| return Resources.createResource( |
| YarnConfiguration.DEFAULT_RM_SCHEDULER_MINIMUM_ALLOCATION_MB); |
| } |
| |
| } |
