test: fixed tests (#44)

Signed-off-by: Carlos A Becker <caarlos0@gmail.com>
caarlos0 · Oct 6, 2021 · f68e9c5 · f68e9c5
1 parent c192293
commit f68e9c5
Showing 1 changed file with 7 additions and 3 deletions.
diff --git a/main_test.go b/main_test.go
@@ -25,13 +25,17 @@ func TestUnsetMetadata(t *testing.T) {
 }
 
 func TestStripPrefixReturnsVersionOnly(t *testing.T) {
-	is.New(t).True(getVersion("v2.3.4", "v", "4.5.6", true) == "4.5.6")
+	is.New(t).True(getVersion("v2.3.4", "v", "4.5.6", "", true) == "4.5.6")
 }
 
 func TestStripPrefixWhenNoPrefixReturnsVersionOnly(t *testing.T) {
-	is.New(t).True(getVersion("2.3.4", "v", "4.5.6", true) == "4.5.6")
+	is.New(t).True(getVersion("2.3.4", "v", "4.5.6", "", true) == "4.5.6")
 }
 
 func TestNoStripPrefixReturnsPrefixAndVersion(t *testing.T) {
-	is.New(t).True(getVersion("v2.3.4", "v", "4.5.6", false) == "v4.5.6")
+	is.New(t).True(getVersion("v2.3.4", "v", "4.5.6", "", false) == "v4.5.6")
+}
+
+func TestSuffix(t *testing.T) {
+	is.New(t).True(getVersion("v2.3.4", "v", "4.5.6", "dev", false) == "v4.5.6-dev")
 }