mirror of
https://github.com/zrepl/zrepl.git
synced 2024-11-29 03:45:27 +01:00
87 lines
1.9 KiB
Go
87 lines
1.9 KiB
Go
package config
|
|
|
|
import (
|
|
"github.com/stretchr/testify/assert"
|
|
"testing"
|
|
"time"
|
|
)
|
|
|
|
func TestRPC(t *testing.T) {
|
|
conf := testValidConfig(t, `
|
|
jobs:
|
|
- name: pull_servers
|
|
type: pull
|
|
connect:
|
|
type: tcp
|
|
address: "server1.foo.bar:8888"
|
|
rpc:
|
|
timeout: 20s # different form default, should merge
|
|
root_dataset: "pool2/backup_servers"
|
|
interval: 10m
|
|
pruning:
|
|
keep_sender:
|
|
- type: not_replicated
|
|
keep_receiver:
|
|
- type: last_n
|
|
count: 100
|
|
|
|
- name: pull_servers2
|
|
type: pull
|
|
connect:
|
|
type: tcp
|
|
address: "server1.foo.bar:8888"
|
|
rpc:
|
|
tx_chunk_size: 0xabcd # different from default, should merge
|
|
root_dataset: "pool2/backup_servers"
|
|
interval: 10m
|
|
pruning:
|
|
keep_sender:
|
|
- type: not_replicated
|
|
keep_receiver:
|
|
- type: last_n
|
|
count: 100
|
|
|
|
- type: sink
|
|
name: "laptop_sink"
|
|
root_dataset: "pool2/backup_laptops"
|
|
serve:
|
|
type: tcp
|
|
listen: "192.168.122.189:8888"
|
|
clients: {
|
|
"10.23.42.23":"client1"
|
|
}
|
|
rpc:
|
|
rx_structured_max: 0x2342
|
|
|
|
- type: sink
|
|
name: "other_sink"
|
|
root_dataset: "pool2/backup_laptops"
|
|
serve:
|
|
type: tcp
|
|
listen: "192.168.122.189:8888"
|
|
clients: {
|
|
"10.23.42.23":"client1"
|
|
}
|
|
rpc:
|
|
send_heartbeat_interval: 10s
|
|
|
|
`)
|
|
|
|
assert.Equal(t, 20*time.Second, conf.Jobs[0].Ret.(*PullJob).Connect.Ret.(*TCPConnect).RPC.Timeout)
|
|
assert.Equal(t, uint32(0xabcd), conf.Jobs[1].Ret.(*PullJob).Connect.Ret.(*TCPConnect).RPC.TxChunkSize)
|
|
assert.Equal(t, uint32(0x2342), conf.Jobs[2].Ret.(*SinkJob).Serve.Ret.(*TCPServe).RPC.RxStructuredMaxLen)
|
|
assert.Equal(t, 10*time.Second, conf.Jobs[3].Ret.(*SinkJob).Serve.Ret.(*TCPServe).RPC.SendHeartbeatInterval)
|
|
defConf := RPCConfig{}
|
|
Default(&defConf)
|
|
assert.Equal(t, defConf.Timeout, conf.Global.RPC.Timeout)
|
|
}
|
|
|
|
func TestGlobal_DefaultRPCConfig(t *testing.T) {
|
|
assert.NotPanics(t, func() {
|
|
var c RPCConfig
|
|
Default(&c)
|
|
assert.NotNil(t, c)
|
|
assert.Equal(t, c.TxChunkSize, uint32(1)<<15)
|
|
})
|
|
}
|