func ConnectToEventStream(conf config.Config) error {
kClient := kubernetesclient.NewClient(conf.KubernetesURL, false)
dClient, err := dockerclient.NewDockerClient()
if err != nil {
return err
}
sh := syncHandler{
kClient: kClient,
dClient: dClient,
}
ph := PingHandler{}
eventHandlers := map[string]revents.EventHandler{
"compute.instance.providelabels": sh.Handler,
"config.update": ph.Handler,
"ping": ph.Handler,
}
router, err := revents.NewEventRouter("", 0, conf.CattleURL, conf.CattleAccessKey, conf.CattleSecretKey, nil, eventHandlers, "", conf.WorkerCount)
if err != nil {
return err
}
err = router.Run(nil)
return err
}
func (s *ListenerTestSuite) SetUpSuite(c *check.C) {
s.publishChan = make(chan client.Publish, 10)
s.kClient = kubernetesclient.NewClient(conf.KubernetesURL, true)
mock := &MockPublishOperations{
publishChan: s.publishChan,
}
s.mockRClient = &client.RancherClient{
Publish: mock,
}
dClient, err := dockerclient.NewDockerClient()
if err != nil {
c.Fatal(err)
}
s.dClient = dClient
}