assertNotNull(future.get()); assertTrue(future.get()); wm.addTestEvent().get(); allSessionProviders = wm.getAllSessionTriggerProviders(); assertEquals(0, allSessionProviders.get("A").getSessions().size()); assertNotNull(future.get()); assertTrue(future.get()); wm.addTestEvent().get(); allSessionProviders = wm.getAllSessionTriggerProviders(); assertEquals(0, allSessionProviders.get("A").getSessions().size()); assertNotNull(future.get()); assertTrue(future.get()); wm.addTestEvent().get(); allSessionProviders = wm.getAllSessionTriggerProviders(); assertEquals(0, allSessionProviders.get("A").getSessions().size()); assertNotNull(future.get()); assertFalse(future.get()); wm.addTestEvent().get(); while(sessionA1.isOpen()) { Thread.sleep(100); wm.addTestEvent().get(); allSessionProviders = wm.getAllSessionTriggerProviders(); assertEquals(0, allSessionProviders.get("A").getSessions().size());
assertNotNull(future.get()); assertTrue(future.get()); wm.addTestEvent().get(); allSessionProviders = wm.getAllSessionTriggerProviders(); assertEquals(0, allSessionProviders.get("A").getSessions().size()); assertNotNull(future.get()); assertTrue(future.get()); wm.addTestEvent().get(); allSessionProviders = wm.getAllSessionTriggerProviders(); assertEquals(0, allSessionProviders.get("A").getSessions().size()); assertNotNull(future.get()); assertFalse(future.get()); wm.addTestEvent().get(); while(sessionA3.isOpen()) { Thread.sleep(100);
plan.setMappings(Lists.newArrayList(mapping("A", "A"))); wm.updateResourcePlanAsync(plan); wm.addTestEvent().get(); assertEquals(0, pool.getCurrentSize()); sessionA1.get().returnToSessionManager(); plan.setMappings(Lists.newArrayList(mapping("A", "A"))); wm.updateResourcePlanAsync(plan); wm.addTestEvent().get(); assertEquals(0, pool.getCurrentSize()); pool.returnSession(oob3);
@Test(timeout = 10000) public void testReopen() throws Exception { // We should always get a different object, and cluster fraction should be propagated. HiveConf conf = createConf(); MockQam qam = new MockQam(); WorkloadManager wm = new WorkloadManagerForTest("test", conf, 1, qam); wm.start(); WmTezSession session = (WmTezSession) wm.getSession( null, mappingInput("user"), conf); assertEquals(1.0, session.getClusterFraction(), EPSILON); qam.assertWasCalledAndReset(); WmTezSession session2 = (WmTezSession) session.reopen(); assertNotSame(session, session2); wm.addTestEvent().get(); assertEquals(session2.toString(), 1.0, session2.getClusterFraction(), EPSILON); assertFalse(session.hasClusterFraction()); qam.assertWasCalledAndReset(); }
mapping("C", "C"), mapping("D", "D"))); wm.updateResourcePlanAsync(plan); wm.addTestEvent().get();