Go: Add UnformatSemVer function

github · May 8, 2024 · dfa2319 · dfa2319
1 parent 8a9fd8c
commit dfa2319
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 0 deletions.
diff --git a/go/extractor/util/util.go b/go/extractor/util/util.go
@@ -418,3 +418,12 @@ func FormatSemVer(version string) string {
 
 	return version
 }
+
+// Removes "v" from `semver`, if present.
+func UnformatSemVer(semver string) string {
+	if strings.HasPrefix(semver, "v") {
+		return semver[1:]
+	}
+
+	return semver
+}
diff --git a/go/extractor/util/util_test.go b/go/extractor/util/util_test.go
@@ -40,4 +40,11 @@ func TestFormatSemVer(t *testing.T) {
 			t.Errorf("Expected FormatSemVer(\"%s\") to be \"%s\", but got \"%s\".", pair.Input, pair.Expected, actual)
 		}
 	}
+
+	for _, pair := range tests {
+		actual := UnformatSemVer(pair.Input)
+		if actual != pair.Expected[1:] {
+			t.Errorf("Expected UnformatSemVer(\"%s\") to be \"%s\", but got \"%s\".", pair.Input, pair.Expected, actual)
+		}
+	}
 }