Mirroristas/etcd
2
0
Fork 0
mirror of https://github.com/etcd-io/etcd.git synced 2024-09-27 06:25:44 +00:00
Code Issues Packages Projects Releases Wiki Activity

tests: Migrate key value Get to common framework

Browse Source
This commit is contained in:
nic-chen 2022-03-19 00:51:29 +08:00
parent dca5874d44
commit 3e657bdc50
1 changed files with 53 additions and 52 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

105
tests/common/alarm_test.go
View File

@ -21,73 +21,74 @@ import (
"time" "time"
"go.etcd.io/etcd/tests/v3/framework/config" "go.etcd.io/etcd/tests/v3/framework/config"
"go.etcd.io/etcd/tests/v3/framework/testutils"
) )
func TestAlarm(t *testing.T) { func TestAlarm(t *testing.T) {
testRunner.BeforeTest(t) testRunner.BeforeTest(t)
clus := testRunner.NewCluster(t, config.ClusterConfig{ClusterSize: 3, QuotaBackendBytes: int64(13 * os.Getpagesize())}) clus := testRunner.NewCluster(t, config.ClusterConfig{ClusterSize: 3, QuotaBackendBytes: int64(13 * os.Getpagesize())})
defer clus.Close() defer clus.Close()
testutils.ExecuteWithTimeout(t, 10*time.Second, func() {
// test small put still works
smallbuf := strings.Repeat("a", 64)
if err := clus.Client().Put("1st_test", smallbuf); err != nil {
t.Fatalf("alarmTest: put kv error (%v)", err)
}
// test small put still works // write some chunks to fill up the database
smallbuf := strings.Repeat("a", 64) buf := strings.Repeat("b", os.Getpagesize())
if err := clus.Client().Put("1st_test", smallbuf); err != nil { for {
t.Fatalf("alarmTest: put kv error (%v)", err) if err := clus.Client().Put("2nd_test", buf); err != nil {
} if !strings.Contains(err.Error(), "etcdserver: mvcc: database space exceeded") {
t.Fatal(err)
}
break
}
}
// write some chunks to fill up the database // quota alarm should now be on
buf := strings.Repeat("b", os.Getpagesize()) _, err := clus.Client().Alarm("list")
for { if err != nil {
if err := clus.Client().Put("2nd_test", buf); err != nil { t.Fatalf("alarmTest: Alarm error (%v)", err)
}
// endpoint should not healthy
if err := clus.Client().Health(); err == nil {
t.Fatalf("endpoint should not healthy")
}
// check that Put is rejected when alarm is on
if err := clus.Client().Put("3rd_test", smallbuf); err != nil {
if !strings.Contains(err.Error(), "etcdserver: mvcc: database space exceeded") { if !strings.Contains(err.Error(), "etcdserver: mvcc: database space exceeded") {
t.Fatal(err) t.Fatal(err)
} }
break
} }
}
// quota alarm should now be on // get latest revision to compact
_, err := clus.Client().Alarm("list") sresp, err := clus.Client().Status()
if err != nil { if err != nil {
t.Fatalf("alarmTest: Alarm error (%v)", err) t.Fatalf("get endpoint status error: %v", err)
} }
// endpoint should not healthy // make some space
if err := clus.Client().Health(); err == nil { _, err = clus.Client().Compact(sresp[0].Header.Revision, config.CompactOption{Physical: true, Timeout: 10 * time.Second})
t.Fatalf("endpoint should not healthy") if err != nil {
} t.Fatalf("alarmTest: Compact error (%v)", err)
}
// check that Put is rejected when alarm is on if err = clus.Client().Defragment(config.DefragOption{Timeout: 10 * time.Second}); err != nil {
if err := clus.Client().Put("3rd_test", smallbuf); err != nil { t.Fatalf("alarmTest: defrag error (%v)", err)
if !strings.Contains(err.Error(), "etcdserver: mvcc: database space exceeded") { }
// turn off alarm
_, err = clus.Client().Alarm("disarm")
if err != nil {
t.Fatalf("alarmTest: Alarm error (%v)", err)
}
// put one more key below quota
if err := clus.Client().Put("4th_test", smallbuf); err != nil {
t.Fatal(err) t.Fatal(err)
} }
} })
// get latest revision to compact
sresp, err := clus.Client().Status()
if err != nil {
t.Fatalf("get endpoint status error: %v", err)
}
// make some space
_, err = clus.Client().Compact(sresp[0].Header.Revision, config.CompactOption{Physical: true, Timeout: 10 * time.Second})
if err != nil {
t.Fatalf("alarmTest: Compact error (%v)", err)
}
if err = clus.Client().Defragment(config.DefragOption{Timeout: 10 * time.Second}); err != nil {
t.Fatalf("alarmTest: defrag error (%v)", err)
}
// turn off alarm
_, err = clus.Client().Alarm("disarm")
if err != nil {
t.Fatalf("alarmTest: Alarm error (%v)", err)
}
// put one more key below quota
if err := clus.Client().Put("4th_test", smallbuf); err != nil {
t.Fatal(err)
}
} }