Merge pull request #85 from hashicorp/base-version-func
add Core func
diff --git a/version.go b/version.go
index 09703e8..8068834 100644
--- a/version.go
+++ b/version.go
@@ -278,6 +278,14 @@
return 0
}
+// Core returns a new version constructed from only the MAJOR.MINOR.PATCH
+// segments of the version, without prerelease or metadata.
+func (v *Version) Core() *Version {
+ segments := v.Segments64()
+ segmentsOnly := fmt.Sprintf("%d.%d.%d", segments[0], segments[1], segments[2])
+ return Must(NewVersion(segmentsOnly))
+}
+
// Equal tests if two versions are equal.
func (v *Version) Equal(o *Version) bool {
if v == nil || o == nil {
diff --git a/version_test.go b/version_test.go
index cbbeac8..9fa34f6 100644
--- a/version_test.go
+++ b/version_test.go
@@ -87,6 +87,38 @@
}
}
+func TestCore(t *testing.T) {
+ cases := []struct {
+ v1 string
+ v2 string
+ }{
+ {"1.2.3", "1.2.3"},
+ {"2.3.4-alpha1", "2.3.4"},
+ {"3.4.5alpha1", "3.4.5"},
+ {"1.2.3-2", "1.2.3"},
+ {"4.5.6-beta1+meta", "4.5.6"},
+ {"5.6.7.1.2.3", "5.6.7"},
+ }
+
+ for _, tc := range cases {
+ v1, err := NewVersion(tc.v1)
+ if err != nil {
+ t.Fatalf("error for version %q: %s", tc.v1, err)
+ }
+ v2, err := NewVersion(tc.v2)
+ if err != nil {
+ t.Fatalf("error for version %q: %s", tc.v2, err)
+ }
+
+ actual := v1.Core()
+ expected := v2
+
+ if !reflect.DeepEqual(actual, expected) {
+ t.Fatalf("expected: %s\nactual: %s", expected, actual)
+ }
+ }
+}
+
func TestVersionCompare(t *testing.T) {
cases := []struct {
v1 string
