Merge pull request #465 from smacker/wrap-process-broken

WrapProcess is broken in v5

example_instrumentation_test.go

@@ -5,6 +5,9 @@ import (
 	"sync/atomic"
 	"time"
 
+	. "github.com/onsi/ginkgo"
+	. "github.com/onsi/gomega"
+
 	redis "gopkg.in/redis.v5"
 )
 
@@ -57,3 +60,35 @@ func wrapRedisProcess(client *redis.Client) {
 		}
 	})
 }
+
+var _ = Describe("Instrumentation", func() {
+	var client *redis.Client
+
+	BeforeEach(func() {
+		client = redis.NewClient(redisOptions())
+		Expect(client.FlushDb().Err()).NotTo(HaveOccurred())
+	})
+
+	AfterEach(func() {
+		Expect(client.Close()).NotTo(HaveOccurred())
+	})
+
+	Describe("WrapProcess", func() {
+
+		It("should call for client", func() {
+			wrapperFnCalled := false
+
+			client.WrapProcess(func(oldProcess func(redis.Cmder) error) func(redis.Cmder) error {
+				return func(cmd redis.Cmder) error {
+					wrapperFnCalled = true
+					return oldProcess(cmd)
+				}
+			})
+
+			client.Ping()
+
+			Expect(wrapperFnCalled).To(Equal(true))
+		})
+
+	})
+})