store_etcd_test.go

package prophet

import (
	"context"
	"fmt"
	"math"
	"strings"
	"testing"
	"time"

	"github.com/golang/mock/gomock"
	"github.com/stretchr/testify/assert"
	"go.etcd.io/etcd/clientv3"
)

func TestAlreadyBootstrapped(t *testing.T) {
	ctrl := gomock.NewController(t)
	defer ctrl.Finish()

	stopC, port, err := startTestSingleEtcd(t)
	if err != nil {
		assert.FailNowf(t, "start embed etcd failed", "error: %+v", err)
	}
	defer close(stopC)

	client, err := clientv3.New(clientv3.Config{
		Endpoints:   strings.Split(fmt.Sprintf("http://127.0.0.1:%d", port), ","),
		DialTimeout: DefaultTimeout,
	})
	assert.Nil(t, err, "create etcd client failed")
	defer client.Close()

	e, err := NewElector(client)
	assert.Nil(t, err, "TestAlreadyBootstrapped failed")
	defer e.Stop(math.MaxUint64)

	go e.ElectionLoop(context.Background(), math.MaxUint64, "node1", func() {}, func() {})
	time.Sleep(time.Millisecond * 200)

	store := newEtcdStore(client, newTestAdapter(ctrl), "node1", e)
	yes, err := store.AlreadyBootstrapped()
	assert.NoError(t, err, "TestAlreadyBootstrapped failed")
	assert.False(t, yes, "TestAlreadyBootstrapped failed")

	var reses []Resource
	for i := 0; i < 100; i++ {
		res := newTestResource()
		res.SetID(uint64(i + 1))
		reses = append(reses, res)
	}
	yes, err = store.PutBootstrapped(newTestContainer(), reses...)
	assert.NoError(t, err, "TestAlreadyBootstrapped failed")
	assert.True(t, yes, "TestAlreadyBootstrapped failed")
	c := 0
	err = store.LoadResources(8, func(res Resource) {
		c++
	})
	assert.NoError(t, err, "TestAlreadyBootstrapped failed")
	assert.Equal(t, 100, c, "TestAlreadyBootstrapped failed")

	yes, err = store.AlreadyBootstrapped()
	assert.NoError(t, err, "TestAlreadyBootstrapped failed")
	assert.True(t, yes, "TestAlreadyBootstrapped failed")
}