|
@@ -52,6 +52,7 @@ import org.apache.hadoop.yarn.api.records.ApplicationReport;
|
|
import org.apache.hadoop.yarn.api.records.YarnApplicationState;
|
|
import org.apache.hadoop.yarn.api.records.YarnApplicationState;
|
|
import org.apache.hadoop.yarn.api.records.impl.pb.ApplicationReportPBImpl;
|
|
import org.apache.hadoop.yarn.api.records.impl.pb.ApplicationReportPBImpl;
|
|
import org.apache.hadoop.yarn.conf.YarnConfiguration;
|
|
import org.apache.hadoop.yarn.conf.YarnConfiguration;
|
|
|
|
+import org.apache.hadoop.yarn.exceptions.ApplicationNotFoundException;
|
|
import org.apache.hadoop.yarn.exceptions.YarnException;
|
|
import org.apache.hadoop.yarn.exceptions.YarnException;
|
|
import org.apache.hadoop.yarn.exceptions.YarnRuntimeException;
|
|
import org.apache.hadoop.yarn.exceptions.YarnRuntimeException;
|
|
import org.apache.hadoop.yarn.webapp.util.WebAppUtils;
|
|
import org.apache.hadoop.yarn.webapp.util.WebAppUtils;
|
|
@@ -149,11 +150,20 @@ public class TestWebAppProxyServlet {
|
|
assertEquals(HttpURLConnection.HTTP_OK, proxyConn.getResponseCode());
|
|
assertEquals(HttpURLConnection.HTTP_OK, proxyConn.getResponseCode());
|
|
assertTrue(isResponseCookiePresent(
|
|
assertTrue(isResponseCookiePresent(
|
|
proxyConn, "checked_application_0_0000", "true"));
|
|
proxyConn, "checked_application_0_0000", "true"));
|
|
- // cannot found application
|
|
|
|
|
|
+ // cannot found application 1: null
|
|
appReportFetcher.answer = 1;
|
|
appReportFetcher.answer = 1;
|
|
proxyConn = (HttpURLConnection) url.openConnection();
|
|
proxyConn = (HttpURLConnection) url.openConnection();
|
|
proxyConn.setRequestProperty("Cookie", "checked_application_0_0000=true");
|
|
proxyConn.setRequestProperty("Cookie", "checked_application_0_0000=true");
|
|
proxyConn.connect();
|
|
proxyConn.connect();
|
|
|
|
+ assertEquals(HttpURLConnection.HTTP_NOT_FOUND,
|
|
|
|
+ proxyConn.getResponseCode());
|
|
|
|
+ assertFalse(isResponseCookiePresent(
|
|
|
|
+ proxyConn, "checked_application_0_0000", "true"));
|
|
|
|
+ // cannot found application 2: ApplicationNotFoundException
|
|
|
|
+ appReportFetcher.answer = 4;
|
|
|
|
+ proxyConn = (HttpURLConnection) url.openConnection();
|
|
|
|
+ proxyConn.setRequestProperty("Cookie", "checked_application_0_0000=true");
|
|
|
|
+ proxyConn.connect();
|
|
assertEquals(HttpURLConnection.HTTP_NOT_FOUND,
|
|
assertEquals(HttpURLConnection.HTTP_NOT_FOUND,
|
|
proxyConn.getResponseCode());
|
|
proxyConn.getResponseCode());
|
|
assertFalse(isResponseCookiePresent(
|
|
assertFalse(isResponseCookiePresent(
|
|
@@ -340,6 +350,8 @@ public class TestWebAppProxyServlet {
|
|
ApplicationReport result = getDefaultApplicationReport(appId);
|
|
ApplicationReport result = getDefaultApplicationReport(appId);
|
|
result.setYarnApplicationState(YarnApplicationState.KILLED);
|
|
result.setYarnApplicationState(YarnApplicationState.KILLED);
|
|
return result;
|
|
return result;
|
|
|
|
+ } else if (answer == 4) {
|
|
|
|
+ throw new ApplicationNotFoundException("Application is not found");
|
|
}
|
|
}
|
|
return null;
|
|
return null;
|
|
}
|
|
}
|