test: fix unbounded context

DoNewsCode · Aug 6, 2021 · f705cf6 · f705cf6
1 parent d95c325
commit f705cf6
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/config/remote/etcd/etcd_test.go b/config/remote/etcd/etcd_test.go
@@ -133,7 +133,10 @@ func put(r *ETCD, val string) error {
 	}
 	defer client.Close()
 
-	_, err = client.Put(context.Background(), r.key, val)
+	ctx, cancel := context.WithTimeout(context.Background(), time.Second)
+	defer cancel()
+
+	_, err = client.Put(ctx, r.key, val)
 	if err != nil {
 		return err
 	}