storage/etcd_test.go at master · rovergulf-archive/storage · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
package storage

import (
	"fmt"
	"github.com/stretchr/testify/suite"
	clientv3 "go.etcd.io/etcd/client/v3"
	"go.uber.org/zap"
	"os"
	"testing"
)

var (
	//cafile   = "./_test/ca.pem"
	//certfile = "./_test/cert.pem"
	//keyfile  = "./_test/key.pem"
	endpoints = os.Getenv("ETCD_ADDR")
)

type CsEtcdSuite struct {
	suite.Suite
	etcd Backend
}

func (c *CsEtcdSuite) SetupSuite() {
	// TODO: renable, see https://github.com/chartmuseum/storage/issues/6
	c.T().Skip()
	etcd, err := NewEtcdStorage(EtcdOptions{
		Logger: zap.S(),
		Config: &clientv3.Config{
			Endpoints: []string{endpoints},
		},
	})
	if err != nil {
		c.Fail(err.Error())
	}

	c.etcd = etcd
	fmt.Println("init success")
}

func (c *CsEtcdSuite) TestListObjecs() {
	c.etcd.PutObject("listtest1", []byte("puttestetcd"))

	c.etcd.PutObject("listtest2", []byte("puttestetcd"))

	if objs, err := c.etcd.ListObjects("listtest"); err != nil {
		c.Fail("etcd list objects err:%s", err)
	} else {
		fmt.Println(objs)
	}

}

func (c *CsEtcdSuite) TestGetObject() {

	c.etcd.PutObject("gettest", []byte("testdate"))

	if obj, err := c.etcd.GetObject("gettest"); err != nil {
		c.Fail("etcd get objects err:%s", err)
	} else {
		fmt.Println(obj)
	}

}

func (c *CsEtcdSuite) TestDeleteObject() {

	c.etcd.PutObject("deletetest", []byte("testdate"))

	if err := c.etcd.DeleteObject("deletetest"); err != nil {
		c.Fail("etcd delete objects err:%s", err)
	}

}

func TestEtcdCSBackend(t *testing.T) {

	suite.Run(t, new(CsEtcdSuite))
}