Function TestPluginPublish

cli/cmd/plugin_publish_test.go:21–117

(t *testing.T)

)

func TestPluginPublish(t *testing.T) {
	type testCase struct {
		name    string
		distDir string
		args    []string
	}

	var testCases = []testCase{
		{
			name:    "old package json with old command line args format",
			distDir: "testdata/dist-v1-no-team-package-json",
			args:    []string{"plugin", "publish", "cloudquery/test", "--dist-dir", "testdata/dist-v1-no-team-package-json"},
		},
		{
			name:    "new package json with old command line args format",
			distDir: "testdata/dist-v1-with-team-package-json",
			args:    []string{"plugin", "publish", "cloudquery/test", "--dist-dir", "testdata/dist-v1-with-team-package-json"},
		},
		{
			name:    "new package json with new command line args format (no args)",
			distDir: "testdata/dist-v1-with-team-package-json",
			args:    []string{"plugin", "publish", "--dist-dir", "testdata/dist-v1-with-team-package-json"},
		},
	}

	for _, tc := range testCases {
		t.Run(tc.name, func(t *testing.T) {
			t.Setenv("CLOUDQUERY_API_KEY", "testkey")
			wantCalls := map[string]int{
				"PUT /plugins/cloudquery/source/test/versions/v1.2.3":                      1,
				"PUT /plugins/cloudquery/source/test/versions/v1.2.3/tables":               1,
				"POST /plugins/cloudquery/source/test/versions/v1.2.3/docs":                1,
				"POST /plugins/cloudquery/source/test/versions/v1.2.3/assets/linux_amd64":  1,
				"POST /plugins/cloudquery/source/test/versions/v1.2.3/assets/darwin_amd64": 1,
				"PUT /upload-linux":  1,
				"PUT /upload-darwin": 1,
			}
			gotCalls := map[string]int{}
			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 "/plugins/cloudquery/source/test/versions/v1.2.3":
					checkAuthHeader(t, r)
					w.WriteHeader(http.StatusCreated)
					_, err := w.Write([]byte(`{"name": "v1.2.3"}`))
					require.NoError(t, err)
					checkCreatePluginVersionRequest(t, r)
				case "/plugins/cloudquery/source/test/versions/v1.2.3/tables":
					checkAuthHeader(t, r)
					w.WriteHeader(http.StatusCreated)
					_, err := w.Write([]byte(`{}`))
					require.NoError(t, err)
					checkCreateTablesRequest(t, r)
				case "/plugins/cloudquery/source/test/versions/v1.2.3/docs":
					checkAuthHeader(t, r)
					w.WriteHeader(http.StatusCreated)
					_, err := w.Write([]byte(`{}`))