Commit 40666d18 by ashutoshm Committed by Madhan Neethiraj

ATLAS-2158: ZipFileResourceTestUtils: Improved failure scenario.

parent 6a64cd9c
...@@ -76,9 +76,8 @@ public class ZipFileResourceTestUtils { ...@@ -76,9 +76,8 @@ public class ZipFileResourceTestUtils {
File[] topModelsDirContents = topModelsDir.exists() ? topModelsDir.listFiles() : null; File[] topModelsDirContents = topModelsDir.exists() ? topModelsDir.listFiles() : null;
assertTrue(topModelsDirContents != null, topModelsDir.getAbsolutePath() + ": unable to find/read directory"); assertTrue(topModelsDirContents != null, topModelsDir.getAbsolutePath() + ": unable to find/read directory");
if(topModelsDirContents != null) {
Arrays.sort(topModelsDirContents); Arrays.sort(topModelsDirContents);
for (File modelDir : topModelsDirContents) { for (File modelDir : topModelsDirContents) {
if (modelDir.exists() && modelDir.isDirectory()) { if (modelDir.exists() && modelDir.isDirectory()) {
ret = getFileContents(modelDir, fileName); ret = getFileContents(modelDir, fileName);
...@@ -94,6 +93,9 @@ public class ZipFileResourceTestUtils { ...@@ -94,6 +93,9 @@ public class ZipFileResourceTestUtils {
} }
assertTrue(ret != null, fileName + ": unable to find model file"); assertTrue(ret != null, fileName + ": unable to find model file");
} else {
throw new IOException("Unable to retrieve model contents.");
}
return ret; return ret;
} }
...@@ -169,15 +171,19 @@ public class ZipFileResourceTestUtils { ...@@ -169,15 +171,19 @@ public class ZipFileResourceTestUtils {
} }
private static void createTypesAsNeeded(AtlasTypesDef typesFromJson, AtlasTypeDefStore typeDefStore, AtlasTypeRegistry typeRegistry) throws AtlasBaseException { private static void createTypesAsNeeded(AtlasTypesDef typesFromJson, AtlasTypeDefStore typeDefStore, AtlasTypeRegistry typeRegistry) throws AtlasBaseException {
AtlasTypesDef typesToCreate = AtlasTypeDefStoreInitializer.getTypesToCreate(typesFromJson, typeRegistry); if(typesFromJson == null) {
return;
}
if (!typesToCreate.isEmpty()) { AtlasTypesDef typesToCreate = AtlasTypeDefStoreInitializer.getTypesToCreate(typesFromJson, typeRegistry);
if (typesToCreate != null && !typesToCreate.isEmpty()) {
typeDefStore.createTypesDef(typesToCreate); typeDefStore.createTypesDef(typesToCreate);
} }
} }
private static AtlasTypesDef getAtlasTypesDefFromFile(String fileName) throws IOException { private static AtlasTypesDef getAtlasTypesDefFromFile(String fileName) throws IOException {
String sampleTypes = ZipFileResourceTestUtils.getModelJson(fileName); String sampleTypes = ZipFileResourceTestUtils.getModelJson(fileName);
if(sampleTypes == null) return null;
return AtlasType.fromJson(sampleTypes, AtlasTypesDef.class); return AtlasType.fromJson(sampleTypes, AtlasTypesDef.class);
} }
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment