@@ -18,29 +18,36 @@ func setupSearchIndex(t *testing.T, addr string) error {
1818 }
1919 defer c .Close ()
2020
21- // create search index, based on https://redis.io/docs/latest/develop/get-started/document-database/
22- if _ , err := doRedisCmd (c , "FT.CREATE" , "idx:bicycle " , "ON" , "JSON" , "PREFIX" , "1" , "bicycle:" , "SCORE" , "1.0" , " SCHEMA" , "$.brand " , "AS " , "brand " , "TEXT " , "WEIGHT " , "1.0 " , "$.model " , "AS " , "model " , "TEXT " , "WEIGHT " , "1.0 " , "$.description " , "AS" , "description" , "TEXT" , "WEIGHT" , "1.0" , "$.price" , "AS" , "price" , "NUMERIC" , "$.condition" , "AS" , "condition" , "TAG" , "SEPARATOR" , ", " ); err != nil {
21+ // create search index, based on https://redis.io/docs/latest/commands/ft.create and https://valkey.io/commands/ft.create
22+ if _ , err := doRedisCmd (c , "FT.CREATE" , "test_index " , "SCHEMA" , "my_hash_field_key " , "VECTOR " , "HNSW " , "10 " , "TYPE " , "FLOAT32 " , "DIM " , "20 " , "DISTANCE_METRIC " , "COSINE " , "M " , "4 " , "EF_CONSTRUCTION " , "100 " ); err != nil {
2323 log .Printf ("setupSearchIndex() - couldn't create search index, err: %s " , err )
2424 return err
2525 }
2626 return nil
2727}
2828
2929func TestExtractSearchIndexesMetrics (t * testing.T ) {
30- if os .Getenv ("TEST_REDIS8_URI" ) == "" {
31- t .Skipf ("TEST_REDIS8_URI not set - skipping" )
30+ test_redis8_uri := os .Getenv ("TEST_REDIS8_URI" )
31+ test_valkey8_bundle_uri := os .Getenv ("TEST_VALKEY8_BUNDLE_URI" )
32+ if test_redis8_uri == "" || test_valkey8_bundle_uri == "" {
33+ t .Skipf ("TEST_REDIS8_URI or TEST_VALKEY8_BUNDLE_URI aren't set - skipping" )
3234 }
33- if err := setupSearchIndex (t , os .Getenv ("TEST_REDIS8_URI" )); err != nil {
34- t .Fatalf ("couldn't create search index, err: %s " , err )
35+ if err := setupSearchIndex (t , test_redis8_uri ); err != nil {
36+ t .Fatalf ("couldn't create search index in TEST_REDIS8_URI (%s), err: %s " , test_redis8_uri , err )
37+ }
38+ if err := setupSearchIndex (t , test_valkey8_bundle_uri ); err != nil {
39+ t .Fatalf ("couldn't create search index in TEST_VALKEY8_BUNDLE_URI (%s), err: %s " , test_valkey8_bundle_uri , err )
3540 }
3641
3742 tsts := []struct {
3843 addr string
3944 inclSearchIndexesMetrics bool
4045 wantSearchIndexesMetrics bool
4146 }{
42- {addr : os .Getenv ("TEST_REDIS8_URI" ), inclSearchIndexesMetrics : true , wantSearchIndexesMetrics : true },
43- {addr : os .Getenv ("TEST_REDIS8_URI" ), inclSearchIndexesMetrics : false , wantSearchIndexesMetrics : false },
47+ {addr : test_redis8_uri , inclSearchIndexesMetrics : true , wantSearchIndexesMetrics : true },
48+ {addr : test_redis8_uri , inclSearchIndexesMetrics : false , wantSearchIndexesMetrics : false },
49+ {addr : test_valkey8_bundle_uri , inclSearchIndexesMetrics : true , wantSearchIndexesMetrics : true },
50+ {addr : test_valkey8_bundle_uri , inclSearchIndexesMetrics : false , wantSearchIndexesMetrics : false },
4451 }
4552
4653 for _ , tst := range tsts {
0 commit comments