From eacb06cd308a876da252fb436e97ceec064fdcd6 Mon Sep 17 00:00:00 2001
From: JuanZoran <1430359574@qq.com>
Date: Mon, 27 Mar 2023 00:36:28 +0800
Subject: [PATCH] test: add simple test for buffer:setline()

---
 lua/Trans/core/buffer.lua | 12 +++++--
 lua/test/base_spec.lua    | 72 +++++++++++++++++++++++++++++++++++++++
 lua/test/demo_spec.lua    |  5 ---
 makefile                  |  2 +-
 4 files changed, 82 insertions(+), 9 deletions(-)
 create mode 100644 lua/test/base_spec.lua
 delete mode 100644 lua/test/demo_spec.lua

diff --git a/lua/Trans/core/buffer.lua b/lua/Trans/core/buffer.lua
index 3ca8b6f..f3ab25b 100644
--- a/lua/Trans/core/buffer.lua
+++ b/lua/Trans/core/buffer.lua
@@ -97,15 +97,21 @@ end
 ---@param nodes string|table|table[] string -> as line content | table -> as a node | table[] -> as node[]
 ---@param one_index number? line number should be set[one index] or let it be nil to append
 function buffer:setline(nodes, one_index)
-    one_index = one_index or self:line_count() + 1
-    if one_index == 2 and self[1] == '' then one_index = 1 end
+    local append_line_index = self:line_count() + 1
+    if append_line_index == 2 and self[1] == '' then append_line_index = 1 end
+    one_index = one_index or append_line_index
+    if one_index > append_line_index then
+        for i = append_line_index, one_index - 1 do
+            self:setline('', i)
+        end
+    end
+
 
     if type(nodes) == 'string' then
         fn.setbufline(self.bufnr, one_index, nodes)
         return
     end
 
-    -- FIXME :set [nodes] type as node
     if type(nodes[1]) == 'string' then
         ---@diagnostic disable-next-line: assign-type-mismatch, param-type-mismatch
         fn.setbufline(self.bufnr, one_index, nodes[1])
diff --git a/lua/test/base_spec.lua b/lua/test/base_spec.lua
new file mode 100644
index 0000000..a95e2b6
--- /dev/null
+++ b/lua/test/base_spec.lua
@@ -0,0 +1,72 @@
+local Trans = require 'Trans'
+local node = Trans.util.node
+local i, t, pr, f = node.item, node.text, node.prompt, node.format
+
+---@param func fun(buffer: TransBuffer)
+---@return fun()
+local function with_buffer(func)
+    return function()
+        local buffer = Trans.buffer.new()
+        func(buffer)
+        buffer:destroy()
+    end
+end
+
+
+describe('buffer:setline()', function()
+    it('and buffer[i] can accept a string as first arg', with_buffer(function(buffer)
+        buffer:setline 'hello world'
+        buffer[2] = 'hello world'
+        assert.are.equal(buffer[1], 'hello world')
+        assert.are.equal(buffer[2], 'hello world')
+    end))
+
+    it('and buffer[i] can accept a node as first arg', with_buffer(function(buffer)
+        buffer:setline(i { 'hello world' })
+        buffer[2] = i { 'hello world' }
+        assert.are.equal(buffer[1], 'hello world')
+        assert.are.equal(buffer[2], 'hello world')
+    end))
+
+    it('and buffer[i] can accept a node list as first arg', with_buffer(function(buffer)
+        buffer:setline {
+            i { 'hello ' },
+            i { 'world' },
+        }
+
+        buffer[2] = {
+            i { 'hello ' },
+            i { 'world' },
+        }
+
+        assert.are.equal(buffer[1], 'hello world')
+        assert.are.equal(buffer[2], 'hello world')
+    end))
+
+    it('and buffer[i] accept linenr more than line_count will fill with empty line', with_buffer(function(buffer)
+        buffer:setline('hello world', 3)
+        buffer[4] = 'hello world'
+        assert.are.equal(buffer[1], '')
+        assert.are.equal(buffer[2], '')
+        assert.are.equal(buffer[3], 'hello world')
+        assert.are.equal(buffer[4], 'hello world')
+    end))
+
+
+    it('can accept one index linenr as second arg', with_buffer(function(buffer)
+        buffer:setline({
+            i { 'hello ' },
+            i { 'world' },
+        }, 1)
+        assert.are.equal(buffer[1], 'hello world')
+    end))
+
+    it('when no second arg, it will append line', with_buffer(function(buffer)
+        buffer[1] = 'hello'
+        buffer:setline 'world'
+
+        assert.are.equal(buffer[2], 'world')
+    end))
+end)
+
+-- TODO :Add node test
diff --git a/lua/test/demo_spec.lua b/lua/test/demo_spec.lua
deleted file mode 100644
index c3a6de4..0000000
--- a/lua/test/demo_spec.lua
+++ /dev/null
@@ -1,5 +0,0 @@
-describe('Test Demo', function()
-    it('should work', function()
-        print('hello')
-    end)
-end)
diff --git a/makefile b/makefile
index 3a80591..b490585 100644
--- a/makefile
+++ b/makefile
@@ -1,4 +1,4 @@
 .PHONE: test
 
 test:
-	nvim -headless -c 'TODO: '
+	@nvim --headless -c "lua require'plenary'" -c "PlenaryBustedDirectory lua/test/"