Test configuration loading more precisely

pkg/config/load_test.go

@@ -22,9 +22,7 @@ import (
 	v1 "github.com/fatedier/frp/pkg/config/v1"
 )
 
-func TestLoadConfigure(t *testing.T) {
+const tomlServerContent = `
-	require := require.New(t)
-	content := `
 bindAddr = "127.0.0.1"
 kcpBindPort = 7000
 quicBindPort = 7001
@@ -33,8 +31,34 @@ custom404Page = "/abc.html"
 transport.tcpKeepalive = 10
 `
 
+const yamlServerContent = `
+bindAddr: 127.0.0.1
+kcpBindPort: 7000
+quicBindPort: 7001
+tcpmuxHTTPConnectPort: 7005
+custom404Page: /abc.html
+transport:
+  tcpKeepalive: 10
+`
+
+const jsonServerContent = `
+{
+  "bindAddr": "127.0.0.1",
+  "kcpBindPort": 7000,
+  "quicBindPort": 7001,
+  "tcpmuxHTTPConnectPort": 7005,
+  "custom404Page": "/abc.html",
+  "transport": {
+    "tcpKeepalive": 10
+  }
+}
+`
+
+func TestLoadServerConfig(t *testing.T) {
+	for _, content := range []string{tomlServerContent, yamlServerContent, jsonServerContent} {
 		svrCfg := v1.ServerConfig{}
 		err := LoadConfigure([]byte(content), &svrCfg)
+		require := require.New(t)
 		require.NoError(err)
 		require.EqualValues("127.0.0.1", svrCfg.BindAddr)
 		require.EqualValues(7000, svrCfg.KCPBindPort)
@@ -42,4 +66,5 @@ transport.tcpKeepalive = 10
 		require.EqualValues(7005, svrCfg.TCPMuxHTTPConnectPort)
 		require.EqualValues("/abc.html", svrCfg.Custom404Page)
 		require.EqualValues(10, svrCfg.Transport.TCPKeepAlive)
+	}
 }