test(f-string): add tests for inserts and not insterting fstrings

Author: joshzcold

lua/python/treesitter/commands.lua

@@ -209,7 +209,7 @@ function PythonTreeSitterCommands.ts_wrap_at_cursor(subtitute_option)
   end)
 end
 
---- 
+---
 ---@param node TSNode the current ts node we are checking for parents
 ---@return string callText check if this node has a "call" type node 3 parents up
 --- this is used for checking on "".format() calls for strings.
@@ -263,7 +263,7 @@ function PythonTreeSitterCommands.pythonFStr()
 
   local isFormatString = callText:find([[^.*["']%.format%(]])
   local isRString = text:find("^r")
-  local isFString = text:find("^r?f")           -- rf -> raw-formatted-string
+  local isFString = text:find("^r?f") -- rf -> raw-formatted-string
   local hasBraces = text:find("{.-[^%d,%s].-}") -- nonRegex-braces, see #12 and #15
 
   if (not isFString and not isFormatString and not isRString) and hasBraces then

tests/test_treesitter.lua

@@ -63,5 +63,31 @@ T["enumerate"] = function()
   eq(get_lines(), { "for idx, i in enumerate([1, 2, 3]):" })
 end
 
+T["f-string"] = MiniTest.new_set({
+  hooks = {
+    pre_case = function()
+      child.cmd("e _not_existing_new_buffer.py")
+      child.type_keys("cc", [["TEST"]], "<Esc>", "0")
+    end,
+  },
+})
+
+T["f-string"]["insert f string"] = function()
+  child.cmd("e! _not_existing_new_buffer.py")
+  child.type_keys("cc", [["{foo}"]], "<Esc>", "hh", "i", "<Esc>")
+  eq(get_lines(), { [[f"{foo}"]] })
+end
+
+T["f-string"]["skip on r"] = function()
+  child.cmd("e! _not_existing_new_buffer.py")
+  child.type_keys("cc", [[r"{foo}"]], "<Esc>", "hh", "i", "<Esc>")
+  eq(get_lines(), { [[r"{foo}"]] })
+end
+
+T["f-string"]["skip on format"] = function()
+  child.cmd("e! _not_existing_new_buffer.py")
+  child.type_keys("cc", [["{foo}".format()]], "<Esc>", "0lll", "i", "<Esc>")
+  eq(get_lines(), { [["{foo}".format()]] })
+end
 -- Return test set which will be collected and execute inside `MiniTest.run()`
 return T