Fix datalake tests

doublek420 · Jul 5, 2016 · 5c82bed · 5c82bed
1 parent 1142160
commit 5c82bed
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 10 deletions.
diff --git a/...om/microsoft/azure/management/datalake/analytics/DataLakeAnalyticsManagementTestBase.java b/...om/microsoft/azure/management/datalake/analytics/DataLakeAnalyticsManagementTestBase.java
@@ -1,6 +1,7 @@
 package com.microsoft.azure.management.datalake.analytics;
 
-import com.microsoft.azure.credentials.AzureEnvironment;
+import com.microsoft.azure.AzureEnvironment;
+import com.microsoft.azure.RestClient;
 import com.microsoft.azure.credentials.UserTokenCredentials;
 import com.microsoft.azure.management.datalake.analytics.implementation.DataLakeAnalyticsAccountManagementClientImpl;
 import com.microsoft.azure.management.datalake.analytics.implementation.DataLakeAnalyticsCatalogManagementClientImpl;
@@ -13,18 +14,15 @@
 import com.microsoft.azure.management.datalake.store.implementation.DataLakeStoreAccountManagementClientImpl;
 import com.microsoft.azure.management.resources.implementation.ResourceManagementClientImpl;
 import com.microsoft.azure.management.storage.implementation.StorageManagementClientImpl;
-
-import com.microsoft.azure.RestClient;
+import okhttp3.OkHttpClient;
+import okhttp3.logging.HttpLoggingInterceptor;
 import org.junit.Assert;
+import retrofit2.Retrofit;
 
 import java.text.MessageFormat;
 import java.util.UUID;
 import java.util.concurrent.TimeUnit;
 
-import okhttp3.OkHttpClient;
-import okhttp3.logging.HttpLoggingInterceptor;
-import retrofit2.Retrofit;
-
 public abstract class DataLakeAnalyticsManagementTestBase {
     protected static DataLakeAnalyticsAccountManagementClientImpl dataLakeAnalyticsAccountManagementClient;
     protected static DataLakeAnalyticsJobManagementClientImpl dataLakeAnalyticsJobManagementClient;
@@ -48,13 +46,21 @@ public static void createClients() {
             case "ppe":
                 armUri = "https://api-dogfood.resources.windows-int.net";
                 adlaSuffix = "konaaccountdogfood.net";
-                authEnv = new AzureEnvironment("https://login.windows-ppe.net/", "https://management.core.windows.net/", true);
+                authEnv = new AzureEnvironment(
+                        "https://login.windows-ppe.net/",
+                        "https://management.core.windows.net/",
+                        true,
+                        "https://api-dogfood.resources.windows-int.net");
                 break;
             case "test":
                 armUri = "https://api-dogfood.resources.windows-int.net";
                 adlaSuffix = "konaaccountdogfood.net";
                 environmentLocation = "westus";
-                authEnv = new AzureEnvironment("https://login.windows-ppe.net/", "https://management.core.windows.net/", true);
+                authEnv = new AzureEnvironment(
+                        "https://login.windows-ppe.net/",
+                        "https://management.core.windows.net/",
+                        true,
+                        "https://api-dogfood.resources.windows-int.net");
                 break;
             default: // default to production
                 armUri = "https://management.azure.com";

diff --git a/...t/java/com/microsoft/azure/management/datalake/store/DataLakeStoreManagementTestBase.java b/...t/java/com/microsoft/azure/management/datalake/store/DataLakeStoreManagementTestBase.java
@@ -1,6 +1,6 @@
 package com.microsoft.azure.management.datalake.store;
 
-import com.microsoft.azure.credentials.AzureEnvironment;
+import com.microsoft.azure.AzureEnvironment;
 import com.microsoft.azure.credentials.UserTokenCredentials;
 import com.microsoft.azure.management.datalake.store.implementation.DataLakeStoreAccountManagementClientImpl;
 import com.microsoft.azure.management.datalake.store.implementation.DataLakeStoreFileSystemManagementClientImpl;