added unit tests

Author: andreaangiolillo

tools/cli/internal/cli/versions/versions.go

@@ -46,7 +46,6 @@ func (o *Opts) Run() error {
 
 	var versions []string
 	versions, err = openapi.ExtractVersionsWithEnv(specInfo.Spec, o.env)
-
 	if err != nil {
 		return err
 	}
@@ -55,6 +54,7 @@ func (o *Opts) Run() error {
 		return fmt.Errorf("no versions found in the OpenAPI specification")
 	}
 
+	versions = o.filterStabilityLevelVersions(versions)
 	bytes, err := o.versionsAsBytes(versions)
 	if err != nil {
 		return err
@@ -68,6 +68,25 @@ func (o *Opts) Run() error {
 	return nil
 }
 
+func (o *Opts) filterStabilityLevelVersions(apiVersions []string) []string {
+	if o.stabilityLevel == "" || apiVersions == nil {
+		return apiVersions
+	}
+
+	var out []string
+	for _, v := range apiVersions {
+		if o.stabilityLevel == apiversion.PreviewStabilityLevel && strings.Contains(v, "preview") {
+			out = append(out, v)
+		}
+
+		if o.stabilityLevel == apiversion.StableStabilityLevel && !strings.Contains(v, "preview") {
+			out = append(out, v)
+		}
+	}
+
+	return out
+}
+
 func (o *Opts) versionsAsBytes(versions []string) ([]byte, error) {
 	data, err := json.MarshalIndent(versions, "", "  ")
 	if err != nil {

tools/cli/internal/cli/versions/versions_test.go

@@ -21,7 +21,7 @@ import (
 	"github.com/stretchr/testify/assert"
 )
 
-func TestVersions(t *testing.T) {
+func TestVersions_Run(t *testing.T) {
 	fs := afero.NewMemMapFs()
 	opts := &Opts{
 		basePath:   "../../../test/data/base_spec.json",
@@ -43,7 +43,7 @@ func TestVersions(t *testing.T) {
 	assert.Contains(t, string(b), "2023-02-01")
 }
 
-func TestVersionWithEnv(t *testing.T) {
+func TestVersion_RunWithEnv(t *testing.T) {
 	fs := afero.NewMemMapFs()
 	opts := &Opts{
 		basePath:   "../../../test/data/base_spec.json",
@@ -65,3 +65,77 @@ func TestVersionWithEnv(t *testing.T) {
 	assert.NotEmpty(t, b)
 	assert.Contains(t, string(b), "2023-02-01")
 }
+
+func TestVersion_RunWithPreview(t *testing.T) {
+	fs := afero.NewMemMapFs()
+	opts := &Opts{
+		basePath:   "../../../test/data/base_spec_with_preview.json",
+		outputPath: "foas.json",
+		fs:         fs,
+		env:        "staging",
+	}
+
+	if err := opts.Run(); err != nil {
+		t.Fatalf("Run() unexpected error: %v", err)
+	}
+
+	b, err := afero.ReadFile(fs, opts.outputPath)
+	if err != nil {
+		t.Fatalf("ReadFile() unexpected error: %v", err)
+	}
+
+	// Check initial versions
+	assert.NotEmpty(t, b)
+	assert.Contains(t, string(b), "2023-02-01")
+	assert.Contains(t, string(b), "preview")
+}
+
+func TestVersion_RunStabilityLevelPreview(t *testing.T) {
+	fs := afero.NewMemMapFs()
+	opts := &Opts{
+		basePath:       "../../../test/data/base_spec_with_preview.json",
+		outputPath:     "foas.json",
+		fs:             fs,
+		env:            "staging",
+		stabilityLevel: "PREVIEW",
+	}
+
+	if err := opts.Run(); err != nil {
+		t.Fatalf("Run() unexpected error: %v", err)
+	}
+
+	b, err := afero.ReadFile(fs, opts.outputPath)
+	if err != nil {
+		t.Fatalf("ReadFile() unexpected error: %v", err)
+	}
+
+	// Check initial versions
+	assert.NotEmpty(t, b)
+	assert.NotContains(t, string(b), "2023-02-01")
+	assert.Contains(t, string(b), "preview")
+}
+
+func TestVersion_RunStabilityLevelStable(t *testing.T) {
+	fs := afero.NewMemMapFs()
+	opts := &Opts{
+		basePath:       "../../../test/data/base_spec_with_preview.json",
+		outputPath:     "foas.json",
+		fs:             fs,
+		env:            "staging",
+		stabilityLevel: "STABLE",
+	}
+
+	if err := opts.Run(); err != nil {
+		t.Fatalf("Run() unexpected error: %v", err)
+	}
+
+	b, err := afero.ReadFile(fs, opts.outputPath)
+	if err != nil {
+		t.Fatalf("ReadFile() unexpected error: %v", err)
+	}
+
+	// Check initial versions
+	assert.NotEmpty(t, b)
+	assert.Contains(t, string(b), "2023-02-01")
+	assert.NotContains(t, string(b), "preview")
+}