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;
     }