test: add SnssaiToNas test

free5gc · Aug 7, 2024 · ed354e5 · ed354e5
1 parent 1f64e9b
commit ed354e5
Show file tree

Hide file tree

Showing 2 changed files with 37 additions and 2 deletions.
diff --git a/nasConvert/Snssai.go b/nasConvert/Snssai.go
@@ -10,8 +10,8 @@ import (
 
 // TS24.501 9.11.2.8 S-NSSAI
 func SnssaiToModels(nasSnssai *nasType.SNSSAI) (snssai models.Snssai) {
-	sD := nasSnssai.GetSD()
-	if nasSnssai.GetLen() >= uint8(3) {
+	if nasSnssai.GetLen() == uint8(4) {
+		sD := nasSnssai.GetSD()
 		snssai.Sd = hex.EncodeToString(sD[:])
 	}
 	snssai.Sst = int32(nasSnssai.GetSST())

diff --git a/nasConvert/Snssai_test.go b/nasConvert/Snssai_test.go
@@ -47,3 +47,38 @@ func TestSnssaiToModels(t *testing.T) {
 		})
 	}
 }
+
+func TestSnssaiToNas(t *testing.T) {
+	testCase := []struct {
+		Name      string
+		Snssai    models.Snssai
+		nasSnssai []uint8
+	}{
+		{
+			Name: "Default",
+			Snssai: models.Snssai{
+				Sst: int32(1),
+				Sd:  "010203",
+			},
+			nasSnssai: []uint8{
+				04, 01, 01, 02, 03,
+			},
+		},
+		{
+			Name: "Empty SD",
+			Snssai: models.Snssai{
+				Sst: int32(1),
+			},
+			nasSnssai: []uint8{
+				01, 01,
+			},
+		},
+	}
+
+	for _, tc := range testCase {
+		t.Run(tc.Name, func(t *testing.T) {
+			result := nasConvert.SnssaiToNas(tc.Snssai)
+			require.Equal(t, tc.nasSnssai, result)
+		})
+	}
+}