Lwm2m2 unitest

pkg/deviceshifu/deviceshifulwm2m/deviceshifulwm2mconfig_test.go

@@ -0,0 +1,60 @@
+package deviceshifulwm2m
+
+import (
+	"testing"
+
+	"github.com/edgenesis/shifu/pkg/deviceshifu/deviceshifubase"
+	"github.com/stretchr/testify/assert"
+)
+
+func TestCreateLwM2MInstructions(t *testing.T) {
+	// Initialize test data
+	dsInstructions := &deviceshifubase.DeviceShifuInstructions{
+		Instructions: map[string]*deviceshifubase.DeviceShifuInstruction{
+			"instruction1": {
+				DeviceShifuProtocolProperties: map[string]string{
+					objectIdStr:      "123",
+					enableObserveStr: "true",
+				},
+			},
+			"instruction2": {
+				DeviceShifuProtocolProperties: map[string]string{
+					objectIdStr:      "456",
+					enableObserveStr: "false",
+				},
+			},
+		},
+	}
+
+	// Call the function under test
+	result := CreateLwM2MInstructions(dsInstructions)
+
+	// Assert that the result is not nil
+	assert.NotNil(t, result)
+	assert.Equal(t, 2, len(result.Instructions))
+
+	// Check if each instruction's properties are correctly mapped
+	instruction1 := result.Instructions["instruction1"]
+	assert.NotNil(t, instruction1)
+	assert.Equal(t, "123", instruction1.ObjectId)
+	assert.True(t, instruction1.EnableObserve)
+
+	instruction2 := result.Instructions["instruction2"]
+	assert.NotNil(t, instruction2)
+	assert.Equal(t, "456", instruction2.ObjectId)
+	assert.False(t, instruction2.EnableObserve)
+}
+
+func TestCreateLwM2MInstructions_EmptyInstructions(t *testing.T) {
+	// Test the case of an empty instruction map
+	dsInstructions := &deviceshifubase.DeviceShifuInstructions{
+		Instructions: map[string]*deviceshifubase.DeviceShifuInstruction{},
+	}
+
+	// Call the function under test
+	result := CreateLwM2MInstructions(dsInstructions)
+
+	// Assert that the result is not nil and the instruction map is empty
+	assert.NotNil(t, result)
+	assert.Equal(t, 0, len(result.Instructions))
+}

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)
+			}
+		})
+	}
+}