client_test.go 22 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670
  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://username:password@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. assert.Equal(t, url.User.Username(), "username")
  191. client.Shutdown()
  192. os.Setenv("https_proxy", envHttpsProxy)
  193. os.Setenv("http_proxy", envHttpProxy)
  194. os.Setenv("no_proxy", envNoProxy)
  195. assert.Equal(t, false, client.isRunning)
  196. }
  197. func Test_DoAction_HTTPSInsecure(t *testing.T) {
  198. client, err := NewClientWithAccessKey("regionid", "acesskeyid", "accesskeysecret")
  199. assert.Nil(t, err)
  200. assert.NotNil(t, client)
  201. client.SetHTTPSInsecure(true)
  202. request := requests.NewCommonRequest()
  203. request.Product = "Ram"
  204. request.Version = "2015-05-01"
  205. request.ApiName = "CreateRole"
  206. request.Domain = "ecs.aliyuncs.com"
  207. request.QueryParams["RegionId"] = os.Getenv("REGION_ID")
  208. request.TransToAcsRequest()
  209. response := responses.NewCommonResponse()
  210. origTestHookDo := hookDo
  211. defer func() { hookDo = origTestHookDo }()
  212. hookDo = func(fn func(req *http.Request) (*http.Response, error)) func(req *http.Request) (*http.Response, error) {
  213. return func(req *http.Request) (*http.Response, error) {
  214. return mockResponse(200, "", nil)
  215. }
  216. }
  217. err = client.DoAction(request, response)
  218. assert.Nil(t, err)
  219. assert.Equal(t, 200, response.GetHttpStatus())
  220. assert.Equal(t, "", response.GetHttpContentString())
  221. trans := client.httpClient.Transport.(*http.Transport)
  222. assert.Equal(t, true, trans.TLSClientConfig.InsecureSkipVerify)
  223. request.SetHTTPSInsecure(false)
  224. err = client.DoAction(request, response)
  225. assert.Nil(t, err)
  226. trans = client.httpClient.Transport.(*http.Transport)
  227. assert.Equal(t, false, trans.TLSClientConfig.InsecureSkipVerify)
  228. // Test when scheme is http, only http proxy is valid.
  229. envHttpsProxy := os.Getenv("HTTPS_PROXY")
  230. os.Setenv("HTTPS_PROXY", "https://127.0.0.1:9000")
  231. err = client.DoAction(request, response)
  232. assert.Nil(t, err)
  233. trans, _ = client.httpClient.Transport.(*http.Transport)
  234. assert.Nil(t, trans.Proxy)
  235. // Test when host is in no_proxy, proxy is invalid
  236. envNoProxy := os.Getenv("NO_PROXY")
  237. os.Setenv("NO_PROXY", "ecs.aliyuncs.com")
  238. envHttpProxy := os.Getenv("HTTP_PROXY")
  239. os.Setenv("HTTP_PROXY", "http://127.0.0.1:8888")
  240. err = client.DoAction(request, response)
  241. assert.Nil(t, err)
  242. trans, _ = client.httpClient.Transport.(*http.Transport)
  243. assert.Nil(t, trans.Proxy)
  244. client.SetNoProxy("ecs.testaliyuncs.com")
  245. err = client.DoAction(request, response)
  246. assert.Nil(t, err)
  247. trans, _ = client.httpClient.Transport.(*http.Transport)
  248. url, _ := trans.Proxy(nil)
  249. assert.Equal(t, url.Scheme, "http")
  250. assert.Equal(t, url.Host, "127.0.0.1:8888")
  251. // Test when setting http proxy, client has a high priority than environment variable
  252. client.SetHttpProxy("http://127.0.0.1:8080")
  253. err = client.DoAction(request, response)
  254. assert.Nil(t, err)
  255. trans, _ = client.httpClient.Transport.(*http.Transport)
  256. url, _ = trans.Proxy(nil)
  257. assert.Equal(t, url.Scheme, "http")
  258. assert.Equal(t, url.Host, "127.0.0.1:8080")
  259. // Test when scheme is https, only https proxy is valid
  260. request.Scheme = "https"
  261. err = client.DoAction(request, response)
  262. assert.Nil(t, err)
  263. trans, _ = client.httpClient.Transport.(*http.Transport)
  264. url, _ = trans.Proxy(nil)
  265. assert.Equal(t, url.Scheme, "https")
  266. assert.Equal(t, url.Host, "127.0.0.1:9000")
  267. // Test when setting https proxy, client has a high priority than environment variable
  268. client.SetHttpsProxy("https://127.0.0.1:6666")
  269. err = client.DoAction(request, response)
  270. assert.Nil(t, err)
  271. trans, _ = client.httpClient.Transport.(*http.Transport)
  272. url, _ = trans.Proxy(nil)
  273. assert.Equal(t, url.Scheme, "https")
  274. assert.Equal(t, url.Host, "127.0.0.1:6666")
  275. client.Shutdown()
  276. os.Setenv("HTTPS_PROXY", envHttpsProxy)
  277. os.Setenv("HTTP_PROXY", envHttpProxy)
  278. os.Setenv("NO_PROXY", envNoProxy)
  279. assert.Equal(t, false, client.isRunning)
  280. }
  281. func Test_DoAction_Timeout(t *testing.T) {
  282. client, err := NewClientWithAccessKey("regionid", "acesskeyid", "accesskeysecret")
  283. assert.Nil(t, err)
  284. assert.NotNil(t, client)
  285. assert.Equal(t, true, client.isRunning)
  286. request := requests.NewCommonRequest()
  287. request.Domain = "ecs.aliyuncs.com"
  288. request.Version = "2014-05-26"
  289. request.ApiName = "DescribeInstanceStatus"
  290. request.QueryParams["PageNumber"] = "1"
  291. request.QueryParams["PageSize"] = "30"
  292. request.TransToAcsRequest()
  293. response := responses.NewCommonResponse()
  294. origTestHookDo := hookDo
  295. defer func() { hookDo = origTestHookDo }()
  296. hookDo = func(fn func(req *http.Request) (*http.Response, error)) func(req *http.Request) (*http.Response, error) {
  297. return func(req *http.Request) (*http.Response, error) {
  298. return mockResponse(400, "Server Internel Error", fmt.Errorf("read tcp"))
  299. }
  300. }
  301. err = client.DoAction(request, response)
  302. assert.NotNil(t, err)
  303. assert.Equal(t, 0, response.GetHttpStatus())
  304. assert.Equal(t, "", response.GetHttpContentString())
  305. // Test set client timeout
  306. client.SetReadTimeout(1 * time.Millisecond)
  307. client.SetConnectTimeout(1 * time.Millisecond)
  308. assert.Equal(t, 1*time.Millisecond, client.GetConnectTimeout())
  309. assert.Equal(t, 1*time.Millisecond, client.GetReadTimeout())
  310. client.config.AutoRetry = false
  311. err = client.DoAction(request, response)
  312. assert.NotNil(t, err)
  313. assert.Equal(t, 0, response.GetHttpStatus())
  314. assert.Equal(t, "", response.GetHttpContentString())
  315. // Test set request timeout
  316. request.SetReadTimeout(1 * time.Millisecond)
  317. request.SetConnectTimeout(1 * time.Millisecond)
  318. err = client.DoAction(request, response)
  319. assert.NotNil(t, err)
  320. assert.Equal(t, 0, response.GetHttpStatus())
  321. assert.Equal(t, "", response.GetHttpContentString())
  322. client.Shutdown()
  323. assert.Equal(t, false, client.isRunning)
  324. }
  325. func Test_ProcessCommonRequest(t *testing.T) {
  326. client, err := NewClientWithAccessKey("regionid", "acesskeyid", "accesskeysecret")
  327. assert.Nil(t, err)
  328. assert.NotNil(t, client)
  329. request := requests.NewCommonRequest()
  330. request.Domain = "ecs.aliyuncs.com"
  331. request.Version = "2014-05-26"
  332. request.ApiName = "DescribeInstanceStatus"
  333. request.QueryParams["PageNumber"] = "1"
  334. request.QueryParams["PageSize"] = "30"
  335. origTestHookDo := hookDo
  336. defer func() { hookDo = origTestHookDo }()
  337. hookDo = func(fn func(req *http.Request) (*http.Response, error)) func(req *http.Request) (*http.Response, error) {
  338. return func(req *http.Request) (*http.Response, error) {
  339. return mockResponse(400, "", fmt.Errorf("test error"))
  340. }
  341. }
  342. resp, err := client.ProcessCommonRequest(request)
  343. assert.NotNil(t, err)
  344. assert.Contains(t, err.Error(), "test error")
  345. assert.Equal(t, 0, resp.GetHttpStatus())
  346. assert.Equal(t, "", resp.GetHttpContentString())
  347. }
  348. func mockServer(status int, json string) (server *httptest.Server) {
  349. // Start a test server locally.
  350. ts := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
  351. w.WriteHeader(status)
  352. w.Write([]byte(json))
  353. return
  354. }))
  355. return ts
  356. }
  357. func Test_DoAction_With500(t *testing.T) {
  358. client, err := NewClientWithAccessKey("regionid", "acesskeyid", "accesskeysecret")
  359. assert.Nil(t, err)
  360. assert.NotNil(t, client)
  361. assert.Equal(t, true, client.isRunning)
  362. request := requests.NewCommonRequest()
  363. request.Version = "2014-05-26"
  364. request.ApiName = "DescribeInstanceStatus"
  365. request.QueryParams["PageNumber"] = "1"
  366. request.QueryParams["PageSize"] = "30"
  367. request.TransToAcsRequest()
  368. response := responses.NewCommonResponse()
  369. ts := mockServer(500, "Server Internel Error")
  370. defer ts.Close()
  371. domain := strings.Replace(ts.URL, "http://", "", 1)
  372. request.Domain = domain
  373. err = client.DoAction(request, response)
  374. assert.NotNil(t, err)
  375. assert.Equal(t, 500, response.GetHttpStatus())
  376. assert.Equal(t, "Server Internel Error", response.GetHttpContentString())
  377. }
  378. func TestClient_BuildRequestWithSigner(t *testing.T) {
  379. client, err := NewClientWithAccessKey("regionid", "acesskeyid", "accesskeysecret")
  380. assert.Nil(t, err)
  381. assert.NotNil(t, client)
  382. assert.Equal(t, true, client.isRunning)
  383. request := requests.NewCommonRequest()
  384. request.Domain = "ecs.aliyuncs.com"
  385. request.Version = "2014-05-26"
  386. request.ApiName = "DescribeInstanceStatus"
  387. request.QueryParams["PageNumber"] = "1"
  388. request.QueryParams["PageSize"] = "30"
  389. request.RegionId = "regionid"
  390. request.TransToAcsRequest()
  391. client.config.UserAgent = "user_agent"
  392. err = client.BuildRequestWithSigner(request, nil)
  393. assert.Nil(t, err)
  394. }
  395. func TestClient_BuildRequestWithSigner1(t *testing.T) {
  396. client, err := NewClientWithAccessKey("regionid", "acesskeyid", "accesskeysecret")
  397. assert.Nil(t, err)
  398. assert.NotNil(t, client)
  399. assert.Equal(t, true, client.isRunning)
  400. request := requests.NewCommonRequest()
  401. request.Domain = "ecs.aliyuncs.com"
  402. request.Version = "2014-05-26"
  403. request.ApiName = "DescribeInstanceStatus"
  404. request.QueryParams["PageNumber"] = "1"
  405. request.QueryParams["PageSize"] = "30"
  406. request.RegionId = "regionid"
  407. request.TransToAcsRequest()
  408. signer := &signertest{
  409. name: "signer",
  410. }
  411. err = client.BuildRequestWithSigner(request, signer)
  412. assert.Nil(t, err)
  413. }
  414. func TestClient_ProcessCommonRequestWithSigner(t *testing.T) {
  415. client, err := NewClientWithAccessKey("regionid", "acesskeyid", "accesskeysecret")
  416. assert.Nil(t, err)
  417. assert.NotNil(t, client)
  418. assert.Equal(t, true, client.isRunning)
  419. request := requests.NewCommonRequest()
  420. request.Domain = "ecs.aliyuncs.com"
  421. request.Version = "2014-05-26"
  422. request.ApiName = "DescribeInstanceStatus"
  423. request.QueryParams["PageNumber"] = "1"
  424. request.QueryParams["PageSize"] = "30"
  425. request.RegionId = "regionid"
  426. signer := &signertest{
  427. name: "signer",
  428. }
  429. origTestHookDo := hookDo
  430. defer func() { hookDo = origTestHookDo }()
  431. hookDo = func(fn func(req *http.Request) (*http.Response, error)) func(req *http.Request) (*http.Response, error) {
  432. return func(req *http.Request) (*http.Response, error) {
  433. return mockResponse(500, "Server Internel Error", fmt.Errorf("test error"))
  434. }
  435. }
  436. resp, err := client.ProcessCommonRequestWithSigner(request, signer)
  437. assert.NotNil(t, err)
  438. assert.Contains(t, err.Error(), "test error")
  439. assert.Equal(t, resp.GetHttpContentString(), "")
  440. }
  441. func TestClient_AppendUserAgent(t *testing.T) {
  442. client, err := NewClientWithAccessKey("regionid", "acesskeyid", "accesskeysecret")
  443. assert.Nil(t, err)
  444. assert.NotNil(t, client)
  445. assert.Equal(t, true, client.isRunning)
  446. request := requests.NewCommonRequest()
  447. request.Domain = "ecs.aliyuncs.com"
  448. request.Version = "2014-05-26"
  449. request.ApiName = "DescribeInstanceStatus"
  450. request.RegionId = "regionid"
  451. signer := &signertest{
  452. name: "signer",
  453. }
  454. request.TransToAcsRequest()
  455. httpRequest, err := client.buildRequestWithSigner(request, signer)
  456. assert.Nil(t, err)
  457. assert.Equal(t, DefaultUserAgent, httpRequest.Header.Get("User-Agent"))
  458. // Test set client useragent.
  459. client.AppendUserAgent("test", "1.01")
  460. httpRequest, err = client.buildRequestWithSigner(request, signer)
  461. assert.Equal(t, DefaultUserAgent+" test/1.01", httpRequest.Header.Get("User-Agent"))
  462. // Test set request useragent. And request useragent has a higner priority than client's.
  463. request.AppendUserAgent("test", "2.01")
  464. httpRequest, err = client.buildRequestWithSigner(request, signer)
  465. assert.Equal(t, DefaultUserAgent+" test/2.01", httpRequest.Header.Get("User-Agent"))
  466. client.AppendUserAgent("test", "2.02")
  467. httpRequest, err = client.buildRequestWithSigner(request, signer)
  468. assert.Equal(t, DefaultUserAgent+" test/2.01", httpRequest.Header.Get("User-Agent"))
  469. // Test update request useragent.
  470. request.AppendUserAgent("test", "2.02")
  471. httpRequest, err = client.buildRequestWithSigner(request, signer)
  472. assert.Equal(t, DefaultUserAgent+" test/2.02", httpRequest.Header.Get("User-Agent"))
  473. // Test client can't modify DefaultUserAgent.
  474. client.AppendUserAgent("core", "1.01")
  475. httpRequest, err = client.buildRequestWithSigner(request, signer)
  476. assert.Equal(t, DefaultUserAgent+" test/2.02", httpRequest.Header.Get("User-Agent"))
  477. // Test request can't modify DefaultUserAgent.
  478. request.AppendUserAgent("core", "1.01")
  479. httpRequest, err = client.buildRequestWithSigner(request, signer)
  480. assert.Equal(t, DefaultUserAgent+" test/2.02", httpRequest.Header.Get("User-Agent"))
  481. request1 := requests.NewCommonRequest()
  482. request1.Domain = "ecs.aliyuncs.com"
  483. request1.Version = "2014-05-26"
  484. request1.ApiName = "DescribeRegions"
  485. request1.RegionId = "regionid"
  486. request1.AppendUserAgent("sys", "1.01")
  487. request1.TransToAcsRequest()
  488. httpRequest, err = client.buildRequestWithSigner(request1, signer)
  489. assert.Nil(t, err)
  490. assert.Equal(t, DefaultUserAgent+" test/2.02 sys/1.01", httpRequest.Header.Get("User-Agent"))
  491. }
  492. func TestClient_ProcessCommonRequestWithSigner_Error(t *testing.T) {
  493. client, err := NewClientWithAccessKey("regionid", "acesskeyid", "accesskeysecret")
  494. assert.Nil(t, err)
  495. assert.NotNil(t, client)
  496. assert.Equal(t, true, client.isRunning)
  497. request := requests.NewCommonRequest()
  498. request.Domain = "ecs.aliyuncs.com"
  499. request.Version = "2014-05-26"
  500. request.ApiName = "DescribeInstanceStatus"
  501. request.QueryParams["PageNumber"] = "1"
  502. request.QueryParams["PageSize"] = "30"
  503. request.RegionId = "regionid"
  504. origTestHookDo := hookDo
  505. defer func() {
  506. hookDo = origTestHookDo
  507. err := recover()
  508. assert.NotNil(t, err)
  509. }()
  510. hookDo = func(fn func(req *http.Request) (*http.Response, error)) func(req *http.Request) (*http.Response, error) {
  511. return func(req *http.Request) (*http.Response, error) {
  512. return mockResponse(500, "Server Internel Error", fmt.Errorf("test error"))
  513. }
  514. }
  515. resp, err := client.ProcessCommonRequestWithSigner(request, nil)
  516. assert.NotNil(t, err)
  517. assert.Contains(t, err.Error(), "test error")
  518. assert.Equal(t, resp.GetHttpContentString(), "Server Internel Error")
  519. }
  520. func TestClient_NewClientWithStsRoleNameOnEcs(t *testing.T) {
  521. client, err := NewClientWithStsRoleNameOnEcs("regionid", "rolename")
  522. assert.Nil(t, err)
  523. assert.NotNil(t, client)
  524. assert.Equal(t, true, client.isRunning)
  525. config := client.GetConfig()
  526. assert.NotNil(t, config)
  527. err = client.AddAsyncTask(nil)
  528. assert.NotNil(t, err)
  529. }
  530. func TestClient_NewClientWithStsRoleArn(t *testing.T) {
  531. client, err := NewClientWithStsRoleArn("regionid", "acesskeyid", "accesskeysecret", "rolearn", "rolesessionname")
  532. assert.Nil(t, err)
  533. assert.NotNil(t, client)
  534. assert.Equal(t, true, client.isRunning)
  535. task := func() {}
  536. client.asyncTaskQueue = make(chan func(), 1)
  537. err = client.AddAsyncTask(task)
  538. assert.Nil(t, err)
  539. client.Shutdown()
  540. assert.Equal(t, false, client.isRunning)
  541. }
  542. func TestInitWithProviderChain(t *testing.T) {
  543. //testcase1: No any environment variable
  544. c, err := NewClientWithProvider("cn-hangzhou")
  545. assert.Empty(t, c)
  546. assert.EqualError(t, err, "No credential found")
  547. //testcase2: AK
  548. os.Setenv(provider.ENVAccessKeyID, "AccessKeyId")
  549. os.Setenv(provider.ENVAccessKeySecret, "AccessKeySecret")
  550. c, err = NewClientWithProvider("cn-hangzhou")
  551. assert.Nil(t, err)
  552. expC, err := NewClientWithAccessKey("cn-hangzhou", "AccessKeyId", "AccessKeySecret")
  553. assert.Nil(t, err)
  554. assert.Equal(t, expC, c)
  555. //testcase3:AK value is ""
  556. os.Setenv(provider.ENVAccessKeyID, "")
  557. os.Setenv(provider.ENVAccessKeySecret, "bbbb")
  558. c, err = NewClientWithProvider("cn-hangzhou")
  559. assert.EqualError(t, err, "Environmental variable (ALIBABACLOUD_ACCESS_KEY_ID or ALIBABACLOUD_ACCESS_KEY_SECRET) is empty")
  560. assert.Empty(t, c)
  561. //testcase4: Profile value is ""
  562. os.Unsetenv(provider.ENVAccessKeyID)
  563. os.Unsetenv(provider.ENVAccessKeySecret)
  564. os.Setenv(provider.ENVCredentialFile, "")
  565. c, err = NewClientWithProvider("cn-hangzhou")
  566. assert.Empty(t, c)
  567. assert.EqualError(t, err, "Environment variable 'ALIBABA_CLOUD_CREDENTIALS_FILE' cannot be empty")
  568. //testcase5: Profile
  569. os.Setenv(provider.ENVCredentialFile, "./profile")
  570. c, err = NewClientWithProvider("cn-hangzhou")
  571. assert.Empty(t, c)
  572. assert.NotNil(t, err)
  573. //testcase6:Instances
  574. os.Unsetenv(provider.ENVCredentialFile)
  575. os.Setenv(provider.ENVEcsMetadata, "")
  576. c, err = NewClientWithProvider("cn-hangzhou")
  577. assert.Empty(t, c)
  578. assert.EqualError(t, err, "Environmental variable 'ALIBABA_CLOUD_ECS_METADATA' are empty")
  579. //testcase7: Custom Providers
  580. c, err = NewClientWithProvider("cn-hangzhou", provider.ProviderProfile, provider.ProviderEnv)
  581. assert.Empty(t, c)
  582. assert.EqualError(t, err, "No credential found")
  583. }
  584. func TestNewClientWithBearerToken(t *testing.T) {
  585. client, err := NewClientWithBearerToken("cn-hangzhou", "Bearer.Token")
  586. assert.Nil(t, err)
  587. assert.NotNil(t, client)
  588. }