add tests for sync message versions

Signed-off-by: Dmitri Dolguikh <dmitri.external@netbird.io>

management/internals/shared/grpc/sync_message_versions.go

@@ -67,7 +67,7 @@ func SyncMessageVersionsFromString(toconvert []string) []SyncMessageVersion {
 	for _, v := range AllSyncMessageVersions {
 		allversions[v.String()] = v
 	}
-	toret := make([]SyncMessageVersion, len(toconvert))
+	toret := make([]SyncMessageVersion, 0)
 	for _, s := range toconvert {
 		toret = append(toret, allversions[s])
 	}
@@ -95,6 +95,8 @@ func SyncMessageVersionsFromMap(toconvert map[string][]string) map[string][]Sync
 	return toret
 }
 
+// these come from the client; peer capabilities are expected to contain all enabled sync message versions.
+// an empty list is interpreted as all but the base version are disabled.
 func SyncMessageVersionsFromProtoEnums(peerCapabilities []int32) []SyncMessageVersion {
 	toret := make([]SyncMessageVersion, 0)
 	for _, pc := range peerCapabilities {

management/internals/shared/grpc/sync_message_versions_test.go

@@ -0,0 +1,45 @@
+package grpc
+
+import (
+	"testing"
+
+	"github.com/stretchr/testify/assert"
+)
+
+func TestValidation(t *testing.T) {
+	assert.NoError(t, ValidateSyncMessageVersions([]string{"Base", "ComponentNetworkMap"}))
+	assert.NoError(t, ValidateSyncMessageVersions([]string{}))
+	assert.ErrorIs(t, ValidateSyncMessageVersions([]string{"Boom"}), ErrorUnrecognizedSyncMessageVersion)
+	assert.ErrorIs(t, ValidateSyncMessageVersions([]string{"Base", "Boom"}), ErrorUnrecognizedSyncMessageVersion)
+}
+
+func TestConversionStringToEnum(t *testing.T) {
+	assert.Equal(t, []SyncMessageVersion{0, 1}, SyncMessageVersionsFromString([]string{}))
+	assert.Equal(t, []SyncMessageVersion{0}, SyncMessageVersionsFromString([]string{"Base"}))
+	assert.Equal(t, []SyncMessageVersion{1}, SyncMessageVersionsFromString([]string{"ComponentNetworkMap"}))
+}
+
+func TestPerAccountConversionStringToEnum(t *testing.T) {
+	assert.Equal(t, map[string][]SyncMessageVersion{"1": {0, 1}}, SyncMessageVersionsFromMap(map[string][]string{"1": {}}))
+	assert.Equal(t, map[string][]SyncMessageVersion{"2": {0}}, SyncMessageVersionsFromMap(map[string][]string{"2": {"Base"}}))
+	assert.Equal(t, map[string][]SyncMessageVersion{"3": {1}, "4": {0, 1}},
+		SyncMessageVersionsFromMap(map[string][]string{
+			"3": {"ComponentNetworkMap"},
+			"4": {"Base", "ComponentNetworkMap"},
+		}))
+}
+
+func TestConversionFromProtoEnums(t *testing.T) {
+	assert.Equal(t, []SyncMessageVersion{}, SyncMessageVersionsFromProtoEnums([]int32{}))
+	assert.Equal(t, []SyncMessageVersion{}, SyncMessageVersionsFromProtoEnums([]int32{0}))
+	assert.Equal(t, []SyncMessageVersion{1}, SyncMessageVersionsFromProtoEnums([]int32{3}))
+}
+
+func TestCommonVersions(t *testing.T) {
+	assert.Equal(t, []SyncMessageVersion{0},
+		CommonSyncMessageVersions([]SyncMessageVersion{0, 1}, []SyncMessageVersion{}))
+	assert.Equal(t, []SyncMessageVersion{0},
+		CommonSyncMessageVersions([]SyncMessageVersion{0}, []SyncMessageVersion{0, 1, 2}))
+	assert.Equal(t, []SyncMessageVersion{5, 1, 0},
+		CommonSyncMessageVersions([]SyncMessageVersion{0, 1, 4, 5}, []SyncMessageVersion{1, 0, 5, 2}))
+}