mirror of
https://github.com/etcd-io/etcd.git
synced 2024-09-27 06:25:44 +00:00
8f4735dfd4
To avoid inconsistant behavior during cluster upgrade we are feature gating persistance behind cluster version. This should ensure that all cluster members are upgraded to v3.6 before changing behavior. To allow backporting this fix to v3.5 we are also introducing flag --experimental-enable-lease-checkpoint-persist that will allow for smooth upgrade in v3.5 clusters with this feature enabled. Signed-off-by: Marek Siarkowicz <siarkowicz@google.com>
130 lines
3.4 KiB
Go
130 lines
3.4 KiB
Go
// 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 leasehttp
|
|
|
|
import (
|
|
"context"
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"os"
|
|
"testing"
|
|
"time"
|
|
|
|
"go.etcd.io/etcd/lease"
|
|
"go.etcd.io/etcd/mvcc/backend"
|
|
"go.uber.org/zap"
|
|
)
|
|
|
|
func TestRenewHTTP(t *testing.T) {
|
|
lg := zap.NewNop()
|
|
be, tmpPath := backend.NewTmpBackend(time.Hour, 10000)
|
|
defer os.Remove(tmpPath)
|
|
defer be.Close()
|
|
|
|
le := lease.NewLessor(lg, be, nil, lease.LessorConfig{MinLeaseTTL: int64(5)})
|
|
le.Promote(time.Second)
|
|
l, err := le.Grant(1, int64(5))
|
|
if err != nil {
|
|
t.Fatalf("failed to create lease: %v", err)
|
|
}
|
|
|
|
ts := httptest.NewServer(NewHandler(le, waitReady))
|
|
defer ts.Close()
|
|
|
|
ttl, err := RenewHTTP(context.TODO(), l.ID, ts.URL+LeasePrefix, http.DefaultTransport)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if ttl != 5 {
|
|
t.Fatalf("ttl expected 5, got %d", ttl)
|
|
}
|
|
}
|
|
|
|
func TestTimeToLiveHTTP(t *testing.T) {
|
|
lg := zap.NewNop()
|
|
be, tmpPath := backend.NewTmpBackend(time.Hour, 10000)
|
|
defer os.Remove(tmpPath)
|
|
defer be.Close()
|
|
|
|
le := lease.NewLessor(lg, be, nil, lease.LessorConfig{MinLeaseTTL: int64(5)})
|
|
le.Promote(time.Second)
|
|
l, err := le.Grant(1, int64(5))
|
|
if err != nil {
|
|
t.Fatalf("failed to create lease: %v", err)
|
|
}
|
|
|
|
ts := httptest.NewServer(NewHandler(le, waitReady))
|
|
defer ts.Close()
|
|
|
|
resp, err := TimeToLiveHTTP(context.TODO(), l.ID, true, ts.URL+LeaseInternalPrefix, http.DefaultTransport)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if resp.LeaseTimeToLiveResponse.ID != 1 {
|
|
t.Fatalf("lease id expected 1, got %d", resp.LeaseTimeToLiveResponse.ID)
|
|
}
|
|
if resp.LeaseTimeToLiveResponse.GrantedTTL != 5 {
|
|
t.Fatalf("granted TTL expected 5, got %d", resp.LeaseTimeToLiveResponse.GrantedTTL)
|
|
}
|
|
}
|
|
|
|
func TestRenewHTTPTimeout(t *testing.T) {
|
|
testApplyTimeout(t, func(l *lease.Lease, serverURL string) error {
|
|
_, err := RenewHTTP(context.TODO(), l.ID, serverURL+LeasePrefix, http.DefaultTransport)
|
|
return err
|
|
})
|
|
}
|
|
|
|
func TestTimeToLiveHTTPTimeout(t *testing.T) {
|
|
testApplyTimeout(t, func(l *lease.Lease, serverURL string) error {
|
|
_, err := TimeToLiveHTTP(context.TODO(), l.ID, true, serverURL+LeaseInternalPrefix, http.DefaultTransport)
|
|
return err
|
|
})
|
|
}
|
|
|
|
func testApplyTimeout(t *testing.T, f func(*lease.Lease, string) error) {
|
|
lg := zap.NewNop()
|
|
be, tmpPath := backend.NewTmpBackend(time.Hour, 10000)
|
|
defer os.Remove(tmpPath)
|
|
defer be.Close()
|
|
|
|
le := lease.NewLessor(lg, be, nil, lease.LessorConfig{MinLeaseTTL: int64(5)})
|
|
le.Promote(time.Second)
|
|
l, err := le.Grant(1, int64(5))
|
|
if err != nil {
|
|
t.Fatalf("failed to create lease: %v", err)
|
|
}
|
|
|
|
ts := httptest.NewServer(NewHandler(le, waitNotReady))
|
|
defer ts.Close()
|
|
err = f(l, ts.URL)
|
|
if err == nil {
|
|
t.Fatalf("expected timeout error, got nil")
|
|
}
|
|
if err.Error() != ErrLeaseHTTPTimeout.Error() {
|
|
t.Fatalf("expected (%v), got (%v)", ErrLeaseHTTPTimeout.Error(), err.Error())
|
|
}
|
|
}
|
|
|
|
func waitReady() <-chan struct{} {
|
|
ch := make(chan struct{})
|
|
close(ch)
|
|
return ch
|
|
}
|
|
|
|
func waitNotReady() <-chan struct{} {
|
|
return nil
|
|
}
|