diff --git a/src/app/vnfs/build-artifacts/reference-dataform/reference-dataform.util.spec.ts b/src/app/vnfs/build-artifacts/reference-dataform/reference-dataform.util.spec.ts
index aa7fbfa..7745035 100644
--- a/src/app/vnfs/build-artifacts/reference-dataform/reference-dataform.util.spec.ts
+++ b/src/app/vnfs/build-artifacts/reference-dataform/reference-dataform.util.spec.ts
@@ -71,4 +71,14 @@
         expect(returnValueVnfcType).toBe(true);
         expect(returnValueVnfcTypeList).toBe(true);
     }));
+
+    it('should test decideExtension function when device-protocol is CLI', inject([ReferenceDataFormUtil], (service: ReferenceDataFormUtil) => {
+        let extension = service.decideExtension({'device-protocol': 'CLI'});
+        expect(extension).toBe('.json');
+    }));
+
+    it('should test decideExtension function when device-protocol is NETCONF-XML', inject([ReferenceDataFormUtil], (service: ReferenceDataFormUtil) => {
+        let extension = service.decideExtension({'device-protocol': 'NETCONF-XML'});
+        expect(extension).toBe('.xml');
+    }));
 });