mirror of
https://github.com/redis/go-redis.git
synced 2026-10-24 07:59:20 +03:00
b506c612a4
* script: avoid client-side SHA1 in FIPS mode * modifying the script.go to add new constructor to explicitly control SHA-1 Behaviour(strict FIPS Builds - FIPS140) * fixing the lint errors * Add unit and integration tests for NewScriptServerSHA * fixing the mod format issue * Update go.mod Co-authored-by: Nedyalko Dyakov <1547186+ndyakov@users.noreply.github.com> * Update go.sum Co-authored-by: Nedyalko Dyakov <1547186+ndyakov@users.noreply.github.com> * Update go.sum Co-authored-by: Nedyalko Dyakov <1547186+ndyakov@users.noreply.github.com> * Update script_server_sha_integration_test.go Co-authored-by: Nedyalko Dyakov <1547186+ndyakov@users.noreply.github.com> * Update go.mod Moved the dependency under require block --------- Co-authored-by: Nedyalko Dyakov <1547186+ndyakov@users.noreply.github.com>
75 lines
1.9 KiB
Go
75 lines
1.9 KiB
Go
package redis_test
|
|
|
|
import (
|
|
. "github.com/bsm/ginkgo/v2"
|
|
. "github.com/bsm/gomega"
|
|
"github.com/redis/go-redis/v9"
|
|
)
|
|
|
|
var _ = Describe("NewScriptServerSHA", func() {
|
|
var client *redis.Client
|
|
|
|
BeforeEach(func() {
|
|
client = redis.NewClient(redisOptions())
|
|
Expect(client.FlushDB(ctx).Err()).NotTo(HaveOccurred())
|
|
})
|
|
|
|
AfterEach(func() {
|
|
Expect(client.Close()).NotTo(HaveOccurred())
|
|
})
|
|
|
|
It("should reload script after SCRIPT FLUSH", func() {
|
|
s := redis.NewScriptServerSHA("return 1")
|
|
|
|
// First run should succeed (loads script via SCRIPT LOAD)
|
|
err := s.Run(ctx, client, nil).Err()
|
|
Expect(err).NotTo(HaveOccurred())
|
|
|
|
// Flush all scripts from Redis
|
|
err = client.Do(ctx, "SCRIPT", "FLUSH").Err()
|
|
Expect(err).NotTo(HaveOccurred())
|
|
|
|
// Second run should still succeed (reloads script after NOSCRIPT)
|
|
err = s.Run(ctx, client, nil).Err()
|
|
Expect(err).NotTo(HaveOccurred())
|
|
})
|
|
|
|
It("should run script with RunRO", func() {
|
|
s := redis.NewScriptServerSHA("return 1")
|
|
|
|
err := s.RunRO(ctx, client, nil).Err()
|
|
Expect(err).NotTo(HaveOccurred())
|
|
})
|
|
|
|
It("should run script with EvalSha", func() {
|
|
s := redis.NewScriptServerSHA("return KEYS[1]")
|
|
|
|
result, err := s.EvalSha(ctx, client, []string{"mykey"}).Result()
|
|
Expect(err).NotTo(HaveOccurred())
|
|
Expect(result).To(Equal("mykey"))
|
|
})
|
|
|
|
It("should run script with EvalShaRO", func() {
|
|
s := redis.NewScriptServerSHA("return ARGV[1]")
|
|
|
|
result, err := s.EvalShaRO(ctx, client, nil, "myarg").Result()
|
|
Expect(err).NotTo(HaveOccurred())
|
|
Expect(result).To(Equal("myarg"))
|
|
})
|
|
|
|
It("should cache hash after first load", func() {
|
|
s := redis.NewScriptServerSHA("return 42")
|
|
|
|
// Hash should be empty before first run
|
|
Expect(s.Hash()).To(Equal(""))
|
|
|
|
// Run the script
|
|
result, err := s.Run(ctx, client, nil).Result()
|
|
Expect(err).NotTo(HaveOccurred())
|
|
Expect(result).To(Equal(int64(42)))
|
|
|
|
// Hash should be populated after run
|
|
Expect(s.Hash()).NotTo(Equal(""))
|
|
})
|
|
})
|