test: rueidislock in case of flush

rueidislock/lock.go

@@ -385,14 +385,6 @@ func (m *locker) WithContext(ctx context.Context, name string) (context.Context,
 			}
 		}
 		if cancel(); err != nil {
-			if err == ErrLockerClosed {
-				m.mu.RLock()
-				closed := m.gates == nil
-				m.mu.RUnlock()
-				if !closed {
-					continue
-				}
-			}
 			return ctx, cancel, err
 		}
 	}

rueidislock/lock_test.go

@@ -12,7 +12,7 @@ import (
 	"github.com/redis/rueidis"
 )
 
-var address = []string{"127.0.0.1:6376"}
+var address = []string{"127.0.0.1:6379"}
 
 func newLocker(t *testing.T, noLoop, setpx, nocsc bool) *locker {
 	impl, err := NewLocker(LockerOption{
@@ -512,3 +512,48 @@ func TestLocker_RetryErrLockerClosed(t *testing.T) {
 		})
 	}
 }
+
+func TestLocker_Flush(t *testing.T) {
+	test := func(t *testing.T, noLoop, setpx, nocsc bool) {
+		client, err := rueidis.NewClient(rueidis.ClientOption{InitAddress: address})
+		if err != nil {
+			t.Fatal(err)
+		}
+		defer client.Close()
+
+		locker := newLocker(t, noLoop, setpx, nocsc)
+
+		lck := strconv.Itoa(rand.Int())
+		ctx, _, err := locker.WithContext(context.Background(), lck)
+		if err != nil {
+			t.Fatal(err)
+		}
+
+		if err := client.Do(context.Background(), client.B().Flushall().Build()).Error(); err != nil {
+			t.Fatal(err)
+		}
+
+		<-ctx.Done()
+
+		if err := ctx.Err(); !errors.Is(err, context.Canceled) {
+			t.Fatal(err)
+		}
+
+		ctx, cancel, err := locker.WithContext(context.Background(), strconv.Itoa(rand.Int()))
+		if err != nil {
+			t.Fatal(err)
+		}
+		cancel()
+	}
+	for _, nocsc := range []bool{false, true} {
+		t.Run("Tracking Loop", func(t *testing.T) {
+			test(t, false, false, nocsc)
+		})
+		t.Run("Tracking NoLoop", func(t *testing.T) {
+			test(t, true, false, nocsc)
+		})
+		t.Run("SET PX", func(t *testing.T) {
+			test(t, true, true, nocsc)
+		})
+	}
+}