feat: add file safety config integration

server/filesafety/config_test.go

@@ -0,0 +1,51 @@
+package filesafety
+
+import (
+	"testing"
+
+	"github.com/tinode/chat/server/store"
+	"github.com/tinode/chat/server/store/types"
+)
+
+type fakeThirdPartyConfig map[string]string
+
+func (f fakeThirdPartyConfig) Get(group string, keys ...string) (map[string]string, error) {
+	result := map[string]string{}
+	for _, key := range keys {
+		if value, ok := f[key]; ok {
+			result[key] = value
+		}
+	}
+	return result, nil
+}
+
+func TestLoadConfig(t *testing.T) {
+	original := store.ThirdPartyConfig
+	store.ThirdPartyConfig = fakeThirdPartyConfig{
+		"file_review_content_enabled":   "1",
+		"file_review_content_provider":  types.FileReviewProviderAliyunGreen,
+		"aliyun_green_region_id":        "cn-shanghai",
+		"aliyun_green_endpoint":         "green-cip.cn-shanghai.aliyuncs.com",
+		"aliyun_green_document_service": "document_detection",
+		"file_review_malware_enabled":   "true",
+		"file_review_malware_provider":  types.FileReviewProviderAliyunSAS,
+		"aliyun_sas_region_id":          "cn-shanghai",
+		"aliyun_access_key_id":          "ak",
+		"aliyun_access_key_secret":      "sk",
+	}
+	defer func() { store.ThirdPartyConfig = original }()
+
+	config, err := LoadConfig()
+	if err != nil {
+		t.Fatalf("LoadConfig failed: %v", err)
+	}
+	if !config.Content.Enabled || config.Content.Provider != types.FileReviewProviderAliyunGreen {
+		t.Fatalf("unexpected content config: %+v", config.Content)
+	}
+	if !config.Malware.Enabled || config.Malware.Provider != types.FileReviewProviderAliyunSAS {
+		t.Fatalf("unexpected malware config: %+v", config.Malware)
+	}
+	if config.Content.Green.AccessKeyID != "ak" || config.Content.Green.RegionID != "cn-shanghai" {
+		t.Fatalf("unexpected green config: %+v", config.Content.Green)
+	}
+}