Fix for Microsemi Unit tests
Change-Id: I1e94c85d19de53f82eb7fc657fff42535dafd9f5
diff --git a/drivers/microsemi/src/test/java/org/onosproject/yang/MockMicrosemiRegistrator.java b/drivers/microsemi/src/test/java/org/onosproject/yang/MockMicrosemiRegistrator.java
index 111ec8a..785ba8d 100644
--- a/drivers/microsemi/src/test/java/org/onosproject/yang/MockMicrosemiRegistrator.java
+++ b/drivers/microsemi/src/test/java/org/onosproject/yang/MockMicrosemiRegistrator.java
@@ -15,9 +15,12 @@
*/
package org.onosproject.yang;
+import com.google.common.collect.ImmutableMap;
import org.onosproject.models.microsemi.MicrosemiModelRegistrator;
import org.onosproject.yang.compiler.datamodel.YangNode;
import org.onosproject.yang.compiler.tool.YangNodeInfo;
+import org.onosproject.yang.model.YangModuleId;
+import org.onosproject.yang.runtime.AppModuleInfo;
import org.onosproject.yang.runtime.DefaultModelRegistrationParam;
import org.onosproject.yang.runtime.ModelRegistrationParam;
import org.onosproject.yang.runtime.YangModelRegistry;
@@ -26,7 +29,9 @@
import java.io.File;
import java.io.IOException;
import java.util.ArrayList;
+import java.util.HashMap;
import java.util.List;
+import java.util.Map;
import static org.junit.Assert.fail;
import static org.onosproject.yang.compiler.tool.YangCompilerManager.deSerializeDataModel;
@@ -66,6 +71,14 @@
}
}
+
+ public void addAppInfo(Map<YangModuleId, AppModuleInfo> map) {
+ Map<YangModuleId, AppModuleInfo> appInfoCopy = new HashMap<>();
+ appInfoCopy.putAll(appInfo);
+ appInfoCopy.putAll(map);
+ appInfo = ImmutableMap.copyOf(appInfoCopy);
+ }
+
public YangModelRegistry registry() {
return modelRegistry;
}