mirror of https://github.com/status-im/consul.git
agent: Testing the KV DELETE with cas
This commit is contained in:
parent
4572e7c8ec
commit
d05ed7baa6
|
@ -3,13 +3,14 @@ package agent
|
||||||
import (
|
import (
|
||||||
"bytes"
|
"bytes"
|
||||||
"fmt"
|
"fmt"
|
||||||
"github.com/hashicorp/consul/consul/structs"
|
|
||||||
"github.com/hashicorp/consul/testutil"
|
|
||||||
"net/http"
|
"net/http"
|
||||||
"net/http/httptest"
|
"net/http/httptest"
|
||||||
"os"
|
"os"
|
||||||
"reflect"
|
"reflect"
|
||||||
"testing"
|
"testing"
|
||||||
|
|
||||||
|
"github.com/hashicorp/consul/consul/structs"
|
||||||
|
"github.com/hashicorp/consul/testutil"
|
||||||
)
|
)
|
||||||
|
|
||||||
func TestKVSEndpoint_PUT_GET_DELETE(t *testing.T) {
|
func TestKVSEndpoint_PUT_GET_DELETE(t *testing.T) {
|
||||||
|
@ -183,6 +184,93 @@ func TestKVSEndpoint_Recurse(t *testing.T) {
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func TestKVSEndpoint_DELETE_CAS(t *testing.T) {
|
||||||
|
dir, srv := makeHTTPServer(t)
|
||||||
|
defer os.RemoveAll(dir)
|
||||||
|
defer srv.Shutdown()
|
||||||
|
defer srv.agent.Shutdown()
|
||||||
|
|
||||||
|
testutil.WaitForLeader(t, srv.agent.RPC, "dc1")
|
||||||
|
|
||||||
|
{
|
||||||
|
buf := bytes.NewBuffer([]byte("test"))
|
||||||
|
req, err := http.NewRequest("PUT", "/v1/kv/test", buf)
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("err: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
resp := httptest.NewRecorder()
|
||||||
|
obj, err := srv.KVSEndpoint(resp, req)
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("err: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
if res := obj.(bool); !res {
|
||||||
|
t.Fatalf("should work")
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
req, err := http.NewRequest("GET", "/v1/kv/test", nil)
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("err: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
resp := httptest.NewRecorder()
|
||||||
|
obj, err := srv.KVSEndpoint(resp, req)
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("err: %v", err)
|
||||||
|
}
|
||||||
|
d := obj.(structs.DirEntries)[0]
|
||||||
|
|
||||||
|
// Create a CAS request, bad index
|
||||||
|
{
|
||||||
|
buf := bytes.NewBuffer([]byte("zip"))
|
||||||
|
req, err := http.NewRequest("DELETE",
|
||||||
|
fmt.Sprintf("/v1/kv/test?cas=%d", d.ModifyIndex-1), buf)
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("err: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
resp := httptest.NewRecorder()
|
||||||
|
obj, err := srv.KVSEndpoint(resp, req)
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("err: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
if res := obj.(bool); res {
|
||||||
|
t.Fatalf("should NOT work")
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Create a CAS request, good index
|
||||||
|
{
|
||||||
|
buf := bytes.NewBuffer([]byte("zip"))
|
||||||
|
req, err := http.NewRequest("DELETE",
|
||||||
|
fmt.Sprintf("/v1/kv/test?cas=%d", d.ModifyIndex), buf)
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("err: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
resp := httptest.NewRecorder()
|
||||||
|
obj, err := srv.KVSEndpoint(resp, req)
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("err: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
if res := obj.(bool); !res {
|
||||||
|
t.Fatalf("should work")
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Verify the delete
|
||||||
|
req, _ = http.NewRequest("GET", "/v1/kv/test", nil)
|
||||||
|
resp = httptest.NewRecorder()
|
||||||
|
obj, _ = srv.KVSEndpoint(resp, req)
|
||||||
|
if obj != nil {
|
||||||
|
t.Fatalf("should be destroyed")
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
func TestKVSEndpoint_CAS(t *testing.T) {
|
func TestKVSEndpoint_CAS(t *testing.T) {
|
||||||
dir, srv := makeHTTPServer(t)
|
dir, srv := makeHTTPServer(t)
|
||||||
defer os.RemoveAll(dir)
|
defer os.RemoveAll(dir)
|
||||||
|
|
Loading…
Reference in New Issue