Exemplo n.º 1
0
func (s *testSuite) TestNewEtcdClient(c *C) {
	if !IsEtcd() {
		c.Skip("not etcd")
	}

	client, err := getEtcdClient()
	c.Assert(err, IsNil)

	volClient, err := etcd.NewClient(etcdHosts, "volplugin")
	c.Assert(err, IsNil)
	c.Assert(volClient, NotNil)

	_, err = client.Get(context.Background(), "/volplugin", nil)
	c.Assert(err, IsNil)

	volClient, err = etcd.NewClient(etcdHosts, "volplugin")
	c.Assert(err, IsNil, Commentf("%v", err))
	c.Assert(volClient, NotNil)

	volClient, err = etcd.NewClient(etcdHosts, "testing")
	c.Assert(err, IsNil)
	c.Assert(volClient, NotNil)

	_, err = client.Get(context.Background(), "/testing", nil)
	c.Assert(err, IsNil)
}
Exemplo n.º 2
0
func (s *testSuite) TestEtcdDown(c *C) {
	if !IsEtcd() {
		c.Skip("not etcd")
	}

	volClient, err := etcd.NewClient(etcdHosts, "volplugin")
	c.Assert(err, IsNil)

	c.Assert(exec.Command("/bin/sh", "-c", "sudo systemctl stop etcd").Run(), IsNil)
	for {
		if err := exec.Command("etcdctl", "cluster-health").Run(); err != nil {
			break
		}

		time.Sleep(time.Second / 4)
	}

	defer func() {
		c.Assert(exec.Command("/bin/sh", "-c", "sudo systemctl start etcd").Run(), IsNil)
		for {
			if err := exec.Command("etcdctl", "cluster-health").Run(); err == nil {
				break
			}

			time.Sleep(time.Second / 4)
		}
	}()

	policy := db.NewPolicy("policy")
	c.Assert(volClient.Set(policy), NotNil)
	c.Assert(policy, DeepEquals, db.NewPolicy("policy"))
}
Exemplo n.º 3
0
func (s *testSuite) TestCRUD(c *C) {
	volClient, err := etcd.NewClient(etcdHosts, "volplugin")
	c.Assert(err, IsNil)

	client, err := getEtcdClient()
	c.Assert(err, IsNil)

	te := newTestEntity("test", "data")
	path, err := te.Path()
	c.Assert(err, IsNil)
	c.Assert(path, Equals, "test/test")
	c.Assert(volClient.Get(te), NotNil)

	c.Assert(volClient.Set(te), IsNil)

	te = newTestEntity("test", "")
	c.Assert(volClient.Get(te), IsNil)
	c.Assert(te.SomeData, Equals, "data")

	_, err = client.Get(context.Background(), "/volplugin/test/test", nil)
	c.Assert(err, IsNil)

	entities, err := volClient.List(te)
	c.Assert(err, IsNil)
	c.Assert(len(entities), Equals, 1)
	c.Assert(entities[0].(*testEntity).Name, Equals, "test")
	c.Assert(entities[0].(*testEntity).SomeData, Equals, "data")

	c.Assert(volClient.Delete(te), IsNil)
	c.Assert(volClient.Get(te), NotNil)
}
Exemplo n.º 4
0
func (s *testSuite) SetUpTest(c *C) {
	errored.AlwaysDebug = true
	errored.AlwaysTrace = true
	exec.Command("/bin/sh", "-c", "etcdctl rm --recursive /volplugin /testing /test /watch").Run()
	var err error
	s.client, err = etcd.NewClient(etcdHosts, "volplugin")
	c.Assert(err, IsNil)
}
Exemplo n.º 5
0
func (s *testSuite) SetUpSuite(c *C) {
	if os.Getenv("DEBUG") != "" {
		errored.AlwaysDebug = true
		errored.AlwaysTrace = true
		logrus.SetLevel(logrus.DebugLevel)
	}
	var err error

	switch Driver {
	case "etcd":
		s.client, err = etcd.NewClient(etcdHosts, "volplugin")
		c.Assert(err, IsNil)
	case "consul":
		s.client, err = consul.NewClient("volplugin", &api.Config{Address: ":8500"})
		c.Assert(err, IsNil)
	}
}
Exemplo n.º 6
0
func (s *testSuite) TestWatch(c *C) {
	volClient, err := etcd.NewClient(etcdHosts, "volplugin")
	c.Assert(err, IsNil)

	te := newTestEntity("test", "")

	channel := make(chan interface{}, 10)

	retChan, errChan := volClient.Watch(te)

	for i := 0; i < 10; i++ {
		te2 := te.Copy().(*testEntity)
		te2.SomeData = fmt.Sprintf("data%d", i)
		setKey(te2)
	}

	for i := 0; i < 10; i++ {
		select {
		case err := <-errChan:
			c.Assert(err, IsNil, Commentf("select: %v", err)) // this will always fail, assert is just to raise the error.
		case ent := <-retChan:
			te, ok := ent.(*testEntity)
			c.Assert(ok, Equals, true)
			c.Assert(te, NotNil)
			c.Assert(te.Name, Equals, "test")
			c.Assert(string(te.SomeData), Equals, fmt.Sprintf("data%d", i))
		}
	}

	c.Assert(volClient.WatchStop(te), IsNil)
	for i := 0; i < 10; i++ {
		te2 := te.Copy().(*testEntity)
		te2.SomeData = fmt.Sprintf("data%d", i)
		setKey(te2)
	}

	var x = 0

	for i := 0; i < 10; i++ {
		select {
		case <-channel:
			x++
		default:
		}
	}
}
Exemplo n.º 7
0
func (s *testSuite) TestDump(c *C) {
	volClient, err := etcd.NewClient(etcdHosts, "volplugin")
	c.Assert(err, IsNil)

	key := "/volplugin/foo"
	value := "baz"

	// add test key to etcd
	result, err := executor.New(exec.Command("/bin/sh", "-c", "etcdctl set "+key+" "+value)).Run(context.Background())
	c.Assert(err, IsNil)
	c.Assert(result.ExitStatus, Equals, 0)

	// remove test key after the test completes
	defer func() {
		result, err := executor.New(exec.Command("/bin/sh", "-c", "etcdctl rm "+key)).Run(context.Background())
		c.Assert(err, IsNil)
		c.Assert(result.ExitStatus, Equals, 0)
	}()

	tarballPath, err := volClient.Dump("")
	c.Assert(err, IsNil)
	defer os.Remove(tarballPath)

	// check that the tarball was created in the expected spot
	_, err = os.Stat(tarballPath)
	c.Assert(os.IsNotExist(err), Equals, false)

	tarballFilename := filepath.Base(tarballPath)
	// can't use filepath.Ext() here because it thinks ".gz" is the extension
	dirName := tarballFilename[:strings.LastIndex(tarballFilename, ".tar.gz")]

	testFilename := dirName + key

	data, err := ioutil.ReadFile(tarballPath)
	c.Assert(err, IsNil)

	// check that our test key is present and has the expected value
	reader := bytes.NewReader(data)
	gzReader, err := gzip.NewReader(reader)
	c.Assert(err, IsNil)
	tarReader := tar.NewReader(gzReader)

	found := false

	for {
		header, err := tarReader.Next()
		if err == io.EOF {
			break
		}
		c.Assert(err, IsNil)

		if header.Name == testFilename {
			var b bytes.Buffer
			_, err = io.Copy(&b, tarReader)
			c.Assert(err, IsNil)
			c.Assert(string(b.Bytes()), Equals, value)
			found = true
			break
		}
	}

	c.Assert(found, Equals, true)
}
Exemplo n.º 8
0
func (s *testSuite) TestHooks(c *C) {
	volClient, err := etcd.NewClient(etcdHosts, "volplugin")
	c.Assert(err, IsNil)

	triggered := &struct{ i int }{i: 0}

	te := newTestEntity("test", "")
	triggerFunc := func(c db.Client, te db.Entity) error {
		triggered.i++
		return nil
	}

	te.hooks.PreSet = triggerFunc
	c.Assert(volClient.Set(te), IsNil)
	c.Assert(triggered.i, Equals, 1)

	triggered.i = 0
	te.hooks.PostSet = triggerFunc
	c.Assert(volClient.Set(te), IsNil)
	c.Assert(triggered.i, Equals, 2)

	triggered.i = 0
	te.hooks.PreSet = nil
	te.hooks.PostSet = nil
	c.Assert(volClient.Set(te), IsNil)
	c.Assert(triggered.i, Equals, 0)

	te.hooks.PreGet = triggerFunc
	c.Assert(volClient.Get(te), IsNil)
	c.Assert(triggered.i, Equals, 1)

	triggered.i = 0
	te.hooks.PostGet = triggerFunc
	c.Assert(volClient.Get(te), IsNil)
	c.Assert(triggered.i, Equals, 2)

	triggered.i = 0
	te.hooks.PreGet = nil
	te.hooks.PostGet = nil
	c.Assert(volClient.Get(te), IsNil)
	c.Assert(triggered.i, Equals, 0)

	te.hooks.PreDelete = triggerFunc
	c.Assert(volClient.Delete(te), IsNil)
	c.Assert(triggered.i, Equals, 1)

	// to test these delete calls, we have to re-set the data.
	c.Assert(volClient.Set(te), IsNil)

	triggered.i = 0
	te.hooks.PostDelete = triggerFunc
	c.Assert(volClient.Delete(te), IsNil)
	c.Assert(triggered.i, Equals, 2)

	// to test these delete calls, we have to re-set the data.
	c.Assert(volClient.Set(te), IsNil)

	triggered.i = 0
	te.hooks.PreDelete = nil
	te.hooks.PostDelete = nil
	c.Assert(volClient.Delete(te), IsNil)
	c.Assert(triggered.i, Equals, 0)
}