mirror of
https://github.com/openimsdk/open-im-server.git
synced 2026-07-04 01:28:17 +08:00
92 lines
3.4 KiB
Go
92 lines
3.4 KiB
Go
package config
|
|
|
|
import (
|
|
"os"
|
|
"path/filepath"
|
|
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func TestLoadLogConfig(t *testing.T) {
|
|
var log Log
|
|
err := LoadConfig("../../../config/log.yml", "IMENV_LOG", &log)
|
|
assert.Nil(t, err)
|
|
assert.Equal(t, "../../../../logs/", log.StorageLocation)
|
|
}
|
|
|
|
func TestLoadMinioConfig(t *testing.T) {
|
|
var storageConfig Minio
|
|
err := LoadConfig("../../../config/minio.yml", "IMENV_MINIO", &storageConfig)
|
|
assert.Nil(t, err)
|
|
assert.Equal(t, "openim", storageConfig.Bucket)
|
|
}
|
|
|
|
func TestLoadWebhooksConfig(t *testing.T) {
|
|
var webhooks Webhooks
|
|
err := LoadConfig("../../../config/webhooks.yml", "IMENV_WEBHOOKS", &webhooks)
|
|
assert.Nil(t, err)
|
|
assert.Equal(t, 5, webhooks.BeforeAddBlack.Timeout)
|
|
|
|
}
|
|
|
|
func TestLoadDiscoveryKubernetesConfig(t *testing.T) {
|
|
path := filepath.Join(t.TempDir(), "discovery.yml")
|
|
err := os.WriteFile(path, []byte(`enable: kubernetes
|
|
kubernetes:
|
|
namespace: openim
|
|
etcd:
|
|
rootDirectory: openim
|
|
address: [localhost:12379]
|
|
`), 0600)
|
|
assert.Nil(t, err)
|
|
|
|
var discovery Discovery
|
|
err = LoadConfig(path, "IMENV_DISCOVERY", &discovery)
|
|
assert.Nil(t, err)
|
|
assert.Equal(t, KUBERNETES, discovery.Enable)
|
|
assert.Equal(t, "openim", discovery.Kubernetes.Namespace)
|
|
}
|
|
|
|
func TestLoadOpenIMRpcUserConfig(t *testing.T) {
|
|
var user User
|
|
err := LoadConfig("../../../config/openim-rpc-user.yml", "IMENV_OPENIM_RPC_USER", &user)
|
|
assert.Nil(t, err)
|
|
//export IMENV_OPENIM_RPC_USER_RPC_LISTENIP="0.0.0.0"
|
|
assert.Equal(t, "0.0.0.0", user.RPC.ListenIP)
|
|
//export IMENV_OPENIM_RPC_USER_RPC_PORTS="10110,10111,10112"
|
|
assert.Equal(t, []int{10110, 10111, 10112}, user.RPC.Ports)
|
|
}
|
|
|
|
func TestLoadNotificationConfig(t *testing.T) {
|
|
var noti Notification
|
|
err := LoadConfig("../../../config/notification.yml", "IMENV_NOTIFICATION", ¬i)
|
|
assert.Nil(t, err)
|
|
assert.Equal(t, "Your friend's profile has been changed", noti.FriendRemarkSet.OfflinePush.Title)
|
|
}
|
|
|
|
func TestLoadOpenIMThirdConfig(t *testing.T) {
|
|
var third Third
|
|
err := LoadConfig("../../../config/openim-rpc-third.yml", "IMENV_OPENIM_RPC_THIRD", &third)
|
|
assert.Nil(t, err)
|
|
assert.Equal(t, "enabled", third.Object.Enable)
|
|
assert.Equal(t, "https://oss-cn-chengdu.aliyuncs.com", third.Object.Oss.Endpoint)
|
|
assert.Equal(t, "my_bucket_name", third.Object.Oss.Bucket)
|
|
assert.Equal(t, "https://my_bucket_name.oss-cn-chengdu.aliyuncs.com", third.Object.Oss.BucketURL)
|
|
assert.Equal(t, "AKID1234567890", third.Object.Oss.AccessKeyID)
|
|
assert.Equal(t, "abc123xyz789", third.Object.Oss.AccessKeySecret)
|
|
assert.Equal(t, "session_token_value", third.Object.Oss.SessionToken) // Uncomment if session token is needed
|
|
assert.Equal(t, true, third.Object.Oss.PublicRead)
|
|
|
|
// Environment: IMENV_OPENIM_RPC_THIRD_OBJECT_ENABLE=enabled;IMENV_OPENIM_RPC_THIRD_OBJECT_OSS_ENDPOINT=https://oss-cn-chengdu.aliyuncs.com;IMENV_OPENIM_RPC_THIRD_OBJECT_OSS_BUCKET=my_bucket_name;IMENV_OPENIM_RPC_THIRD_OBJECT_OSS_BUCKETURL=https://my_bucket_name.oss-cn-chengdu.aliyuncs.com;IMENV_OPENIM_RPC_THIRD_OBJECT_OSS_ACCESSKEYID=AKID1234567890;IMENV_OPENIM_RPC_THIRD_OBJECT_OSS_ACCESSKEYSECRET=abc123xyz789;IMENV_OPENIM_RPC_THIRD_OBJECT_OSS_SESSIONTOKEN=session_token_value;IMENV_OPENIM_RPC_THIRD_OBJECT_OSS_PUBLICREAD=true
|
|
}
|
|
|
|
func TestTransferConfig(t *testing.T) {
|
|
var tran MsgTransfer
|
|
err := LoadConfig("../../../config/openim-msgtransfer.yml", "IMENV_OPENIM-MSGTRANSFER", &tran)
|
|
assert.Nil(t, err)
|
|
assert.Equal(t, true, tran.Prometheus.Enable)
|
|
assert.Equal(t, true, tran.Prometheus.AutoSetPorts)
|
|
}
|