Add unit tests

Author: blva

tools/cli/internal/openapi/versions_test.go

@@ -27,6 +27,18 @@ func TestVersions(t *testing.T) {
 	assert.Equal(t, []string{"2023-01-01", "2023-02-01"}, versions)
 }
 
+func TestVersions_PrivatePreview(t *testing.T) {
+	versions, err := ExtractVersionsWithEnv(NewVersionedResponses(t), "dev")
+	require.NoError(t, err)
+	assert.ElementsMatch(t, []string{"2023-01-01", "2023-02-01", "private-preview-info-resource", "preview"}, versions)
+}
+
+func TestVersions_PublicPreview(t *testing.T) {
+	versions, err := ExtractVersionsWithEnv(NewVersionedResponses(t), "qa")
+	require.NoError(t, err)
+	assert.Equal(t, []string{"2023-01-01", "2023-02-01", "preview"}, versions)
+}
+
 func NewVersionedResponses(t *testing.T) *openapi3.T {
 	t.Helper()
 	inputPath := &openapi3.Paths{}
@@ -83,6 +95,83 @@ func NewVersionedResponses(t *testing.T) *openapi3.T {
 		},
 	})
 
+	extensionThree := map[string]any{
+		"x-xgen-version": "preview",
+		"x-xgen-preview": map[string]any{
+			"name": "info-resource",
+		},
+	}
+
+	responseThree := &openapi3.ResponseRef{
+		Value: &openapi3.Response{
+			Extensions: nil,
+			Content: map[string]*openapi3.MediaType{
+				"application/vnd.atlas.preview+json": {
+					Extensions: extensionThree,
+				},
+			},
+		},
+	}
+
+	hiddenEnvExtension := map[string]any{
+		"x-xgen-hidden-env": map[string]any{
+			"envs": "qa,prod",
+		},
+	}
+
+	responsesThree := &openapi3.Responses{}
+	responsesThree.Set("200", responseThree)
+
+	inputPath.Set("pathBase3", &openapi3.PathItem{
+		Extensions:  hiddenEnvExtension,
+		Ref:         "",
+		Summary:     "pathBase3",
+		Description: "pathBase3Description",
+		Delete: &openapi3.Operation{
+			Tags:      []string{"tag1"},
+			Responses: responsesThree,
+		},
+	})
+
+	extensionFour := map[string]any{
+		"x-xgen-version": "preview",
+		"x-xgen-preview": map[string]any{
+			"public": "true",
+		},
+	}
+
+	responseFour := &openapi3.ResponseRef{
+		Value: &openapi3.Response{
+			Extensions: nil,
+			Content: map[string]*openapi3.MediaType{
+				"application/vnd.atlas.preview+json": {
+					Extensions: extensionFour,
+				},
+			},
+		},
+	}
+
+	hiddenEnvExtensionTwo := map[string]any{
+		"x-xgen-hidden-env": map[string]any{
+			"envs": "prod",
+		},
+	}
+
+	responsesFour := &openapi3.Responses{}
+	responsesFour.Set("200", responseFour)
+
+	inputPath.Set("pathBase4", &openapi3.PathItem{
+		Extensions:  hiddenEnvExtensionTwo,
+		Ref:         "",
+		Summary:     "pathBase4",
+		Description: "pathBase4Description",
+		Post: &openapi3.Operation{
+
+			Tags:      []string{"tag1"},
+			Responses: responsesFour,
+		},
+	})
+
 	oas := &openapi3.T{
 		Paths: inputPath,
 	}