From 89e2daf2cddfc5106c010b76b8eb9aef51638867 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/test/base_spec.lua | 72 ++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 72 insertions(+)
 create mode 100644 lua/test/base_spec.lua

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