source_update_test.go

package unstructured

import (
	"errors"
	"net/http"
	"strconv"
	"testing"
	"time"
)

func TestUpdateSource(t *testing.T) {
	t.Parallel()

	client, mux := testclient(t)

	id := "a15d4161-77a0-4e08-b65e-86f398ce15ad"

	mux.UpdateSource = func(w http.ResponseWriter, r *http.Request) {
		if val := r.PathValue("id"); val != id {
			http.Error(w, "source ID "+val+" not found", http.StatusNotFound)
			return
		}

		response := []byte(`{` +
			`  "config": {` +
			`    "client_id": "foo",` +
			`    "tenant": "foo",` +
			`    "authority_url": "foo",` +
			`    "user_pname": "foo",` +
			`    "client_cred": "foo",` +
			`    "recursive": false,` +
			`    "path": "foo"` +
			`  },` +
			`  "created_at": "2023-09-15T01:06:53.146Z",` +
			`  "id": "a15d4161-77a0-4e08-b65e-86f398ce15ad",` +
			`  "name": "test_source_name",` +
			`  "type": "onedrive"` +
			`}`)

		w.Header().Set("Content-Type", "application/json")
		w.Header().Set("Content-Length", strconv.Itoa(len(response)))
		w.WriteHeader(http.StatusOK)
		w.Write(response)
	}

	source, err := client.UpdateSource(testContext(t), UpdateSourceRequest{
		ID: id,
		Config: &OneDriveConnectorConfig{
			ClientID:     "foo",
			Tenant:       "foo",
			AuthorityURL: "foo",
			UserPName:    "foo",
			ClientCred:   "foo",
			Recursive:    Bool(false),
			Path:         String("foo"),
		},
	})
	if err != nil {
		t.Fatalf("failed to update source: %v", err)
	}

	if err := errors.Join(
		eq("updated_source.id", source.ID, id),
		eq("updated_source.name", source.Name, "test_source_name"),
		equal("updated_source.created_at", source.CreatedAt, time.Date(2023, 9, 15, 1, 6, 53, 146000000, time.UTC)),
	); err != nil {
		t.Error(err)
	}

	cfg, ok := source.Config.(*OneDriveConnectorConfig)
	if !ok {
		t.Errorf("expected source config to be %T, got %T", cfg, source.Config)
	}
}