exclude ID attribute and update reference model source

Author: xiwenc

mpr/mpr_test.go

@@ -0,0 +1,100 @@
+// mpr_test.go
+package mpr
+
+import (
+	"fmt"
+	"os"
+	"path/filepath"
+	"strings"
+	"testing"
+
+	"gopkg.in/yaml.v3"
+)
+
+// TestAdd tests the Add function to ensure it returns correct results.
+func TestMPRMetadata(t *testing.T) {
+	t.Run("single-mpr", func(t *testing.T) {
+		if err := exportMetadata("./../resources/app-mpr-v1", "./../tmp", nil); err != nil {
+			t.Errorf("Failed to export metadata from MPR file")
+		}
+
+		// open metadata file
+		metadataFile, err := os.ReadFile("./../tmp/Metadata.yaml")
+		if err != nil {
+			t.Errorf("Failed to read metadata file")
+		}
+		// read metadata file
+		var metadataObj MxMetadata
+		var node yaml.Node
+		if err := yaml.Unmarshal(metadataFile, &node); err != nil {
+			t.Errorf("Failed to unmarshal metadata file: %v", err)
+		}
+		if err := node.Decode(&metadataObj); err != nil {
+			t.Errorf("Failed to decode metadata file: %v", err)
+		}
+		// check metadata
+		expectedProductVersion := "10.18.3.58900"
+		if metadataObj.ProductVersion != expectedProductVersion {
+			t.Errorf("ProductVersion is incorrect. Expected: %s, Got: %s", expectedProductVersion, metadataObj.ProductVersion)
+		}
+	})
+}
+
+func TestMPRUnits(t *testing.T) {
+	t.Run("single-mpr", func(t *testing.T) {
+		if err := exportUnits("./../resources/app-mpr-v1", "./../tmp", false, "basic"); err != nil {
+			t.Errorf("Failed to export units from MPR file")
+		}
+	})
+}
+
+func TestIDAttributesExclusion(t *testing.T) {
+	t.Run("verify-id-attributes-excluded", func(t *testing.T) {
+		// Export units with ID attributes excluded
+		if err := exportUnits("./../resources/app-mpr-v1", "./../tmp", false, "basic"); err != nil {
+			t.Errorf("Failed to export units from MPR file: %v", err)
+			return
+		}
+
+		// Read a sample unit file
+		files, err := os.ReadDir("./../tmp/MyFirstModule")
+		if err != nil {
+			t.Errorf("Failed to read directory: %v", err)
+			return
+		}
+
+		// Find a file to test
+		var filePath string
+		for _, file := range files {
+			if !file.IsDir() && strings.HasSuffix(file.Name(), ".yaml") {
+				filePath = filepath.Join("./../tmp/MyFirstModule", file.Name())
+				break
+			}
+		}
+
+		if filePath == "" {
+			t.Errorf("No unit files found to test")
+			return
+		}
+
+		// Read the file content
+		content, err := os.ReadFile(filePath)
+		if err != nil {
+			t.Errorf("Failed to read file %s: %v", filePath, err)
+			return
+		}
+
+		// Check that ID attributes are excluded
+		contentStr := string(content)
+		if strings.Contains(contentStr, "\"ID\":") || strings.Contains(contentStr, "\"$ID\":") {
+			t.Errorf("ID attributes were not excluded from unit document: %s", filePath)
+		}
+
+		// Also check for other ignored attributes
+		for _, attr := range ignoredAttributes {
+			if strings.Contains(contentStr, fmt.Sprintf("\"%s\":", attr)) {
+				t.Errorf("Ignored attribute '%s' was not excluded from unit document: %s", attr, filePath)
+			}
+		}
+	})
+}

mpr/mpr_v1_test.go

@@ -17,7 +17,7 @@ func SetLogger(logger *logrus.Logger) {
 	log = logger
 }
 
-var ignoredAttributes = []string{"$ID", "Flows", "OriginPointer", "Type", "LineType", "DestinationPointer", "Image", "ImageData", "GUID", "StableId", "Size", "RelativeMiddlePoint", "Location", "OriginBezierVector", "DestinationBezierVector", "OriginConnectionIndex", "DestinationConnectionIndex"}
+var ignoredAttributes = []string{"ID", "$ID", "Flows", "OriginPointer", "Type", "LineType", "DestinationPointer", "Image", "ImageData", "GUID", "StableId", "Size", "RelativeMiddlePoint", "Location", "OriginBezierVector", "DestinationBezierVector", "OriginConnectionIndex", "DestinationConnectionIndex"}
 
 func Contains(slice []string, str string) bool {
 	for _, item := range slice {