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,13 +31,40 @@ custom404Page = "/abc.html"
 transport.tcpKeepalive = 10
 `
 
-	svrCfg := v1.ServerConfig{}
+const yamlServerContent = `
-	err := LoadConfigure([]byte(content), &svrCfg)
+bindAddr: 127.0.0.1
-	require.NoError(err)
+kcpBindPort: 7000
-	require.EqualValues("127.0.0.1", svrCfg.BindAddr)
+quicBindPort: 7001
-	require.EqualValues(7000, svrCfg.KCPBindPort)
+tcpmuxHTTPConnectPort: 7005
-	require.EqualValues(7001, svrCfg.QUICBindPort)
+custom404Page: /abc.html
-	require.EqualValues(7005, svrCfg.TCPMuxHTTPConnectPort)
+transport:
-	require.EqualValues("/abc.html", svrCfg.Custom404Page)
+  tcpKeepalive: 10
-	require.EqualValues(10, svrCfg.Transport.TCPKeepAlive)
+`
+
+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)
+		require.EqualValues(7001, svrCfg.QUICBindPort)
+		require.EqualValues(7005, svrCfg.TCPMuxHTTPConnectPort)
+		require.EqualValues("/abc.html", svrCfg.Custom404Page)
+		require.EqualValues(10, svrCfg.Transport.TCPKeepAlive)
+	}
 }