func startClient(URLs ...string) *broker.Client { var client *broker.Client f := func() error { var err error client, err = broker.NewClient(URLs...) return err } Eventually(f, 5).ShouldNot(HaveOccurred()) return client }
client *broker.Client ) BeforeEach(func() { mockServers = nil servers = nil var urls []string for i := 0; i < 3; i++ { mockServer, server := startMockServer() mockServers = append(mockServers, mockServer) servers = append(servers, server) urls = append(urls, "ws"+server.URL[4:]) } var err error client, err = broker.NewClient(urls...) Expect(err).ToNot(HaveOccurred()) }) AfterEach(func() { client.Close() for _, server := range mockServers { close(server.serverCh) } for _, server := range servers { server.CloseClientConnections() server.Close() } })