public virtual void TestApplicationInit2() { TestApplication.WrappedApplication wa = null; try { wa = new TestApplication.WrappedApplication(this, 2, 314159265358979L, "yak", 3); wa.InitApplication(); wa.InitContainer(0); NUnit.Framework.Assert.AreEqual(ApplicationState.Initing, wa.app.GetApplicationState ()); NUnit.Framework.Assert.AreEqual(1, wa.app.GetContainers().Count); wa.ApplicationInited(); NUnit.Framework.Assert.AreEqual(ApplicationState.Running, wa.app.GetApplicationState ()); Org.Mockito.Mockito.Verify(wa.containerBus).Handle(Matchers.ArgThat(new TestApplication.ContainerInitMatcher (this, wa.containers[0].GetContainerId()))); wa.InitContainer(1); wa.InitContainer(2); NUnit.Framework.Assert.AreEqual(ApplicationState.Running, wa.app.GetApplicationState ()); NUnit.Framework.Assert.AreEqual(3, wa.app.GetContainers().Count); for (int i = 1; i < wa.containers.Count; i++) { Org.Mockito.Mockito.Verify(wa.containerBus).Handle(Matchers.ArgThat(new TestApplication.ContainerInitMatcher (this, wa.containers[i].GetContainerId()))); } } finally { if (wa != null) { wa.Finished(); } } }
public virtual void TestNMTokenSecretManagerCleanup() { TestApplication.WrappedApplication wa = null; try { wa = new TestApplication.WrappedApplication(this, 1, 314159265358979L, "yak", 1); wa.InitApplication(); wa.InitContainer(0); NUnit.Framework.Assert.AreEqual(ApplicationState.Initing, wa.app.GetApplicationState ()); NUnit.Framework.Assert.AreEqual(1, wa.app.GetContainers().Count); wa.AppFinished(); wa.ContainerFinished(0); wa.AppResourcesCleanedup(); NUnit.Framework.Assert.AreEqual(ApplicationState.Finished, wa.app.GetApplicationState ()); Org.Mockito.Mockito.Verify(wa.nmTokenSecretMgr).AppFinished(Matchers.Eq(wa.appId) ); } finally { if (wa != null) { wa.Finished(); } } }
public virtual void TestContainersCompleteDuringAppInit2() { TestApplication.WrappedApplication wa = null; try { wa = new TestApplication.WrappedApplication(this, 3, 314159265358979L, "yak", 3); wa.InitApplication(); wa.InitContainer(-1); NUnit.Framework.Assert.AreEqual(ApplicationState.Initing, wa.app.GetApplicationState ()); wa.ContainerFinished(0); NUnit.Framework.Assert.AreEqual(ApplicationState.Initing, wa.app.GetApplicationState ()); wa.ApplicationInited(); NUnit.Framework.Assert.AreEqual(ApplicationState.Running, wa.app.GetApplicationState ()); NUnit.Framework.Assert.AreEqual(2, wa.app.GetContainers().Count); wa.ContainerFinished(1); wa.ContainerFinished(2); NUnit.Framework.Assert.AreEqual(ApplicationState.Running, wa.app.GetApplicationState ()); NUnit.Framework.Assert.AreEqual(0, wa.app.GetContainers().Count); } finally { if (wa != null) { wa.Finished(); } } }
public virtual void TestAppFinishedOnCompletedContainers() { TestApplication.WrappedApplication wa = null; try { wa = new TestApplication.WrappedApplication(this, 5, 314159265358979L, "yak", 3); wa.InitApplication(); wa.InitContainer(-1); NUnit.Framework.Assert.AreEqual(ApplicationState.Initing, wa.app.GetApplicationState ()); wa.ApplicationInited(); NUnit.Framework.Assert.AreEqual(ApplicationState.Running, wa.app.GetApplicationState ()); Org.Mockito.Mockito.Reset(wa.localizerBus); wa.ContainerFinished(0); wa.ContainerFinished(1); wa.ContainerFinished(2); NUnit.Framework.Assert.AreEqual(ApplicationState.Running, wa.app.GetApplicationState ()); NUnit.Framework.Assert.AreEqual(0, wa.app.GetContainers().Count); wa.AppFinished(); NUnit.Framework.Assert.AreEqual(ApplicationState.ApplicationResourcesCleaningup, wa.app.GetApplicationState()); Org.Mockito.Mockito.Verify(wa.localizerBus).Handle(Matchers.RefEq(new ApplicationLocalizationEvent (LocalizationEventType.DestroyApplicationResources, wa.app))); wa.AppResourcesCleanedup(); foreach (Org.Apache.Hadoop.Yarn.Server.Nodemanager.Containermanager.Container.Container container in wa.containers) { ContainerTokenIdentifier identifier = wa.GetContainerTokenIdentifier(container.GetContainerId ()); WaitForContainerTokenToExpire(identifier); NUnit.Framework.Assert.IsTrue(wa.context.GetContainerTokenSecretManager().IsValidStartContainerRequest (identifier)); } NUnit.Framework.Assert.AreEqual(ApplicationState.Finished, wa.app.GetApplicationState ()); } finally { if (wa != null) { wa.Finished(); } } }
//TODO Re-work after Application transitions are changed. // @Test public virtual void TestAppFinishedOnIniting() { // AM may send a startContainer() - AM APP_FINIHSED processed after // APP_FINISHED on another NM TestApplication.WrappedApplication wa = null; try { wa = new TestApplication.WrappedApplication(this, 1, 314159265358979L, "yak", 3); wa.InitApplication(); wa.InitContainer(0); NUnit.Framework.Assert.AreEqual(ApplicationState.Initing, wa.app.GetApplicationState ()); NUnit.Framework.Assert.AreEqual(1, wa.app.GetContainers().Count); Org.Mockito.Mockito.Reset(wa.localizerBus); wa.AppFinished(); Org.Mockito.Mockito.Verify(wa.containerBus).Handle(Matchers.ArgThat(new TestApplication.ContainerKillMatcher (this, wa.containers[0].GetContainerId()))); NUnit.Framework.Assert.AreEqual(ApplicationState.FinishingContainersWait, wa.app. GetApplicationState()); wa.ContainerFinished(0); NUnit.Framework.Assert.AreEqual(ApplicationState.ApplicationResourcesCleaningup, wa.app.GetApplicationState()); Org.Mockito.Mockito.Verify(wa.localizerBus).Handle(Matchers.RefEq(new ApplicationLocalizationEvent (LocalizationEventType.DestroyApplicationResources, wa.app))); wa.InitContainer(1); NUnit.Framework.Assert.AreEqual(ApplicationState.ApplicationResourcesCleaningup, wa.app.GetApplicationState()); NUnit.Framework.Assert.AreEqual(0, wa.app.GetContainers().Count); wa.AppResourcesCleanedup(); NUnit.Framework.Assert.AreEqual(ApplicationState.Finished, wa.app.GetApplicationState ()); } finally { if (wa != null) { wa.Finished(); } } }
//TODO Re-work after Application transitions are changed. // @Test public virtual void TestStartContainerAfterAppFinished() { TestApplication.WrappedApplication wa = null; try { wa = new TestApplication.WrappedApplication(this, 5, 314159265358979L, "yak", 3); wa.InitApplication(); wa.InitContainer(-1); NUnit.Framework.Assert.AreEqual(ApplicationState.Initing, wa.app.GetApplicationState ()); wa.ApplicationInited(); NUnit.Framework.Assert.AreEqual(ApplicationState.Running, wa.app.GetApplicationState ()); Org.Mockito.Mockito.Reset(wa.localizerBus); wa.ContainerFinished(0); wa.ContainerFinished(1); wa.ContainerFinished(2); NUnit.Framework.Assert.AreEqual(ApplicationState.Running, wa.app.GetApplicationState ()); NUnit.Framework.Assert.AreEqual(0, wa.app.GetContainers().Count); wa.AppFinished(); NUnit.Framework.Assert.AreEqual(ApplicationState.ApplicationResourcesCleaningup, wa.app.GetApplicationState()); Org.Mockito.Mockito.Verify(wa.localizerBus).Handle(Matchers.RefEq(new ApplicationLocalizationEvent (LocalizationEventType.DestroyApplicationResources, wa.app))); wa.AppResourcesCleanedup(); NUnit.Framework.Assert.AreEqual(ApplicationState.Finished, wa.app.GetApplicationState ()); } finally { if (wa != null) { wa.Finished(); } } }