|
22 | 22 | import org.apache.hadoop.fs.CommonConfigurationKeysPublic; |
23 | 23 | import org.apache.hadoop.metrics2.lib.DefaultMetricsSystem; |
24 | 24 | import org.apache.hadoop.test.LambdaTestUtils; |
| 25 | +import org.apache.hadoop.yarn.api.records.ApplicationId; |
25 | 26 | import org.apache.hadoop.yarn.api.records.DecommissionType; |
26 | 27 | import org.apache.hadoop.yarn.api.records.NodeId; |
27 | 28 | import org.apache.hadoop.yarn.api.records.Resource; |
|
31 | 32 | import org.apache.hadoop.yarn.api.records.NodeAttributeType; |
32 | 33 | import org.apache.hadoop.yarn.conf.YarnConfiguration; |
33 | 34 | import org.apache.hadoop.yarn.exceptions.YarnException; |
34 | | -import org.apache.hadoop.yarn.server.api.protocolrecords.*; |
| 35 | +import org.apache.hadoop.yarn.server.api.protocolrecords.RefreshNodesRequest; |
| 36 | +import org.apache.hadoop.yarn.server.api.protocolrecords.RefreshQueuesRequest; |
| 37 | +import org.apache.hadoop.yarn.server.api.protocolrecords.RefreshQueuesResponse; |
| 38 | +import org.apache.hadoop.yarn.server.api.protocolrecords.RefreshSuperUserGroupsConfigurationRequest; |
| 39 | +import org.apache.hadoop.yarn.server.api.protocolrecords.RefreshSuperUserGroupsConfigurationResponse; |
| 40 | +import org.apache.hadoop.yarn.server.api.protocolrecords.RefreshUserToGroupsMappingsRequest; |
| 41 | +import org.apache.hadoop.yarn.server.api.protocolrecords.RefreshUserToGroupsMappingsResponse; |
| 42 | +import org.apache.hadoop.yarn.server.api.protocolrecords.RefreshAdminAclsRequest; |
| 43 | +import org.apache.hadoop.yarn.server.api.protocolrecords.RefreshAdminAclsResponse; |
| 44 | +import org.apache.hadoop.yarn.server.api.protocolrecords.RefreshServiceAclsRequest; |
| 45 | +import org.apache.hadoop.yarn.server.api.protocolrecords.RefreshServiceAclsResponse; |
| 46 | +import org.apache.hadoop.yarn.server.api.protocolrecords.UpdateNodeResourceRequest; |
| 47 | +import org.apache.hadoop.yarn.server.api.protocolrecords.UpdateNodeResourceResponse; |
| 48 | +import org.apache.hadoop.yarn.server.api.protocolrecords.RefreshNodesResourcesRequest; |
| 49 | +import org.apache.hadoop.yarn.server.api.protocolrecords.RefreshNodesResourcesResponse; |
| 50 | +import org.apache.hadoop.yarn.server.api.protocolrecords.AddToClusterNodeLabelsRequest; |
| 51 | +import org.apache.hadoop.yarn.server.api.protocolrecords.AddToClusterNodeLabelsResponse; |
| 52 | +import org.apache.hadoop.yarn.server.api.protocolrecords.RemoveFromClusterNodeLabelsRequest; |
| 53 | +import org.apache.hadoop.yarn.server.api.protocolrecords.RemoveFromClusterNodeLabelsResponse; |
| 54 | +import org.apache.hadoop.yarn.server.api.protocolrecords.ReplaceLabelsOnNodeRequest; |
| 55 | +import org.apache.hadoop.yarn.server.api.protocolrecords.ReplaceLabelsOnNodeResponse; |
| 56 | +import org.apache.hadoop.yarn.server.api.protocolrecords.CheckForDecommissioningNodesRequest; |
| 57 | +import org.apache.hadoop.yarn.server.api.protocolrecords.CheckForDecommissioningNodesResponse; |
| 58 | +import org.apache.hadoop.yarn.server.api.protocolrecords.NodesToAttributesMappingRequest; |
| 59 | +import org.apache.hadoop.yarn.server.api.protocolrecords.AttributeMappingOperationType; |
| 60 | +import org.apache.hadoop.yarn.server.api.protocolrecords.NodeToAttributes; |
| 61 | +import org.apache.hadoop.yarn.server.api.protocolrecords.FederationQueueWeight; |
| 62 | +import org.apache.hadoop.yarn.server.api.protocolrecords.SaveFederationQueuePolicyRequest; |
| 63 | +import org.apache.hadoop.yarn.server.api.protocolrecords.SaveFederationQueuePolicyResponse; |
| 64 | +import org.apache.hadoop.yarn.server.api.protocolrecords.BatchSaveFederationQueuePoliciesRequest; |
| 65 | +import org.apache.hadoop.yarn.server.api.protocolrecords.BatchSaveFederationQueuePoliciesResponse; |
| 66 | +import org.apache.hadoop.yarn.server.api.protocolrecords.QueryFederationQueuePoliciesRequest; |
| 67 | +import org.apache.hadoop.yarn.server.api.protocolrecords.QueryFederationQueuePoliciesResponse; |
| 68 | +import org.apache.hadoop.yarn.server.api.protocolrecords.DeleteFederationApplicationRequest; |
| 69 | +import org.apache.hadoop.yarn.server.api.protocolrecords.DeleteFederationApplicationResponse; |
35 | 70 | import org.apache.hadoop.yarn.server.federation.policies.dao.WeightedPolicyInfo; |
36 | 71 | import org.apache.hadoop.yarn.server.federation.policies.manager.WeightedLocalityPolicyManager; |
37 | 72 | import org.apache.hadoop.yarn.server.federation.store.impl.MemoryFederationStateStore; |
38 | 73 | import org.apache.hadoop.yarn.server.federation.store.records.SubClusterId; |
39 | 74 | import org.apache.hadoop.yarn.server.federation.store.records.SubClusterIdInfo; |
40 | 75 | import org.apache.hadoop.yarn.server.federation.store.records.SubClusterPolicyConfiguration; |
| 76 | +import org.apache.hadoop.yarn.server.federation.store.records.ApplicationHomeSubCluster; |
41 | 77 | import org.apache.hadoop.yarn.server.federation.utils.FederationStateStoreFacade; |
42 | 78 | import org.apache.hadoop.yarn.server.federation.utils.FederationStateStoreTestUtil; |
43 | 79 | import org.junit.Assert; |
@@ -960,10 +996,25 @@ public void testFilterPoliciesConfigurationsByQueues() throws Exception { |
960 | 996 |
|
961 | 997 |
|
962 | 998 | @Test |
963 | | - public void testDeleteFederationApplication() throws IOException, YarnException { |
964 | | - DeleteFederationApplicationRequest request = |
965 | | - DeleteFederationApplicationRequest.newInstance("application_1"); |
966 | | - DeleteFederationApplicationResponse response = interceptor.deleteFederationApplication(request); |
967 | | - assertNotNull(response); |
| 999 | + public void testDeleteFederationApplication() throws Exception { |
| 1000 | + ApplicationId applicationId = ApplicationId.newInstance(10, 1); |
| 1001 | + DeleteFederationApplicationRequest request1 = |
| 1002 | + DeleteFederationApplicationRequest.newInstance(applicationId.toString()); |
| 1003 | + LambdaTestUtils.intercept(YarnException.class, |
| 1004 | + "Application application_10_0001 does not exist.", |
| 1005 | + () -> interceptor.deleteFederationApplication(request1)); |
| 1006 | + |
| 1007 | + ApplicationId applicationId2 = ApplicationId.newInstance(10, 2); |
| 1008 | + SubClusterId homeSubCluster = SubClusterId.newInstance("SC-1"); |
| 1009 | + ApplicationHomeSubCluster appHomeSubCluster = |
| 1010 | + ApplicationHomeSubCluster.newInstance(applicationId2, homeSubCluster); |
| 1011 | + facade.addApplicationHomeSubCluster(appHomeSubCluster); |
| 1012 | + DeleteFederationApplicationRequest request2 = |
| 1013 | + DeleteFederationApplicationRequest.newInstance(applicationId2.toString()); |
| 1014 | + DeleteFederationApplicationResponse deleteFederationApplicationResponse = |
| 1015 | + interceptor.deleteFederationApplication(request2); |
| 1016 | + assertNotNull(deleteFederationApplicationResponse); |
| 1017 | + assertEquals("applicationId = " + applicationId2 + " delete success.", |
| 1018 | + deleteFederationApplicationResponse.getMessage()); |
968 | 1019 | } |
969 | 1020 | } |
0 commit comments