From 5aa0a87032d717adf283758a9a3e54a73064ed5a Mon Sep 17 00:00:00 2001
From: Kamal Sacranie <kamalsjunkmail@gmail.com>
Date: Fri, 19 Jan 2024 17:03:07 +0000
Subject: [PATCH] Add tests for treesitter context in markdown sample

- Add 'ts_context markdown `*` success md_context'
- Add 'ts_context codeblock `*` fail js_context'
---
 lua/nvim-autopairs/ts-conds.lua | 11 +++++------
 tests/endwise/sample.md         |  9 +++++++++
 tests/test_utils.lua            |  4 ++++
 tests/treesitter_spec.lua       | 32 +++++++++++++++++++++++++++++++-
 4 files changed, 49 insertions(+), 7 deletions(-)
 create mode 100644 tests/endwise/sample.md

diff --git a/lua/nvim-autopairs/ts-conds.lua b/lua/nvim-autopairs/ts-conds.lua
index a6f62f24..d7ef8e0e 100644
--- a/lua/nvim-autopairs/ts-conds.lua
+++ b/lua/nvim-autopairs/ts-conds.lua
@@ -148,12 +148,11 @@ conds.is_not_ts_node_comment = function()
 end
 
 conds.is_not_in_context = function()
-    return function()
-        local rules = require("nvim-autopairs").get_buf_rules()
-        for _, rule in pairs(rules) do
-            local context = require("nvim-autopairs.ts-utils")
-                .get_language_tree_at_position({ utils.get_cursor() })
-            return vim.tbl_contains(rule.filetypes, context:lang())
+    return function(opts)
+        local context = require("nvim-autopairs.ts-utils")
+            .get_language_tree_at_position({ utils.get_cursor() })
+        if not vim.tbl_contains(opts.rule.filetypes, context:lang()) then
+            return false
         end
     end
 end
diff --git a/tests/endwise/sample.md b/tests/endwise/sample.md
new file mode 100644
index 00000000..490b6d83
--- /dev/null
+++ b/tests/endwise/sample.md
@@ -0,0 +1,9 @@
+# Example Markdown File
+
+```javascript
+let;
+let;
+let;
+let;
+let;
+```
diff --git a/tests/test_utils.lua b/tests/test_utils.lua
index 02b9a6ee..e259d99e 100644
--- a/tests/test_utils.lua
+++ b/tests/test_utils.lua
@@ -133,6 +133,10 @@ _G.Test_withfile = function(test_data, cb)
                     vim.bo.filetype = value.filetype
                 end
             end
+            local status, parser = pcall(vim.treesitter.get_parser, 0)
+            if status then
+                parser:parse(true)
+            end
             vim.api.nvim_buf_set_lines(
                 0,
                 value.linenr - 1,
diff --git a/tests/treesitter_spec.lua b/tests/treesitter_spec.lua
index 4c877731..d86dd61a 100644
--- a/tests/treesitter_spec.lua
+++ b/tests/treesitter_spec.lua
@@ -13,7 +13,7 @@ vim.api.nvim_set_keymap(
 )
 
 ts.setup({
-    ensure_installed = { 'lua', 'javascript', 'rust' },
+    ensure_installed = { 'lua', 'javascript', 'rust', 'markdown', 'markdown_inline' },
     highlight = { enable = true },
     autopairs = { enable = true },
 })
@@ -93,6 +93,36 @@ local data = {
         before = [[pub fn noop(_inp: Vec|) {]],
         after = [[pub fn noop(_inp: Vec<|>) {]],
     },
+    {
+        setup_func = function()
+            npairs.add_rules({
+                Rule('*', '*', { 'markdown', 'markdown_inline' })
+                    :with_pair(ts_conds.is_not_in_context()),
+            })
+        end,
+        name = 'ts_context markdown `*` success md_context',
+        filepath = './tests/endwise/sample.md',
+        linenr = 2,
+        filetype = 'markdown',
+        key = '*',
+        before = [[|]],
+        after = [[*|*]],
+    },
+    {
+        setup_func = function()
+            npairs.add_rules({
+                Rule('*', '*', { 'markdown', 'markdown_inline' })
+                    :with_pair(ts_conds.is_not_in_context()),
+            })
+        end,
+        name = 'ts_context codeblock `*` fail js_context',
+        filepath = './tests/endwise/sample.md',
+        linenr = 6,
+        filetype = 'markdown',
+        key = '*',
+        before = [[let calc = 1  |]],
+        after = [[let calc = 1 *|]],
+    },
 }
 
 local run_data = _G.Test_filter(data)