2018-03-19 14:47:44 +03:00
|
|
|
// Copyright 2018 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 (
|
|
|
|
"fmt"
|
|
|
|
"testing"
|
|
|
|
|
2018-08-29 03:13:25 +03:00
|
|
|
pb "go.etcd.io/etcd/etcdserver/etcdserverpb"
|
2018-03-19 14:47:44 +03:00
|
|
|
)
|
|
|
|
|
|
|
|
func TestV3CurlLeaseGrantNoTLS(t *testing.T) {
|
|
|
|
for _, p := range apiPrefix {
|
|
|
|
testCtl(t, testV3CurlLeaseGrant, withApiPrefix(p), withCfg(configNoTLS))
|
|
|
|
}
|
|
|
|
}
|
|
|
|
func TestV3CurlLeaseRevokeNoTLS(t *testing.T) {
|
|
|
|
for _, p := range apiPrefix {
|
|
|
|
testCtl(t, testV3CurlLeaseRevoke, withApiPrefix(p), withCfg(configNoTLS))
|
|
|
|
}
|
|
|
|
}
|
|
|
|
func TestV3CurlLeaseLeasesNoTLS(t *testing.T) {
|
|
|
|
for _, p := range apiPrefix {
|
|
|
|
testCtl(t, testV3CurlLeaseLeases, withApiPrefix(p), withCfg(configNoTLS))
|
|
|
|
}
|
|
|
|
}
|
|
|
|
func TestV3CurlLeaseKeepAliveNoTLS(t *testing.T) {
|
|
|
|
for _, p := range apiPrefix {
|
|
|
|
testCtl(t, testV3CurlLeaseKeepAlive, withApiPrefix(p), withCfg(configNoTLS))
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
type v3cURLTest struct {
|
|
|
|
endpoint string
|
|
|
|
value string
|
|
|
|
expected string
|
|
|
|
}
|
|
|
|
|
2018-03-17 21:59:35 +03:00
|
|
|
// TODO remove /kv/lease/timetolive, /kv/lease/revoke, /kv/lease/leases tests in 3.5 release
|
|
|
|
|
2018-03-19 14:47:44 +03:00
|
|
|
func testV3CurlLeaseGrant(cx ctlCtx) {
|
|
|
|
leaseID := randomLeaseID()
|
|
|
|
|
|
|
|
tests := []v3cURLTest{
|
|
|
|
{
|
|
|
|
endpoint: "/lease/grant",
|
|
|
|
value: gwLeaseGrant(cx, leaseID, 0),
|
|
|
|
expected: gwLeaseIDExpected(leaseID),
|
|
|
|
},
|
|
|
|
{
|
|
|
|
endpoint: "/lease/grant",
|
|
|
|
value: gwLeaseGrant(cx, 0, 20),
|
|
|
|
expected: `"TTL":"20"`,
|
|
|
|
},
|
|
|
|
{
|
|
|
|
endpoint: "/kv/put",
|
|
|
|
value: gwKVPutLease(cx, "foo", "bar", leaseID),
|
|
|
|
expected: `"revision":"`,
|
|
|
|
},
|
2018-03-17 21:59:35 +03:00
|
|
|
{
|
|
|
|
endpoint: "/lease/timetolive",
|
|
|
|
value: gwLeaseTTLWithKeys(cx, leaseID),
|
|
|
|
expected: `"grantedTTL"`,
|
|
|
|
},
|
2018-03-19 14:47:44 +03:00
|
|
|
{
|
|
|
|
endpoint: "/kv/lease/timetolive",
|
|
|
|
value: gwLeaseTTLWithKeys(cx, leaseID),
|
|
|
|
expected: `"grantedTTL"`,
|
|
|
|
},
|
|
|
|
}
|
|
|
|
if err := cURLWithExpected(cx, tests); err != nil {
|
|
|
|
cx.t.Fatalf("testV3CurlLeaseGrant: %v", err)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
func testV3CurlLeaseRevoke(cx ctlCtx) {
|
|
|
|
leaseID := randomLeaseID()
|
|
|
|
|
|
|
|
tests := []v3cURLTest{
|
|
|
|
{
|
|
|
|
endpoint: "/lease/grant",
|
|
|
|
value: gwLeaseGrant(cx, leaseID, 0),
|
|
|
|
expected: gwLeaseIDExpected(leaseID),
|
|
|
|
},
|
|
|
|
{
|
2018-03-17 21:59:35 +03:00
|
|
|
endpoint: "/lease/revoke",
|
2018-03-19 14:47:44 +03:00
|
|
|
value: gwLeaseRevoke(cx, leaseID),
|
|
|
|
expected: `"revision":"`,
|
|
|
|
},
|
2018-03-17 21:59:35 +03:00
|
|
|
{
|
|
|
|
endpoint: "/kv/lease/revoke",
|
|
|
|
value: gwLeaseRevoke(cx, leaseID),
|
|
|
|
expected: `etcdserver: requested lease not found`,
|
|
|
|
},
|
2018-03-19 14:47:44 +03:00
|
|
|
}
|
|
|
|
if err := cURLWithExpected(cx, tests); err != nil {
|
|
|
|
cx.t.Fatalf("testV3CurlLeaseRevoke: %v", err)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
func testV3CurlLeaseLeases(cx ctlCtx) {
|
|
|
|
leaseID := randomLeaseID()
|
|
|
|
|
|
|
|
tests := []v3cURLTest{
|
|
|
|
{
|
|
|
|
endpoint: "/lease/grant",
|
|
|
|
value: gwLeaseGrant(cx, leaseID, 0),
|
|
|
|
expected: gwLeaseIDExpected(leaseID),
|
|
|
|
},
|
2018-03-17 21:59:35 +03:00
|
|
|
{
|
|
|
|
endpoint: "/lease/leases",
|
|
|
|
value: "{}",
|
|
|
|
expected: gwLeaseIDExpected(leaseID),
|
|
|
|
},
|
2018-03-19 14:47:44 +03:00
|
|
|
{
|
|
|
|
endpoint: "/kv/lease/leases",
|
|
|
|
value: "{}",
|
|
|
|
expected: gwLeaseIDExpected(leaseID),
|
|
|
|
},
|
|
|
|
}
|
|
|
|
if err := cURLWithExpected(cx, tests); err != nil {
|
|
|
|
cx.t.Fatalf("testV3CurlLeaseGrant: %v", err)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
func testV3CurlLeaseKeepAlive(cx ctlCtx) {
|
|
|
|
leaseID := randomLeaseID()
|
|
|
|
|
|
|
|
tests := []v3cURLTest{
|
|
|
|
{
|
|
|
|
endpoint: "/lease/grant",
|
|
|
|
value: gwLeaseGrant(cx, leaseID, 0),
|
|
|
|
expected: gwLeaseIDExpected(leaseID),
|
|
|
|
},
|
|
|
|
{
|
|
|
|
endpoint: "/lease/keepalive",
|
|
|
|
value: gwLeaseKeepAlive(cx, leaseID),
|
|
|
|
expected: gwLeaseIDExpected(leaseID),
|
|
|
|
},
|
|
|
|
}
|
|
|
|
if err := cURLWithExpected(cx, tests); err != nil {
|
|
|
|
cx.t.Fatalf("testV3CurlLeaseGrant: %v", err)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
func gwLeaseIDExpected(leaseID int64) string {
|
|
|
|
return fmt.Sprintf(`"ID":"%d"`, leaseID)
|
|
|
|
}
|
|
|
|
|
|
|
|
func gwLeaseTTLWithKeys(cx ctlCtx, leaseID int64) string {
|
|
|
|
d := &pb.LeaseTimeToLiveRequest{ID: leaseID, Keys: true}
|
|
|
|
s, err := dataMarshal(d)
|
|
|
|
if err != nil {
|
|
|
|
cx.t.Fatalf("gwLeaseTTLWithKeys: error (%v)", err)
|
|
|
|
}
|
|
|
|
return s
|
|
|
|
}
|
|
|
|
|
|
|
|
func gwLeaseKeepAlive(cx ctlCtx, leaseID int64) string {
|
|
|
|
d := &pb.LeaseKeepAliveRequest{ID: leaseID}
|
|
|
|
s, err := dataMarshal(d)
|
|
|
|
if err != nil {
|
|
|
|
cx.t.Fatalf("gwLeaseKeepAlive: Marshal error (%v)", err)
|
|
|
|
}
|
|
|
|
return s
|
|
|
|
}
|
|
|
|
|
|
|
|
func gwLeaseGrant(cx ctlCtx, leaseID int64, ttl int64) string {
|
|
|
|
d := &pb.LeaseGrantRequest{ID: leaseID, TTL: ttl}
|
|
|
|
s, err := dataMarshal(d)
|
|
|
|
if err != nil {
|
|
|
|
cx.t.Fatalf("gwLeaseGrant: Marshal error (%v)", err)
|
|
|
|
}
|
|
|
|
return s
|
|
|
|
}
|
|
|
|
|
|
|
|
func gwLeaseRevoke(cx ctlCtx, leaseID int64) string {
|
|
|
|
d := &pb.LeaseRevokeRequest{ID: leaseID}
|
|
|
|
s, err := dataMarshal(d)
|
|
|
|
if err != nil {
|
|
|
|
cx.t.Fatalf("gwLeaseRevoke: Marshal error (%v)", err)
|
|
|
|
}
|
|
|
|
return s
|
|
|
|
}
|
|
|
|
|
|
|
|
func gwKVPutLease(cx ctlCtx, k string, v string, leaseID int64) string {
|
|
|
|
d := pb.PutRequest{Key: []byte(k), Value: []byte(v), Lease: leaseID}
|
|
|
|
s, err := dataMarshal(d)
|
|
|
|
if err != nil {
|
|
|
|
cx.t.Fatalf("gwKVPutLease: Marshal error (%v)", err)
|
|
|
|
}
|
|
|
|
return s
|
|
|
|
}
|