97 lines
2.8 KiB
Go
97 lines
2.8 KiB
Go
package processor
|
|
|
|
import (
|
|
"cook/utils"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func TestProcessJSON(t *testing.T) {
|
|
tests := []struct {
|
|
name string
|
|
input string
|
|
luaExpression string
|
|
expectedOutput string
|
|
expectedMods int
|
|
}{
|
|
{
|
|
name: "Basic JSON object modification",
|
|
input: `{"name": "test", "value": 42}`,
|
|
luaExpression: `data.value = data.value * 2; return true`,
|
|
expectedOutput: `{"name": "test", "value": 84}`,
|
|
expectedMods: 1,
|
|
},
|
|
{
|
|
name: "JSON array modification",
|
|
input: `{"items": [{"name": "item1", "value": 10}, {"name": "item2", "value": 20}]}`,
|
|
luaExpression: `for i, item in ipairs(data.items) do item.value = item.value * 2 end modified = true`,
|
|
expectedOutput: `{"items": [{"name": "item1", "value": 20}, {"name": "item2", "value": 40}]}`,
|
|
expectedMods: 2,
|
|
},
|
|
{
|
|
name: "JSON nested object modification",
|
|
input: `{"config": {"setting1": {"enabled": true, "value": 5}, "setting2": {"enabled": false, "value": 10}}}`,
|
|
luaExpression: `data.config.setting1.enabled = false data.config.setting2.value = 15 modified = true`,
|
|
expectedOutput: `{"config": {"setting1": {"enabled": false, "value": 5}, "setting2": {"enabled": false, "value": 15}}}`,
|
|
expectedMods: 2,
|
|
},
|
|
{
|
|
name: "JSON no modification",
|
|
input: `{"name": "test", "value": 42}`,
|
|
luaExpression: `return false`,
|
|
expectedOutput: `{"name": "test", "value": 42}`,
|
|
expectedMods: 0,
|
|
},
|
|
}
|
|
|
|
for _, tt := range tests {
|
|
t.Run(tt.name, func(t *testing.T) {
|
|
command := utils.ModifyCommand{
|
|
Name: tt.name,
|
|
JSON: true,
|
|
Lua: tt.luaExpression,
|
|
}
|
|
|
|
modifications, err := ProcessJSON(tt.input, command, "test.json")
|
|
assert.NoError(t, err, "ProcessJSON failed: %v", err)
|
|
|
|
if len(modifications) > 0 {
|
|
// Execute modifications
|
|
result, count := utils.ExecuteModifications(modifications, tt.input)
|
|
assert.Equal(t, tt.expectedMods, count, "Expected %d modifications, got %d", tt.expectedMods, count)
|
|
assert.Equal(t, tt.expectedOutput, result, "Expected output: %s, got: %s", tt.expectedOutput, result)
|
|
} else {
|
|
assert.Equal(t, 0, tt.expectedMods, "Expected no modifications but got some")
|
|
}
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestToLuaValue(t *testing.T) {
|
|
L, err := NewLuaState()
|
|
if err != nil {
|
|
t.Fatalf("Failed to create Lua state: %v", err)
|
|
}
|
|
defer L.Close()
|
|
|
|
tests := []struct {
|
|
name string
|
|
input interface{}
|
|
expected string
|
|
}{
|
|
{"string", "hello", "hello"},
|
|
{"number", 42.0, "42"},
|
|
{"boolean", true, "true"},
|
|
{"nil", nil, "nil"},
|
|
}
|
|
|
|
for _, tt := range tests {
|
|
t.Run(tt.name, func(t *testing.T) {
|
|
result, err := ToLuaValue(L, tt.input)
|
|
assert.NoError(t, err)
|
|
assert.Equal(t, tt.expected, result.String())
|
|
})
|
|
}
|
|
}
|