You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
43 lines
1.0 KiB
Go
43 lines
1.0 KiB
Go
package zrpc
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/tal-tech/go-zero/core/discov"
|
|
"github.com/tal-tech/go-zero/core/service"
|
|
"github.com/tal-tech/go-zero/core/stores/redis"
|
|
)
|
|
|
|
func TestRpcClientConf(t *testing.T) {
|
|
conf := NewDirectClientConf([]string{"localhost:1234"}, "foo", "bar")
|
|
assert.True(t, conf.HasCredential())
|
|
conf = NewEtcdClientConf([]string{"localhost:1234", "localhost:5678"}, "key", "foo", "bar")
|
|
assert.True(t, conf.HasCredential())
|
|
}
|
|
|
|
func TestRpcServerConf(t *testing.T) {
|
|
conf := RpcServerConf{
|
|
ServiceConf: service.ServiceConf{},
|
|
ListenOn: "",
|
|
Etcd: discov.EtcdConf{
|
|
Hosts: []string{"localhost:1234"},
|
|
Key: "key",
|
|
},
|
|
Auth: true,
|
|
Redis: redis.RedisKeyConf{
|
|
RedisConf: redis.RedisConf{
|
|
Type: redis.NodeType,
|
|
},
|
|
Key: "foo",
|
|
},
|
|
StrictControl: false,
|
|
Timeout: 0,
|
|
CpuThreshold: 0,
|
|
}
|
|
assert.True(t, conf.HasEtcd())
|
|
assert.NotNil(t, conf.Validate())
|
|
conf.Redis.Host = "localhost:5678"
|
|
assert.Nil(t, conf.Validate())
|
|
}
|