CLOUDP-297221: Add preview support to split command

mongodb · Feb 6, 2025 · cebc24e · cebc24e
1 parent 82d24b8
commit cebc24e
Show file tree

Hide file tree

Showing 2 changed files with 21 additions and 0 deletions.
diff --git a/tools/cli/internal/apiversion/version.go b/tools/cli/internal/apiversion/version.go
@@ -18,6 +18,7 @@ import (
 	"fmt"
 	"log"
 	"regexp"
+	"strings"
 	"time"
 
 	"github.com/getkin/kin-openapi/openapi3"
@@ -54,6 +55,13 @@ func New(opts ...Option) (*APIVersion, error) {
 // WithVersion sets the version on the APIVersion.
 func WithVersion(version string) Option {
 	return func(v *APIVersion) error {
+		if strings.EqualFold(version, PreviewStabilityLevel) {
+			v.version = version
+			v.stabilityVersion = PreviewStabilityLevel
+			v.versionDate = time.Now() // make preview look like the latest version
+			return nil
+		}
+
 		versionDate, err := DateFromVersion(version)
 		if err != nil {
 			return err
@@ -87,6 +95,7 @@ func WithContent(contentType string) Option {
 		v.stabilityVersion = StableStabilityLevel
 		if version == PreviewStabilityLevel {
 			v.stabilityVersion = PreviewStabilityLevel
+			v.versionDate = time.Now() // make preview look like the latest version
 			return nil
 		}
 

diff --git a/tools/cli/internal/apiversion/version_test.go b/tools/cli/internal/apiversion/version_test.go
@@ -48,6 +48,12 @@ func TestParseVersion(t *testing.T) {
 			expectedMatch: "2030-02-20",
 			wantErr:       false,
 		},
+		{
+			name:          "preview",
+			contentType:   "application/vnd.atlas.preview+json",
+			expectedMatch: "preview",
+			wantErr:       false,
+		},
 		{
 			name:          "invalid",
 			contentType:   "application/vnd.test.2023-01-01",
@@ -94,6 +100,12 @@ func TestNewAPIVersionFromContentType(t *testing.T) {
 			expectedMatch: "2030-02-20",
 			wantErr:       false,
 		},
+		{
+			name:          "preview",
+			contentType:   "application/vnd.atlas.preview+json",
+			expectedMatch: "preview",
+			wantErr:       false,
+		},
 		{
 			name:          "invalid",
 			contentType:   "application/vnd.test.2023-01-01",