Function TestAddonPublishFinalize

cli/cmd/addon_publish_test.go:106–161

(t *testing.T)

}

func TestAddonPublishFinalize(t *testing.T) {
	t.Setenv("CLOUDQUERY_API_KEY", "testkey")

	wantCalls := map[string]int{
		"PUT /addons/cloudquery/visualization/test/versions/v1.2.3":         1,
		"POST /addons/cloudquery/visualization/test/versions/v1.2.3/assets": 1,
		"PUT /upload-zip": 1,
		"PATCH /addons/cloudquery/visualization/test/versions/v1.2.3": 1,
	}
	gotCalls := map[string]int{}
	gotUploads := 0
	ts := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
		w.Header().Set("Content-Type", "application/json")
		gotCalls[r.Method+" "+r.URL.Path]++
		switch r.URL.Path {
		case "/addons/cloudquery/visualization/test/versions/v1.2.3":
			checkAuthHeader(t, r)
			if r.Method == "PATCH" {
				checkUpdateAddonVersionRequest(t, r)
				if gotUploads != 1 {
					t.Fatalf("expected 1 upload before draft=false, got %d", gotUploads)
				}
				w.WriteHeader(http.StatusOK)
			} else {
				checkCreateAddonVersionRequest(t, r)
				w.WriteHeader(http.StatusCreated)
			}
			_, err := w.Write([]byte(`{"name": "v1.2.3"}`))
			require.NoError(t, err)
		case "/addons/cloudquery/visualization/test/versions/v1.2.3/assets":
			checkAuthHeader(t, r)
			w.WriteHeader(http.StatusCreated)
			_, err := w.Write([]byte(fmt.Sprintf(`{"url": "%s"}`, "http://"+r.Host+"/upload-zip")))
			require.NoError(t, err)
		case "/upload-zip":
			w.WriteHeader(http.StatusOK)
			_, err := w.Write([]byte(`{}`))
			require.NoError(t, err)
			gotUploads++
		}
	}))
	defer ts.Close()

	t.Setenv(envAPIURL, ts.URL)

	cmd := NewCmdRoot()
	args := append([]string{"addon", "publish", "testdata/addon-v1/manifest.json", "v1.2.3", "--finalize"}, testCommandArgs(t)...)
	cmd.SetArgs(args)
	err := cmd.Execute()
	if err != nil {
		t.Fatal(err)
	}
	if diff := cmp.Diff(wantCalls, gotCalls); diff != "" {
		t.Fatalf("mismatch (-want +got):\n%s", diff)
	}
}

func TestAddonPublish_Unauthorized(t *testing.T) {