Task header should no longer be converted to test cases header (#221)

* task header should no longer be converted to test cases header

* black

Author: bhirsz

CHANGES.md

@@ -4,6 +4,7 @@
 
 ### Fixes
 - Trailing whitespace is no longer added to blank lines in multiline statements when using align transformers ([#219](https://github.com/MarketSquare/robotframework-tidy/issues/219))
+- `*** Tasks ***` is no longer converted to `*** Test Cases ***` by NormalizeSectionHeaderName ([#218](https://github.com/MarketSquare/robotframework-tidy/issues/218))
 
 ## 1.6.0
 

robotidy/transformers/NormalizeSectionHeaderName.py

@@ -1,4 +1,4 @@
-from robot.api.parsing import ModelTransformer, SectionHeader
+from robot.api.parsing import ModelTransformer, SectionHeader, Token
 
 from robotidy.decorators import check_start_end_line
 
@@ -35,10 +35,18 @@ def __init__(self, uppercase: bool = False):
 
     @check_start_end_line
     def visit_SectionHeader(self, node):  # noqa
-        normalized_section = SectionHeader.from_params(type=node.type)
-        normalized_name = normalized_section.data_tokens[0].value
+        if node.name and "task" in node.name:
+            name = "*** Tasks ***"
+        else:
+            name = {
+                Token.SETTING_HEADER: "*** Settings ***",
+                Token.VARIABLE_HEADER: "*** Variables ***",
+                Token.TESTCASE_HEADER: "*** Test Cases ***",
+                Token.KEYWORD_HEADER: "*** Keywords ***",
+                Token.COMMENT_HEADER: "*** Comments ***",
+            }[node.type]
         if self.uppercase:
-            normalized_name = normalized_name.upper()
+            name = name.upper()
         # we only modify header token value in order to preserver optional data driven testing column names
-        node.data_tokens[0].value = normalized_name
+        node.data_tokens[0].value = name
         return node

tests/atest/transformers/NormalizeSectionHeaderName/expected/task.robot

@@ -0,0 +1,3 @@
+*** Settings ***
+
+*** Tasks ***

tests/atest/transformers/NormalizeSectionHeaderName/source/task.robot

@@ -0,0 +1,3 @@
+*** Settings ***
+
+*** task ***

tests/atest/transformers/NormalizeSectionHeaderName/test_transformer.py

@@ -22,3 +22,6 @@ def test_normalize_names_selected(self):
             expected='selected.robot',
             config=' --startline 5 --endline 6'
         )
+
+    def test_tasks(self):
+        run_tidy_and_compare(self.TRANSFORMER_NAME, source='task.robot')