test(bump): add test cast that fails to bump correctly

Lee-W · Lee-W · commit b7741290513e · 2021-05-26T22:11:53.000+08:00
commitizen-tools#383
diff --git a/tests/test_bump_update_version_in_files.py b/tests/test_bump_update_version_in_files.py
@@ -59,6 +59,36 @@
   command: my-command
 """
 
+MULTIPLE_VERSIONS_TO_UPDATE_POETRY_LOCK = """
+[[package]]
+name = "to-update-1"
+version = "1.2.9"
+
+[[package]]
+name = "not-to-update"
+version = "1.3.3"
+
+[[package]]
+name = "not-to-update"
+version = "1.3.3"
+
+[[package]]
+name = "not-to-update"
+version = "1.3.3"
+
+[[package]]
+name = "not-to-update"
+version = "1.3.3"
+
+[[package]]
+name = "not-to-update"
+version = "1.3.3"
+
+[[package]]
+name = "to-update-2"
+version = "1.2.9"
+"""
+
 MULTIPLE_VERSIONS_INCREASE_STRING = 'version = "1.2.9"\n' * 30
 MULTIPLE_VERSIONS_REDUCE_STRING = 'version = "1.2.10"\n' * 30
 
@@ -119,6 +149,13 @@ def docker_compose_file(tmpdir):
     return str(tmp_file)
 
 
+@pytest.fixture(scope="function")
+def multiple_versions_to_update_poetry_lock(tmpdir):
+    tmp_file = tmpdir.join("pyproject.toml")
+    tmp_file.write(MULTIPLE_VERSIONS_TO_UPDATE_POETRY_LOCK)
+    return str(tmp_file)
+
+
 @pytest.fixture(scope="function")
 def version_files(
     commitizen_config_file,
@@ -228,3 +265,15 @@ def test_file_version_inconsistent_error(
         "version_files are possibly inconsistent."
     )
     assert expected_msg in str(excinfo.value)
+
+
+def test_multiplt_versions_to_bump(multiple_versions_to_update_poetry_lock):
+    old_version = "1.2.9"
+    new_version = "1.2.10"
+    location = f"{multiple_versions_to_update_poetry_lock}:version"
+
+    bump.update_version_in_files(old_version, new_version, [location])
+    with open(multiple_versions_to_update_poetry_lock, "r") as f:
+        data = f.read()
+        assert len(re.findall(old_version, data)) == 0
+        assert len(re.findall(new_version, data)) == 2
