client_test.go 22 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673
  1. /*
  2. * Licensed under the Apache License, Version 2.0 (the "License");
  3. * you may not use this file except in compliance with the License.
  4. * You may obtain a copy of the License at
  5. *
  6. * http://www.apache.org/licenses/LICENSE-2.0
  7. *
  8. * Unless required by applicable law or agreed to in writing, software
  9. * distributed under the License is distributed on an "AS IS" BASIS,
  10. * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  11. * See the License for the specific language governing permissions and
  12. * limitations under the License.
  13. */
  14. package sdk
  15. import (
  16. "bytes"
  17. "fmt"
  18. "io/ioutil"
  19. "net/http"
  20. "net/http/httptest"
  21. "os"
  22. "strconv"
  23. "strings"
  24. "testing"
  25. "time"
  26. "github.com/aliyun/alibaba-cloud-sdk-go/sdk/auth/credentials/provider"
  27. "github.com/aliyun/alibaba-cloud-sdk-go/sdk/responses"
  28. "github.com/aliyun/alibaba-cloud-sdk-go/sdk/auth/credentials"
  29. "github.com/aliyun/alibaba-cloud-sdk-go/sdk/requests"
  30. "github.com/stretchr/testify/assert"
  31. )
  32. type signertest struct {
  33. name string
  34. }
  35. func (s *signertest) GetName() string {
  36. return ""
  37. }
  38. func (s *signertest) GetType() string {
  39. return ""
  40. }
  41. func (s *signertest) GetVersion() string {
  42. return ""
  43. }
  44. func (s *signertest) GetAccessKeyId() (string, error) {
  45. return "", nil
  46. }
  47. func (s *signertest) GetExtraParam() map[string]string {
  48. return nil
  49. }
  50. func (s *signertest) Sign(stringToSign, secretSuffix string) string {
  51. return ""
  52. }
  53. func Test_Client(t *testing.T) {
  54. defer func() {
  55. err := recover()
  56. assert.NotNil(t, err)
  57. assert.Equal(t, "not support yet", err)
  58. }()
  59. NewClient()
  60. }
  61. func Test_NewClientWithOptions(t *testing.T) {
  62. c := NewConfig()
  63. c.HttpTransport = &http.Transport{
  64. IdleConnTimeout: time.Duration(10 * time.Second),
  65. }
  66. c.EnableAsync = true
  67. c.GoRoutinePoolSize = 1
  68. c.MaxTaskQueueSize = 1
  69. credential := credentials.NewAccessKeyCredential("acesskeyid", "accesskeysecret")
  70. client, err := NewClientWithOptions("regionid", c, credential)
  71. assert.Nil(t, err)
  72. assert.NotNil(t, client)
  73. }
  74. func Test_NewClientWithPolicy(t *testing.T) {
  75. client, err := NewClientWithRamRoleArnAndPolicy("regionid", "acesskeyid", "accesskeysecret", "roleArn", "sessionName", "policy")
  76. assert.Nil(t, err)
  77. assert.NotNil(t, client)
  78. }
  79. func Test_NewClientWithAccessKey(t *testing.T) {
  80. client, err := NewClientWithAccessKey("regionid", "acesskeyid", "accesskeysecret")
  81. assert.Nil(t, err)
  82. assert.NotNil(t, client)
  83. }
  84. func Test_NewClientWithStsToken(t *testing.T) {
  85. client, err := NewClientWithStsToken("regionid", "acesskeyid", "accesskeysecret", "token")
  86. assert.Nil(t, err)
  87. assert.NotNil(t, client)
  88. }
  89. func Test_NewClientWithRamRoleArn(t *testing.T) {
  90. client, err := NewClientWithRamRoleArn("regionid", "acesskeyid", "accesskeysecret", "roleArn", "roleSessionName")
  91. assert.Nil(t, err)
  92. assert.NotNil(t, client)
  93. config := client.InitClientConfig()
  94. assert.NotNil(t, config)
  95. }
  96. func Test_NewClientWithEcsRamRole(t *testing.T) {
  97. client, err := NewClientWithEcsRamRole("regionid", "roleName")
  98. assert.Nil(t, err)
  99. assert.NotNil(t, client)
  100. }
  101. func Test_NewClientWithRsaKeyPair(t *testing.T) {
  102. client, err := NewClientWithRsaKeyPair("regionid", "publicKey", "privateKey", 3600)
  103. assert.Nil(t, err)
  104. assert.NotNil(t, client)
  105. }
  106. func mockResponse(statusCode int, content string, mockerr error) (res *http.Response, err error) {
  107. status := strconv.Itoa(statusCode)
  108. res = &http.Response{
  109. Proto: "HTTP/1.1",
  110. ProtoMajor: 1,
  111. Header: make(http.Header),
  112. StatusCode: statusCode,
  113. Status: status + " " + http.StatusText(statusCode),
  114. }
  115. res.Body = ioutil.NopCloser(bytes.NewReader([]byte(content)))
  116. err = mockerr
  117. return
  118. }
  119. func Test_DoActionWithProxy(t *testing.T) {
  120. client, err := NewClientWithAccessKey("regionid", "acesskeyid", "accesskeysecret")
  121. assert.Nil(t, err)
  122. assert.NotNil(t, client)
  123. assert.Equal(t, true, client.isRunning)
  124. request := requests.NewCommonRequest()
  125. request.Domain = "ecs.aliyuncs.com"
  126. request.Version = "2014-05-26"
  127. request.ApiName = "DescribeInstanceStatus"
  128. request.QueryParams["PageNumber"] = "1"
  129. request.QueryParams["PageSize"] = "30"
  130. request.TransToAcsRequest()
  131. response := responses.NewCommonResponse()
  132. origTestHookDo := hookDo
  133. defer func() { hookDo = origTestHookDo }()
  134. hookDo = func(fn func(req *http.Request) (*http.Response, error)) func(req *http.Request) (*http.Response, error) {
  135. return func(req *http.Request) (*http.Response, error) {
  136. return mockResponse(200, "", nil)
  137. }
  138. }
  139. err = client.DoAction(request, response)
  140. assert.Nil(t, err)
  141. assert.Equal(t, 200, response.GetHttpStatus())
  142. assert.Equal(t, "", response.GetHttpContentString())
  143. // Test when scheme is http, only http proxy is valid.
  144. envHttpsProxy := os.Getenv("https_proxy")
  145. os.Setenv("https_proxy", "https://127.0.0.1:9000")
  146. err = client.DoAction(request, response)
  147. assert.Nil(t, err)
  148. trans, _ := client.httpClient.Transport.(*http.Transport)
  149. assert.Nil(t, trans.Proxy)
  150. // Test when host is in no_proxy, proxy is invalid
  151. envNoProxy := os.Getenv("no_proxy")
  152. os.Setenv("no_proxy", "ecs.aliyuncs.com")
  153. envHttpProxy := os.Getenv("http_proxy")
  154. os.Setenv("http_proxy", "http://127.0.0.1:8888")
  155. err = client.DoAction(request, response)
  156. assert.Nil(t, err)
  157. trans, _ = client.httpClient.Transport.(*http.Transport)
  158. assert.Nil(t, trans.Proxy)
  159. client.SetNoProxy("ecs.testaliyuncs.com")
  160. err = client.DoAction(request, response)
  161. assert.Nil(t, err)
  162. trans, _ = client.httpClient.Transport.(*http.Transport)
  163. url, _ := trans.Proxy(nil)
  164. assert.Equal(t, url.Scheme, "http")
  165. assert.Equal(t, url.Host, "127.0.0.1:8888")
  166. // Test when setting http proxy, client has a high priority than environment variable
  167. client.SetHttpProxy("http://127.0.0.1:8080")
  168. err = client.DoAction(request, response)
  169. assert.Nil(t, err)
  170. trans, _ = client.httpClient.Transport.(*http.Transport)
  171. url, _ = trans.Proxy(nil)
  172. assert.Equal(t, url.Scheme, "http")
  173. assert.Equal(t, url.Host, "127.0.0.1:8080")
  174. // Test when scheme is https, only https proxy is valid
  175. request.Scheme = "https"
  176. err = client.DoAction(request, response)
  177. assert.Nil(t, err)
  178. trans, _ = client.httpClient.Transport.(*http.Transport)
  179. url, _ = trans.Proxy(nil)
  180. assert.Equal(t, url.Scheme, "https")
  181. assert.Equal(t, url.Host, "127.0.0.1:9000")
  182. // Test when setting https proxy, client has a high priority than environment variable
  183. client.SetHttpsProxy("https://127.0.0.1:6666")
  184. err = client.DoAction(request, response)
  185. assert.Nil(t, err)
  186. trans, _ = client.httpClient.Transport.(*http.Transport)
  187. url, _ = trans.Proxy(nil)
  188. assert.Equal(t, url.Scheme, "https")
  189. assert.Equal(t, url.Host, "127.0.0.1:6666")
  190. client.Shutdown()
  191. os.Setenv("https_proxy", envHttpsProxy)
  192. os.Setenv("http_proxy", envHttpProxy)
  193. os.Setenv("no_proxy", envNoProxy)
  194. assert.Equal(t, false, client.isRunning)
  195. }
  196. func Test_DoAction_HTTPSInsecure(t *testing.T) {
  197. client, err := NewClientWithAccessKey("regionid", "acesskeyid", "accesskeysecret")
  198. assert.Nil(t, err)
  199. assert.NotNil(t, client)
  200. client.SetHTTPSInsecure(true)
  201. request := requests.NewCommonRequest()
  202. request.Product = "Ram"
  203. request.Version = "2015-05-01"
  204. request.ApiName = "CreateRole"
  205. request.Domain = "ecs.aliyuncs.com"
  206. request.QueryParams["RegionId"] = os.Getenv("REGION_ID")
  207. request.TransToAcsRequest()
  208. response := responses.NewCommonResponse()
  209. origTestHookDo := hookDo
  210. defer func() { hookDo = origTestHookDo }()
  211. hookDo = func(fn func(req *http.Request) (*http.Response, error)) func(req *http.Request) (*http.Response, error) {
  212. return func(req *http.Request) (*http.Response, error) {
  213. return mockResponse(200, "", nil)
  214. }
  215. }
  216. err = client.DoAction(request, response)
  217. assert.Nil(t, err)
  218. assert.Equal(t, 200, response.GetHttpStatus())
  219. assert.Equal(t, "", response.GetHttpContentString())
  220. trans := client.httpClient.Transport.(*http.Transport)
  221. assert.Equal(t, true, trans.TLSClientConfig.InsecureSkipVerify)
  222. request.SetHTTPSInsecure(false)
  223. err = client.DoAction(request, response)
  224. assert.Nil(t, err)
  225. trans = client.httpClient.Transport.(*http.Transport)
  226. assert.Equal(t, false, trans.TLSClientConfig.InsecureSkipVerify)
  227. // Test when scheme is http, only http proxy is valid.
  228. envHttpsProxy := os.Getenv("HTTPS_PROXY")
  229. os.Setenv("HTTPS_PROXY", "https://127.0.0.1:9000")
  230. err = client.DoAction(request, response)
  231. assert.Nil(t, err)
  232. trans, _ = client.httpClient.Transport.(*http.Transport)
  233. assert.Nil(t, trans.Proxy)
  234. // Test when host is in no_proxy, proxy is invalid
  235. envNoProxy := os.Getenv("NO_PROXY")
  236. os.Setenv("NO_PROXY", "ecs.aliyuncs.com")
  237. envHttpProxy := os.Getenv("HTTP_PROXY")
  238. os.Setenv("HTTP_PROXY", "http://127.0.0.1:8888")
  239. err = client.DoAction(request, response)
  240. assert.Nil(t, err)
  241. trans, _ = client.httpClient.Transport.(*http.Transport)
  242. assert.Nil(t, trans.Proxy)
  243. client.SetNoProxy("ecs.testaliyuncs.com")
  244. err = client.DoAction(request, response)
  245. assert.Nil(t, err)
  246. trans, _ = client.httpClient.Transport.(*http.Transport)
  247. url, _ := trans.Proxy(nil)
  248. assert.Equal(t, url.Scheme, "http")
  249. assert.Equal(t, url.Host, "127.0.0.1:8888")
  250. // Test when setting http proxy, client has a high priority than environment variable
  251. client.SetHttpProxy("http://127.0.0.1:8080")
  252. err = client.DoAction(request, response)
  253. assert.Nil(t, err)
  254. trans, _ = client.httpClient.Transport.(*http.Transport)
  255. url, _ = trans.Proxy(nil)
  256. assert.Equal(t, url.Scheme, "http")
  257. assert.Equal(t, url.Host, "127.0.0.1:8080")
  258. // Test when scheme is https, only https proxy is valid
  259. request.Scheme = "https"
  260. err = client.DoAction(request, response)
  261. assert.Nil(t, err)
  262. trans, _ = client.httpClient.Transport.(*http.Transport)
  263. url, _ = trans.Proxy(nil)
  264. assert.Equal(t, url.Scheme, "https")
  265. assert.Equal(t, url.Host, "127.0.0.1:9000")
  266. // Test when setting https proxy, client has a high priority than environment variable
  267. client.SetHttpsProxy("https://127.0.0.1:6666")
  268. err = client.DoAction(request, response)
  269. assert.Nil(t, err)
  270. trans, _ = client.httpClient.Transport.(*http.Transport)
  271. url, _ = trans.Proxy(nil)
  272. assert.Equal(t, url.Scheme, "https")
  273. assert.Equal(t, url.Host, "127.0.0.1:6666")
  274. client.Shutdown()
  275. os.Setenv("HTTPS_PROXY", envHttpsProxy)
  276. os.Setenv("HTTP_PROXY", envHttpProxy)
  277. os.Setenv("NO_PROXY", envNoProxy)
  278. assert.Equal(t, false, client.isRunning)
  279. }
  280. func Test_DoAction_Timeout(t *testing.T) {
  281. client, err := NewClientWithAccessKey("regionid", "acesskeyid", "accesskeysecret")
  282. assert.Nil(t, err)
  283. assert.NotNil(t, client)
  284. assert.Equal(t, true, client.isRunning)
  285. request := requests.NewCommonRequest()
  286. request.Domain = "ecs.aliyuncs.com"
  287. request.Version = "2014-05-26"
  288. request.ApiName = "DescribeInstanceStatus"
  289. request.QueryParams["PageNumber"] = "1"
  290. request.QueryParams["PageSize"] = "30"
  291. request.TransToAcsRequest()
  292. response := responses.NewCommonResponse()
  293. origTestHookDo := hookDo
  294. defer func() { hookDo = origTestHookDo }()
  295. hookDo = func(fn func(req *http.Request) (*http.Response, error)) func(req *http.Request) (*http.Response, error) {
  296. return func(req *http.Request) (*http.Response, error) {
  297. return mockResponse(400, "Server Internel Error", fmt.Errorf("read tcp"))
  298. }
  299. }
  300. err = client.DoAction(request, response)
  301. assert.NotNil(t, err)
  302. assert.Equal(t, 0, response.GetHttpStatus())
  303. assert.Equal(t, "", response.GetHttpContentString())
  304. // Test set client timeout
  305. client.SetReadTimeout(1 * time.Millisecond)
  306. client.SetConnectTimeout(1 * time.Millisecond)
  307. assert.Equal(t, 1*time.Millisecond, client.GetConnectTimeout())
  308. assert.Equal(t, 1*time.Millisecond, client.GetReadTimeout())
  309. client.config.AutoRetry = false
  310. err = client.DoAction(request, response)
  311. assert.NotNil(t, err)
  312. assert.Equal(t, 0, response.GetHttpStatus())
  313. assert.Equal(t, "", response.GetHttpContentString())
  314. // Test set request timeout
  315. request.SetReadTimeout(1 * time.Millisecond)
  316. request.SetConnectTimeout(1 * time.Millisecond)
  317. err = client.DoAction(request, response)
  318. assert.NotNil(t, err)
  319. assert.Equal(t, 0, response.GetHttpStatus())
  320. assert.Equal(t, "", response.GetHttpContentString())
  321. client.Shutdown()
  322. assert.Equal(t, false, client.isRunning)
  323. }
  324. func Test_ProcessCommonRequest(t *testing.T) {
  325. client, err := NewClientWithAccessKey("regionid", "acesskeyid", "accesskeysecret")
  326. assert.Nil(t, err)
  327. assert.NotNil(t, client)
  328. request := requests.NewCommonRequest()
  329. request.Domain = "ecs.aliyuncs.com"
  330. request.Version = "2014-05-26"
  331. request.ApiName = "DescribeInstanceStatus"
  332. request.QueryParams["PageNumber"] = "1"
  333. request.QueryParams["PageSize"] = "30"
  334. origTestHookDo := hookDo
  335. defer func() { hookDo = origTestHookDo }()
  336. hookDo = func(fn func(req *http.Request) (*http.Response, error)) func(req *http.Request) (*http.Response, error) {
  337. return func(req *http.Request) (*http.Response, error) {
  338. return mockResponse(400, "", fmt.Errorf("test error"))
  339. }
  340. }
  341. resp, err := client.ProcessCommonRequest(request)
  342. assert.NotNil(t, err)
  343. assert.Contains(t, err.Error(), "test error")
  344. assert.Equal(t, 0, resp.GetHttpStatus())
  345. assert.Equal(t, "", resp.GetHttpContentString())
  346. }
  347. func mockServer(status int, json string) (server *httptest.Server) {
  348. // Start a test server locally.
  349. ts := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
  350. w.WriteHeader(status)
  351. w.Write([]byte(json))
  352. return
  353. }))
  354. return ts
  355. }
  356. func Test_DoAction_With500(t *testing.T) {
  357. client, err := NewClientWithAccessKey("regionid", "acesskeyid", "accesskeysecret")
  358. assert.Nil(t, err)
  359. assert.NotNil(t, client)
  360. assert.Equal(t, true, client.isRunning)
  361. request := requests.NewCommonRequest()
  362. request.Version = "2014-05-26"
  363. request.ApiName = "DescribeInstanceStatus"
  364. request.QueryParams["PageNumber"] = "1"
  365. request.QueryParams["PageSize"] = "30"
  366. request.TransToAcsRequest()
  367. response := responses.NewCommonResponse()
  368. ts := mockServer(500, "Server Internel Error")
  369. defer ts.Close()
  370. domain := strings.Replace(ts.URL, "http://", "", 1)
  371. request.Domain = domain
  372. err = client.DoAction(request, response)
  373. assert.NotNil(t, err)
  374. assert.Equal(t, 500, response.GetHttpStatus())
  375. assert.Equal(t, "Server Internel Error", response.GetHttpContentString())
  376. }
  377. func TestClient_BuildRequestWithSigner(t *testing.T) {
  378. client, err := NewClientWithAccessKey("regionid", "acesskeyid", "accesskeysecret")
  379. assert.Nil(t, err)
  380. assert.NotNil(t, client)
  381. assert.Equal(t, true, client.isRunning)
  382. request := requests.NewCommonRequest()
  383. request.Domain = "ecs.aliyuncs.com"
  384. request.Version = "2014-05-26"
  385. request.ApiName = "DescribeInstanceStatus"
  386. request.QueryParams["PageNumber"] = "1"
  387. request.QueryParams["PageSize"] = "30"
  388. request.RegionId = "regionid"
  389. request.TransToAcsRequest()
  390. client.config.UserAgent = "user_agent"
  391. err = client.BuildRequestWithSigner(request, nil)
  392. assert.Nil(t, err)
  393. }
  394. func TestClient_BuildRequestWithSigner1(t *testing.T) {
  395. client, err := NewClientWithAccessKey("regionid", "acesskeyid", "accesskeysecret")
  396. assert.Nil(t, err)
  397. assert.NotNil(t, client)
  398. assert.Equal(t, true, client.isRunning)
  399. request := requests.NewCommonRequest()
  400. request.Domain = "ecs.aliyuncs.com"
  401. request.Version = "2014-05-26"
  402. request.ApiName = "DescribeInstanceStatus"
  403. request.QueryParams["PageNumber"] = "1"
  404. request.QueryParams["PageSize"] = "30"
  405. request.RegionId = "regionid"
  406. request.TransToAcsRequest()
  407. signer := &signertest{
  408. name: "signer",
  409. }
  410. err = client.BuildRequestWithSigner(request, signer)
  411. assert.Nil(t, err)
  412. }
  413. func TestClient_ProcessCommonRequestWithSigner(t *testing.T) {
  414. client, err := NewClientWithAccessKey("regionid", "acesskeyid", "accesskeysecret")
  415. assert.Nil(t, err)
  416. assert.NotNil(t, client)
  417. assert.Equal(t, true, client.isRunning)
  418. request := requests.NewCommonRequest()
  419. request.Domain = "ecs.aliyuncs.com"
  420. request.Version = "2014-05-26"
  421. request.ApiName = "DescribeInstanceStatus"
  422. request.QueryParams["PageNumber"] = "1"
  423. request.QueryParams["PageSize"] = "30"
  424. request.RegionId = "regionid"
  425. signer := &signertest{
  426. name: "signer",
  427. }
  428. origTestHookDo := hookDo
  429. defer func() { hookDo = origTestHookDo }()
  430. hookDo = func(fn func(req *http.Request) (*http.Response, error)) func(req *http.Request) (*http.Response, error) {
  431. return func(req *http.Request) (*http.Response, error) {
  432. return mockResponse(500, "Server Internel Error", fmt.Errorf("test error"))
  433. }
  434. }
  435. resp, err := client.ProcessCommonRequestWithSigner(request, signer)
  436. assert.NotNil(t, err)
  437. assert.Contains(t, err.Error(), "test error")
  438. assert.Equal(t, resp.GetHttpContentString(), "")
  439. }
  440. func TestClient_AppendUserAgent(t *testing.T) {
  441. client, err := NewClientWithAccessKey("regionid", "acesskeyid", "accesskeysecret")
  442. assert.Nil(t, err)
  443. assert.NotNil(t, client)
  444. assert.Equal(t, true, client.isRunning)
  445. request := requests.NewCommonRequest()
  446. request.Domain = "ecs.aliyuncs.com"
  447. request.Version = "2014-05-26"
  448. request.ApiName = "DescribeInstanceStatus"
  449. request.RegionId = "regionid"
  450. signer := &signertest{
  451. name: "signer",
  452. }
  453. request.TransToAcsRequest()
  454. httpRequest, err := client.buildRequestWithSigner(request, signer)
  455. assert.Nil(t, err)
  456. assert.Equal(t, DefaultUserAgent, httpRequest.Header.Get("User-Agent"))
  457. // Test set client useragent.
  458. client.AppendUserAgent("test", "1.01")
  459. httpRequest, err = client.buildRequestWithSigner(request, signer)
  460. assert.Equal(t, DefaultUserAgent+" test/1.01", httpRequest.Header.Get("User-Agent"))
  461. // Test set request useragent. And request useragent has a higner priority than client's.
  462. request.AppendUserAgent("test", "2.01")
  463. httpRequest, err = client.buildRequestWithSigner(request, signer)
  464. assert.Equal(t, DefaultUserAgent+" test/2.01", httpRequest.Header.Get("User-Agent"))
  465. client.AppendUserAgent("test", "2.02")
  466. httpRequest, err = client.buildRequestWithSigner(request, signer)
  467. assert.Equal(t, DefaultUserAgent+" test/2.01", httpRequest.Header.Get("User-Agent"))
  468. // Test update request useragent.
  469. request.AppendUserAgent("test", "2.02")
  470. httpRequest, err = client.buildRequestWithSigner(request, signer)
  471. assert.Equal(t, DefaultUserAgent+" test/2.02", httpRequest.Header.Get("User-Agent"))
  472. // Test client can't modify DefaultUserAgent.
  473. client.AppendUserAgent("core", "1.01")
  474. httpRequest, err = client.buildRequestWithSigner(request, signer)
  475. assert.Equal(t, DefaultUserAgent+" test/2.02", httpRequest.Header.Get("User-Agent"))
  476. // Test request can't modify DefaultUserAgent.
  477. request.AppendUserAgent("core", "1.01")
  478. httpRequest, err = client.buildRequestWithSigner(request, signer)
  479. assert.Equal(t, DefaultUserAgent+" test/2.02", httpRequest.Header.Get("User-Agent"))
  480. request1 := requests.NewCommonRequest()
  481. request1.Domain = "ecs.aliyuncs.com"
  482. request1.Version = "2014-05-26"
  483. request1.ApiName = "DescribeRegions"
  484. request1.RegionId = "regionid"
  485. request1.AppendUserAgent("sys", "1.01")
  486. request1.TransToAcsRequest()
  487. httpRequest, err = client.buildRequestWithSigner(request1, signer)
  488. assert.Nil(t, err)
  489. assert.Equal(t, DefaultUserAgent+" test/2.02 sys/1.01", httpRequest.Header.Get("User-Agent"))
  490. }
  491. func TestClient_ProcessCommonRequestWithSigner_Error(t *testing.T) {
  492. client, err := NewClientWithAccessKey("regionid", "acesskeyid", "accesskeysecret")
  493. assert.Nil(t, err)
  494. assert.NotNil(t, client)
  495. assert.Equal(t, true, client.isRunning)
  496. request := requests.NewCommonRequest()
  497. request.Domain = "ecs.aliyuncs.com"
  498. request.Version = "2014-05-26"
  499. request.ApiName = "DescribeInstanceStatus"
  500. request.QueryParams["PageNumber"] = "1"
  501. request.QueryParams["PageSize"] = "30"
  502. request.RegionId = "regionid"
  503. origTestHookDo := hookDo
  504. defer func() {
  505. hookDo = origTestHookDo
  506. err := recover()
  507. assert.NotNil(t, err)
  508. }()
  509. hookDo = func(fn func(req *http.Request) (*http.Response, error)) func(req *http.Request) (*http.Response, error) {
  510. return func(req *http.Request) (*http.Response, error) {
  511. return mockResponse(500, "Server Internel Error", fmt.Errorf("test error"))
  512. }
  513. }
  514. resp, err := client.ProcessCommonRequestWithSigner(request, nil)
  515. assert.NotNil(t, err)
  516. assert.Contains(t, err.Error(), "test error")
  517. assert.Equal(t, resp.GetHttpContentString(), "Server Internel Error")
  518. }
  519. func TestClient_NewClientWithStsRoleNameOnEcs(t *testing.T) {
  520. client, err := NewClientWithStsRoleNameOnEcs("regionid", "rolename")
  521. assert.Nil(t, err)
  522. assert.NotNil(t, client)
  523. assert.Equal(t, true, client.isRunning)
  524. config := client.GetConfig()
  525. assert.NotNil(t, config)
  526. err = client.AddAsyncTask(nil)
  527. assert.NotNil(t, err)
  528. }
  529. func TestClient_NewClientWithStsRoleArn(t *testing.T) {
  530. client, err := NewClientWithStsRoleArn("regionid", "acesskeyid", "accesskeysecret", "rolearn", "rolesessionname")
  531. assert.Nil(t, err)
  532. assert.NotNil(t, client)
  533. assert.Equal(t, true, client.isRunning)
  534. task := func() {}
  535. client.asyncTaskQueue = make(chan func(), 1)
  536. err = client.AddAsyncTask(task)
  537. assert.Nil(t, err)
  538. client.Shutdown()
  539. assert.Equal(t, false, client.isRunning)
  540. }
  541. //func Test_EnableAsync(t *testing.T) {
  542. // client, err := NewClientWithAccessKey("regionid", "acesskeyid", "accesskeysecret")
  543. // assert.Nil(t, err)
  544. // assert.NotNil(t, client)
  545. // assert.Equal(t, true, client.isRunning)
  546. // client.EnableAsync(2, 8)
  547. // client.Shutdown()
  548. // assert.Equal(t, false, client.isRunning)
  549. //}
  550. func TestInitWithProviderChain(t *testing.T) {
  551. //testcase1: No any environment variable
  552. c, err := NewClientWithProvider("cn-hangzhou")
  553. assert.Empty(t, c)
  554. assert.EqualError(t, err, "No credential found")
  555. //testcase2: AK
  556. os.Setenv(provider.ENVAccessKeyID, "AccessKeyId")
  557. os.Setenv(provider.ENVAccessKeySecret, "AccessKeySecret")
  558. c, err = NewClientWithProvider("cn-hangzhou")
  559. assert.Nil(t, err)
  560. expC, err := NewClientWithAccessKey("cn-hangzhou", "AccessKeyId", "AccessKeySecret")
  561. assert.Nil(t, err)
  562. assert.Equal(t, expC, c)
  563. //testcase3:AK value is ""
  564. os.Setenv(provider.ENVAccessKeyID, "")
  565. os.Setenv(provider.ENVAccessKeySecret, "bbbb")
  566. c, err = NewClientWithProvider("cn-hangzhou")
  567. assert.EqualError(t, err, "Environmental variable (ALIBABACLOUD_ACCESS_KEY_ID or ALIBABACLOUD_ACCESS_KEY_SECRET) is empty")
  568. assert.Empty(t, c)
  569. //testcase4: Profile value is ""
  570. os.Unsetenv(provider.ENVAccessKeyID)
  571. os.Unsetenv(provider.ENVAccessKeySecret)
  572. os.Setenv(provider.ENVCredentialFile, "")
  573. c, err = NewClientWithProvider("cn-hangzhou")
  574. assert.Empty(t, c)
  575. assert.EqualError(t, err, "Environment variable 'ALIBABA_CLOUD_CREDENTIALS_FILE' cannot be empty")
  576. //testcase5: Profile
  577. os.Setenv(provider.ENVCredentialFile, "./profile")
  578. c, err = NewClientWithProvider("cn-hangzhou")
  579. assert.Empty(t, c)
  580. assert.NotNil(t, err)
  581. //testcase6:Instances
  582. os.Unsetenv(provider.ENVCredentialFile)
  583. os.Setenv(provider.ENVEcsMetadata, "")
  584. c, err = NewClientWithProvider("cn-hangzhou")
  585. assert.Empty(t, c)
  586. assert.EqualError(t, err, "Environmental variable 'ALIBABA_CLOUD_ECS_METADATA' are empty")
  587. //testcase7: Custom Providers
  588. c, err = NewClientWithProvider("cn-hangzhou", provider.ProviderProfile, provider.ProviderEnv)
  589. assert.Empty(t, c)
  590. assert.EqualError(t, err, "No credential found")
  591. }