nginx-proxy-manager-zh/backend/internal/util/strings_test.go
2023-07-27 21:17:08 +10:00

68 lines
1.5 KiB
Go

package util
import (
"testing"
"github.com/stretchr/testify/assert"
)
func TestCleanupWhitespace(t *testing.T) {
tests := []struct {
name string
input string
want string
}{
{
name: "test a",
input: `# ------------------------------------------------------------
# Upstream 5: API servers 2
# ------------------------------------------------------------
upstream npm_upstream_5 {` + ` ` + /* this adds whitespace to the end without the ide trimming it */ `
server 192.168.0.10:80 weight=100 ;
server 192.168.0.11:80 weight=50 ;
}`,
want: `# ------------------------------------------------------------
# Upstream 5: API servers 2
# ------------------------------------------------------------
upstream npm_upstream_5 {
server 192.168.0.10:80 weight=100 ;
server 192.168.0.11:80 weight=50 ;
}`,
},
}
for _, tt := range tests {
t.Run(tt.name, func(t *testing.T) {
output := CleanupWhitespace(tt.input)
assert.Equal(t, tt.want, output)
})
}
}
func TestPrettyPrintJSON(t *testing.T) {
testCases := []struct {
input string
expected string
}{
{`{"name":"John","age":30,"city":"New York"}`, "{\n \"name\": \"John\",\n \"age\": 30,\n \"city\": \"New York\"\n}"},
{`{"fruit":"apple","color":"red"}`, "{\n \"fruit\": \"apple\",\n \"color\": \"red\"\n}"},
{"invalid-json", "invalid-json"}, // non-JSON input should return the original string unchanged
}
for _, tc := range testCases {
result := PrettyPrintJSON(tc.input)
assert.Equal(t, tc.expected, result)
}
}