First implementation of next_version

tomschr · tomschr · commit 171fc9758b6c · 2020-02-24T22:02:15.000+01:00
* Implement semver.next_version(version, part, prerelease_token="rc")
* Add test cases
  * test_next_version
  * test_next_version_with_invalid_parts
* Reformat code with black

Co-authored-by: George Sakkis &lt;gsakkis&gt;
diff --git a/semver.py b/semver.py
@@ -684,6 +684,53 @@ def finalize_version(version):
     return format_version(verinfo["major"], verinfo["minor"], verinfo["patch"])
 
 
+def next_version(version, part, prerelease_token="rc"):
+    """
+    Determines the next version, takeing prereleases into account.
+
+    The "major", "minor", and "patch" raises the respective parts like
+    the ``bump_*`` functions. The real difference is using the
+    "preprelease" part. It gives you the next patch version of the prerelease,
+    for example:
+
+    >>> semver.next_version("0.1.4", "prerelease")
+    '0.1.5-rc.1'
+
+    :param version: a semver version string
+    :param part: One of "major", "minor", "patch", or "prerelease"
+    :param prerelease_token: prefix string of prerelease, defaults to 'rc'
+    :return:
+    """
+    validparts = {
+        "major",
+        "minor",
+        "patch",
+        "prerelease",
+        # "build", # ???
+    }
+    if part not in validparts:
+        raise ValueError(
+            "Invalid part. Expected one of {validparts}, but got {part!r}".format(
+                validparts=validparts, part=part
+            )
+        )
+    version = VersionInfo.parse(version)
+    if version.prerelease and (
+        part == "patch"
+        or (part == "minor" and version.patch == 0)
+        or (part == "major" and version.minor == version.patch == 0)
+    ):
+        return str(version.replace(prerelease=None))
+
+    if part in ("major", "minor", "patch"):
+        return str(getattr(version, "bump_" + part)())
+
+    if not version.prerelease:
+        version = version.bump_patch()
+    return str(version.bump_prerelease(prerelease_token))
+
+
+# ---- CLI
 def cmd_bump(args):
     """
     Subcommand: Bumps a version.
diff --git a/test_semver.py b/test_semver.py
@@ -20,6 +20,7 @@
     match,
     max_ver,
     min_ver,
+    next_version,
     parse,
     parse_version_info,
     process,
@@ -39,6 +40,7 @@
     match,
     max_ver,
     min_ver,
+    next_version,
     parse,
     process,
     replace,
@@ -827,3 +829,33 @@ def test_replace_raises_ValueError_for_non_numeric_values():
 def test_should_versioninfo_isvalid():
     assert VersionInfo.isvalid("1.0.0") is True
     assert VersionInfo.isvalid("foo") is False
+
+
+@pytest.mark.parametrize(
+    "version, part, expected",
+    [
+        # major
+        ("1.0.4-rc.1", "major", "2.0.0"),
+        ("1.1.0-rc.1", "major", "2.0.0"),
+        ("1.1.4-rc.1", "major", "2.0.0"),
+        ("1.2.3", "major", "2.0.0"),
+        ("1.0.0-rc.1", "major", "1.0.0"),
+        # minor
+        ("0.2.0-rc.1", "minor", "0.2.0"),
+        ("0.2.5-rc.1", "minor", "0.3.0"),
+        ("1.3.1", "minor", "1.4.0"),
+        # patch
+        ("1.3.2", "patch", "1.3.3"),
+        ("0.1.5-rc.2", "patch", "0.1.5"),
+        # prerelease
+        ("0.1.4", "prerelease", "0.1.5-rc.1"),
+        ("0.1.5-rc.1", "prerelease", "0.1.5-rc.2"),
+    ],
+)
+def test_next_version(version, part, expected):
+    assert next_version(version, part) == expected
+
+
+def test_next_version_with_invalid_parts():
+    with pytest.raises(ValueError):
+        next_version("1.0.1", "invalid")
