Пример #1
0
				MetadataRetries: 30,
				WaitForElection: time.Second,
			})
			Expect(err).NotTo(HaveOccurred())

			subject, err = NewConsumerGroup(client, zk, tnG, tnT, nil, consumerConfig)
			Expect(err).NotTo(HaveOccurred())
		})

		AfterEach(func() {
			if subject != nil {
				subject.Close()
				subject = nil
			}
			if client != nil {
				client.Close()
				client = nil
			}
			if zk != nil {
				zk.Close()
				zk = nil
			}
		})

		It("can be created & closed", func() {
			lst, _, err := zk.Consumers(tnG)
			Expect(err).NotTo(HaveOccurred())
			Expect(lst).To(HaveLen(1))
			Expect(subject.Close()).To(BeNil())
			subject = nil
		})
Пример #2
0
func unmountOnInt(c chan os.Signal, server *fuse.Server, client *sarama.Client) {
	s := <-c
	fmt.Println("Got signal:", s)
	server.Unmount()
	client.Close()
}
Пример #3
0
		testClient, err = sarama.NewClient(testKafkaAddrs, nil)
		return err
	}, "10s", "1s").ShouldNot(HaveOccurred())

	// Ensure we can retrieve partition info
	Eventually(func() error {
		_, err := testClient.Partitions(testTopics[0])
		return err
	}, "10s", "500ms").ShouldNot(HaveOccurred())

	// Seed a few messages
	Expect(testSeed(1000)).NotTo(HaveOccurred())
})

var _ = AfterSuite(func() {
	_ = testClient.Close()

	_ = testKafkaCmd.Process.Kill()
	_ = testZkCmd.Process.Kill()
	_ = testKafkaCmd.Wait()
	_ = testZkCmd.Wait()
	_ = os.RemoveAll(testKafkaData)
})

// --------------------------------------------------------------------

func TestSuite(t *testing.T) {
	RegisterFailHandler(Fail)
	RunSpecs(t, "sarama/cluster")
}