clientoptions_test.go

package signalr

import (
	"context"
	"github.com/cenkalti/backoff/v4"

	. "github.com/onsi/ginkgo"
	. "github.com/onsi/gomega"
)

var _ = Describe("Client options", func() {

	Describe("WithConnection and WithConnector option", func() {
		Context("none of them is given", func() {
			It("NewClient should fail", func() {
				_, err := NewClient(context.TODO())
				Expect(err).To(HaveOccurred())
			}, 3.0)
		})
		Context("both are given", func() {
			It("NewClient should fail", func() {
				conn := NewNetConnection(context.TODO(), nil)
				_, err := NewClient(context.TODO(), WithConnection(conn), WithConnector(func() (Connection, error) {
					return conn, nil
				}))
				Expect(err).To(HaveOccurred())
			}, 3.0)
		})
		Context("only WithConnection is given", func() {
			It("NewClient should not fail", func() {
				conn := NewNetConnection(context.TODO(), nil)
				_, err := NewClient(context.TODO(), WithConnection(conn))
				Expect(err).NotTo(HaveOccurred())
			}, 3.0)
		})
		Context("only WithConnector is given", func() {
			It("NewClient should not fail", func() {
				conn := NewNetConnection(context.TODO(), nil)
				_, err := NewClient(context.TODO(), WithConnector(func() (Connection, error) {
					return conn, nil
				}))
				Expect(err).NotTo(HaveOccurred())
			}, 3.0)
		})
		Context("only WithBackoff is given", func() {
			It("NewClient should not fail", func() {
				conn := NewNetConnection(context.TODO(), nil)
				_, err := NewClient(context.TODO(), WithConnection(conn), WithBackoff(func() backoff.BackOff {
					return backoff.NewExponentialBackOff()
				}))
				Expect(err).NotTo(HaveOccurred())
			}, 3.0)
		})
	})

})