func (s *ExternalSuite) TestSwapWithDifferentRouterKinds(c *check.C) {
config.Set("hipache:redis-server", "127.0.0.1:6379")
config.Set("hipache:redis-db", 5)
backend1 := "bb1"
backend2 := "bb2"
r1, err := router.Get("fake")
c.Assert(err, check.IsNil)
r2, err := router.Get("hipache")
c.Assert(err, check.IsNil)
err = r1.AddBackend(backend1)
c.Assert(err, check.IsNil)
defer r1.RemoveBackend(backend1)
addr1, _ := url.Parse("http://127.0.0.1")
err = r1.AddRoute(backend1, addr1)
c.Assert(err, check.IsNil)
defer r1.RemoveRoute(backend1, addr1)
err = r2.AddBackend(backend2)
c.Assert(err, check.IsNil)
defer r2.RemoveBackend(backend2)
addr2, _ := url.Parse("http://10.10.10.10")
err = r2.AddRoute(backend2, addr2)
c.Assert(err, check.IsNil)
defer r2.RemoveRoute(backend2, addr2)
err = router.Swap(r1, backend1, backend2, false)
c.Assert(err, check.ErrorMatches, `swap is only allowed between routers of the same kind. "bb1" uses "fake", "bb2" uses "hipache"`)
err = router.Swap(r2, backend1, backend2, false)
c.Assert(err, check.ErrorMatches, `swap is only allowed between routers of the same kind. "bb1" uses "fake", "bb2" uses "hipache"`)
}
func (s *ExternalSuite) TestSwapCnameOnly(c *check.C) {
backend1 := "bx1"
backend2 := "bx2"
r, err := router.Get("fake")
c.Assert(err, check.IsNil)
cnameRouter, ok := r.(router.CNameRouter)
c.Assert(ok, check.Equals, true)
err = r.AddBackend(backend1)
c.Assert(err, check.IsNil)
addr1, err := url.Parse("http://127.0.0.1")
c.Assert(err, check.IsNil)
r.AddRoute(backend1, addr1)
err = cnameRouter.SetCName("cname.com", backend1)
c.Assert(err, check.IsNil)
err = r.AddBackend(backend2)
c.Assert(err, check.IsNil)
addr2, err := url.Parse("http://10.10.10.10")
c.Assert(err, check.IsNil)
r.AddRoute(backend2, addr2)
err = router.Swap(r, backend1, backend2, true)
c.Assert(err, check.IsNil)
routes1, err := r.Routes(backend1)
c.Assert(err, check.IsNil)
c.Assert(routes1, check.DeepEquals, []*url.URL{addr1})
routes2, err := r.Routes(backend2)
c.Assert(err, check.IsNil)
c.Assert(routes2, check.DeepEquals, []*url.URL{addr2})
name1, err := router.Retrieve(backend1)
c.Assert(err, check.IsNil)
c.Assert(name1, check.Equals, backend1)
name2, err := router.Retrieve(backend2)
c.Assert(err, check.IsNil)
c.Assert(name2, check.Equals, backend2)
cnames, err := cnameRouter.CNames(backend1)
c.Assert(err, check.IsNil)
c.Assert(cnames, check.HasLen, 0)
expected := []*url.URL{{Host: "cname.com"}}
cnames, err = cnameRouter.CNames(backend2)
c.Assert(err, check.IsNil)
c.Assert(expected, check.DeepEquals, cnames)
err = router.Swap(r, backend1, backend2, true)
c.Assert(err, check.IsNil)
cnames, err = cnameRouter.CNames(backend1)
c.Assert(err, check.IsNil)
c.Assert(expected, check.DeepEquals, cnames)
cnames, err = cnameRouter.CNames(backend2)
c.Assert(err, check.IsNil)
c.Assert(cnames, check.HasLen, 0)
}
func (s *ExternalSuite) TestSwap(c *check.C) {
backend1 := "b1"
backend2 := "b2"
r, err := router.Get("fake")
c.Assert(err, check.IsNil)
r.AddBackend(backend1)
addr1, _ := url.Parse("http://127.0.0.1")
r.AddRoute(backend1, addr1)
r.AddBackend(backend2)
addr2, _ := url.Parse("http://10.10.10.10")
r.AddRoute(backend2, addr2)
err = router.Swap(r, backend1, backend2)
c.Assert(err, check.IsNil)
routes1, err := r.Routes(backend1)
c.Assert(err, check.IsNil)
c.Assert(routes1, check.DeepEquals, []*url.URL{addr1})
routes2, err := r.Routes(backend2)
c.Assert(err, check.IsNil)
c.Assert(routes2, check.DeepEquals, []*url.URL{addr2})
name1, err := router.Retrieve(backend1)
c.Assert(err, check.IsNil)
c.Assert(name1, check.Equals, backend2)
name2, err := router.Retrieve(backend2)
c.Assert(err, check.IsNil)
c.Assert(name2, check.Equals, backend1)
}
func (s *ExternalSuite) TestSwapWithDifferentRouterKinds(c *check.C) {
backend1 := "bb1"
backend2 := "bb2"
r1, err := router.Get("fake")
c.Assert(err, check.IsNil)
r2, err := router.Get("hipache")
c.Assert(err, check.IsNil)
err = r1.AddBackend(backend1)
c.Assert(err, check.IsNil)
addr1, _ := url.Parse("http://127.0.0.1")
err = r1.AddRoute(backend1, addr1)
c.Assert(err, check.IsNil)
err = r2.AddBackend(backend2)
c.Assert(err, check.IsNil)
addr2, _ := url.Parse("http://10.10.10.10")
err = r2.AddRoute(backend2, addr2)
c.Assert(err, check.IsNil)
err = router.Swap(r1, backend1, backend2)
c.Assert(err, check.ErrorMatches, `swap is only allowed between routers of the same kind. "bb1" uses "fake", "bb2" uses "hipache"`)
err = router.Swap(r2, backend1, backend2)
c.Assert(err, check.ErrorMatches, `swap is only allowed between routers of the same kind. "bb1" uses "fake", "bb2" uses "hipache"`)
}
func (r *galebRouter) Swap(backend1, backend2 string) error {
return router.Swap(r, backend1, backend2)
}
func (r *fusisRouter) Swap(backend1 string, backend2 string, cnameOnly bool) error {
return router.Swap(r, backend1, backend2, cnameOnly)
}
func (r *hipacheRouter) Swap(backend1, backend2 string, cnameOnly bool) error {
return router.Swap(r, backend1, backend2, cnameOnly)
}
func (r hipacheRouter) Swap(backend1, backend2 string) error {
return router.Swap(r, backend1, backend2)
}
func (r *vulcandRouter) Swap(backend1, backend2 string) error {
return router.Swap(r, backend1, backend2)
}
