Break the seed retry using the context

toggle/client.go

@@ -186,12 +186,12 @@ func (c *Client) Connect(ctx context.Context) chan error {
 				c.opts.log.Printf("Error sending the seed flags: %v. Retry in %s", err, retry)
 				backoff++
 
-				time.Sleep(retry)
-				continue
-			}
-
-			if ctx.Err() != nil {
-				return
+				select {
+				case <-ctx.Done():
+					return
+				case <-time.After(retry):
+					continue
+				}
 			}
 
 			break

toggle/toggle.go

@@ -20,8 +20,9 @@ func Initialize(ctx context.Context, name string, opts ...ClientOption) {
 	DefaultClient = New(name, opts...)
 	DefaultClient.ParseEnv(os.Environ())
 
+	errC := DefaultClient.Connect(ctx)
 	go func() {
-		for err := range DefaultClient.Connect(ctx) {
+		for err := range errC {
 			DefaultClient.opts.log.Println("Error listening for updates:", err)
 		}
 	}()

toggle/toggle_test.go

@@ -20,7 +20,6 @@ func TestGet(t *testing.T) {
 		cname   string
 		seed    []string
 		args    args
-		sub     bool
 		want    bool
 		wantRaw string
 	}{
@@ -49,9 +48,6 @@ func TestGet(t *testing.T) {
 
 			toggle.Initialize(ctx, tt.cname)
 
-			if tt.sub {
-				toggle.DefaultClient = toggle.New(tt.cname)
-			}
 			if got := toggle.Get(tt.args.name, tt.args.opts...); got != tt.want {
 				t.Errorf("Get() = %v, want %v", got, tt.want)
 			}