test: add simple test for buffer:setline()

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
+    local append_line_index = self:line_count() + 1
-    if one_index == 2 and self[1] == '' then one_index = 1 end
+    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])

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

lua/test/demo_spec.lua

@@ -1,5 +0,0 @@
-describe('Test Demo', function()
-    it('should work', function()
-        print('hello')
-    end)
-end)

makefile

@@ -1,4 +1,4 @@
 .PHONE: test
 
 test:
-	nvim -headless -c 'TODO: '
+	@nvim --headless -c "lua require'plenary'" -c "PlenaryBustedDirectory lua/test/"