98 lines
3.6 KiB
Go
98 lines
3.6 KiB
Go
package configread
|
|
|
|
import (
|
|
"context"
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"reflect"
|
|
"strings"
|
|
"testing"
|
|
)
|
|
|
|
const eventTestDispatcherID = "11111111-1111-4111-8111-111111111111"
|
|
|
|
func eventTestClient(t *testing.T, server *httptest.Server) *Client {
|
|
t.Helper()
|
|
client, err := NewClient(server.URL, eventTestDispatcherID, mockTestSecret, server.Client())
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
return client
|
|
}
|
|
|
|
func TestV04SnapshotAndLiveChangesUseSeparateRequests(t *testing.T) {
|
|
var paths []string
|
|
server := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
|
paths = append(paths, r.URL.RequestURI())
|
|
if r.Header.Get(dispatcherIDHeader) != eventTestDispatcherID || r.Header.Get(dispatcherSecretHeader) != mockTestSecret {
|
|
t.Error("missing Dispatcher credentials")
|
|
}
|
|
var fixture string
|
|
switch r.URL.RawQuery {
|
|
case "mode=snapshot":
|
|
fixture = "task-discovery-snapshot-page1-v0.4.json"
|
|
case "page_token=page-2&snapshot_id=22222222-2222-4222-8222-222222222222":
|
|
fixture = "task-discovery-snapshot-page2-v0.4.json"
|
|
case "after=42":
|
|
fixture = "task-discovery-changes-v0.4.json"
|
|
default:
|
|
t.Errorf("unexpected request %s", r.URL.RequestURI())
|
|
w.WriteHeader(http.StatusBadRequest)
|
|
return
|
|
}
|
|
w.Header().Set("Content-Type", "application/json")
|
|
_, _ = w.Write(readConfigFixture(t, fixture))
|
|
}))
|
|
defer server.Close()
|
|
client := eventTestClient(t, server)
|
|
first, err := client.ReadTaskSnapshot(context.Background(), "", "")
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if first.SnapshotID != "22222222-2222-4222-8222-222222222222" || first.Watermark != "42" || first.NextPageToken != "page-2" || len(first.Tasks) != 1 {
|
|
t.Fatalf("first page=%+v", first)
|
|
}
|
|
second, err := client.ReadTaskSnapshot(context.Background(), first.SnapshotID, first.NextPageToken)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if second.SnapshotID != first.SnapshotID || second.Watermark != first.Watermark || second.NextPageToken != "" || len(second.Tasks) != 1 {
|
|
t.Fatalf("second page=%+v", second)
|
|
}
|
|
changes, err := client.ReadTaskChanges(context.Background(), first.Watermark)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if changes.NextCursor != "43" || len(changes.Tasks) != 1 || changes.Tasks[0].Status != "removed" {
|
|
t.Fatalf("changes=%+v", changes)
|
|
}
|
|
want := []string{"/internal/v1/dispatcher/tasks?mode=snapshot", "/internal/v1/dispatcher/tasks?page_token=page-2&snapshot_id=22222222-2222-4222-8222-222222222222", "/internal/v1/dispatcher/tasks?after=42"}
|
|
if !reflect.DeepEqual(paths, want) {
|
|
t.Fatalf("queries=%v want=%v", paths, want)
|
|
}
|
|
}
|
|
|
|
func TestV04SnapshotRejectsInvalidPayloadOrRequest(t *testing.T) {
|
|
for name, fixture := range map[string]string{
|
|
"extra field": "task-discovery-invalid-v0.4.json",
|
|
"changes in snapshot": "task-discovery-changes-v0.4.json",
|
|
"old cursor page": "task-discovery-page-v0.3.json",
|
|
} {
|
|
t.Run(name, func(t *testing.T) {
|
|
server := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, _ *http.Request) { _, _ = w.Write(readConfigFixture(t, fixture)) }))
|
|
defer server.Close()
|
|
if _, err := eventTestClient(t, server).ReadTaskSnapshot(context.Background(), "", ""); err == nil {
|
|
t.Fatal("invalid snapshot accepted")
|
|
}
|
|
})
|
|
}
|
|
server := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, _ *http.Request) {
|
|
w.WriteHeader(http.StatusConflict)
|
|
_, _ = w.Write(readConfigFixture(t, "task-discovery-error-v0.4.json"))
|
|
}))
|
|
defer server.Close()
|
|
if _, err := eventTestClient(t, server).ReadTaskSnapshot(context.Background(), "", ""); err == nil || !strings.Contains(err.Error(), "409") {
|
|
t.Fatalf("unavailable snapshot error=%v", err)
|
|
}
|
|
}
|