tests: Handle multiple cluster configurations

This commit is contained in:
Marek Siarkowicz 2022-02-18 12:48:56 +01:00
parent 1c8adcd830
commit 58c3808919

View File

@ -24,28 +24,41 @@ import (
func TestKVPut(t *testing.T) { func TestKVPut(t *testing.T) {
testRunner.BeforeTest(t) testRunner.BeforeTest(t)
clus := testRunner.NewCluster(t, config.ClusterConfig{ClusterSize: 1, PeerTLS: config.AutoTLS}) tcs := []struct {
defer clus.Close() name string
cc := clus.Client() config config.ClusterConfig
}{
{
name: "PeerAutoTLS",
config: config.ClusterConfig{ClusterSize: 1, PeerTLS: config.AutoTLS},
},
}
for _, tc := range tcs {
t.Run(tc.name, func(t *testing.T) {
clus := testRunner.NewCluster(t, tc.config)
defer clus.Close()
cc := clus.Client()
testutils.ExecuteWithTimeout(t, 10*time.Second, func() { testutils.ExecuteWithTimeout(t, 10*time.Second, func() {
key, value := "foo", "bar" key, value := "foo", "bar"
if err := cc.Put(key, value); err != nil { if err := cc.Put(key, value); err != nil {
t.Fatalf("count not put key %q, err: %s", key, err) t.Fatalf("count not put key %q, err: %s", key, err)
} }
resp, err := cc.Get(key, config.GetOptions{Serializable: true}) resp, err := cc.Get(key, config.GetOptions{Serializable: true})
if err != nil { if err != nil {
t.Fatalf("count not get key %q, err: %s", key, err) t.Fatalf("count not get key %q, err: %s", key, err)
} }
if len(resp.Kvs) != 1 { if len(resp.Kvs) != 1 {
t.Errorf("Unexpected lenth of response, got %d", len(resp.Kvs)) t.Errorf("Unexpected lenth of response, got %d", len(resp.Kvs))
} }
if string(resp.Kvs[0].Key) != key { if string(resp.Kvs[0].Key) != key {
t.Errorf("Unexpected key, want %q, got %q", key, resp.Kvs[0].Key) t.Errorf("Unexpected key, want %q, got %q", key, resp.Kvs[0].Key)
} }
if string(resp.Kvs[0].Value) != value { if string(resp.Kvs[0].Value) != value {
t.Errorf("Unexpected value, want %q, got %q", value, resp.Kvs[0].Value) t.Errorf("Unexpected value, want %q, got %q", value, resp.Kvs[0].Value)
} }
}) })
})
}
} }