1 files changed, 116 insertions, 0 deletions
diff --git a/dcaedt_be/src/test/java/org/onap/sdc/dcae/VesStructureLoaderMock.java b/dcaedt_be/src/test/java/org/onap/sdc/dcae/VesStructureLoaderMock.java
new file mode 100644
index 0000000..34db9fa
--- /dev/null
+++ b/dcaedt_be/src/test/java/org/onap/sdc/dcae/VesStructureLoaderMock.java
@@ -0,0 +1,116 @@
+package org.onap.sdc.dcae;
+
+import com.google.gson.Gson;
+import com.google.gson.GsonBuilder;
+import com.google.gson.JsonIOException;
+import com.google.gson.JsonSyntaxException;
+import com.google.gson.reflect.TypeToken;
+import org.apache.commons.lang.ArrayUtils;
+import org.apache.commons.lang.StringUtils;
+import org.onap.sdc.dcae.ves.EventListenerDefinition;
+import org.onap.sdc.dcae.ves.VesDataItemsDefinition;
+import org.onap.sdc.dcae.ves.VesDataTypeDefinition;
+import org.onap.sdc.dcae.ves.VesJsonDeserializer;
+
+import java.io.File;
+import java.io.FileNotFoundException;
+import java.io.FileReader;
+import java.io.FilenameFilter;
+import java.lang.reflect.Type;
+import java.util.*;
+import java.util.stream.Collectors;
+
+public class VesStructureLoaderMock {
+
+	private Map<String, EventListenerDefinition> eventListeners = new HashMap<>();
+	private Type type = new TypeToken<VesDataItemsDefinition>() {
+	}.getType();
+	private Gson gson = new GsonBuilder().registerTypeAdapter(type, new VesJsonDeserializer()).create();
+	private final String schemaNamePrefix = "CommonEventFormat_v";
+	private final String schemaNameSuffix = ".json";
+	private List<String> initErrors;
+
+	public VesStructureLoaderMock() {
+		this(true);
+	}
+
+	public VesStructureLoaderMock(boolean validateAndResolve) {
+		this(validateAndResolve, System.getProperty("user.dir") + "/src/test/resources/ves-schema");
+	}
+
+	public VesStructureLoaderMock(boolean validateAndResolve, String path) {
+		initErrors = init(validateAndResolve, path);
+	}
+
+	public List<String> init(boolean validateAndResolve, String pathToSchemaDir) {
+
+		List<String> parseErrors = new ArrayList<>();
+		File dir = new File(pathToSchemaDir);
+		File[] files = dir.listFiles(new FilenameFilter() {
+			@Override public boolean accept(File dir, String name) {
+				return name.startsWith(schemaNamePrefix) && name.endsWith(schemaNameSuffix);
+			}
+		});
+		if (ArrayUtils.isEmpty(files)) {
+			parseErrors.add("No VES schema files found");
+		} else {
+			for (File f : files) {
+				String error = parseJsonFileAndSaveToMap(f, validateAndResolve);
+				if (StringUtils.isNotBlank(error)) {
+					parseErrors.add("Error: parsing VES schema file " + f.getName() + " failed due to " + error);
+				}
+			}
+		}
+		return parseErrors;
+
+	}
+
+	public Map<String, VesDataTypeDefinition> getEventListenerDefinitionByVersion(String version) {
+		return eventListeners.get(version).getProperties().get(EventListenerDefinition.EVENT_ROOT).getProperties();
+	}
+
+	public Set<String> getAvailableVersionsList() {
+		return eventListeners.keySet();
+	}
+
+	public Map<String, Set<String>> getAvailableVersionsAndEventTypes() {
+		return eventListeners.entrySet().stream().collect(Collectors.toMap(Map.Entry::getKey, e -> getEventListenerDefinitionByVersion(e.getKey()).keySet()));
+	}
+
+	public Set<String> getEventTypeListByVersion(String version) {
+		return getEventListenerDefinitionByVersion(version).keySet();
+	}
+
+	public String getVersionFromFileName(String fileName) {
+		return fileName.replace(schemaNamePrefix, "").replace(schemaNameSuffix, "");
+	}
+
+	private String parseJsonFileAndSaveToMap(File file, boolean validateAndResolve) {
+		String validationError = null;
+		try {
+			EventListenerDefinition eventListener = gson.fromJson(new FileReader(file), EventListenerDefinition.class);
+			if (validateAndResolve)
+				validationError = getValidatorMessage(eventListener);
+			if (StringUtils.isEmpty(validationError))
+				eventListeners.put(getVersionFromFileName(file.getName()), eventListener);
+		} catch (FileNotFoundException | JsonIOException | JsonSyntaxException e) {
+			validationError = e.getMessage();
+		}
+		return validationError;
+	}
+
+	public Map<String, EventListenerDefinition> getEventListeners() {
+		return eventListeners;
+	}
+
+	public List<String> getInitErrors() {
+		return initErrors;
+	}
+
+	private String getValidatorMessage(EventListenerDefinition eventListenerDefinition) {
+		String validationError = eventListenerDefinition.validate();
+		if (StringUtils.isBlank(validationError))
+			validationError = eventListenerDefinition.resolveRefTypes();
+		return validationError;
+	}
+}