Browse Source

Merge pull request #6688 from gyuho/compact-rev

e2e: compact with latest rev in alarm test
Gyu-Ho Lee 9 years ago
parent
commit
e719d8641e
1 changed files with 22 additions and 3 deletions
  1. 22 3
      e2e/ctl_v3_alarm_test.go

+ 22 - 3
e2e/ctl_v3_alarm_test.go

@@ -18,6 +18,10 @@ import (
 	"os"
 	"strings"
 	"testing"
+	"time"
+
+	"github.com/coreos/etcd/clientv3"
+	"golang.org/x/net/context"
 )
 
 func TestCtlV3Alarm(t *testing.T) {
@@ -34,8 +38,7 @@ func alarmTest(cx ctlCtx) {
 
 	// write some chunks to fill up the database
 	buf := strings.Repeat("b", int(os.Getpagesize()))
-	var rev int64
-	for ; ; rev++ {
+	for {
 		if err := ctlV3Put(cx, "2nd_test", buf, ""); err != nil {
 			if !strings.Contains(err.Error(), "etcdserver: mvcc: database space exceeded") {
 				cx.t.Fatal(err)
@@ -56,8 +59,24 @@ func alarmTest(cx ctlCtx) {
 		}
 	}
 
+	eps := cx.epc.grpcEndpoints()
+
+	// get latest revision to compact
+	cli, err := clientv3.New(clientv3.Config{
+		Endpoints:   eps,
+		DialTimeout: 3 * time.Second,
+	})
+	if err != nil {
+		cx.t.Fatal(err)
+	}
+	defer cli.Close()
+	sresp, err := cli.Status(context.TODO(), eps[0])
+	if err != nil {
+		cx.t.Fatal(err)
+	}
+
 	// make some space
-	if err := ctlV3Compact(cx, rev, true); err != nil {
+	if err := ctlV3Compact(cx, sresp.Header.Revision, true); err != nil {
 		cx.t.Fatal(err)
 	}
 	if err := ctlV3Defrag(cx); err != nil {