Skip to content

Commit aa04a0d

Browse files
committed
test: fix remote test
1 parent 63496e3 commit aa04a0d

File tree

1 file changed

+10
-6
lines changed

1 file changed

+10
-6
lines changed

c_test.go

Lines changed: 10 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -10,7 +10,6 @@ import (
1010
"time"
1111

1212
"github.com/DoNewsCode/core/config"
13-
"github.com/DoNewsCode/core/config/remote"
1413
"github.com/DoNewsCode/core/contract"
1514
"github.com/DoNewsCode/core/di"
1615
"github.com/DoNewsCode/core/events"
@@ -135,14 +134,14 @@ func TestC_Remote(t *testing.T) {
135134
addr := os.Getenv("ETCD_ADDR")
136135
if addr == "" {
137136
t.Skip("set ETCD_ADDR for run remote test")
137+
return
138138
}
139139
key := "core.yaml"
140140
envEtcdAddrs := strings.Split(addr, ",")
141141
cfg := clientv3.Config{
142142
Endpoints: envEtcdAddrs,
143143
DialTimeout: 2 * time.Second,
144144
}
145-
_ = remote.Provider(key, &cfg)
146145
if err := put(cfg, key, "name: remote"); err != nil {
147146
t.Fatal(err)
148147
}
@@ -210,9 +209,14 @@ func put(cfg clientv3.Config, key, val string) error {
210209
}
211210
defer client.Close()
212211

213-
_, err = client.Put(context.Background(), key, val)
214-
if err != nil {
215-
return err
212+
ctx, cancel := context.WithTimeout(context.Background(), time.Second)
213+
defer cancel()
214+
215+
for i := 0; i < 2; i++ {
216+
_, err = client.Put(ctx, key, val)
217+
if err != nil {
218+
continue
219+
}
216220
}
217-
return nil
221+
return err
218222
}

0 commit comments

Comments
 (0)