fix(commands/init): fix clean up file when initialize commitizen config

Lee-W · Lee-W · commit 9fb34761877f · 2020-03-01T18:06:45.000+08:00
#138
diff --git a/commitizen/commands/init.py b/commitizen/commands/init.py
@@ -6,7 +6,7 @@
 from commitizen.cz import registry
 from commitizen.config import BaseConfig, TomlConfig, IniConfig
 from commitizen.git import get_latest_tag_name, get_tag_names
-from commitizen.defaults import config_files
+from commitizen.defaults import long_term_support_config_files
 
 
 class Init:
@@ -17,7 +17,7 @@ def __init__(self, config: BaseConfig, *args):
     def __call__(self):
         values_to_add = {}
 
-        # No config file exist
+        # No config for commitizen exist
         if not self.config.path:
             config_path = self._ask_config_path()
 
@@ -26,7 +26,7 @@ def __call__(self):
             else:
                 self.config = IniConfig(data="", path=config_path)
 
-            self.config.init_empty_config_file()
+            self.config.init_empty_config_content()
 
             values_to_add["name"] = self._ask_name()
             tag = self._ask_tag()
@@ -41,7 +41,7 @@ def __call__(self):
     def _ask_config_path(self) -> str:
         name = questionary.select(
             "Please choose a supported config file: (default: pyproject.tml)",
-            choices=config_files,
+            choices=long_term_support_config_files,
             default="pyproject.toml",
             style=self.cz.style,
         ).ask()
diff --git a/commitizen/config/toml_config.py b/commitizen/config/toml_config.py
@@ -10,8 +10,8 @@ def __init__(self, *, data: str, path: str):
         self._parse_setting(data)
         self.add_path(path)
 
-    def init_empty_config_file(self):
-        with open(self.path, "w") as toml_file:
+    def init_empty_config_content(self):
+        with open(self.path, "a") as toml_file:
             toml_file.write("[tool.commitizen]")
 
     def set_key(self, key, value):
diff --git a/tests/test_conf.py b/tests/test_conf.py
@@ -144,10 +144,21 @@ def test_read_cfg_when_not_in_a_git_project(tmpdir):
 
 
 class TestTomlConfig:
-    def test_init_empty_config_file(self, tmpdir):
+    def test_init_empty_config_content(self, tmpdir):
         path = tmpdir.mkdir("commitizen").join(".cz.toml")
         toml_config = config.TomlConfig(data="", path=path)
-        toml_config.init_empty_config_file()
+        toml_config.init_empty_config_content()
 
         with open(path, "r") as toml_file:
             assert toml_file.read() == "[tool.commitizen]"
+
+    def test_init_empty_config_content_with_existing_content(self, tmpdir):
+        existing_content = "[tool.black]\n" "line-length = 88\n"
+
+        path = tmpdir.mkdir("commitizen").join(".cz.toml")
+        path.write(existing_content)
+        toml_config = config.TomlConfig(data="", path=path)
+        toml_config.init_empty_config_content()
+
+        with open(path, "r") as toml_file:
+            assert toml_file.read() == existing_content + "[tool.commitizen]"
