feat: publish v1.0.0-rc.36-custom.1

setting/model_setting/gemini_test.go

@@ -0,0 +1,99 @@
+package model_setting
+
+import (
+	"testing"
+
+	"github.com/stretchr/testify/assert"
+	"github.com/stretchr/testify/require"
+)
+
+func TestGeminiSafetySettingsReadNormalization(t *testing.T) {
+	original := geminiSettings.SafetySettings
+	t.Cleanup(func() {
+		geminiSettings.SafetySettings = original
+	})
+
+	tests := []struct {
+		name     string
+		settings map[string]string
+		key      string
+		want     string
+	}{
+		{
+			name:     "nil map gets OFF default",
+			settings: nil,
+			key:      "HARM_CATEGORY_HATE_SPEECH",
+			want:     "OFF",
+		},
+		{
+			name: "missing default gets OFF without replacing existing values",
+			settings: map[string]string{
+				"HARM_CATEGORY_HATE_SPEECH": "BLOCK_SOME",
+			},
+			key:  "HARM_CATEGORY_HATE_SPEECH",
+			want: "BLOCK_SOME",
+		},
+		{
+			name: "empty default gets OFF",
+			settings: map[string]string{
+				"default": "",
+			},
+			key:  "HARM_CATEGORY_HATE_SPEECH",
+			want: "OFF",
+		},
+		{
+			name: "empty override falls back to configured default",
+			settings: map[string]string{
+				"default":                   "BLOCK_ONLY_HIGH",
+				"HARM_CATEGORY_HATE_SPEECH": "",
+			},
+			key:  "HARM_CATEGORY_HATE_SPEECH",
+			want: "BLOCK_ONLY_HIGH",
+		},
+		{
+			name: "historical invalid nonempty default is preserved",
+			settings: map[string]string{
+				"default": "BLOCK_SOME",
+			},
+			key:  "HARM_CATEGORY_HATE_SPEECH",
+			want: "BLOCK_SOME",
+		},
+	}
+
+	for _, test := range tests {
+		t.Run(test.name, func(t *testing.T) {
+			geminiSettings.SafetySettings = test.settings
+
+			assert.Equal(t, test.want, GetGeminiSafetySetting(test.key))
+		})
+	}
+}
+
+func TestValidateGeminiSafetySettings(t *testing.T) {
+	valid := []string{
+		`{}`,
+		`{"default":""}`,
+		`{"HARM_CATEGORY_HATE_SPEECH":""}`,
+		`{"default":"OFF"}`,
+		`{"default":"BLOCK_NONE"}`,
+		`{"default":"BLOCK_ONLY_HIGH"}`,
+		`{"default":"BLOCK_MEDIUM_AND_ABOVE"}`,
+		`{"default":"BLOCK_LOW_AND_ABOVE"}`,
+		`{"default":"HARM_BLOCK_THRESHOLD_UNSPECIFIED"}`,
+	}
+	for _, value := range valid {
+		require.NoError(t, ValidateGeminiSafetySettings(value), value)
+	}
+
+	invalid := []string{
+		`null`,
+		`[]`,
+		`{"default":1}`,
+		`{"default":"BLOCK_SOME"}`,
+		`{"default":" off "}`,
+		`{"default":`,
+	}
+	for _, value := range invalid {
+		assert.Error(t, ValidateGeminiSafetySettings(value), value)
+	}
+}