seata
/
seata-go
Not watched
1
0
Fork 0
Code
Releases 11 Wiki evaluate Activity
Issues 0 Pull Requests 0 Datasets Model Cloudbrain HPC
You can not select more than 25 topics Topics must start with a chinese character,a letter or number, can include dashes ('-') and can be up to 35 characters long.
Tree: 05db834335
Branches Tags
${ item.name }
Create branch ${ searchTerm }
from '05db834335'
${ noResults }
seata-go/pkg/client/config_test.go

227 lines
14 kB
Raw Blame History 

  1. /*

  2.  * Licensed to the Apache Software Foundation (ASF) under one or more

  3.  * contributor license agreements.  See the NOTICE file distributed with

  4.  * this work for additional information regarding copyright ownership.

  5.  * The ASF licenses this file to You under the Apache License, Version 2.0

  6.  * (the "License"); you may not use this file except in compliance with

  7.  * the License.  You may obtain a copy of the License at

  8.  *

  9.  *     http://www.apache.org/licenses/LICENSE-2.0

  10.  *

  11.  * Unless required by applicable law or agreed to in writing, software

  12.  * distributed under the License is distributed on an "AS IS" BASIS,

  13.  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.

  14.  * See the License for the specific language governing permissions and

  15.  * limitations under the License.

  16.  */

  17. 

  18. package client

  19. 

  20. import (

  21. 	"flag"

  22. 	"os"

  23. 	"testing"

  24. 	"time"

  25. 

  26. 	"github.com/stretchr/testify/assert"

  27. )

  28. 

  29. func TestLoadPath(t *testing.T) {

  30. 	cfg := LoadPath("../../testdata/conf/seatago.yml")

  31. 	assert.NotNil(t, cfg)

  32. 	assert.Equal(t, true, cfg.Enabled)

  33. 	assert.Equal(t, "applicationName", cfg.ApplicationID)

  34. 	assert.Equal(t, "default_tx_group", cfg.TxServiceGroup)

  35. 	assert.Equal(t, "aliyunAccessKey", cfg.AccessKey)

  36. 	assert.Equal(t, "aliyunSecretKey", cfg.SecretKey)

  37. 	assert.Equal(t, true, cfg.EnableAutoDataSourceProxy)

  38. 	assert.Equal(t, "AT", cfg.DataSourceProxyMode)

  39. 

  40. 	assert.NotNil(t, cfg.TCCConfig)

  41. 	assert.NotNil(t, cfg.TCCConfig.FenceConfig)

  42. 	assert.Equal(t, false, cfg.TCCConfig.FenceConfig.Enable)

  43. 	assert.Equal(t, "root:12345678@tcp(127.0.0.1:3306)/seata_client1?charset=utf8&parseTime=True", cfg.TCCConfig.FenceConfig.Url)

  44. 	assert.Equal(t, "tcc_fence_log_test", cfg.TCCConfig.FenceConfig.LogTableName)

  45. 	assert.Equal(t, time.Second*60, cfg.TCCConfig.FenceConfig.CleanPeriod)

  46. 

  47. 	assert.NotNil(t, cfg.ClientConfig)

  48. 	assert.NotNil(t, cfg.ClientConfig.TmConfig)

  49. 	assert.Equal(t, 5, cfg.ClientConfig.TmConfig.CommitRetryCount)

  50. 	assert.Equal(t, 5, cfg.ClientConfig.TmConfig.RollbackRetryCount)

  51. 	assert.Equal(t, time.Second*60, cfg.ClientConfig.TmConfig.DefaultGlobalTransactionTimeout)

  52. 	assert.Equal(t, false, cfg.ClientConfig.TmConfig.DegradeCheck)

  53. 	assert.Equal(t, 2000, cfg.ClientConfig.TmConfig.DegradeCheckPeriod)

  54. 	assert.Equal(t, time.Second*10, cfg.ClientConfig.TmConfig.DegradeCheckAllowTimes)

  55. 	assert.Equal(t, -2147482648, cfg.ClientConfig.TmConfig.InterceptorOrder)

  56. 

  57. 	assert.Equal(t, 10000, cfg.ClientConfig.RmConfig.AsyncCommitBufferLimit)

  58. 	assert.Equal(t, 5, cfg.ClientConfig.RmConfig.ReportRetryCount)

  59. 	assert.Equal(t, false, cfg.ClientConfig.RmConfig.TableMetaCheckEnable)

  60. 	assert.Equal(t, false, cfg.ClientConfig.RmConfig.ReportSuccessEnable)

  61. 	assert.Equal(t, false, cfg.ClientConfig.RmConfig.SagaBranchRegisterEnable)

  62. 	assert.Equal(t, "fastjson", cfg.ClientConfig.RmConfig.SagaJsonParser)

  63. 	assert.Equal(t, false, cfg.ClientConfig.RmConfig.SagaCompensatePersistModeUpdate)

  64. 	assert.Equal(t, false, cfg.ClientConfig.RmConfig.SagaRetryPersistModeUpdate)

  65. 	assert.Equal(t, -2147482648, cfg.ClientConfig.RmConfig.TccActionInterceptorOrder)

  66. 	assert.Equal(t, "druid", cfg.ClientConfig.RmConfig.SqlParserType)

  67. 	assert.Equal(t, 30*time.Second, cfg.ClientConfig.RmConfig.LockConfig.RetryInterval)

  68. 	assert.Equal(t, 10, cfg.ClientConfig.RmConfig.LockConfig.RetryTimes)

  69. 	assert.Equal(t, true, cfg.ClientConfig.RmConfig.LockConfig.RetryPolicyBranchRollbackOnConflict)

  70. 

  71. 	assert.NotNil(t, cfg.ClientConfig.UndoConfig)

  72. 	assert.Equal(t, true, cfg.ClientConfig.UndoConfig.DataValidation)

  73. 	assert.Equal(t, "json", cfg.ClientConfig.UndoConfig.LogSerialization)

  74. 	assert.Equal(t, "undo_log", cfg.ClientConfig.UndoConfig.LogTable)

  75. 	assert.Equal(t, true, cfg.ClientConfig.UndoConfig.OnlyCareUpdateColumns)

  76. 	assert.NotNil(t, cfg.ClientConfig.UndoConfig.CompressConfig)

  77. 	assert.Equal(t, true, cfg.ClientConfig.UndoConfig.CompressConfig.Enable)

  78. 	assert.Equal(t, "zip", cfg.ClientConfig.UndoConfig.CompressConfig.Type)

  79. 	assert.Equal(t, "64k", cfg.ClientConfig.UndoConfig.CompressConfig.Threshold)

  80. 

  81. 	assert.NotNil(t, cfg.GettyConfig)

  82. 	assert.NotNil(t, cfg.GettyConfig.SessionConfig)

  83. 	assert.Equal(t, 0, cfg.GettyConfig.ReconnectInterval)

  84. 	assert.Equal(t, 1, cfg.GettyConfig.ConnectionNum)

  85. 	assert.Equal(t, false, cfg.GettyConfig.SessionConfig.CompressEncoding)

  86. 	assert.Equal(t, true, cfg.GettyConfig.SessionConfig.TCPNoDelay)

  87. 	assert.Equal(t, true, cfg.GettyConfig.SessionConfig.TCPKeepAlive)

  88. 	assert.Equal(t, time.Minute*2, cfg.GettyConfig.SessionConfig.KeepAlivePeriod)

  89. 	assert.Equal(t, 262144, cfg.GettyConfig.SessionConfig.TCPRBufSize)

  90. 	assert.Equal(t, 65536, cfg.GettyConfig.SessionConfig.TCPWBufSize)

  91. 	assert.Equal(t, time.Second, cfg.GettyConfig.SessionConfig.TCPReadTimeout)

  92. 	assert.Equal(t, time.Second*5, cfg.GettyConfig.SessionConfig.TCPWriteTimeout)

  93. 	assert.Equal(t, time.Second, cfg.GettyConfig.SessionConfig.WaitTimeout)

  94. 	assert.Equal(t, 16498688, cfg.GettyConfig.SessionConfig.MaxMsgLen)

  95. 	assert.Equal(t, "client_test", cfg.GettyConfig.SessionConfig.SessionName)

  96. 	assert.Equal(t, time.Second, cfg.GettyConfig.SessionConfig.CronPeriod)

  97. 

  98. 	assert.NotNil(t, cfg.TransportConfig)

  99. 	assert.NotNil(t, cfg.TransportConfig.ShutdownConfig)

  100. 	assert.Equal(t, time.Second*3, cfg.TransportConfig.ShutdownConfig.Wait)

  101. 	assert.Equal(t, "TCP", cfg.TransportConfig.Type)

  102. 	assert.Equal(t, "NIO", cfg.TransportConfig.Server)

  103. 	assert.Equal(t, true, cfg.TransportConfig.Heartbeat)

  104. 	assert.Equal(t, "seata", cfg.TransportConfig.Serialization)

  105. 	assert.Equal(t, "none", cfg.TransportConfig.Compressor)

  106. 	assert.Equal(t, false, cfg.TransportConfig.EnableTmClientBatchSendRequest)

  107. 	assert.Equal(t, true, cfg.TransportConfig.EnableRmClientBatchSendRequest)

  108. 	assert.Equal(t, time.Second*30, cfg.TransportConfig.RPCRmRequestTimeout)

  109. 	assert.Equal(t, time.Second*30, cfg.TransportConfig.RPCTmRequestTimeout)

  110. 

  111. 	assert.NotNil(t, cfg.ServiceConfig)

  112. 	assert.Equal(t, false, cfg.ServiceConfig.EnableDegrade)

  113. 	assert.Equal(t, false, cfg.ServiceConfig.DisableGlobalTransaction)

  114. 	assert.Equal(t, "default", cfg.ServiceConfig.VgroupMapping["default_tx_group"])

  115. 	assert.Equal(t, "127.0.0.1:8091", cfg.ServiceConfig.Grouplist["default"])

  116. 

  117. 	assert.NotNil(t, cfg.RegistryConfig)

  118. 	assert.Equal(t, "file", cfg.RegistryConfig.Type)

  119. 	assert.Equal(t, "seatago.yml", cfg.RegistryConfig.File.Name)

  120. 	assert.Equal(t, "seata-server", cfg.RegistryConfig.Nacos.Application)

  121. 	assert.Equal(t, "127.0.0.1:8848", cfg.RegistryConfig.Nacos.ServerAddr)

  122. 	assert.Equal(t, "SEATA_GROUP", cfg.RegistryConfig.Nacos.Group)

  123. 	assert.Equal(t, "test-namespace", cfg.RegistryConfig.Nacos.Namespace)

  124. 	assert.Equal(t, "test-username", cfg.RegistryConfig.Nacos.Username)

  125. 	assert.Equal(t, "test-password", cfg.RegistryConfig.Nacos.Password)

  126. 	assert.Equal(t, "test-access-key", cfg.RegistryConfig.Nacos.AccessKey)

  127. 	assert.Equal(t, "test-secret-key", cfg.RegistryConfig.Nacos.SecretKey)

  128. 	assert.Equal(t, "default", cfg.RegistryConfig.Etcd3.Cluster)

  129. 	assert.Equal(t, "http://localhost:2379", cfg.RegistryConfig.Etcd3.ServerAddr)

  130. 

  131. 	// reset flag.CommandLine

  132. 	flag.CommandLine = flag.NewFlagSet(os.Args[0], flag.ExitOnError)

  133. }

  134. 

  135. func TestLoadJson(t *testing.T) {

  136. 	confJson := `{"enabled":false,"application-id":"application_test","tx-service-group":"default_tx_group","access-key":"test","secret-key":"test","enable-auto-data-source-proxy":false,"data-source-proxy-mode":"AT","client":{"rm":{"async-commit-buffer-limit":10000,"report-retry-count":5,"table-meta-check-enable":false,"report-success-enable":false,"saga-branch-register-enable":false,"saga-json-parser":"fastjson","saga-retry-persist-mode-update":false,"saga-compensate-persist-mode-update":false,"tcc-action-interceptor-order":-2147482648,"sql-parser-type":"druid","lock":{"retry-interval":"30s","retry-times":10,"retry-policy-branch-rollback-on-conflict":true}},"tm":{"commit-retry-count":5,"rollback-retry-count":5,"default-global-transaction-timeout":"60s","degrade-check":false,"degrade-check-period":2000,"degrade-check-allow-times":"10s","interceptor-order":-2147482648},"undo":{"data-validation":false,"log-serialization":"jackson222","only-care-update-columns":false,"log-table":"undo_log333","compress":{"enable":false,"type":"zip111","threshold":"128k"}}},"tcc":{"fence":{"enable":false,"url":"root:12345678@tcp(127.0.0.1:3306)/seata_client1?charset=utf8&parseTime=True","log-table-name":"tcc_fence_log_test2","clean-period":80000000000}},"getty":{"reconnect-interval":1,"connection-num":10,"session":{"compress-encoding":true,"tcp-no-delay":false,"tcp-keep-alive":false,"keep-alive-period":"120s","tcp-r-buf-size":261120,"tcp-w-buf-size":32768,"tcp-read-timeout":"2s","tcp-write-timeout":"8s","wait-timeout":"2s","max-msg-len":261120,"session-name":"client_test","cron-period":"2s"}},"transport":{"shutdown":{"wait":"3s"},"type":"TCP","server":"NIO","heartbeat":true,"serialization":"seata","compressor":"none"," enable-tm-client-batch-send-request":false,"enable-rm-client-batch-send-request":true,"rpc-rm-request-timeout":"30s","rpc-tm-request-timeout":"30s"},"service":{"enable-degrade":true,"disable-global-transaction":true,"vgroup-mapping":{"default_tx_group":"default_test"},"grouplist":{"default":"127.0.0.1:8092"}}}`

  137. 	cfg := LoadJson([]byte(confJson))

  138. 	assert.NotNil(t, cfg)

  139. 	assert.Equal(t, false, cfg.Enabled)

  140. 	assert.Equal(t, "application_test", cfg.ApplicationID)

  141. 	assert.Equal(t, "default_tx_group", cfg.TxServiceGroup)

  142. 	assert.Equal(t, "test", cfg.AccessKey)

  143. 	assert.Equal(t, "test", cfg.SecretKey)

  144. 	assert.Equal(t, false, cfg.EnableAutoDataSourceProxy)

  145. 	assert.Equal(t, "AT", cfg.DataSourceProxyMode)

  146. 

  147. 	assert.Equal(t, 10000, cfg.ClientConfig.RmConfig.AsyncCommitBufferLimit)

  148. 	assert.Equal(t, 5, cfg.ClientConfig.RmConfig.ReportRetryCount)

  149. 	assert.Equal(t, false, cfg.ClientConfig.RmConfig.TableMetaCheckEnable)

  150. 	assert.Equal(t, false, cfg.ClientConfig.RmConfig.ReportSuccessEnable)

  151. 	assert.Equal(t, false, cfg.ClientConfig.RmConfig.SagaBranchRegisterEnable)

  152. 	assert.Equal(t, "fastjson", cfg.ClientConfig.RmConfig.SagaJsonParser)

  153. 	assert.Equal(t, false, cfg.ClientConfig.RmConfig.SagaCompensatePersistModeUpdate)

  154. 	assert.Equal(t, false, cfg.ClientConfig.RmConfig.SagaRetryPersistModeUpdate)

  155. 	assert.Equal(t, -2147482648, cfg.ClientConfig.RmConfig.TccActionInterceptorOrder)

  156. 	assert.Equal(t, "druid", cfg.ClientConfig.RmConfig.SqlParserType)

  157. 	assert.Equal(t, 30*time.Second, cfg.ClientConfig.RmConfig.LockConfig.RetryInterval)

  158. 	assert.Equal(t, 10, cfg.ClientConfig.RmConfig.LockConfig.RetryTimes)

  159. 	assert.Equal(t, true, cfg.ClientConfig.RmConfig.LockConfig.RetryPolicyBranchRollbackOnConflict)

  160. 

  161. 	assert.NotNil(t, cfg.ClientConfig.UndoConfig)

  162. 	assert.Equal(t, false, cfg.ClientConfig.UndoConfig.DataValidation)

  163. 	assert.Equal(t, "jackson222", cfg.ClientConfig.UndoConfig.LogSerialization)

  164. 	assert.Equal(t, "undo_log333", cfg.ClientConfig.UndoConfig.LogTable)

  165. 	assert.Equal(t, false, cfg.ClientConfig.UndoConfig.OnlyCareUpdateColumns)

  166. 	assert.NotNil(t, cfg.ClientConfig.UndoConfig.CompressConfig)

  167. 	assert.Equal(t, false, cfg.ClientConfig.UndoConfig.CompressConfig.Enable)

  168. 	assert.Equal(t, "zip111", cfg.ClientConfig.UndoConfig.CompressConfig.Type)

  169. 	assert.Equal(t, "128k", cfg.ClientConfig.UndoConfig.CompressConfig.Threshold)

  170. 

  171. 	assert.NotNil(t, cfg.TCCConfig)

  172. 	assert.NotNil(t, cfg.TCCConfig.FenceConfig)

  173. 	assert.Equal(t, false, cfg.TCCConfig.FenceConfig.Enable)

  174. 	assert.Equal(t, "root:12345678@tcp(127.0.0.1:3306)/seata_client1?charset=utf8&parseTime=True", cfg.TCCConfig.FenceConfig.Url)

  175. 	assert.Equal(t, "tcc_fence_log_test2", cfg.TCCConfig.FenceConfig.LogTableName)

  176. 	assert.Equal(t, time.Second*80, cfg.TCCConfig.FenceConfig.CleanPeriod)

  177. 

  178. 	assert.NotNil(t, cfg.ClientConfig)

  179. 	assert.NotNil(t, cfg.ClientConfig.TmConfig)

  180. 	assert.Equal(t, 5, cfg.ClientConfig.TmConfig.CommitRetryCount)

  181. 	assert.Equal(t, 5, cfg.ClientConfig.TmConfig.RollbackRetryCount)

  182. 	assert.Equal(t, time.Second*60, cfg.ClientConfig.TmConfig.DefaultGlobalTransactionTimeout)

  183. 	assert.Equal(t, false, cfg.ClientConfig.TmConfig.DegradeCheck)

  184. 	assert.Equal(t, 2000, cfg.ClientConfig.TmConfig.DegradeCheckPeriod)

  185. 	assert.Equal(t, time.Second*10, cfg.ClientConfig.TmConfig.DegradeCheckAllowTimes)

  186. 	assert.Equal(t, -2147482648, cfg.ClientConfig.TmConfig.InterceptorOrder)

  187. 

  188. 	assert.NotNil(t, cfg.GettyConfig)

  189. 	assert.NotNil(t, cfg.GettyConfig.SessionConfig)

  190. 	assert.Equal(t, 1, cfg.GettyConfig.ReconnectInterval)

  191. 	assert.Equal(t, 10, cfg.GettyConfig.ConnectionNum)

  192. 	assert.Equal(t, true, cfg.GettyConfig.SessionConfig.CompressEncoding)

  193. 	assert.Equal(t, false, cfg.GettyConfig.SessionConfig.TCPNoDelay)

  194. 	assert.Equal(t, false, cfg.GettyConfig.SessionConfig.TCPKeepAlive)

  195. 	assert.Equal(t, time.Minute*2, cfg.GettyConfig.SessionConfig.KeepAlivePeriod)

  196. 	assert.Equal(t, 261120, cfg.GettyConfig.SessionConfig.TCPRBufSize)

  197. 	assert.Equal(t, 32768, cfg.GettyConfig.SessionConfig.TCPWBufSize)

  198. 	assert.Equal(t, time.Second*2, cfg.GettyConfig.SessionConfig.TCPReadTimeout)

  199. 	assert.Equal(t, time.Second*8, cfg.GettyConfig.SessionConfig.TCPWriteTimeout)

  200. 	assert.Equal(t, time.Second*2, cfg.GettyConfig.SessionConfig.WaitTimeout)

  201. 	assert.Equal(t, 261120, cfg.GettyConfig.SessionConfig.MaxMsgLen)

  202. 	assert.Equal(t, "client_test", cfg.GettyConfig.SessionConfig.SessionName)

  203. 	assert.Equal(t, time.Second*2, cfg.GettyConfig.SessionConfig.CronPeriod)

  204. 

  205. 	assert.NotNil(t, cfg.TransportConfig)

  206. 	assert.NotNil(t, cfg.TransportConfig.ShutdownConfig)

  207. 	assert.Equal(t, time.Second*3, cfg.TransportConfig.ShutdownConfig.Wait)

  208. 	assert.Equal(t, "TCP", cfg.TransportConfig.Type)

  209. 	assert.Equal(t, "NIO", cfg.TransportConfig.Server)

  210. 	assert.Equal(t, true, cfg.TransportConfig.Heartbeat)

  211. 	assert.Equal(t, "seata", cfg.TransportConfig.Serialization)

  212. 	assert.Equal(t, "none", cfg.TransportConfig.Compressor)

  213. 	assert.Equal(t, false, cfg.TransportConfig.EnableTmClientBatchSendRequest)

  214. 	assert.Equal(t, true, cfg.TransportConfig.EnableRmClientBatchSendRequest)

  215. 	assert.Equal(t, time.Second*30, cfg.TransportConfig.RPCRmRequestTimeout)

  216. 	assert.Equal(t, time.Second*30, cfg.TransportConfig.RPCTmRequestTimeout)

  217. 

  218. 	assert.NotNil(t, cfg.ServiceConfig)

  219. 	assert.Equal(t, true, cfg.ServiceConfig.EnableDegrade)

  220. 	assert.Equal(t, true, cfg.ServiceConfig.DisableGlobalTransaction)

  221. 	assert.Equal(t, "default_test", cfg.ServiceConfig.VgroupMapping["default_tx_group"])

  222. 	assert.Equal(t, "127.0.0.1:8092", cfg.ServiceConfig.Grouplist["default"])

  223. 

  224. 	// reset flag.CommandLine

  225. 	flag.CommandLine = flag.NewFlagSet(os.Args[0], flag.ExitOnError)

  226. }