How to mock Connect API for test?

[max-min]

go-zookeeper/zk/conn.go

Line 175 in c4fab1a

func Connect(servers []string, sessionTimeout time.Duration, options ...connOption) (*Conn, <-chan Event, error) {

connOption is not friendly ...

[khuddlefish]

We just left the connOption part out in our mock since it is optional. This is kind of what our connect looks like

type stubZkClient struct {
	ConnectStub struct {
		called         int
		servers        []string
		sessionTimeout time.Duration
		zkConn         ZkConn
		eventChan      chan zk.Event
		err            error
	}
}

func (c *stubZkClient) Connect(servers []string, sessionTimeout time.Duration) (ZkConn, <-chan zk.Event, error) {
        c.ConnectStub.called++
        c.ConnectStub.servers = servers
	c.ConnectStub.sessionTimeout = sessionTimeout
	return c.ConnectStub.zkConn, c.ConnectStub.eventChan, c.ConnectStub.err
}