Skip to content

Commit 2b820c3

Browse files
committed
add null check
1 parent 6c516e5 commit 2b820c3

File tree

2 files changed

+10
-5
lines changed

2 files changed

+10
-5
lines changed

Utils/azuretools-core/src/com/microsoft/azuretools/core/mvp/model/webapp/AzureWebAppMvpModel.java

Lines changed: 9 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -48,6 +48,7 @@
4848

4949
public class AzureWebAppMvpModel {
5050

51+
public static final String CANNOT_GET_WEB_APP_WITH_ID = "Cannot get Web App with ID: ";
5152
private final Map<String, List<ResourceEx<WebApp>>> subscriptionIdToWebAppsMap;
5253
private final Map<String, List<ResourceEx<SiteInner>>> subscriptionIdToWebAppsOnLinuxMap;
5354

@@ -63,9 +64,13 @@ public static AzureWebAppMvpModel getInstance() {
6364
/**
6465
* get the web app by ID.
6566
*/
66-
public WebApp getWebAppById(String sid, String id) throws IOException {
67+
public WebApp getWebAppById(String sid, String id) throws Exception {
6768
Azure azure = AuthMethodManager.getInstance().getAzureClient(sid);
68-
return azure.webApps().getById(id);
69+
WebApp app = azure.webApps().getById(id);
70+
if (app == null) {
71+
throw new Exception(CANNOT_GET_WEB_APP_WITH_ID + id);
72+
}
73+
return app;
6974
}
7075

7176
/**
@@ -233,8 +238,8 @@ public WebApp createWebAppOnLinux(WebAppOnLinuxDeployModel model)
233238
* @param imageSetting new container settings
234239
* @return instance of the updated Web App on Linux
235240
*/
236-
public WebApp updateWebAppOnLinux(String sid, String webAppId, ImageSetting imageSetting) throws IOException {
237-
WebApp app = AzureWebAppMvpModel.getInstance().getWebAppById(sid, webAppId);
241+
public WebApp updateWebAppOnLinux(String sid, String webAppId, ImageSetting imageSetting) throws Exception {
242+
WebApp app = getWebAppById(sid, webAppId);
238243
clearTags(app);
239244
if (imageSetting instanceof PrivateRegistryImageSetting) {
240245
PrivateRegistryImageSetting pr = (PrivateRegistryImageSetting) imageSetting;

Utils/azuretools-core/test/com/microsoft/azuretools/core/mvp/model/webapp/AzureWebAppMvpModelTest.java

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -107,7 +107,7 @@ public void testGetWebAppById() {
107107
try {
108108
WebApp result = azureWebAppMvpModel.getWebAppById(MOCK_SUBSCRIPTION, "test");
109109
assert(result.toString()).equals("testApp");
110-
} catch (IOException e) {
110+
} catch (Exception e) {
111111
fail();
112112
}
113113
}

0 commit comments

Comments
 (0)