123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134 |
- // Copyright 2018 The etcd Authors
- //
- // Licensed under the Apache License, Version 2.0 (the "License");
- // you may not use this file except in compliance with the License.
- // You may obtain a copy of the License at
- //
- // http://www.apache.org/licenses/LICENSE-2.0
- //
- // Unless required by applicable law or agreed to in writing, software
- // distributed under the License is distributed on an "AS IS" BASIS,
- // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
- // See the License for the specific language governing permissions and
- // limitations under the License.
- // +build cov
- package e2e
- import (
- "os"
- "testing"
- )
- func TestCtlV3Watch(t *testing.T) { testCtl(t, watchTest) }
- func TestCtlV3WatchNoTLS(t *testing.T) { testCtl(t, watchTest, withCfg(configNoTLS)) }
- func TestCtlV3WatchClientTLS(t *testing.T) { testCtl(t, watchTest, withCfg(configClientTLS)) }
- func TestCtlV3WatchPeerTLS(t *testing.T) { testCtl(t, watchTest, withCfg(configPeerTLS)) }
- func TestCtlV3WatchTimeout(t *testing.T) { testCtl(t, watchTest, withDialTimeout(0)) }
- func TestCtlV3WatchInteractive(t *testing.T) {
- testCtl(t, watchTest, withInteractive())
- }
- func TestCtlV3WatchInteractiveNoTLS(t *testing.T) {
- testCtl(t, watchTest, withInteractive(), withCfg(configNoTLS))
- }
- func TestCtlV3WatchInteractiveClientTLS(t *testing.T) {
- testCtl(t, watchTest, withInteractive(), withCfg(configClientTLS))
- }
- func TestCtlV3WatchInteractivePeerTLS(t *testing.T) {
- testCtl(t, watchTest, withInteractive(), withCfg(configPeerTLS))
- }
- func watchTest(cx ctlCtx) {
- tests := []struct {
- puts []kv
- envKey string
- envRange string
- args []string
- wkv []kvExec
- }{
- { // watch 1 key
- puts: []kv{{"sample", "value"}},
- args: []string{"sample", "--rev", "1"},
- wkv: []kvExec{{key: "sample", val: "value"}},
- },
- { // watch 1 key with env
- puts: []kv{{"sample", "value"}},
- envKey: "sample",
- args: []string{"--rev", "1"},
- wkv: []kvExec{{key: "sample", val: "value"}},
- },
- // coverage tests get extra arguments:
- // ./bin/etcdctl_test -test.coverprofile=e2e.1525392462795198897.coverprofile -test.outputdir=../..
- // do not test watch exec commands
- { // watch 3 keys by prefix
- puts: []kv{{"key1", "val1"}, {"key2", "val2"}, {"key3", "val3"}},
- args: []string{"key", "--rev", "1", "--prefix"},
- wkv: []kvExec{{key: "key1", val: "val1"}, {key: "key2", val: "val2"}, {key: "key3", val: "val3"}},
- },
- { // watch 3 keys by prefix, with env
- puts: []kv{{"key1", "val1"}, {"key2", "val2"}, {"key3", "val3"}},
- envKey: "key",
- args: []string{"--rev", "1", "--prefix"},
- wkv: []kvExec{{key: "key1", val: "val1"}, {key: "key2", val: "val2"}, {key: "key3", val: "val3"}},
- },
- { // watch by revision
- puts: []kv{{"etcd", "revision_1"}, {"etcd", "revision_2"}, {"etcd", "revision_3"}},
- args: []string{"etcd", "--rev", "2"},
- wkv: []kvExec{{key: "etcd", val: "revision_2"}, {key: "etcd", val: "revision_3"}},
- },
- { // watch 3 keys by range
- puts: []kv{{"key1", "val1"}, {"key3", "val3"}, {"key2", "val2"}},
- args: []string{"key", "key3", "--rev", "1"},
- wkv: []kvExec{{key: "key1", val: "val1"}, {key: "key2", val: "val2"}},
- },
- { // watch 3 keys by range, with env
- puts: []kv{{"key1", "val1"}, {"key3", "val3"}, {"key2", "val2"}},
- envKey: "key",
- envRange: "key3",
- args: []string{"--rev", "1"},
- wkv: []kvExec{{key: "key1", val: "val1"}, {key: "key2", val: "val2"}},
- },
- }
- for i, tt := range tests {
- donec := make(chan struct{})
- go func(i int, puts []kv) {
- for j := range puts {
- if err := ctlV3Put(cx, puts[j].key, puts[j].val, ""); err != nil {
- cx.t.Fatalf("watchTest #%d-%d: ctlV3Put error (%v)", i, j, err)
- }
- }
- close(donec)
- }(i, tt.puts)
- unsetEnv := func() {}
- if tt.envKey != "" || tt.envRange != "" {
- if tt.envKey != "" {
- os.Setenv("ETCDCTL_WATCH_KEY", tt.envKey)
- unsetEnv = func() { os.Unsetenv("ETCDCTL_WATCH_KEY") }
- }
- if tt.envRange != "" {
- os.Setenv("ETCDCTL_WATCH_RANGE_END", tt.envRange)
- unsetEnv = func() { os.Unsetenv("ETCDCTL_WATCH_RANGE_END") }
- }
- if tt.envKey != "" && tt.envRange != "" {
- unsetEnv = func() {
- os.Unsetenv("ETCDCTL_WATCH_KEY")
- os.Unsetenv("ETCDCTL_WATCH_RANGE_END")
- }
- }
- }
- if err := ctlV3Watch(cx, tt.args, tt.wkv...); err != nil {
- if cx.dialTimeout > 0 && !isGRPCTimedout(err) {
- cx.t.Errorf("watchTest #%d: ctlV3Watch error (%v)", i, err)
- }
- }
- unsetEnv()
- <-donec
- }
- }
|