Added test setup that adds keys for RANDOMKEY command instead of relying on keys from other tests.

echovault/keyspace.go

@@ -642,22 +642,22 @@ func (server *EchoVault) randomKey(ctx context.Context) string {
 	defer server.storeLock.RUnlock()
 
 	database := ctx.Value("Database").(int)
-    
-	    _max := len(server.store[database])
-	    if _max == 0 {
-	        return ""
-	    }
 
-	    randnum := rand.Intn(_max)
-	    i := 0
-	    var randkey string
+	_max := len(server.store[database])
+	if _max == 0 {
+		return ""
+	}
 
-	    for key, _ := range server.store[database] {
-	        if i == randnum {
-	            randkey = key
-	        } else {
-	            i++
-	        }
+	randnum := rand.Intn(_max)
+	i := 0
+	var randkey string
+
+	for key, _ := range server.store[database] {
+		if i == randnum {
+			randkey = key
+		} else {
+			i++
+		}
 	}
 
 	return randkey

internal/modules/generic/commands_test.go

@@ -2021,7 +2021,6 @@ func Test_Generic(t *testing.T) {
 	t.Run("Test_HandlerDECR", func(t *testing.T) {
 		t.Parallel()
 		conn, err := internal.GetConnection("localhost", port)
-
 		if err != nil {
 			t.Error(err)
 			return
@@ -2760,6 +2759,20 @@ func Test_Generic(t *testing.T) {
 
 	t.Run("Test_HandleRANDOMKEY", func(t *testing.T) {
 		t.Parallel()
+
+		// Populate the store with keys first
+		for i := 0; i < 10; i++ {
+			_, _, err := mockServer.Set(
+				fmt.Sprintf("RandomKey%d", i),
+				fmt.Sprintf("Value%d", i),
+				echovault.SetOptions{},
+			)
+			if err != nil {
+				t.Error(err)
+				return
+			}
+		}
+
 		conn, err := internal.GetConnection("localhost", port)
 		if err != nil {
 			t.Error(err)
@@ -2770,23 +2783,18 @@ func Test_Generic(t *testing.T) {
 		}()
 		client := resp.NewConn(conn)
 
-        // Reuse existing keys to test RANDOMKEY
-        // All keys used for other tests contain the substring 'Key'
-        expected := "Key"
-        // Test RANDOMKEY
-        if err = client.WriteArray([]resp.Value{resp.StringValue("RANDOMKEY")}); err != nil {
-            t.Error(err)
-        }
+		expected := "Key"
+		if err = client.WriteArray([]resp.Value{resp.StringValue("RANDOMKEY")}); err != nil {
+			t.Error(err)
+		}
 
-        res, _, err := client.ReadValue()
-        if err != nil {
-            t.Error(err)
-        }
-
-        if !strings.Contains(res.String(), expected) {
-            t.Errorf("expected a key containing substring '%s', got %s", expected, res.String())
-        }
+		res, _, err := client.ReadValue()
+		if err != nil {
+			t.Error(err)
+		}
 
+		if !strings.Contains(res.String(), expected) {
+			t.Errorf("expected a key containing substring '%s', got %s", expected, res.String())
+		}
 	})
-
 }