| // Copyright 2016 The etcd Authors |
| // |
| // Licensed under the Apache License, Version 2.0 (the "License"); |
| // you may not use this file except in compliance with the License. |
| // You may obtain a copy of the License at |
| // |
| // http://www.apache.org/licenses/LICENSE-2.0 |
| // |
| // Unless required by applicable law or agreed to in writing, software |
| // distributed under the License is distributed on an "AS IS" BASIS, |
| // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| // See the License for the specific language governing permissions and |
| // limitations under the License. |
| |
| package e2e |
| |
| import ( |
| "context" |
| "os" |
| "strings" |
| "testing" |
| "time" |
| |
| "github.com/coreos/etcd/clientv3" |
| ) |
| |
| func TestCtlV3Alarm(t *testing.T) { |
| // The boltdb minimum working set is six pages. |
| testCtl(t, alarmTest, withQuota(int64(13*os.Getpagesize()))) |
| } |
| |
| func alarmTest(cx ctlCtx) { |
| // test small put still works |
| smallbuf := strings.Repeat("a", 64) |
| if err := ctlV3Put(cx, "1st_test", smallbuf, ""); err != nil { |
| cx.t.Fatal(err) |
| } |
| |
| // write some chunks to fill up the database |
| buf := strings.Repeat("b", int(os.Getpagesize())) |
| for { |
| if err := ctlV3Put(cx, "2nd_test", buf, ""); err != nil { |
| if !strings.Contains(err.Error(), "etcdserver: mvcc: database space exceeded") { |
| cx.t.Fatal(err) |
| } |
| break |
| } |
| } |
| |
| // quota alarm should now be on |
| if err := ctlV3Alarm(cx, "list", "alarm:NOSPACE"); err != nil { |
| cx.t.Fatal(err) |
| } |
| |
| // '/health' handler should return 'false' |
| if err := cURLGet(cx.epc, cURLReq{endpoint: "/health", expected: `{"health":"false"}`}); err != nil { |
| cx.t.Fatalf("failed get with curl (%v)", err) |
| } |
| |
| // check that Put is rejected when alarm is on |
| if err := ctlV3Put(cx, "3rd_test", smallbuf, ""); err != nil { |
| if !strings.Contains(err.Error(), "etcdserver: mvcc: database space exceeded") { |
| cx.t.Fatal(err) |
| } |
| } |
| |
| eps := cx.epc.EndpointsV3() |
| |
| // 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, sresp.Header.Revision, true); err != nil { |
| cx.t.Fatal(err) |
| } |
| if err := ctlV3Defrag(cx); err != nil { |
| cx.t.Fatal(err) |
| } |
| |
| // turn off alarm |
| if err := ctlV3Alarm(cx, "disarm", "alarm:NOSPACE"); err != nil { |
| cx.t.Fatal(err) |
| } |
| |
| // put one more key below quota |
| if err := ctlV3Put(cx, "4th_test", smallbuf, ""); err != nil { |
| cx.t.Fatal(err) |
| } |
| } |
| |
| func ctlV3Alarm(cx ctlCtx, cmd string, as ...string) error { |
| cmdArgs := append(cx.PrefixArgs(), "alarm", cmd) |
| return spawnWithExpects(cmdArgs, as...) |
| } |