util: improve mask key function and add tests

pkg/util/string.go

@@ -13,8 +13,17 @@ func StringSliceContains(slice []string, needle string) bool {
 }
 
 func MaskKey(key string) string {
-	maskKey := key[0:5]
+	if len(key) == 0 {
-	maskKey += strings.Repeat("*", len(key)-1-5-5)
+		return "{empty}"
-	maskKey += key[len(key)-5-1:]
+	}
+
+	h := len(key) / 3
+	if h > 5 {
+		h = 5
+	}
+
+	maskKey := key[0:h]
+	maskKey += strings.Repeat("*", len(key)-h*2)
+	maskKey += key[len(key)-h:]
 	return maskKey
 }

pkg/util/string_test.go

@@ -0,0 +1,42 @@
+package util
+
+import "testing"
+
+func TestMaskKey(t *testing.T) {
+	type args struct {
+		key string
+	}
+	tests := []struct {
+		name string
+		args args
+		want string
+	}{
+		{
+			name: "key length more than 5",
+			args: args{key: "abcdefghijklmnopqr"},
+			want: "abcde********nopqr",
+		},
+		{
+			name: "key length less than 10",
+			args: args{key: "12345678"},
+			want: "12****78",
+		},
+		{
+			name: "even",
+			args: args{key: "1234567"},
+			want: "12***67",
+		},
+		{
+			name: "empty",
+			args: args{key: ""},
+			want: "{empty}",
+		},
+	}
+	for _, tt := range tests {
+		t.Run(tt.name, func(t *testing.T) {
+			if got := MaskKey(tt.args.key); got != tt.want {
+				t.Errorf("MaskKey() = %v, want %v", got, tt.want)
+			}
+		})
+	}
+}