Lwm2m2 unitest

pkg/deviceshifu/deviceshifulwm2m/deviceshifulwm2mconfig_test.go

@@ -0,0 +1,54 @@
+package deviceshifulwm2m
+
+import (
+	"reflect"
+	"testing"
+
+	"github.com/edgenesis/shifu/pkg/deviceshifu/deviceshifubase"
+	"github.com/stretchr/testify/assert"
+)
+
+func TestCreateLwM2MInstructions(t *testing.T) {
+	tests := []struct {
+		name           string
+		input          *deviceshifubase.DeviceShifuInstructions
+		expected       *LwM2MInstruction
+		expectingPanic bool
+	}{
+		{
+			name: "Valid Instructions",
+			input: &deviceshifubase.DeviceShifuInstructions{
+				Instructions: map[string]*deviceshifubase.DeviceShifuInstruction{
+					"instruction1": {
+						DeviceShifuProtocolProperties: map[string]string{
+							objectIdStr:      "1",
+							enableObserveStr: "true",
+						},
+					},
+				},
+			},
+			expected: &LwM2MInstruction{
+				Instructions: map[string]*LwM2MProtocolProperty{
+					"instruction1": {
+						ObjectId:      "1",
+						EnableObserve: true,
+					},
+				},
+			},
+			expectingPanic: false,
+		},
+	}
+
+	for _, tt := range tests {
+		t.Run(tt.name, func(t *testing.T) {
+			if tt.expectingPanic {
+				assert.Panics(t, func() { CreateLwM2MInstructions(tt.input) })
+			} else {
+				result := CreateLwM2MInstructions(tt.input)
+				if !reflect.DeepEqual(tt.expected, result) {
+					t.Errorf("expected %v, got %v", tt.expected, result)
+				}
+			}
+		})
+	}
+}

pkg/deviceshifu/deviceshifulwm2m/lwm2m/ciphersuite_test.go

@@ -0,0 +1,66 @@
+package lwm2m
+
+import (
+	"reflect"
+	"testing"
+
+	"github.com/edgenesis/shifu/pkg/k8s/api/v1alpha1"
+	"github.com/pion/dtls/v2"
+)
+
+func TestCipherSuiteStringsToCodes(t *testing.T) {
+	tests := []struct {
+		name           string
+		input          []v1alpha1.CipherSuite
+		expectedOutput []dtls.CipherSuiteID
+		expectError    bool
+	}{
+		{
+			name: "Valid cipher suites",
+			input: []v1alpha1.CipherSuite{
+				v1alpha1.CipherSuite_TLS_ECDHE_ECDSA_WITH_AES_128_CCM,
+				v1alpha1.CipherSuite_TLS_ECDHE_RSA_WITH_AES_128_GCM_SHA256,
+			},
+			expectedOutput: []dtls.CipherSuiteID{
+				TLS_ECDHE_ECDSA_WITH_AES_128_CCM,
+				TLS_ECDHE_RSA_WITH_AES_128_GCM_SHA256,
+			},
+			expectError: false,
+		},
+		{
+			name: "Invalid cipher suite",
+			input: []v1alpha1.CipherSuite{
+				v1alpha1.CipherSuite("INVALID_CIPHER_SUITE"),
+			},
+			expectedOutput: nil,
+			expectError:    true,
+		},
+		{
+			name: "Mixed valid and invalid cipher suites",
+			input: []v1alpha1.CipherSuite{
+				v1alpha1.CipherSuite_TLS_ECDHE_ECDSA_WITH_AES_128_CCM,
+				v1alpha1.CipherSuite("INVALID_CIPHER_SUITE"),
+			},
+			expectedOutput: nil,
+			expectError:    true,
+		},
+		{
+			name:           "Empty input",
+			input:          []v1alpha1.CipherSuite{},
+			expectedOutput: []dtls.CipherSuiteID{},
+			expectError:    false,
+		},
+	}
+
+	for _, tt := range tests {
+		t.Run(tt.name, func(t *testing.T) {
+			output, err := CipherSuiteStringsToCodes(tt.input)
+			if (err != nil) != tt.expectError {
+				t.Errorf("expected error: %v, got: %v", tt.expectError, err)
+			}
+			if !tt.expectError && !reflect.DeepEqual(output, tt.expectedOutput) {
+				t.Errorf("expected output: %v, got: %v", tt.expectedOutput, output)
+			}
+		})
+	}
+}