Package org.apache.tez.dag.app

Examples of org.apache.tez.dag.app.AppContext

44
45
46
47
48
49
50
51
52
53
  public AppContext createMockAppContext() {

    ApplicationId appId = ApplicationId.newInstance(2000, 1);
    ApplicationAttemptId appAttemptId = ApplicationAttemptId.newInstance(appId, 1);

    AppContext appContext = mock(AppContext.class);
    doReturn(appAttemptId).when(appContext).getApplicationAttemptId();

    return appContext;
  }
View Full Code Here
81
82
83
84
85
86
87
88
89
90
91
    dispatcher.stop();
  }
 
  @Test(timeout=5000)
  public void testHealthUpdateKnownNode() {
    AppContext appContext = mock(AppContext.class);

    AMNodeMap amNodeMap = new AMNodeMap(eventHandler, appContext);
    amNodeMap.init(new Configuration(false));
    amNodeMap.start();
View Full Code Here
99
100
101
102
103
104
105
106
107
108
109
    amNodeMap.stop();
  }

  @Test(timeout=5000)
  public void testHealthUpdateUnknownNode() {
    AppContext appContext = mock(AppContext.class);

    AMNodeMap amNodeMap = new AMNodeMap(eventHandler, appContext);
    amNodeMap.init(new Configuration(false));
    amNodeMap.start();
View Full Code Here
118
119
120
121
122
123
124
125
126
127
128
    // the log message for verification.
  }
 
  @Test(timeout=10000)
  public void testNodeSelfBlacklist() throws InterruptedException {
    AppContext appContext = mock(AppContext.class);
    Configuration conf = new Configuration(false);
    conf.setInt(TezConfiguration.TEZ_AM_MAX_TASK_FAILURES_PER_NODE, 2);
    TestEventHandler handler = new TestEventHandler();
    AMNodeMap amNodeMap = new AMNodeMap(handler, appContext);
    AMContainerMap amContainerMap = mock(AMContainerMap.class);
View Full Code Here
47
48
49
50
51
52
53
54
    doReturn(socketAddr).when(tal).getAddress();
    return tal;
  }

  private AppContext mockAppContext() {
    AppContext appContext = mock(AppContext.class);
    return appContext;
  }
View Full Code Here
91
92
93
94
95
96
97
98
99
100
101
102
  @SuppressWarnings({ "unchecked" })
  @Test(timeout=10000)
  public void testTaskSchedulerNoReuse() throws Exception {
    RackResolver.init(new YarnConfiguration());
    TaskSchedulerAppCallback mockApp = mock(TaskSchedulerAppCallback.class);
    AppContext mockAppContext = mock(AppContext.class);
    when(mockAppContext.getAMState()).thenReturn(DAGAppMasterState.RUNNING);

    TezAMRMClientAsync<CookieContainerRequest> mockRMClient =
                                                  mock(TezAMRMClientAsync.class);

    String appHost = "host";
View Full Code Here
497
498
499
500
501
502
503
504
505
506
507
508
  @SuppressWarnings({ "unchecked" })
  @Test(timeout=10000)
  public void testTaskSchedulerWithReuse() throws Exception {
    RackResolver.init(new YarnConfiguration());
    TaskSchedulerAppCallback mockApp = mock(TaskSchedulerAppCallback.class);
    AppContext mockAppContext = mock(AppContext.class);
    when(mockAppContext.getAMState()).thenReturn(DAGAppMasterState.RUNNING);

    TezAMRMClientAsync<CookieContainerRequest> mockRMClient =
                                                  mock(TezAMRMClientAsync.class);

    String appHost = "host";
View Full Code Here
993
994
995
996
997
998
999
1000
1001
1002
1003
1004
  @SuppressWarnings("unchecked")
  @Test(timeout=5000)
  public void testTaskSchedulerDetermineMinHeldContainers() throws Exception {
    RackResolver.init(new YarnConfiguration());
    TaskSchedulerAppCallback mockApp = mock(TaskSchedulerAppCallback.class);
    AppContext mockAppContext = mock(AppContext.class);
    when(mockAppContext.getAMState()).thenReturn(DAGAppMasterState.RUNNING);

    TezAMRMClientAsync<CookieContainerRequest> mockRMClient =
                                                  mock(TezAMRMClientAsync.class);

    String appHost = "host";
View Full Code Here
1127
1128
1129
1130
1131
1132
1133
1134
1135
1136
1137
1138
  @SuppressWarnings("unchecked")
  @Test(timeout=5000)
  public void testTaskSchedulerRandomReuseExpireTime() throws Exception {
    RackResolver.init(new YarnConfiguration());
    TaskSchedulerAppCallback mockApp = mock(TaskSchedulerAppCallback.class);
    AppContext mockAppContext = mock(AppContext.class);
    when(mockAppContext.getAMState()).thenReturn(DAGAppMasterState.RUNNING);

    TezAMRMClientAsync<CookieContainerRequest> mockRMClient =
                                                  mock(TezAMRMClientAsync.class);

    String appHost = "host";
View Full Code Here
1204
1205
1206
1207
1208
1209
1210
1211
1212
1213
1214
1215
  @SuppressWarnings({ "unchecked", "rawtypes" })
  @Test
  public void testTaskSchedulerPreemption() throws Exception {
    RackResolver.init(new YarnConfiguration());
    TaskSchedulerAppCallback mockApp = mock(TaskSchedulerAppCallback.class);
    AppContext mockAppContext = mock(AppContext.class);
    when(mockAppContext.getAMState()).thenReturn(DAGAppMasterState.RUNNING);

    TezAMRMClientAsync<CookieContainerRequest> mockRMClient =
                                                  mock(TezAMRMClientAsync.class);

    String appHost = "host";
View Full Code Here
TOP

Related Classes of org.apache.tez.dag.app.AppContext

  • org.apache.tez.client.LocalClient
  • org.apache.tez.dag.api.client.TestDAGClientHandler
  • org.apache.tez.dag.app.dag.impl.TestTaskAttempt
  • org.apache.tez.dag.app.rm.container.TestAMContainerMap
  • org.apache.tez.dag.app.rm.node.TestAMNodeMap
  • org.apache.tez.dag.app.rm.TestContainerReuse
  • org.apache.tez.dag.app.rm.TestLocalTaskScheduler
  • org.apache.tez.dag.app.rm.TestTaskScheduler
    • Copyright © 2018 www.massapicom. All rights reserved.
    All source code are property of their respective owners. Java is a trademark of Sun Microsystems, Inc and owned by ORACLE Inc. Contact coftware#gmail.com.