client_test.go 103 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276127712781279128012811282128312841285128612871288128912901291129212931294129512961297129812991300130113021303130413051306130713081309131013111312131313141315131613171318131913201321132213231324132513261327132813291330133113321333133413351336133713381339134013411342134313441345134613471348134913501351135213531354135513561357135813591360136113621363136413651366136713681369137013711372137313741375137613771378137913801381138213831384138513861387138813891390139113921393139413951396139713981399140014011402140314041405140614071408140914101411141214131414141514161417141814191420142114221423142414251426142714281429143014311432143314341435143614371438143914401441144214431444144514461447144814491450145114521453145414551456145714581459146014611462146314641465146614671468146914701471147214731474147514761477147814791480148114821483148414851486148714881489149014911492149314941495149614971498149915001501150215031504150515061507150815091510151115121513151415151516151715181519152015211522152315241525152615271528152915301531153215331534153515361537153815391540154115421543154415451546154715481549155015511552155315541555155615571558155915601561156215631564156515661567156815691570157115721573157415751576157715781579158015811582158315841585158615871588158915901591159215931594159515961597159815991600160116021603160416051606160716081609161016111612161316141615161616171618161916201621162216231624162516261627162816291630163116321633163416351636163716381639164016411642164316441645164616471648164916501651165216531654165516561657165816591660166116621663166416651666166716681669167016711672167316741675167616771678167916801681168216831684168516861687168816891690169116921693169416951696169716981699170017011702170317041705170617071708170917101711171217131714171517161717171817191720172117221723172417251726172717281729173017311732173317341735173617371738173917401741174217431744174517461747174817491750175117521753175417551756175717581759176017611762176317641765176617671768176917701771177217731774177517761777177817791780178117821783178417851786178717881789179017911792179317941795179617971798179918001801180218031804180518061807180818091810181118121813181418151816181718181819182018211822182318241825182618271828182918301831183218331834183518361837183818391840184118421843184418451846184718481849185018511852185318541855185618571858185918601861186218631864186518661867186818691870187118721873187418751876187718781879188018811882188318841885188618871888188918901891189218931894189518961897189818991900190119021903190419051906190719081909191019111912191319141915191619171918191919201921192219231924192519261927192819291930193119321933193419351936193719381939194019411942194319441945194619471948194919501951195219531954195519561957195819591960196119621963196419651966196719681969197019711972197319741975197619771978197919801981198219831984198519861987198819891990199119921993199419951996199719981999200020012002200320042005200620072008200920102011201220132014201520162017201820192020202120222023202420252026202720282029203020312032203320342035203620372038203920402041204220432044204520462047204820492050205120522053205420552056205720582059206020612062206320642065206620672068206920702071207220732074207520762077207820792080208120822083208420852086208720882089209020912092209320942095209620972098209921002101210221032104210521062107210821092110211121122113211421152116211721182119212021212122212321242125212621272128212921302131213221332134213521362137213821392140214121422143214421452146214721482149215021512152215321542155215621572158215921602161216221632164216521662167216821692170217121722173217421752176217721782179218021812182218321842185218621872188218921902191219221932194219521962197219821992200220122022203220422052206220722082209221022112212221322142215221622172218221922202221222222232224222522262227222822292230223122322233223422352236223722382239224022412242224322442245224622472248224922502251225222532254225522562257225822592260226122622263226422652266226722682269227022712272227322742275227622772278227922802281228222832284228522862287228822892290229122922293229422952296229722982299230023012302230323042305230623072308230923102311231223132314231523162317231823192320232123222323232423252326232723282329233023312332233323342335233623372338233923402341234223432344234523462347234823492350235123522353235423552356235723582359236023612362236323642365236623672368236923702371237223732374237523762377237823792380238123822383238423852386238723882389239023912392239323942395239623972398239924002401240224032404240524062407240824092410241124122413241424152416241724182419242024212422242324242425242624272428242924302431243224332434243524362437243824392440244124422443244424452446244724482449245024512452245324542455245624572458245924602461246224632464246524662467246824692470247124722473247424752476247724782479248024812482248324842485248624872488248924902491249224932494249524962497249824992500250125022503250425052506250725082509251025112512251325142515251625172518251925202521252225232524252525262527252825292530253125322533253425352536253725382539254025412542254325442545254625472548254925502551255225532554255525562557255825592560256125622563256425652566256725682569257025712572257325742575257625772578257925802581258225832584258525862587258825892590259125922593259425952596259725982599260026012602260326042605260626072608260926102611261226132614261526162617261826192620262126222623262426252626262726282629263026312632263326342635263626372638263926402641264226432644264526462647264826492650265126522653265426552656265726582659266026612662266326642665266626672668266926702671267226732674267526762677267826792680268126822683268426852686268726882689269026912692269326942695269626972698269927002701270227032704270527062707270827092710271127122713271427152716271727182719272027212722272327242725272627272728272927302731273227332734273527362737273827392740274127422743274427452746274727482749275027512752275327542755275627572758275927602761276227632764276527662767276827692770277127722773277427752776277727782779278027812782278327842785278627872788278927902791279227932794279527962797279827992800280128022803280428052806280728082809281028112812281328142815281628172818281928202821282228232824282528262827282828292830283128322833283428352836283728382839284028412842284328442845284628472848284928502851285228532854285528562857285828592860286128622863286428652866286728682869287028712872287328742875287628772878287928802881288228832884288528862887288828892890289128922893289428952896289728982899290029012902290329042905290629072908290929102911291229132914291529162917291829192920292129222923292429252926292729282929293029312932293329342935293629372938293929402941294229432944294529462947294829492950295129522953295429552956295729582959296029612962296329642965296629672968296929702971297229732974297529762977297829792980298129822983298429852986298729882989299029912992299329942995299629972998299930003001300230033004300530063007300830093010301130123013301430153016301730183019302030213022302330243025302630273028302930303031303230333034303530363037303830393040304130423043304430453046304730483049305030513052305330543055305630573058305930603061306230633064306530663067306830693070307130723073307430753076307730783079308030813082308330843085308630873088308930903091309230933094309530963097309830993100310131023103310431053106310731083109311031113112311331143115311631173118311931203121312231233124312531263127312831293130313131323133313431353136313731383139314031413142314331443145314631473148314931503151315231533154315531563157315831593160316131623163316431653166316731683169317031713172317331743175317631773178317931803181318231833184318531863187318831893190319131923193319431953196319731983199320032013202320332043205320632073208320932103211321232133214321532163217321832193220322132223223322432253226322732283229323032313232323332343235323632373238323932403241324232433244324532463247324832493250325132523253325432553256325732583259326032613262326332643265326632673268326932703271327232733274327532763277327832793280328132823283328432853286328732883289329032913292329332943295329632973298329933003301330233033304330533063307330833093310331133123313331433153316331733183319332033213322332333243325332633273328332933303331333233333334333533363337333833393340334133423343334433453346334733483349335033513352335333543355335633573358335933603361336233633364336533663367336833693370337133723373337433753376337733783379338033813382338333843385338633873388338933903391339233933394339533963397339833993400340134023403340434053406340734083409341034113412341334143415341634173418341934203421342234233424342534263427342834293430343134323433343434353436343734383439344034413442344334443445344634473448344934503451345234533454345534563457345834593460346134623463346434653466346734683469347034713472347334743475347634773478347934803481348234833484348534863487348834893490349134923493349434953496349734983499350035013502350335043505350635073508350935103511351235133514351535163517351835193520352135223523352435253526352735283529353035313532353335343535353635373538353935403541354235433544
  1. // client test
  2. // use gocheck, install gocheck to execute "go get gopkg.in/check.v1",
  3. // see https://labix.org/gocheck
  4. package oss
  5. import (
  6. "encoding/json"
  7. "encoding/xml"
  8. "fmt"
  9. "io/ioutil"
  10. "log"
  11. "math/rand"
  12. "net"
  13. "net/http"
  14. "os"
  15. "reflect"
  16. "runtime"
  17. "strconv"
  18. "strings"
  19. "testing"
  20. "time"
  21. . "gopkg.in/check.v1"
  22. )
  23. // Test hooks up gocheck into the "go test" runner.
  24. func Test(t *testing.T) {
  25. TestingT(t)
  26. }
  27. type OssClientSuite struct{}
  28. var _ = Suite(&OssClientSuite{})
  29. var (
  30. // Endpoint/ID/Key
  31. endpoint = os.Getenv("OSS_TEST_ENDPOINT")
  32. accessID = os.Getenv("OSS_TEST_ACCESS_KEY_ID")
  33. accessKey = os.Getenv("OSS_TEST_ACCESS_KEY_SECRET")
  34. accountID = os.Getenv("OSS_TEST_ACCOUNT_ID")
  35. // Proxy
  36. proxyHost = os.Getenv("OSS_TEST_PROXY_HOST")
  37. proxyUser = os.Getenv("OSS_TEST_PROXY_USER")
  38. proxyPasswd = os.Getenv("OSS_TEST_PROXY_PASSWORD")
  39. // STS
  40. stsaccessID = os.Getenv("OSS_TEST_STS_ID")
  41. stsaccessKey = os.Getenv("OSS_TEST_STS_KEY")
  42. stsARN = os.Getenv("OSS_TEST_STS_ARN")
  43. // Credential
  44. credentialAccessID = os.Getenv("OSS_CREDENTIAL_KEY_ID")
  45. credentialAccessKey = os.Getenv("OSS_CREDENTIAL_KEY_SECRET")
  46. credentialUID = os.Getenv("OSS_CREDENTIAL_UID")
  47. )
  48. var (
  49. // prefix of bucket name for bucket ops test
  50. bucketNamePrefix = "go-sdk-test-bucket-"
  51. // bucket name for object ops test
  52. bucketName = bucketNamePrefix + RandLowStr(6)
  53. archiveBucketName = bucketNamePrefix + "arch-" + RandLowStr(6)
  54. // object name for object ops test
  55. objectNamePrefix = "go-sdk-test-object-"
  56. // sts region is one and only hangzhou
  57. stsRegion = "cn-hangzhou"
  58. // Credentials
  59. credentialBucketName = bucketNamePrefix + RandLowStr(6)
  60. )
  61. var (
  62. logPath = "go_sdk_test_" + time.Now().Format("20060102_150405") + ".log"
  63. testLogFile, _ = os.OpenFile(logPath, os.O_RDWR|os.O_CREATE, 0664)
  64. testLogger = log.New(testLogFile, "", log.Ldate|log.Ltime|log.Lshortfile)
  65. letters = []rune("abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ")
  66. timeoutInOperation = 3 * time.Second
  67. )
  68. func RandStr(n int) string {
  69. b := make([]rune, n)
  70. randMarker := rand.New(rand.NewSource(time.Now().UnixNano()))
  71. for i := range b {
  72. b[i] = letters[randMarker.Intn(len(letters))]
  73. }
  74. return string(b)
  75. }
  76. func CreateFile(fileName, content string, c *C) {
  77. fout, err := os.Create(fileName)
  78. defer fout.Close()
  79. c.Assert(err, IsNil)
  80. _, err = fout.WriteString(content)
  81. c.Assert(err, IsNil)
  82. }
  83. func RandLowStr(n int) string {
  84. return strings.ToLower(RandStr(n))
  85. }
  86. func ForceDeleteBucket(client *Client, bucketName string, c *C) {
  87. bucket, err := client.Bucket(bucketName)
  88. c.Assert(err, IsNil)
  89. // Delete Object
  90. marker := Marker("")
  91. for {
  92. lor, err := bucket.ListObjects(marker)
  93. c.Assert(err, IsNil)
  94. for _, object := range lor.Objects {
  95. err = bucket.DeleteObject(object.Key)
  96. c.Assert(err, IsNil)
  97. }
  98. marker = Marker(lor.NextMarker)
  99. if !lor.IsTruncated {
  100. break
  101. }
  102. }
  103. // Delete Object Versions and DeleteMarks
  104. keyMarker := KeyMarker("")
  105. versionIdMarker := VersionIdMarker("")
  106. options := []Option{keyMarker, versionIdMarker}
  107. for {
  108. lor, err := bucket.ListObjectVersions(options...)
  109. if err != nil {
  110. break
  111. }
  112. for _, object := range lor.ObjectDeleteMarkers {
  113. err = bucket.DeleteObject(object.Key, VersionId(object.VersionId))
  114. c.Assert(err, IsNil)
  115. }
  116. for _, object := range lor.ObjectVersions {
  117. err = bucket.DeleteObject(object.Key, VersionId(object.VersionId))
  118. c.Assert(err, IsNil)
  119. }
  120. keyMarker = KeyMarker(lor.NextKeyMarker)
  121. versionIdMarker := VersionIdMarker(lor.NextVersionIdMarker)
  122. options = []Option{keyMarker, versionIdMarker}
  123. if !lor.IsTruncated {
  124. break
  125. }
  126. }
  127. // Delete Part
  128. keyMarker = KeyMarker("")
  129. uploadIDMarker := UploadIDMarker("")
  130. for {
  131. lmur, err := bucket.ListMultipartUploads(keyMarker, uploadIDMarker)
  132. c.Assert(err, IsNil)
  133. for _, upload := range lmur.Uploads {
  134. var imur = InitiateMultipartUploadResult{Bucket: bucketName,
  135. Key: upload.Key, UploadID: upload.UploadID}
  136. err = bucket.AbortMultipartUpload(imur)
  137. c.Assert(err, IsNil)
  138. }
  139. keyMarker = KeyMarker(lmur.NextKeyMarker)
  140. uploadIDMarker = UploadIDMarker(lmur.NextUploadIDMarker)
  141. if !lmur.IsTruncated {
  142. break
  143. }
  144. }
  145. // delete live channel
  146. strMarker := ""
  147. for {
  148. result, err := bucket.ListLiveChannel(Marker(strMarker))
  149. c.Assert(err, IsNil)
  150. for _, channel := range result.LiveChannel {
  151. err := bucket.DeleteLiveChannel(channel.Name)
  152. c.Assert(err, IsNil)
  153. }
  154. if result.IsTruncated {
  155. strMarker = result.NextMarker
  156. } else {
  157. break
  158. }
  159. }
  160. // Delete Bucket
  161. err = client.DeleteBucket(bucketName)
  162. c.Assert(err, IsNil)
  163. }
  164. // SetUpSuite runs once when the suite starts running
  165. func (s *OssClientSuite) SetUpSuite(c *C) {
  166. client, err := New(endpoint, accessID, accessKey)
  167. c.Assert(err, IsNil)
  168. lbr, err := client.ListBuckets(Prefix(bucketNamePrefix), MaxKeys(1000))
  169. c.Assert(err, IsNil)
  170. for _, bucket := range lbr.Buckets {
  171. ForceDeleteBucket(client, bucket.Name, c)
  172. }
  173. testLogger.Println("test client started")
  174. }
  175. // TearDownSuite runs before each test or benchmark starts running
  176. func (s *OssClientSuite) TearDownSuite(c *C) {
  177. client, err := New(endpoint, accessID, accessKey)
  178. c.Assert(err, IsNil)
  179. lbr, err := client.ListBuckets(Prefix(bucketNamePrefix), MaxKeys(1000))
  180. c.Assert(err, IsNil)
  181. for _, bucket := range lbr.Buckets {
  182. s.deleteBucket(client, bucket.Name, c)
  183. }
  184. testLogger.Println("test client completed")
  185. }
  186. func (s *OssClientSuite) deleteBucket(client *Client, bucketName string, c *C) {
  187. ForceDeleteBucket(client, bucketName, c)
  188. }
  189. // SetUpTest runs after each test or benchmark runs
  190. func (s *OssClientSuite) SetUpTest(c *C) {
  191. }
  192. // TearDownTest runs once after all tests or benchmarks have finished running
  193. func (s *OssClientSuite) TearDownTest(c *C) {
  194. }
  195. // TestCreateBucket
  196. func (s *OssClientSuite) TestCreateBucket(c *C) {
  197. var bucketNameTest = bucketNamePrefix + RandLowStr(6)
  198. client, err := New(endpoint, accessID, accessKey)
  199. c.Assert(err, IsNil)
  200. // Create
  201. client.DeleteBucket(bucketNameTest)
  202. err = client.CreateBucket(bucketNameTest)
  203. c.Assert(err, IsNil)
  204. //sleep 3 seconds after create bucket
  205. time.Sleep(timeoutInOperation)
  206. // verify bucket is exist
  207. found, err := client.IsBucketExist(bucketNameTest)
  208. c.Assert(err, IsNil)
  209. c.Assert(found, Equals, true)
  210. res, err := client.GetBucketACL(bucketNameTest)
  211. c.Assert(err, IsNil)
  212. c.Assert(res.ACL, Equals, string(ACLPrivate))
  213. err = client.DeleteBucket(bucketNameTest)
  214. c.Assert(err, IsNil)
  215. time.Sleep(timeoutInOperation)
  216. // CreateBucket creates with ACLPublicRead
  217. err = client.CreateBucket(bucketNameTest, ACL(ACLPublicRead))
  218. c.Assert(err, IsNil)
  219. time.Sleep(timeoutInOperation)
  220. res, err = client.GetBucketACL(bucketNameTest)
  221. c.Assert(err, IsNil)
  222. c.Assert(res.ACL, Equals, string(ACLPublicRead))
  223. err = client.DeleteBucket(bucketNameTest)
  224. c.Assert(err, IsNil)
  225. time.Sleep(timeoutInOperation)
  226. // ACLPublicReadWrite
  227. err = client.CreateBucket(bucketNameTest, ACL(ACLPublicReadWrite))
  228. c.Assert(err, IsNil)
  229. time.Sleep(timeoutInOperation)
  230. res, err = client.GetBucketACL(bucketNameTest)
  231. c.Assert(err, IsNil)
  232. c.Assert(res.ACL, Equals, string(ACLPublicReadWrite))
  233. err = client.DeleteBucket(bucketNameTest)
  234. c.Assert(err, IsNil)
  235. time.Sleep(timeoutInOperation)
  236. // ACLPrivate
  237. err = client.CreateBucket(bucketNameTest, ACL(ACLPrivate))
  238. c.Assert(err, IsNil)
  239. time.Sleep(timeoutInOperation)
  240. res, err = client.GetBucketACL(bucketNameTest)
  241. c.Assert(err, IsNil)
  242. c.Assert(res.ACL, Equals, string(ACLPrivate))
  243. // Delete
  244. err = client.DeleteBucket(bucketNameTest)
  245. c.Assert(err, IsNil)
  246. // Create bucket with configuration and test GetBucketInfo
  247. for _, storage := range []StorageClassType{StorageStandard, StorageIA, StorageArchive, StorageColdArchive} {
  248. bucketNameTest := bucketNamePrefix + RandLowStr(6)
  249. err = client.CreateBucket(bucketNameTest, StorageClass(storage), ACL(ACLPublicRead))
  250. c.Assert(err, IsNil)
  251. time.Sleep(timeoutInOperation)
  252. res, err := client.GetBucketInfo(bucketNameTest)
  253. c.Assert(err, IsNil)
  254. c.Assert(res.BucketInfo.Name, Equals, bucketNameTest)
  255. c.Assert(res.BucketInfo.StorageClass, Equals, string(storage))
  256. c.Assert(res.BucketInfo.ACL, Equals, string(ACLPublicRead))
  257. // Delete
  258. err = client.DeleteBucket(bucketNameTest)
  259. c.Assert(err, IsNil)
  260. }
  261. // Error put bucket with configuration
  262. err = client.CreateBucket("ERRORBUCKETNAME", StorageClass(StorageArchive))
  263. c.Assert(err, NotNil)
  264. // Create bucket with configuration and test ListBuckets
  265. for _, storage := range []StorageClassType{StorageStandard, StorageIA, StorageArchive, StorageColdArchive} {
  266. bucketNameTest := bucketNamePrefix + RandLowStr(6)
  267. err = client.CreateBucket(bucketNameTest, StorageClass(storage))
  268. c.Assert(err, IsNil)
  269. time.Sleep(timeoutInOperation)
  270. res, err := client.GetBucketInfo(bucketNameTest)
  271. c.Assert(err, IsNil)
  272. c.Assert(res.BucketInfo.Name, Equals, bucketNameTest)
  273. c.Assert(res.BucketInfo.StorageClass, Equals, string(storage))
  274. // Delete
  275. err = client.DeleteBucket(bucketNameTest)
  276. c.Assert(err, IsNil)
  277. }
  278. }
  279. func (s *OssClientSuite) TestCreateBucketRedundancyType(c *C) {
  280. bucketNameTest := bucketNamePrefix + RandLowStr(6)
  281. client, err := New(endpoint, accessID, accessKey)
  282. c.Assert(err, IsNil)
  283. // CreateBucket creates without property
  284. err = client.CreateBucket(bucketNameTest)
  285. c.Assert(err, IsNil)
  286. client.DeleteBucket(bucketNameTest)
  287. time.Sleep(timeoutInOperation)
  288. // CreateBucket creates with RedundancyZRS
  289. err = client.CreateBucket(bucketNameTest, RedundancyType(RedundancyZRS))
  290. c.Assert(err, IsNil)
  291. res, err := client.GetBucketInfo(bucketNameTest)
  292. c.Assert(err, IsNil)
  293. c.Assert(res.BucketInfo.RedundancyType, Equals, string(RedundancyZRS))
  294. client.DeleteBucket(bucketNameTest)
  295. time.Sleep(timeoutInOperation)
  296. // CreateBucket creates with RedundancyLRS
  297. err = client.CreateBucket(bucketNameTest, RedundancyType(RedundancyLRS))
  298. c.Assert(err, IsNil)
  299. res, err = client.GetBucketInfo(bucketNameTest)
  300. c.Assert(err, IsNil)
  301. c.Assert(res.BucketInfo.RedundancyType, Equals, string(RedundancyLRS))
  302. c.Assert(res.BucketInfo.StorageClass, Equals, string(StorageStandard))
  303. client.DeleteBucket(bucketNameTest)
  304. time.Sleep(timeoutInOperation)
  305. // CreateBucket creates with ACLPublicRead RedundancyZRS
  306. err = client.CreateBucket(bucketNameTest, ACL(ACLPublicRead), RedundancyType(RedundancyZRS))
  307. c.Assert(err, IsNil)
  308. res, err = client.GetBucketInfo(bucketNameTest)
  309. c.Assert(err, IsNil)
  310. c.Assert(res.BucketInfo.RedundancyType, Equals, string(RedundancyZRS))
  311. c.Assert(res.BucketInfo.ACL, Equals, string(ACLPublicRead))
  312. client.DeleteBucket(bucketNameTest)
  313. }
  314. // TestCreateBucketNegative
  315. func (s *OssClientSuite) TestCreateBucketNegative(c *C) {
  316. client, err := New(endpoint, accessID, accessKey)
  317. c.Assert(err, IsNil)
  318. // Bucket name invalid
  319. err = client.CreateBucket("xx")
  320. c.Assert(err, NotNil)
  321. err = client.CreateBucket("XXXX")
  322. c.Assert(err, NotNil)
  323. testLogger.Println(err)
  324. err = client.CreateBucket("_bucket")
  325. c.Assert(err, NotNil)
  326. testLogger.Println(err)
  327. // ACL invalid
  328. err = client.CreateBucket(bucketNamePrefix+RandLowStr(6), ACL("InvaldAcl"))
  329. c.Assert(err, NotNil)
  330. testLogger.Println(err)
  331. }
  332. // TestDeleteBucket
  333. func (s *OssClientSuite) TestDeleteBucket(c *C) {
  334. var bucketNameTest = bucketNamePrefix + RandLowStr(6)
  335. client, err := New(endpoint, accessID, accessKey)
  336. c.Assert(err, IsNil)
  337. // Create
  338. err = client.CreateBucket(bucketNameTest)
  339. c.Assert(err, IsNil)
  340. time.Sleep(timeoutInOperation)
  341. // Check
  342. found, err := client.IsBucketExist(bucketNameTest)
  343. c.Assert(err, IsNil)
  344. c.Assert(found, Equals, true)
  345. // Delete
  346. err = client.DeleteBucket(bucketNameTest)
  347. c.Assert(err, IsNil)
  348. time.Sleep(timeoutInOperation)
  349. // Check
  350. found, err = client.IsBucketExist(bucketNameTest)
  351. c.Assert(err, IsNil)
  352. c.Assert(found, Equals, false)
  353. err = client.DeleteBucket(bucketNameTest)
  354. c.Assert(err, NotNil)
  355. }
  356. // TestDeleteBucketNegative
  357. func (s *OssClientSuite) TestDeleteBucketNegative(c *C) {
  358. var bucketNameTest = bucketNamePrefix + RandLowStr(6)
  359. client, err := New(endpoint, accessID, accessKey)
  360. c.Assert(err, IsNil)
  361. // Bucket name invalid
  362. err = client.DeleteBucket("xx")
  363. c.Assert(err, NotNil)
  364. err = client.DeleteBucket("XXXX")
  365. c.Assert(err, NotNil)
  366. err = client.DeleteBucket("_bucket")
  367. c.Assert(err, NotNil)
  368. // Delete no exist bucket
  369. err = client.DeleteBucket("notexist")
  370. c.Assert(err, NotNil)
  371. // No permission to delete, this ak/sk for js sdk
  372. err = client.CreateBucket(bucketNameTest)
  373. c.Assert(err, IsNil)
  374. accessID := "<accessKeyId>"
  375. accessKey := "<accessKeySecret>"
  376. clientOtherUser, err := New(endpoint, accessID, accessKey)
  377. c.Assert(err, IsNil)
  378. err = clientOtherUser.DeleteBucket(bucketNameTest)
  379. c.Assert(err, NotNil)
  380. err = client.DeleteBucket(bucketNameTest)
  381. c.Assert(err, IsNil)
  382. }
  383. // TestListBucket
  384. func (s *OssClientSuite) TestListBucket(c *C) {
  385. var prefix = bucketNamePrefix + RandLowStr(6)
  386. var bucketNameLbOne = prefix + "tlb1"
  387. var bucketNameLbTwo = prefix + "tlb2"
  388. var bucketNameLbThree = prefix + "tlb3"
  389. client, err := New(endpoint, accessID, accessKey)
  390. c.Assert(err, IsNil)
  391. // CreateBucket
  392. err = client.CreateBucket(bucketNameLbOne)
  393. c.Assert(err, IsNil)
  394. err = client.CreateBucket(bucketNameLbTwo)
  395. c.Assert(err, IsNil)
  396. err = client.CreateBucket(bucketNameLbThree)
  397. c.Assert(err, IsNil)
  398. // ListBuckets, specified prefix
  399. var respHeader http.Header
  400. lbr, err := client.ListBuckets(Prefix(prefix), MaxKeys(2), GetResponseHeader(&respHeader))
  401. c.Assert(GetRequestId(respHeader) != "", Equals, true)
  402. c.Assert(err, IsNil)
  403. c.Assert(len(lbr.Buckets), Equals, 2)
  404. // ListBuckets, specified max keys
  405. lbr, err = client.ListBuckets(MaxKeys(2))
  406. c.Assert(err, IsNil)
  407. c.Assert(len(lbr.Buckets), Equals, 2)
  408. // ListBuckets, specified max keys
  409. lbr, err = client.ListBuckets(Marker(bucketNameLbOne), MaxKeys(1))
  410. c.Assert(err, IsNil)
  411. c.Assert(len(lbr.Buckets), Equals, 1)
  412. // ListBuckets, specified max keys
  413. lbr, err = client.ListBuckets(Marker(bucketNameLbOne))
  414. c.Assert(err, IsNil)
  415. c.Assert(len(lbr.Buckets) >= 2, Equals, true)
  416. // DeleteBucket
  417. err = client.DeleteBucket(bucketNameLbOne)
  418. c.Assert(err, IsNil)
  419. err = client.DeleteBucket(bucketNameLbTwo)
  420. c.Assert(err, IsNil)
  421. err = client.DeleteBucket(bucketNameLbThree)
  422. c.Assert(err, IsNil)
  423. }
  424. // TestListBucket
  425. func (s *OssClientSuite) TestIsBucketExist(c *C) {
  426. var prefix = bucketNamePrefix + RandLowStr(6)
  427. var bucketNameLbOne = prefix + "tibe1"
  428. var bucketNameLbTwo = prefix + "tibe11"
  429. var bucketNameLbThree = prefix + "tibe111"
  430. client, err := New(endpoint, accessID, accessKey)
  431. c.Assert(err, IsNil)
  432. // CreateBucket
  433. err = client.CreateBucket(bucketNameLbOne)
  434. c.Assert(err, IsNil)
  435. err = client.CreateBucket(bucketNameLbTwo)
  436. c.Assert(err, IsNil)
  437. err = client.CreateBucket(bucketNameLbThree)
  438. c.Assert(err, IsNil)
  439. // Exist
  440. exist, err := client.IsBucketExist(bucketNameLbTwo)
  441. c.Assert(err, IsNil)
  442. c.Assert(exist, Equals, true)
  443. exist, err = client.IsBucketExist(bucketNameLbThree)
  444. c.Assert(err, IsNil)
  445. c.Assert(exist, Equals, true)
  446. exist, err = client.IsBucketExist(bucketNameLbOne)
  447. c.Assert(err, IsNil)
  448. c.Assert(exist, Equals, true)
  449. // Not exist
  450. exist, err = client.IsBucketExist(prefix + "tibe")
  451. c.Assert(err, IsNil)
  452. c.Assert(exist, Equals, false)
  453. exist, err = client.IsBucketExist(prefix + "tibe1111")
  454. c.Assert(err, IsNil)
  455. c.Assert(exist, Equals, false)
  456. // Negative
  457. exist, err = client.IsBucketExist("BucketNameInvalid")
  458. c.Assert(err, NotNil)
  459. // DeleteBucket
  460. err = client.DeleteBucket(bucketNameLbOne)
  461. c.Assert(err, IsNil)
  462. err = client.DeleteBucket(bucketNameLbTwo)
  463. c.Assert(err, IsNil)
  464. err = client.DeleteBucket(bucketNameLbThree)
  465. c.Assert(err, IsNil)
  466. }
  467. // TestSetBucketAcl
  468. func (s *OssClientSuite) TestSetBucketAcl(c *C) {
  469. var bucketNameTest = bucketNamePrefix + RandLowStr(6)
  470. client, err := New(endpoint, accessID, accessKey)
  471. c.Assert(err, IsNil)
  472. // Private
  473. err = client.CreateBucket(bucketNameTest)
  474. c.Assert(err, IsNil)
  475. res, err := client.GetBucketACL(bucketNameTest)
  476. c.Assert(err, IsNil)
  477. c.Assert(res.ACL, Equals, string(ACLPrivate))
  478. // Set ACL_PUBLIC_R
  479. err = client.SetBucketACL(bucketNameTest, ACLPublicRead)
  480. c.Assert(err, IsNil)
  481. time.Sleep(timeoutInOperation)
  482. res, err = client.GetBucketACL(bucketNameTest)
  483. c.Assert(err, IsNil)
  484. c.Assert(res.ACL, Equals, string(ACLPublicRead))
  485. // Set ACL_PUBLIC_RW
  486. err = client.SetBucketACL(bucketNameTest, ACLPublicReadWrite)
  487. c.Assert(err, IsNil)
  488. time.Sleep(timeoutInOperation)
  489. res, err = client.GetBucketACL(bucketNameTest)
  490. c.Assert(err, IsNil)
  491. c.Assert(res.ACL, Equals, string(ACLPublicReadWrite))
  492. // Set ACL_PUBLIC_RW
  493. err = client.SetBucketACL(bucketNameTest, ACLPrivate)
  494. c.Assert(err, IsNil)
  495. time.Sleep(timeoutInOperation)
  496. res, err = client.GetBucketACL(bucketNameTest)
  497. c.Assert(err, IsNil)
  498. c.Assert(res.ACL, Equals, string(ACLPrivate))
  499. err = client.DeleteBucket(bucketNameTest)
  500. c.Assert(err, IsNil)
  501. }
  502. // TestSetBucketAclNegative
  503. func (s *OssClientSuite) TestBucketAclNegative(c *C) {
  504. var bucketNameTest = bucketNamePrefix + RandLowStr(6)
  505. client, err := New(endpoint, accessID, accessKey)
  506. c.Assert(err, IsNil)
  507. err = client.CreateBucket(bucketNameTest)
  508. c.Assert(err, IsNil)
  509. err = client.SetBucketACL(bucketNameTest, "InvalidACL")
  510. c.Assert(err, NotNil)
  511. testLogger.Println(err)
  512. err = client.DeleteBucket(bucketNameTest)
  513. c.Assert(err, IsNil)
  514. }
  515. // TestGetBucketAcl
  516. func (s *OssClientSuite) TestGetBucketAcl(c *C) {
  517. var bucketNameTest = bucketNamePrefix + RandLowStr(6)
  518. client, err := New(endpoint, accessID, accessKey)
  519. c.Assert(err, IsNil)
  520. // Private
  521. err = client.CreateBucket(bucketNameTest)
  522. c.Assert(err, IsNil)
  523. time.Sleep(timeoutInOperation)
  524. res, err := client.GetBucketACL(bucketNameTest)
  525. c.Assert(err, IsNil)
  526. c.Assert(res.ACL, Equals, string(ACLPrivate))
  527. err = client.DeleteBucket(bucketNameTest)
  528. c.Assert(err, IsNil)
  529. time.Sleep(timeoutInOperation)
  530. // PublicRead
  531. err = client.CreateBucket(bucketNameTest, ACL(ACLPublicRead))
  532. c.Assert(err, IsNil)
  533. time.Sleep(timeoutInOperation)
  534. res, err = client.GetBucketACL(bucketNameTest)
  535. c.Assert(err, IsNil)
  536. c.Assert(res.ACL, Equals, string(ACLPublicRead))
  537. err = client.DeleteBucket(bucketNameTest)
  538. c.Assert(err, IsNil)
  539. time.Sleep(timeoutInOperation)
  540. // PublicReadWrite
  541. err = client.CreateBucket(bucketNameTest, ACL(ACLPublicReadWrite))
  542. c.Assert(err, IsNil)
  543. time.Sleep(timeoutInOperation)
  544. res, err = client.GetBucketACL(bucketNameTest)
  545. c.Assert(err, IsNil)
  546. c.Assert(res.ACL, Equals, string(ACLPublicReadWrite))
  547. err = client.DeleteBucket(bucketNameTest)
  548. c.Assert(err, IsNil)
  549. }
  550. // TestGetBucketAcl
  551. func (s *OssClientSuite) TestGetBucketLocation(c *C) {
  552. var bucketNameTest = bucketNamePrefix + RandLowStr(6)
  553. client, err := New(endpoint, accessID, accessKey)
  554. c.Assert(err, IsNil)
  555. // Private
  556. err = client.CreateBucket(bucketNameTest)
  557. c.Assert(err, IsNil)
  558. loc, err := client.GetBucketLocation(bucketNameTest)
  559. c.Assert(strings.HasPrefix(loc, "oss-"), Equals, true)
  560. err = client.DeleteBucket(bucketNameTest)
  561. c.Assert(err, IsNil)
  562. }
  563. // TestGetBucketLocationNegative
  564. func (s *OssClientSuite) TestGetBucketLocationNegative(c *C) {
  565. var bucketNameTest = bucketNamePrefix + RandLowStr(6)
  566. client, err := New(endpoint, accessID, accessKey)
  567. c.Assert(err, IsNil)
  568. // Not exist
  569. _, err = client.GetBucketLocation(bucketNameTest)
  570. c.Assert(err, NotNil)
  571. // Not exist
  572. _, err = client.GetBucketLocation("InvalidBucketName_")
  573. c.Assert(err, NotNil)
  574. }
  575. // TestSetBucketLifecycle
  576. func (s *OssClientSuite) TestSetBucketLifecycle(c *C) {
  577. var bucketNameTest = bucketNamePrefix + RandLowStr(6)
  578. var rule1 = BuildLifecycleRuleByDate("rule1", "one", true, 2015, 11, 11)
  579. var rule2 = BuildLifecycleRuleByDays("rule2", "two", true, 3)
  580. client, err := New(endpoint, accessID, accessKey)
  581. c.Assert(err, IsNil)
  582. err = client.CreateBucket(bucketNameTest)
  583. c.Assert(err, IsNil)
  584. // Set single rule
  585. var rules = []LifecycleRule{rule1}
  586. err = client.SetBucketLifecycle(bucketNameTest, rules)
  587. c.Assert(err, IsNil)
  588. // Double set rule
  589. err = client.SetBucketLifecycle(bucketNameTest, rules)
  590. c.Assert(err, IsNil)
  591. res, err := client.GetBucketLifecycle(bucketNameTest)
  592. c.Assert(err, IsNil)
  593. c.Assert(len(res.Rules), Equals, 1)
  594. c.Assert(res.Rules[0].ID, Equals, "rule1")
  595. err = client.DeleteBucketLifecycle(bucketNameTest)
  596. c.Assert(err, IsNil)
  597. // Set two rules
  598. rules = []LifecycleRule{rule1, rule2}
  599. err = client.SetBucketLifecycle(bucketNameTest, rules)
  600. c.Assert(err, IsNil)
  601. // Eliminate effect of cache
  602. time.Sleep(timeoutInOperation)
  603. res, err = client.GetBucketLifecycle(bucketNameTest)
  604. c.Assert(err, IsNil)
  605. c.Assert(len(res.Rules), Equals, 2)
  606. c.Assert(res.Rules[0].ID, Equals, "rule1")
  607. c.Assert(res.Rules[1].ID, Equals, "rule2")
  608. err = client.DeleteBucket(bucketNameTest)
  609. c.Assert(err, IsNil)
  610. }
  611. // TestSetBucketLifecycleNew
  612. func (s *OssClientSuite) TestSetBucketLifecycleNew(c *C) {
  613. var bucketNameTest = bucketNamePrefix + RandLowStr(6)
  614. client, err := New(endpoint, accessID, accessKey)
  615. c.Assert(err, IsNil)
  616. err = client.CreateBucket(bucketNameTest)
  617. c.Assert(err, IsNil)
  618. //invalid status of lifecyclerule
  619. expiration := LifecycleExpiration{
  620. Days: 30,
  621. }
  622. rule := LifecycleRule{
  623. ID: "rule1",
  624. Prefix: "one",
  625. Status: "Invalid",
  626. Expiration: &expiration,
  627. }
  628. rules := []LifecycleRule{rule}
  629. err = client.SetBucketLifecycle(bucketNameTest, rules)
  630. c.Assert(err, NotNil)
  631. //invalid value of CreatedBeforeDate
  632. expiration = LifecycleExpiration{
  633. CreatedBeforeDate: RandStr(10),
  634. }
  635. rule = LifecycleRule{
  636. ID: "rule1",
  637. Prefix: "one",
  638. Status: "Enabled",
  639. Expiration: &expiration,
  640. }
  641. rules = []LifecycleRule{rule}
  642. err = client.SetBucketLifecycle(bucketNameTest, rules)
  643. c.Assert(err, NotNil)
  644. //invalid value of Days
  645. abortMPU := LifecycleAbortMultipartUpload{
  646. Days: -30,
  647. }
  648. rule = LifecycleRule{
  649. ID: "rule1",
  650. Prefix: "one",
  651. Status: "Enabled",
  652. AbortMultipartUpload: &abortMPU,
  653. }
  654. rules = []LifecycleRule{rule}
  655. err = client.SetBucketLifecycle(bucketNameTest, rules)
  656. c.Assert(err, NotNil)
  657. expiration = LifecycleExpiration{
  658. CreatedBeforeDate: "2015-11-11T00:00:00.000Z",
  659. }
  660. rule1 := LifecycleRule{
  661. ID: "rule1",
  662. Prefix: "one",
  663. Status: "Enabled",
  664. Expiration: &expiration,
  665. }
  666. abortMPU = LifecycleAbortMultipartUpload{
  667. Days: 30,
  668. }
  669. rule2 := LifecycleRule{
  670. ID: "rule2",
  671. Prefix: "two",
  672. Status: "Enabled",
  673. Expiration: &expiration,
  674. AbortMultipartUpload: &abortMPU,
  675. }
  676. transition1 := LifecycleTransition{
  677. Days: 3,
  678. StorageClass: StorageIA,
  679. }
  680. transition2 := LifecycleTransition{
  681. Days: 30,
  682. StorageClass: StorageArchive,
  683. }
  684. transitions := []LifecycleTransition{transition1, transition2}
  685. rule3 := LifecycleRule{
  686. ID: "rule3",
  687. Prefix: "three",
  688. Status: "Enabled",
  689. AbortMultipartUpload: &abortMPU,
  690. Transitions: transitions,
  691. }
  692. // Set single rule
  693. rules = []LifecycleRule{rule1}
  694. err = client.SetBucketLifecycle(bucketNameTest, rules)
  695. c.Assert(err, IsNil)
  696. res, err := client.GetBucketLifecycle(bucketNameTest)
  697. c.Assert(err, IsNil)
  698. c.Assert(len(res.Rules), Equals, 1)
  699. c.Assert(res.Rules[0].ID, Equals, "rule1")
  700. c.Assert(res.Rules[0].Expiration, NotNil)
  701. c.Assert(res.Rules[0].Expiration.CreatedBeforeDate, Equals, "2015-11-11T00:00:00.000Z")
  702. err = client.DeleteBucketLifecycle(bucketNameTest)
  703. c.Assert(err, IsNil)
  704. // Set two rule: rule1 and rule2
  705. rules = []LifecycleRule{rule1, rule2}
  706. err = client.SetBucketLifecycle(bucketNameTest, rules)
  707. c.Assert(err, IsNil)
  708. res, err = client.GetBucketLifecycle(bucketNameTest)
  709. c.Assert(err, IsNil)
  710. c.Assert(len(res.Rules), Equals, 2)
  711. c.Assert(res.Rules[0].ID, Equals, "rule1")
  712. c.Assert(res.Rules[0].Expiration, NotNil)
  713. c.Assert(res.Rules[0].Expiration.CreatedBeforeDate, Equals, "2015-11-11T00:00:00.000Z")
  714. c.Assert(res.Rules[1].ID, Equals, "rule2")
  715. c.Assert(res.Rules[1].Expiration, NotNil)
  716. c.Assert(res.Rules[1].Expiration.CreatedBeforeDate, Equals, "2015-11-11T00:00:00.000Z")
  717. c.Assert(res.Rules[1].AbortMultipartUpload, NotNil)
  718. c.Assert(res.Rules[1].AbortMultipartUpload.Days, Equals, 30)
  719. err = client.DeleteBucketLifecycle(bucketNameTest)
  720. c.Assert(err, IsNil)
  721. // Set two rule: rule2 and rule3
  722. rules = []LifecycleRule{rule2, rule3}
  723. err = client.SetBucketLifecycle(bucketNameTest, rules)
  724. c.Assert(err, IsNil)
  725. res, err = client.GetBucketLifecycle(bucketNameTest)
  726. c.Assert(err, IsNil)
  727. c.Assert(len(res.Rules), Equals, 2)
  728. c.Assert(res.Rules[0].ID, Equals, "rule2")
  729. c.Assert(res.Rules[0].Expiration, NotNil)
  730. c.Assert(res.Rules[0].Expiration.CreatedBeforeDate, Equals, "2015-11-11T00:00:00.000Z")
  731. c.Assert(res.Rules[0].AbortMultipartUpload, NotNil)
  732. c.Assert(res.Rules[0].AbortMultipartUpload.Days, Equals, 30)
  733. c.Assert(res.Rules[1].ID, Equals, "rule3")
  734. c.Assert(res.Rules[1].AbortMultipartUpload, NotNil)
  735. c.Assert(res.Rules[1].AbortMultipartUpload.Days, Equals, 30)
  736. c.Assert(len(res.Rules[1].Transitions), Equals, 2)
  737. c.Assert(res.Rules[1].Transitions[0].StorageClass, Equals, StorageIA)
  738. c.Assert(res.Rules[1].Transitions[0].Days, Equals, 3)
  739. c.Assert(res.Rules[1].Transitions[1].StorageClass, Equals, StorageArchive)
  740. c.Assert(res.Rules[1].Transitions[1].Days, Equals, 30)
  741. err = client.DeleteBucketLifecycle(bucketNameTest)
  742. c.Assert(err, IsNil)
  743. // Set two rule: rule1 and rule3
  744. rules = []LifecycleRule{rule1, rule3}
  745. err = client.SetBucketLifecycle(bucketNameTest, rules)
  746. c.Assert(err, IsNil)
  747. res, err = client.GetBucketLifecycle(bucketNameTest)
  748. c.Assert(err, IsNil)
  749. c.Assert(len(res.Rules), Equals, 2)
  750. c.Assert(res.Rules[0].ID, Equals, "rule1")
  751. c.Assert(res.Rules[0].Expiration, NotNil)
  752. c.Assert(res.Rules[0].Expiration.CreatedBeforeDate, Equals, "2015-11-11T00:00:00.000Z")
  753. c.Assert(res.Rules[1].ID, Equals, "rule3")
  754. c.Assert(res.Rules[1].AbortMultipartUpload, NotNil)
  755. c.Assert(res.Rules[1].AbortMultipartUpload.Days, Equals, 30)
  756. c.Assert(len(res.Rules[1].Transitions), Equals, 2)
  757. c.Assert(res.Rules[1].Transitions[0].StorageClass, Equals, StorageIA)
  758. c.Assert(res.Rules[1].Transitions[0].Days, Equals, 3)
  759. c.Assert(res.Rules[1].Transitions[1].StorageClass, Equals, StorageArchive)
  760. c.Assert(res.Rules[1].Transitions[1].Days, Equals, 30)
  761. err = client.DeleteBucketLifecycle(bucketNameTest)
  762. c.Assert(err, IsNil)
  763. // Set three rules
  764. rules = []LifecycleRule{rule1, rule2, rule3}
  765. err = client.SetBucketLifecycle(bucketNameTest, rules)
  766. c.Assert(err, IsNil)
  767. res, err = client.GetBucketLifecycle(bucketNameTest)
  768. c.Assert(err, IsNil)
  769. c.Assert(len(res.Rules), Equals, 3)
  770. c.Assert(res.Rules[0].ID, Equals, "rule1")
  771. c.Assert(res.Rules[0].Expiration, NotNil)
  772. c.Assert(res.Rules[0].Expiration.CreatedBeforeDate, Equals, "2015-11-11T00:00:00.000Z")
  773. c.Assert(res.Rules[1].ID, Equals, "rule2")
  774. c.Assert(res.Rules[1].Expiration, NotNil)
  775. c.Assert(res.Rules[1].Expiration.CreatedBeforeDate, Equals, "2015-11-11T00:00:00.000Z")
  776. c.Assert(res.Rules[1].AbortMultipartUpload, NotNil)
  777. c.Assert(res.Rules[1].AbortMultipartUpload.Days, Equals, 30)
  778. c.Assert(res.Rules[2].ID, Equals, "rule3")
  779. c.Assert(res.Rules[2].AbortMultipartUpload, NotNil)
  780. c.Assert(res.Rules[2].AbortMultipartUpload.Days, Equals, 30)
  781. c.Assert(len(res.Rules[2].Transitions), Equals, 2)
  782. c.Assert(res.Rules[2].Transitions[0].StorageClass, Equals, StorageIA)
  783. c.Assert(res.Rules[2].Transitions[0].Days, Equals, 3)
  784. c.Assert(res.Rules[2].Transitions[1].StorageClass, Equals, StorageArchive)
  785. c.Assert(res.Rules[2].Transitions[1].Days, Equals, 30)
  786. err = client.DeleteBucket(bucketNameTest)
  787. c.Assert(err, IsNil)
  788. }
  789. // TestSetBucketLifecycleAboutVersionObject
  790. func (s *OssClientSuite) TestSetBucketLifecycleAboutVersionObject(c *C) {
  791. var bucketNameTest = bucketNamePrefix + RandLowStr(6)
  792. client, err := New(endpoint, accessID, accessKey)
  793. c.Assert(err, IsNil)
  794. err = client.CreateBucket(bucketNameTest)
  795. c.Assert(err, IsNil)
  796. deleteMark := true
  797. expiration := LifecycleExpiration{
  798. ExpiredObjectDeleteMarker: &deleteMark,
  799. }
  800. versionExpiration := LifecycleVersionExpiration{
  801. NoncurrentDays: 20,
  802. }
  803. versionTransition := LifecycleVersionTransition{
  804. NoncurrentDays: 10,
  805. StorageClass: "IA",
  806. }
  807. rule := LifecycleRule{
  808. Status: "Enabled",
  809. Expiration: &expiration,
  810. NonVersionExpiration: &versionExpiration,
  811. NonVersionTransition: &versionTransition,
  812. }
  813. rules := []LifecycleRule{rule}
  814. err = client.SetBucketLifecycle(bucketNameTest, rules)
  815. c.Assert(err, IsNil)
  816. res, err := client.GetBucketLifecycle(bucketNameTest)
  817. c.Assert(err, IsNil)
  818. c.Assert(res.Rules[0].Expiration, NotNil)
  819. c.Assert(res.Rules[0].Expiration.Days, Equals, 0)
  820. c.Assert(res.Rules[0].Expiration.Date, Equals, "")
  821. c.Assert(*(res.Rules[0].Expiration.ExpiredObjectDeleteMarker), Equals, true)
  822. c.Assert(res.Rules[0].NonVersionExpiration.NoncurrentDays, Equals, 20)
  823. c.Assert(res.Rules[0].NonVersionTransition.NoncurrentDays, Equals, 10)
  824. c.Assert(res.Rules[0].NonVersionTransition.StorageClass, Equals, StorageClassType("IA"))
  825. err = client.DeleteBucket(bucketNameTest)
  826. c.Assert(err, IsNil)
  827. }
  828. // TestDeleteBucketLifecycle
  829. func (s *OssClientSuite) TestDeleteBucketLifecycle(c *C) {
  830. var bucketNameTest = bucketNamePrefix + RandLowStr(6)
  831. var rule1 = BuildLifecycleRuleByDate("rule1", "one", true, 2015, 11, 11)
  832. var rule2 = BuildLifecycleRuleByDays("rule2", "two", true, 3)
  833. var rules = []LifecycleRule{rule1, rule2}
  834. client, err := New(endpoint, accessID, accessKey)
  835. c.Assert(err, IsNil)
  836. err = client.CreateBucket(bucketNameTest)
  837. c.Assert(err, IsNil)
  838. //time.Sleep(timeoutInOperation)
  839. err = client.DeleteBucketLifecycle(bucketNameTest)
  840. c.Assert(err, IsNil)
  841. err = client.SetBucketLifecycle(bucketNameTest, rules)
  842. c.Assert(err, IsNil)
  843. //time.Sleep(timeoutInOperation)
  844. res, err := client.GetBucketLifecycle(bucketNameTest)
  845. c.Assert(err, IsNil)
  846. c.Assert(len(res.Rules), Equals, 2)
  847. // Delete
  848. err = client.DeleteBucketLifecycle(bucketNameTest)
  849. c.Assert(err, IsNil)
  850. //time.Sleep(timeoutInOperation)
  851. res, err = client.GetBucketLifecycle(bucketNameTest)
  852. c.Assert(err, NotNil)
  853. // Eliminate effect of cache
  854. //time.Sleep(timeoutInOperation)
  855. // Delete when not set
  856. err = client.DeleteBucketLifecycle(bucketNameTest)
  857. c.Assert(err, IsNil)
  858. err = client.DeleteBucket(bucketNameTest)
  859. c.Assert(err, IsNil)
  860. }
  861. // TestSetBucketLifecycleNegative
  862. func (s *OssClientSuite) TestBucketLifecycleNegative(c *C) {
  863. var bucketNameTest = bucketNamePrefix + RandLowStr(6)
  864. var rules = []LifecycleRule{}
  865. client, err := New(endpoint, accessID, accessKey)
  866. c.Assert(err, IsNil)
  867. err = client.CreateBucket(bucketNameTest)
  868. c.Assert(err, IsNil)
  869. // Set with no rule
  870. err = client.SetBucketLifecycle(bucketNameTest, rules)
  871. c.Assert(err, NotNil)
  872. err = client.DeleteBucket(bucketNameTest)
  873. c.Assert(err, IsNil)
  874. // Not exist
  875. err = client.SetBucketLifecycle(bucketNameTest, rules)
  876. c.Assert(err, NotNil)
  877. // Not exist
  878. _, err = client.GetBucketLifecycle(bucketNameTest)
  879. c.Assert(err, NotNil)
  880. // Not exist
  881. err = client.DeleteBucketLifecycle(bucketNameTest)
  882. c.Assert(err, NotNil)
  883. }
  884. // TestSetBucketReferer
  885. func (s *OssClientSuite) TestSetBucketReferer(c *C) {
  886. var bucketNameTest = bucketNamePrefix + RandLowStr(6)
  887. var referers = []string{"http://www.aliyun.com", "https://www.aliyun.com"}
  888. client, err := New(endpoint, accessID, accessKey)
  889. c.Assert(err, IsNil)
  890. err = client.CreateBucket(bucketNameTest)
  891. c.Assert(err, IsNil)
  892. time.Sleep(timeoutInOperation)
  893. res, err := client.GetBucketReferer(bucketNameTest)
  894. c.Assert(res.AllowEmptyReferer, Equals, true)
  895. c.Assert(len(res.RefererList), Equals, 0)
  896. // Set referers
  897. err = client.SetBucketReferer(bucketNameTest, referers, false)
  898. c.Assert(err, IsNil)
  899. time.Sleep(timeoutInOperation)
  900. res, err = client.GetBucketReferer(bucketNameTest)
  901. c.Assert(res.AllowEmptyReferer, Equals, false)
  902. c.Assert(len(res.RefererList), Equals, 2)
  903. c.Assert(res.RefererList[0], Equals, "http://www.aliyun.com")
  904. c.Assert(res.RefererList[1], Equals, "https://www.aliyun.com")
  905. // Reset referer, referers empty
  906. referers = []string{""}
  907. err = client.SetBucketReferer(bucketNameTest, referers, true)
  908. c.Assert(err, IsNil)
  909. referers = []string{}
  910. err = client.SetBucketReferer(bucketNameTest, referers, true)
  911. c.Assert(err, IsNil)
  912. res, err = client.GetBucketReferer(bucketNameTest)
  913. c.Assert(res.AllowEmptyReferer, Equals, true)
  914. c.Assert(len(res.RefererList), Equals, 0)
  915. err = client.DeleteBucket(bucketNameTest)
  916. c.Assert(err, IsNil)
  917. }
  918. // TestSetBucketRefererNegative
  919. func (s *OssClientSuite) TestBucketRefererNegative(c *C) {
  920. var bucketNameTest = bucketNamePrefix + RandLowStr(6)
  921. var referers = []string{""}
  922. client, err := New(endpoint, accessID, accessKey)
  923. c.Assert(err, IsNil)
  924. // Not exist
  925. _, err = client.GetBucketReferer(bucketNameTest)
  926. c.Assert(err, NotNil)
  927. testLogger.Println(err)
  928. // Not exist
  929. err = client.SetBucketReferer(bucketNameTest, referers, true)
  930. c.Assert(err, NotNil)
  931. testLogger.Println(err)
  932. }
  933. // TestSetBucketLogging
  934. func (s *OssClientSuite) TestSetBucketLogging(c *C) {
  935. var bucketNameTest = bucketNamePrefix + RandLowStr(6)
  936. var bucketNameTarget = bucketNameTest + "-target"
  937. client, err := New(endpoint, accessID, accessKey)
  938. c.Assert(err, IsNil)
  939. err = client.CreateBucket(bucketNameTest)
  940. c.Assert(err, IsNil)
  941. err = client.CreateBucket(bucketNameTarget)
  942. c.Assert(err, IsNil)
  943. time.Sleep(timeoutInOperation)
  944. // Set logging
  945. err = client.SetBucketLogging(bucketNameTest, bucketNameTarget, "prefix", true)
  946. c.Assert(err, IsNil)
  947. // Reset
  948. err = client.SetBucketLogging(bucketNameTest, bucketNameTarget, "prefix", false)
  949. c.Assert(err, IsNil)
  950. time.Sleep(timeoutInOperation)
  951. res, err := client.GetBucketLogging(bucketNameTest)
  952. c.Assert(err, IsNil)
  953. c.Assert(res.LoggingEnabled.TargetBucket, Equals, "")
  954. c.Assert(res.LoggingEnabled.TargetPrefix, Equals, "")
  955. err = client.DeleteBucketLogging(bucketNameTest)
  956. c.Assert(err, IsNil)
  957. // Set to self
  958. err = client.SetBucketLogging(bucketNameTest, bucketNameTest, "prefix", true)
  959. c.Assert(err, IsNil)
  960. err = client.DeleteBucket(bucketNameTest)
  961. c.Assert(err, IsNil)
  962. err = client.DeleteBucket(bucketNameTarget)
  963. c.Assert(err, IsNil)
  964. }
  965. // TestDeleteBucketLogging
  966. func (s *OssClientSuite) TestDeleteBucketLogging(c *C) {
  967. var bucketNameTest = bucketNamePrefix + RandLowStr(6)
  968. var bucketNameTarget = bucketNameTest + "-target"
  969. client, err := New(endpoint, accessID, accessKey)
  970. c.Assert(err, IsNil)
  971. err = client.CreateBucket(bucketNameTest)
  972. c.Assert(err, IsNil)
  973. err = client.CreateBucket(bucketNameTarget)
  974. c.Assert(err, IsNil)
  975. time.Sleep(timeoutInOperation)
  976. // Get when not set
  977. res, err := client.GetBucketLogging(bucketNameTest)
  978. c.Assert(err, IsNil)
  979. c.Assert(res.LoggingEnabled.TargetBucket, Equals, "")
  980. c.Assert(res.LoggingEnabled.TargetPrefix, Equals, "")
  981. // Set
  982. err = client.SetBucketLogging(bucketNameTest, bucketNameTarget, "prefix", true)
  983. c.Assert(err, IsNil)
  984. // Get
  985. time.Sleep(timeoutInOperation)
  986. res, err = client.GetBucketLogging(bucketNameTest)
  987. c.Assert(err, IsNil)
  988. c.Assert(res.LoggingEnabled.TargetBucket, Equals, bucketNameTarget)
  989. c.Assert(res.LoggingEnabled.TargetPrefix, Equals, "prefix")
  990. // Set
  991. err = client.SetBucketLogging(bucketNameTest, bucketNameTarget, "prefix", false)
  992. c.Assert(err, IsNil)
  993. // Get
  994. time.Sleep(timeoutInOperation)
  995. res, err = client.GetBucketLogging(bucketNameTest)
  996. c.Assert(err, IsNil)
  997. c.Assert(res.LoggingEnabled.TargetBucket, Equals, "")
  998. c.Assert(res.LoggingEnabled.TargetPrefix, Equals, "")
  999. // Delete
  1000. err = client.DeleteBucketLogging(bucketNameTest)
  1001. c.Assert(err, IsNil)
  1002. // Get after delete
  1003. time.Sleep(timeoutInOperation)
  1004. res, err = client.GetBucketLogging(bucketNameTest)
  1005. c.Assert(err, IsNil)
  1006. c.Assert(res.LoggingEnabled.TargetBucket, Equals, "")
  1007. c.Assert(res.LoggingEnabled.TargetPrefix, Equals, "")
  1008. err = client.DeleteBucket(bucketNameTest)
  1009. c.Assert(err, IsNil)
  1010. err = client.DeleteBucket(bucketNameTarget)
  1011. c.Assert(err, IsNil)
  1012. }
  1013. // TestSetBucketLoggingNegative
  1014. func (s *OssClientSuite) TestSetBucketLoggingNegative(c *C) {
  1015. var bucketNameTest = bucketNamePrefix + RandLowStr(6)
  1016. var bucketNameTarget = bucketNameTest + "-target"
  1017. client, err := New(endpoint, accessID, accessKey)
  1018. c.Assert(err, IsNil)
  1019. // Not exist
  1020. _, err = client.GetBucketLogging(bucketNameTest)
  1021. c.Assert(err, NotNil)
  1022. // Not exist
  1023. err = client.SetBucketLogging(bucketNameTest, "targetbucket", "prefix", true)
  1024. c.Assert(err, NotNil)
  1025. // Not exist
  1026. err = client.DeleteBucketLogging(bucketNameTest)
  1027. c.Assert(err, NotNil)
  1028. err = client.CreateBucket(bucketNameTest)
  1029. c.Assert(err, IsNil)
  1030. time.Sleep(timeoutInOperation)
  1031. // Target bucket not exist
  1032. err = client.SetBucketLogging(bucketNameTest, bucketNameTarget, "prefix", true)
  1033. c.Assert(err, NotNil)
  1034. // Parameter invalid
  1035. err = client.SetBucketLogging(bucketNameTest, "XXXX", "prefix", true)
  1036. c.Assert(err, NotNil)
  1037. err = client.SetBucketLogging(bucketNameTest, "xx", "prefix", true)
  1038. c.Assert(err, NotNil)
  1039. err = client.DeleteBucket(bucketNameTest)
  1040. c.Assert(err, IsNil)
  1041. }
  1042. // TestSetBucketWebsite
  1043. func (s *OssClientSuite) TestSetBucketWebsite(c *C) {
  1044. var bucketNameTest = bucketNamePrefix + RandLowStr(6)
  1045. var indexWebsite = "myindex.html"
  1046. var errorWebsite = "myerror.html"
  1047. client, err := New(endpoint, accessID, accessKey)
  1048. c.Assert(err, IsNil)
  1049. time.Sleep(timeoutInOperation)
  1050. err = client.CreateBucket(bucketNameTest)
  1051. c.Assert(err, IsNil)
  1052. // Set
  1053. err = client.SetBucketWebsite(bucketNameTest, indexWebsite, errorWebsite)
  1054. c.Assert(err, IsNil)
  1055. // Double set
  1056. err = client.SetBucketWebsite(bucketNameTest, indexWebsite, errorWebsite)
  1057. c.Assert(err, IsNil)
  1058. res, err := client.GetBucketWebsite(bucketNameTest)
  1059. c.Assert(err, IsNil)
  1060. c.Assert(res.IndexDocument.Suffix, Equals, indexWebsite)
  1061. c.Assert(res.ErrorDocument.Key, Equals, errorWebsite)
  1062. // Reset
  1063. err = client.SetBucketWebsite(bucketNameTest, "your"+indexWebsite, "your"+errorWebsite)
  1064. c.Assert(err, IsNil)
  1065. time.Sleep(timeoutInOperation)
  1066. res, err = client.GetBucketWebsite(bucketNameTest)
  1067. c.Assert(err, IsNil)
  1068. c.Assert(res.IndexDocument.Suffix, Equals, "your"+indexWebsite)
  1069. c.Assert(res.ErrorDocument.Key, Equals, "your"+errorWebsite)
  1070. err = client.DeleteBucketWebsite(bucketNameTest)
  1071. c.Assert(err, IsNil)
  1072. // Set after delete
  1073. err = client.SetBucketWebsite(bucketNameTest, indexWebsite, errorWebsite)
  1074. c.Assert(err, IsNil)
  1075. // Eliminate effect of cache
  1076. time.Sleep(timeoutInOperation)
  1077. res, err = client.GetBucketWebsite(bucketNameTest)
  1078. c.Assert(err, IsNil)
  1079. c.Assert(res.IndexDocument.Suffix, Equals, indexWebsite)
  1080. c.Assert(res.ErrorDocument.Key, Equals, errorWebsite)
  1081. err = client.DeleteBucket(bucketNameTest)
  1082. c.Assert(err, IsNil)
  1083. }
  1084. // TestDeleteBucketWebsite
  1085. func (s *OssClientSuite) TestDeleteBucketWebsite(c *C) {
  1086. var bucketNameTest = bucketNamePrefix + RandLowStr(6)
  1087. var indexWebsite = "myindex.html"
  1088. var errorWebsite = "myerror.html"
  1089. client, err := New(endpoint, accessID, accessKey)
  1090. c.Assert(err, IsNil)
  1091. err = client.CreateBucket(bucketNameTest)
  1092. c.Assert(err, IsNil)
  1093. time.Sleep(timeoutInOperation)
  1094. // Get
  1095. res, err := client.GetBucketWebsite(bucketNameTest)
  1096. c.Assert(err, NotNil)
  1097. // Detele without set
  1098. err = client.DeleteBucketWebsite(bucketNameTest)
  1099. c.Assert(err, IsNil)
  1100. // Set
  1101. err = client.SetBucketWebsite(bucketNameTest, indexWebsite, errorWebsite)
  1102. c.Assert(err, IsNil)
  1103. time.Sleep(timeoutInOperation)
  1104. res, err = client.GetBucketWebsite(bucketNameTest)
  1105. c.Assert(err, IsNil)
  1106. c.Assert(res.IndexDocument.Suffix, Equals, indexWebsite)
  1107. c.Assert(res.ErrorDocument.Key, Equals, errorWebsite)
  1108. // Detele
  1109. time.Sleep(timeoutInOperation)
  1110. err = client.DeleteBucketWebsite(bucketNameTest)
  1111. c.Assert(err, IsNil)
  1112. time.Sleep(timeoutInOperation)
  1113. res, err = client.GetBucketWebsite(bucketNameTest)
  1114. c.Assert(err, NotNil)
  1115. // Detele after delete
  1116. err = client.DeleteBucketWebsite(bucketNameTest)
  1117. c.Assert(err, IsNil)
  1118. err = client.DeleteBucket(bucketNameTest)
  1119. c.Assert(err, IsNil)
  1120. }
  1121. // TestSetBucketWebsiteNegative
  1122. func (s *OssClientSuite) TestSetBucketWebsiteNegative(c *C) {
  1123. var bucketNameTest = bucketNamePrefix + RandLowStr(6)
  1124. var indexWebsite = "myindex.html"
  1125. var errorWebsite = "myerror.html"
  1126. client, err := New(endpoint, accessID, accessKey)
  1127. c.Assert(err, IsNil)
  1128. err = client.DeleteBucket(bucketNameTest)
  1129. // Not exist
  1130. _, err = client.GetBucketWebsite(bucketNameTest)
  1131. c.Assert(err, NotNil)
  1132. err = client.DeleteBucketWebsite(bucketNameTest)
  1133. c.Assert(err, NotNil)
  1134. err = client.SetBucketWebsite(bucketNameTest, indexWebsite, errorWebsite)
  1135. c.Assert(err, NotNil)
  1136. err = client.CreateBucket(bucketNameTest)
  1137. c.Assert(err, IsNil)
  1138. // Set
  1139. time.Sleep(timeoutInOperation)
  1140. err = client.SetBucketWebsite(bucketNameTest, "myindex", "myerror")
  1141. c.Assert(err, IsNil)
  1142. res, err := client.GetBucketWebsite(bucketNameTest)
  1143. c.Assert(err, IsNil)
  1144. c.Assert(res.IndexDocument.Suffix, Equals, "myindex")
  1145. c.Assert(res.ErrorDocument.Key, Equals, "myerror")
  1146. // Detele
  1147. err = client.DeleteBucketWebsite(bucketNameTest)
  1148. c.Assert(err, IsNil)
  1149. time.Sleep(timeoutInOperation)
  1150. _, err = client.GetBucketWebsite(bucketNameTest)
  1151. c.Assert(err, NotNil)
  1152. // Detele after delete
  1153. err = client.DeleteBucketWebsite(bucketNameTest)
  1154. c.Assert(err, IsNil)
  1155. err = client.DeleteBucket(bucketNameTest)
  1156. c.Assert(err, IsNil)
  1157. }
  1158. // TestSetBucketWebsiteDetail
  1159. func (s *OssClientSuite) TestSetBucketWebsiteDetail(c *C) {
  1160. var bucketNameTest = bucketNamePrefix + RandLowStr(6)
  1161. var indexWebsite = "myindex.html"
  1162. var errorWebsite = "myerror.html"
  1163. client, err := New(endpoint, accessID, accessKey)
  1164. c.Assert(err, IsNil)
  1165. err = client.CreateBucket(bucketNameTest)
  1166. c.Assert(err, IsNil)
  1167. time.Sleep(timeoutInOperation)
  1168. btrue := true
  1169. bfalse := false
  1170. // Define one routing rule
  1171. ruleOk := RoutingRule{
  1172. RuleNumber: 1,
  1173. Condition: Condition{
  1174. KeyPrefixEquals: "",
  1175. HTTPErrorCodeReturnedEquals: 404,
  1176. },
  1177. Redirect: Redirect{
  1178. RedirectType: "Mirror",
  1179. // PassQueryString: &btrue, // set default value
  1180. MirrorURL: "http://www.test.com/",
  1181. // MirrorPassQueryString:&btrue, // set default value
  1182. // MirrorFollowRedirect:&bfalse, // set default value
  1183. // MirrorCheckMd5:&bfalse, // set default value
  1184. MirrorHeaders: MirrorHeaders{
  1185. // PassAll:&bfalse, // set default value
  1186. Pass: []string{"myheader-key1", "myheader-key2"},
  1187. Remove: []string{"myheader-key3", "myheader-key4"},
  1188. Set: []MirrorHeaderSet{
  1189. MirrorHeaderSet{
  1190. Key: "myheader-key5",
  1191. Value: "myheader-value5",
  1192. },
  1193. },
  1194. },
  1195. },
  1196. }
  1197. // Define array routing rule
  1198. ruleArrOk := []RoutingRule{
  1199. RoutingRule{
  1200. RuleNumber: 2,
  1201. Condition: Condition{
  1202. KeyPrefixEquals: "abc/",
  1203. HTTPErrorCodeReturnedEquals: 404,
  1204. IncludeHeader: []IncludeHeader{
  1205. IncludeHeader{
  1206. Key: "host",
  1207. Equals: "test.oss-cn-beijing-internal.aliyuncs.com",
  1208. },
  1209. },
  1210. },
  1211. Redirect: Redirect{
  1212. RedirectType: "AliCDN",
  1213. Protocol: "http",
  1214. HostName: "www.test.com",
  1215. PassQueryString: &bfalse,
  1216. ReplaceKeyWith: "prefix/${key}.suffix",
  1217. HttpRedirectCode: 301,
  1218. },
  1219. },
  1220. RoutingRule{
  1221. RuleNumber: 3,
  1222. Condition: Condition{
  1223. KeyPrefixEquals: "",
  1224. HTTPErrorCodeReturnedEquals: 404,
  1225. },
  1226. Redirect: Redirect{
  1227. RedirectType: "Mirror",
  1228. PassQueryString: &btrue,
  1229. MirrorURL: "http://www.test.com/",
  1230. MirrorPassQueryString: &btrue,
  1231. MirrorFollowRedirect: &bfalse,
  1232. MirrorCheckMd5: &bfalse,
  1233. MirrorHeaders: MirrorHeaders{
  1234. PassAll: &btrue,
  1235. Pass: []string{"myheader-key1", "myheader-key2"},
  1236. Remove: []string{"myheader-key3", "myheader-key4"},
  1237. Set: []MirrorHeaderSet{
  1238. MirrorHeaderSet{
  1239. Key: "myheader-key5",
  1240. Value: "myheader-value5",
  1241. },
  1242. },
  1243. },
  1244. },
  1245. },
  1246. }
  1247. // Set one routing rule
  1248. wxmlOne := WebsiteXML{}
  1249. wxmlOne.RoutingRules = append(wxmlOne.RoutingRules, ruleOk)
  1250. var responseHeader http.Header
  1251. err = client.SetBucketWebsiteDetail(bucketNameTest, wxmlOne, GetResponseHeader(&responseHeader))
  1252. c.Assert(err, IsNil)
  1253. requestId := GetRequestId(responseHeader)
  1254. c.Assert(len(requestId) > 0, Equals, true)
  1255. res, err := client.GetBucketWebsite(bucketNameTest)
  1256. c.Assert(err, IsNil)
  1257. c.Assert(res.RoutingRules[0].Redirect.RedirectType, Equals, "Mirror")
  1258. c.Assert(*res.RoutingRules[0].Redirect.PassQueryString, Equals, false)
  1259. c.Assert(*res.RoutingRules[0].Redirect.MirrorPassQueryString, Equals, false)
  1260. c.Assert(*res.RoutingRules[0].Redirect.MirrorFollowRedirect, Equals, true)
  1261. c.Assert(*res.RoutingRules[0].Redirect.MirrorCheckMd5, Equals, false)
  1262. c.Assert(*res.RoutingRules[0].Redirect.MirrorHeaders.PassAll, Equals, false)
  1263. // Set one routing rule and IndexDocument, IndexDocument
  1264. wxml := WebsiteXML{
  1265. IndexDocument: IndexDocument{Suffix: indexWebsite},
  1266. ErrorDocument: ErrorDocument{Key: errorWebsite},
  1267. }
  1268. wxml.RoutingRules = append(wxml.RoutingRules, ruleOk)
  1269. err = client.SetBucketWebsiteDetail(bucketNameTest, wxml, GetResponseHeader(&responseHeader))
  1270. c.Assert(err, IsNil)
  1271. requestId = GetRequestId(responseHeader)
  1272. c.Assert(len(requestId) > 0, Equals, true)
  1273. res, err = client.GetBucketWebsite(bucketNameTest)
  1274. c.Assert(err, IsNil)
  1275. c.Assert(res.IndexDocument.Suffix, Equals, indexWebsite)
  1276. c.Assert(res.ErrorDocument.Key, Equals, errorWebsite)
  1277. c.Assert(res.RoutingRules[0].Redirect.RedirectType, Equals, "Mirror")
  1278. // Set array routing rule
  1279. wxml.RoutingRules = append(wxml.RoutingRules, ruleArrOk...)
  1280. err = client.SetBucketWebsiteDetail(bucketNameTest, wxml, GetResponseHeader(&responseHeader))
  1281. c.Assert(err, IsNil)
  1282. requestId = GetRequestId(responseHeader)
  1283. c.Assert(len(requestId) > 0, Equals, true)
  1284. res, err = client.GetBucketWebsite(bucketNameTest)
  1285. c.Assert(err, IsNil)
  1286. c.Assert(res.IndexDocument.Suffix, Equals, indexWebsite)
  1287. c.Assert(res.ErrorDocument.Key, Equals, errorWebsite)
  1288. c.Assert(len(res.RoutingRules), Equals, 3)
  1289. c.Assert(res.RoutingRules[1].Redirect.RedirectType, Equals, "AliCDN")
  1290. c.Assert(*res.RoutingRules[2].Redirect.MirrorPassQueryString, Equals, true)
  1291. c.Assert(*res.RoutingRules[2].Redirect.MirrorFollowRedirect, Equals, false)
  1292. // Define one error routing rule
  1293. ruleErr := RoutingRule{
  1294. RuleNumber: 1,
  1295. Redirect: Redirect{
  1296. RedirectType: "Mirror",
  1297. PassQueryString: &btrue,
  1298. },
  1299. }
  1300. // Define array error routing rule
  1301. rulesErrArr := []RoutingRule{
  1302. RoutingRule{
  1303. RuleNumber: 1,
  1304. Redirect: Redirect{
  1305. RedirectType: "Mirror",
  1306. PassQueryString: &btrue,
  1307. },
  1308. },
  1309. RoutingRule{
  1310. RuleNumber: 2,
  1311. Redirect: Redirect{
  1312. RedirectType: "Mirror",
  1313. PassQueryString: &btrue,
  1314. },
  1315. },
  1316. }
  1317. ruleIntErr := RoutingRule{
  1318. // RuleNumber:0, // set NULL value
  1319. Condition: Condition{
  1320. KeyPrefixEquals: "",
  1321. HTTPErrorCodeReturnedEquals: 404,
  1322. },
  1323. Redirect: Redirect{
  1324. RedirectType: "Mirror",
  1325. // PassQueryString: &btrue, // set default value
  1326. MirrorURL: "http://www.test.com/",
  1327. // MirrorPassQueryString:&btrue, // set default value
  1328. // MirrorFollowRedirect:&bfalse, // set default value
  1329. // MirrorCheckMd5:&bfalse, // set default value
  1330. MirrorHeaders: MirrorHeaders{
  1331. // PassAll:&bfalse, // set default value
  1332. Pass: []string{"myheader-key1", "myheader-key2"},
  1333. Remove: []string{"myheader-key3", "myheader-key4"},
  1334. Set: []MirrorHeaderSet{
  1335. MirrorHeaderSet{
  1336. Key: "myheader-key5",
  1337. Value: "myheader-value5",
  1338. },
  1339. },
  1340. },
  1341. },
  1342. }
  1343. // Set one int type error rule
  1344. wxmlIntErr := WebsiteXML{}
  1345. wxmlIntErr.RoutingRules = append(wxmlIntErr.RoutingRules, ruleIntErr)
  1346. err = client.SetBucketWebsiteDetail(bucketNameTest, wxmlIntErr)
  1347. c.Assert(err, NotNil)
  1348. // Set one error rule
  1349. wxmlErr := WebsiteXML{}
  1350. wxmlErr.RoutingRules = append(wxmlErr.RoutingRules, ruleErr)
  1351. err = client.SetBucketWebsiteDetail(bucketNameTest, wxmlErr)
  1352. c.Assert(err, NotNil)
  1353. // Set one error rule and one correct rule
  1354. wxmlErr.RoutingRules = append(wxmlErr.RoutingRules, ruleOk)
  1355. err = client.SetBucketWebsiteDetail(bucketNameTest, wxmlErr)
  1356. c.Assert(err, NotNil)
  1357. wxmlErrRuleArr := WebsiteXML{}
  1358. wxmlErrRuleArr.RoutingRules = append(wxmlErrRuleArr.RoutingRules, rulesErrArr...)
  1359. // Set array error routing rule
  1360. err = client.SetBucketWebsiteDetail(bucketNameTest, wxmlErrRuleArr)
  1361. c.Assert(err, NotNil)
  1362. err = client.DeleteBucketWebsite(bucketNameTest)
  1363. c.Assert(err, IsNil)
  1364. err = client.DeleteBucket(bucketNameTest)
  1365. c.Assert(err, IsNil)
  1366. }
  1367. // TestSetBucketWebsiteXml
  1368. func (s *OssClientSuite) TestSetBucketWebsiteXml(c *C) {
  1369. var bucketNameTest = bucketNamePrefix + RandLowStr(6)
  1370. client, err := New(endpoint, accessID, accessKey)
  1371. c.Assert(err, IsNil)
  1372. err = client.CreateBucket(bucketNameTest)
  1373. c.Assert(err, IsNil)
  1374. time.Sleep(timeoutInOperation)
  1375. // Define one routing rule
  1376. ruleOk := RoutingRule{
  1377. RuleNumber: 1,
  1378. Condition: Condition{
  1379. KeyPrefixEquals: "",
  1380. HTTPErrorCodeReturnedEquals: 404,
  1381. },
  1382. Redirect: Redirect{
  1383. RedirectType: "Mirror",
  1384. // PassQueryString: &btrue, // set default value
  1385. MirrorURL: "http://www.test.com/",
  1386. // MirrorPassQueryString:&btrue, // set default value
  1387. // MirrorFollowRedirect:&bfalse, // set default value
  1388. // MirrorCheckMd5:&bfalse, // set default value
  1389. MirrorHeaders: MirrorHeaders{
  1390. // PassAll:&bfalse, // set default value
  1391. Pass: []string{"myheader-key1", "myheader-key2"},
  1392. Remove: []string{"myheader-key3", "myheader-key4"},
  1393. Set: []MirrorHeaderSet{
  1394. MirrorHeaderSet{
  1395. Key: "myheader-key5",
  1396. Value: "myheader-value5",
  1397. },
  1398. },
  1399. },
  1400. },
  1401. }
  1402. // Set one routing rule
  1403. wxmlOne := WebsiteXML{}
  1404. wxmlOne.RoutingRules = append(wxmlOne.RoutingRules, ruleOk)
  1405. bs, err := xml.Marshal(wxmlOne)
  1406. c.Assert(err, IsNil)
  1407. var responseHeader http.Header
  1408. err = client.SetBucketWebsiteXml(bucketNameTest, string(bs), GetResponseHeader(&responseHeader))
  1409. c.Assert(err, IsNil)
  1410. requestId := GetRequestId(responseHeader)
  1411. c.Assert(len(requestId) > 0, Equals, true)
  1412. res, err := client.GetBucketWebsite(bucketNameTest)
  1413. c.Assert(err, IsNil)
  1414. c.Assert(res.RoutingRules[0].Redirect.RedirectType, Equals, "Mirror")
  1415. c.Assert(*res.RoutingRules[0].Redirect.PassQueryString, Equals, false)
  1416. c.Assert(*res.RoutingRules[0].Redirect.MirrorPassQueryString, Equals, false)
  1417. c.Assert(*res.RoutingRules[0].Redirect.MirrorFollowRedirect, Equals, true)
  1418. c.Assert(*res.RoutingRules[0].Redirect.MirrorCheckMd5, Equals, false)
  1419. c.Assert(*res.RoutingRules[0].Redirect.MirrorHeaders.PassAll, Equals, false)
  1420. }
  1421. // TestSetBucketCORS
  1422. func (s *OssClientSuite) TestSetBucketCORS(c *C) {
  1423. var bucketNameTest = bucketNamePrefix + RandLowStr(6)
  1424. var rule1 = CORSRule{
  1425. AllowedOrigin: []string{"*"},
  1426. AllowedMethod: []string{"PUT", "GET", "POST"},
  1427. AllowedHeader: []string{},
  1428. ExposeHeader: []string{},
  1429. MaxAgeSeconds: 100,
  1430. }
  1431. var rule2 = CORSRule{
  1432. AllowedOrigin: []string{"http://www.a.com", "http://www.b.com"},
  1433. AllowedMethod: []string{"GET"},
  1434. AllowedHeader: []string{"Authorization"},
  1435. ExposeHeader: []string{"x-oss-test", "x-oss-test1"},
  1436. MaxAgeSeconds: 200,
  1437. }
  1438. client, err := New(endpoint, accessID, accessKey)
  1439. c.Assert(err, IsNil)
  1440. err = client.CreateBucket(bucketNameTest)
  1441. c.Assert(err, IsNil)
  1442. time.Sleep(timeoutInOperation)
  1443. // Set
  1444. err = client.SetBucketCORS(bucketNameTest, []CORSRule{rule1})
  1445. c.Assert(err, IsNil)
  1446. gbcr, err := client.GetBucketCORS(bucketNameTest)
  1447. c.Assert(err, IsNil)
  1448. c.Assert(len(gbcr.CORSRules), Equals, 1)
  1449. c.Assert(len(gbcr.CORSRules[0].AllowedOrigin), Equals, 1)
  1450. c.Assert(len(gbcr.CORSRules[0].AllowedMethod), Equals, 3)
  1451. c.Assert(len(gbcr.CORSRules[0].AllowedHeader), Equals, 0)
  1452. c.Assert(len(gbcr.CORSRules[0].ExposeHeader), Equals, 0)
  1453. c.Assert(gbcr.CORSRules[0].MaxAgeSeconds, Equals, 100)
  1454. // Double set
  1455. err = client.SetBucketCORS(bucketNameTest, []CORSRule{rule1})
  1456. c.Assert(err, IsNil)
  1457. gbcr, err = client.GetBucketCORS(bucketNameTest)
  1458. c.Assert(err, IsNil)
  1459. c.Assert(len(gbcr.CORSRules), Equals, 1)
  1460. c.Assert(len(gbcr.CORSRules[0].AllowedOrigin), Equals, 1)
  1461. c.Assert(len(gbcr.CORSRules[0].AllowedMethod), Equals, 3)
  1462. c.Assert(len(gbcr.CORSRules[0].AllowedHeader), Equals, 0)
  1463. c.Assert(len(gbcr.CORSRules[0].ExposeHeader), Equals, 0)
  1464. c.Assert(gbcr.CORSRules[0].MaxAgeSeconds, Equals, 100)
  1465. // Set rule2
  1466. err = client.SetBucketCORS(bucketNameTest, []CORSRule{rule2})
  1467. c.Assert(err, IsNil)
  1468. time.Sleep(timeoutInOperation)
  1469. gbcr, err = client.GetBucketCORS(bucketNameTest)
  1470. c.Assert(err, IsNil)
  1471. c.Assert(len(gbcr.CORSRules), Equals, 1)
  1472. c.Assert(len(gbcr.CORSRules[0].AllowedOrigin), Equals, 2)
  1473. c.Assert(len(gbcr.CORSRules[0].AllowedMethod), Equals, 1)
  1474. c.Assert(len(gbcr.CORSRules[0].AllowedHeader), Equals, 1)
  1475. c.Assert(len(gbcr.CORSRules[0].ExposeHeader), Equals, 2)
  1476. c.Assert(gbcr.CORSRules[0].MaxAgeSeconds, Equals, 200)
  1477. // Reset
  1478. err = client.SetBucketCORS(bucketNameTest, []CORSRule{rule1, rule2})
  1479. c.Assert(err, IsNil)
  1480. time.Sleep(timeoutInOperation)
  1481. gbcr, err = client.GetBucketCORS(bucketNameTest)
  1482. c.Assert(err, IsNil)
  1483. c.Assert(len(gbcr.CORSRules), Equals, 2)
  1484. // Set after delete
  1485. err = client.DeleteBucketCORS(bucketNameTest)
  1486. c.Assert(err, IsNil)
  1487. err = client.SetBucketCORS(bucketNameTest, []CORSRule{rule1, rule2})
  1488. c.Assert(err, IsNil)
  1489. time.Sleep(timeoutInOperation)
  1490. gbcr, err = client.GetBucketCORS(bucketNameTest)
  1491. c.Assert(err, IsNil)
  1492. c.Assert(len(gbcr.CORSRules), Equals, 2)
  1493. err = client.DeleteBucketCORS(bucketNameTest)
  1494. c.Assert(err, IsNil)
  1495. err = client.DeleteBucket(bucketNameTest)
  1496. c.Assert(err, IsNil)
  1497. }
  1498. // TestDeleteBucketCORS
  1499. func (s *OssClientSuite) TestDeleteBucketCORS(c *C) {
  1500. var bucketNameTest = bucketNamePrefix + RandLowStr(6)
  1501. var rule = CORSRule{
  1502. AllowedOrigin: []string{"*"},
  1503. AllowedMethod: []string{"PUT", "GET", "POST"},
  1504. AllowedHeader: []string{},
  1505. ExposeHeader: []string{},
  1506. MaxAgeSeconds: 100,
  1507. }
  1508. client, err := New(endpoint, accessID, accessKey)
  1509. c.Assert(err, IsNil)
  1510. err = client.CreateBucket(bucketNameTest)
  1511. c.Assert(err, IsNil)
  1512. time.Sleep(timeoutInOperation)
  1513. // Delete not set
  1514. err = client.DeleteBucketCORS(bucketNameTest)
  1515. c.Assert(err, IsNil)
  1516. // Set
  1517. err = client.SetBucketCORS(bucketNameTest, []CORSRule{rule})
  1518. c.Assert(err, IsNil)
  1519. time.Sleep(timeoutInOperation)
  1520. _, err = client.GetBucketCORS(bucketNameTest)
  1521. c.Assert(err, IsNil)
  1522. // Detele
  1523. err = client.DeleteBucketCORS(bucketNameTest)
  1524. c.Assert(err, IsNil)
  1525. time.Sleep(timeoutInOperation)
  1526. _, err = client.GetBucketCORS(bucketNameTest)
  1527. c.Assert(err, NotNil)
  1528. // Detele after deleting
  1529. err = client.DeleteBucketCORS(bucketNameTest)
  1530. c.Assert(err, IsNil)
  1531. err = client.DeleteBucket(bucketNameTest)
  1532. c.Assert(err, IsNil)
  1533. }
  1534. // TestSetBucketCORSNegative
  1535. func (s *OssClientSuite) TestSetBucketCORSNegative(c *C) {
  1536. var bucketNameTest = bucketNamePrefix + RandLowStr(6)
  1537. var rule = CORSRule{
  1538. AllowedOrigin: []string{"*"},
  1539. AllowedMethod: []string{"PUT", "GET", "POST"},
  1540. AllowedHeader: []string{},
  1541. ExposeHeader: []string{},
  1542. MaxAgeSeconds: 100,
  1543. }
  1544. client, err := New(endpoint, accessID, accessKey)
  1545. c.Assert(err, IsNil)
  1546. err = client.DeleteBucket(bucketNameTest)
  1547. // Not exist
  1548. _, err = client.GetBucketCORS(bucketNameTest)
  1549. c.Assert(err, NotNil)
  1550. err = client.DeleteBucketCORS(bucketNameTest)
  1551. c.Assert(err, NotNil)
  1552. err = client.SetBucketCORS(bucketNameTest, []CORSRule{rule})
  1553. c.Assert(err, NotNil)
  1554. err = client.CreateBucket(bucketNameTest)
  1555. c.Assert(err, IsNil)
  1556. time.Sleep(timeoutInOperation)
  1557. _, err = client.GetBucketCORS(bucketNameTest)
  1558. c.Assert(err, NotNil)
  1559. // Set
  1560. err = client.SetBucketCORS(bucketNameTest, []CORSRule{rule})
  1561. c.Assert(err, IsNil)
  1562. time.Sleep(timeoutInOperation)
  1563. _, err = client.GetBucketCORS(bucketNameTest)
  1564. c.Assert(err, IsNil)
  1565. // Delete
  1566. err = client.DeleteBucketCORS(bucketNameTest)
  1567. c.Assert(err, IsNil)
  1568. time.Sleep(timeoutInOperation)
  1569. _, err = client.GetBucketCORS(bucketNameTest)
  1570. c.Assert(err, NotNil)
  1571. // Delete after deleting
  1572. err = client.DeleteBucketCORS(bucketNameTest)
  1573. c.Assert(err, IsNil)
  1574. err = client.DeleteBucket(bucketNameTest)
  1575. c.Assert(err, IsNil)
  1576. }
  1577. // TestGetBucketInfo
  1578. func (s *OssClientSuite) TestGetBucketInfo(c *C) {
  1579. var bucketNameTest = bucketNamePrefix + RandLowStr(6)
  1580. client, err := New(endpoint, accessID, accessKey)
  1581. c.Assert(err, IsNil)
  1582. err = client.CreateBucket(bucketNameTest)
  1583. c.Assert(err, IsNil)
  1584. res, err := client.GetBucketInfo(bucketNameTest)
  1585. c.Assert(err, IsNil)
  1586. c.Assert(res.BucketInfo.Name, Equals, bucketNameTest)
  1587. c.Assert(strings.HasPrefix(res.BucketInfo.Location, "oss-"), Equals, true)
  1588. c.Assert(res.BucketInfo.ACL, Equals, "private")
  1589. c.Assert(strings.HasSuffix(res.BucketInfo.ExtranetEndpoint, ".com"), Equals, true)
  1590. c.Assert(strings.HasSuffix(res.BucketInfo.IntranetEndpoint, ".com"), Equals, true)
  1591. c.Assert(res.BucketInfo.CreationDate, NotNil)
  1592. err = client.DeleteBucket(bucketNameTest)
  1593. c.Assert(err, IsNil)
  1594. }
  1595. // TestGetBucketInfoNegative
  1596. func (s *OssClientSuite) TestGetBucketInfoNegative(c *C) {
  1597. var bucketNameTest = bucketNamePrefix + RandLowStr(6)
  1598. client, err := New(endpoint, accessID, accessKey)
  1599. c.Assert(err, IsNil)
  1600. // Not exist
  1601. _, err = client.GetBucketInfo(bucketNameTest)
  1602. c.Assert(err, NotNil)
  1603. // Bucket name invalid
  1604. _, err = client.GetBucketInfo("InvalidBucketName_")
  1605. c.Assert(err, NotNil)
  1606. }
  1607. // TestEndpointFormat
  1608. func (s *OssClientSuite) TestEndpointFormat(c *C) {
  1609. var bucketNameTest = bucketNamePrefix + RandLowStr(6)
  1610. // http://host
  1611. client, err := New(endpoint, accessID, accessKey)
  1612. c.Assert(err, IsNil)
  1613. err = client.CreateBucket(bucketNameTest)
  1614. c.Assert(err, IsNil)
  1615. time.Sleep(timeoutInOperation)
  1616. res, err := client.GetBucketACL(bucketNameTest)
  1617. c.Assert(err, IsNil)
  1618. c.Assert(res.ACL, Equals, string(ACLPrivate))
  1619. err = client.DeleteBucket(bucketNameTest)
  1620. c.Assert(err, IsNil)
  1621. time.Sleep(timeoutInOperation)
  1622. // http://host:port
  1623. client, err = New(endpoint+":80", accessID, accessKey)
  1624. c.Assert(err, IsNil)
  1625. err = client.CreateBucket(bucketNameTest)
  1626. c.Assert(err, IsNil)
  1627. time.Sleep(timeoutInOperation)
  1628. res, err = client.GetBucketACL(bucketNameTest)
  1629. c.Assert(err, IsNil)
  1630. c.Assert(res.ACL, Equals, string(ACLPrivate))
  1631. err = client.DeleteBucket(bucketNameTest)
  1632. c.Assert(err, IsNil)
  1633. }
  1634. // TestCname
  1635. func (s *OssClientSuite) _TestCname(c *C) {
  1636. var bucketNameTest = "<my-bucket-cname>"
  1637. client, err := New("<endpoint>", "<accessKeyId>", "<accessKeySecret>", UseCname(true))
  1638. c.Assert(err, IsNil)
  1639. err = client.CreateBucket(bucketNameTest)
  1640. c.Assert(err, IsNil)
  1641. _, err = client.ListBuckets()
  1642. c.Assert(err, NotNil)
  1643. res, err := client.GetBucketACL(bucketNameTest)
  1644. c.Assert(err, IsNil)
  1645. c.Assert(res.ACL, Equals, string(ACLPrivate))
  1646. }
  1647. // TestCnameNegative
  1648. func (s *OssClientSuite) _TestCnameNegative(c *C) {
  1649. var bucketNameTest = "<my-bucket-cname>"
  1650. client, err := New("<endpoint>", "<accessKeyId>", "<accessKeySecret>", UseCname(true))
  1651. c.Assert(err, IsNil)
  1652. err = client.CreateBucket(bucketNameTest)
  1653. c.Assert(err, NotNil)
  1654. _, err = client.ListBuckets()
  1655. c.Assert(err, NotNil)
  1656. _, err = client.GetBucketACL(bucketNameTest)
  1657. c.Assert(err, NotNil)
  1658. }
  1659. // _TestHTTPS
  1660. func (s *OssClientSuite) _TestHTTPS(c *C) {
  1661. var bucketNameTest = "<my-bucket-https>"
  1662. client, err := New("<endpoint>", "<accessKeyId>", "<accessKeySecret>")
  1663. c.Assert(err, IsNil)
  1664. err = client.CreateBucket(bucketNameTest)
  1665. c.Assert(err, IsNil)
  1666. res, err := client.GetBucketACL(bucketNameTest)
  1667. c.Assert(err, IsNil)
  1668. c.Assert(res.ACL, Equals, string(ACLPrivate))
  1669. err = client.DeleteBucket(bucketNameTest)
  1670. c.Assert(err, IsNil)
  1671. }
  1672. // TestClientOption
  1673. func (s *OssClientSuite) TestClientOption(c *C) {
  1674. var bucketNameTest = bucketNamePrefix + RandLowStr(6)
  1675. client, err := New(endpoint, accessID, accessKey, UseCname(true),
  1676. Timeout(11, 12), SecurityToken("token"), Proxy(proxyHost))
  1677. c.Assert(err, IsNil)
  1678. // CreateBucket timeout
  1679. err = client.CreateBucket(bucketNameTest)
  1680. c.Assert(err, NotNil)
  1681. c.Assert(client.Conn.config.HTTPTimeout.ConnectTimeout, Equals, time.Second*11)
  1682. c.Assert(client.Conn.config.HTTPTimeout.ReadWriteTimeout, Equals, time.Second*12)
  1683. c.Assert(client.Conn.config.HTTPTimeout.HeaderTimeout, Equals, time.Second*12)
  1684. c.Assert(client.Conn.config.HTTPTimeout.IdleConnTimeout, Equals, time.Second*12)
  1685. c.Assert(client.Conn.config.HTTPTimeout.LongTimeout, Equals, time.Second*12*10)
  1686. c.Assert(client.Conn.config.SecurityToken, Equals, "token")
  1687. c.Assert(client.Conn.config.IsCname, Equals, true)
  1688. c.Assert(client.Conn.config.IsUseProxy, Equals, true)
  1689. c.Assert(client.Config.ProxyHost, Equals, proxyHost)
  1690. client, err = New(endpoint, accessID, accessKey, AuthProxy(proxyHost, proxyUser, proxyPasswd))
  1691. c.Assert(client.Conn.config.IsUseProxy, Equals, true)
  1692. c.Assert(client.Config.ProxyHost, Equals, proxyHost)
  1693. c.Assert(client.Conn.config.IsAuthProxy, Equals, true)
  1694. c.Assert(client.Conn.config.ProxyUser, Equals, proxyUser)
  1695. c.Assert(client.Conn.config.ProxyPassword, Equals, proxyPasswd)
  1696. client, err = New(endpoint, accessID, accessKey, UserAgent("go sdk user agent"))
  1697. c.Assert(client.Conn.config.UserAgent, Equals, "go sdk user agent")
  1698. // Check we can overide the http.Client
  1699. httpClient := new(http.Client)
  1700. client, err = New(endpoint, accessID, accessKey, HTTPClient(httpClient))
  1701. c.Assert(client.HTTPClient, Equals, httpClient)
  1702. c.Assert(client.Conn.client, Equals, httpClient)
  1703. client, err = New(endpoint, accessID, accessKey)
  1704. c.Assert(client.HTTPClient, IsNil)
  1705. }
  1706. // TestProxy
  1707. func (s *OssClientSuite) ProxyTestFunc(c *C, authVersion AuthVersionType, extraHeaders []string) {
  1708. bucketNameTest := bucketNamePrefix + RandLowStr(6)
  1709. objectName := "体育/奥运/首金"
  1710. objectValue := "大江东去,浪淘尽,千古风流人物。 故垒西边,人道是、三国周郎赤壁。"
  1711. client, err := New(endpoint, accessID, accessKey, AuthProxy(proxyHost, proxyUser, proxyPasswd))
  1712. oldType := client.Config.AuthVersion
  1713. oldHeaders := client.Config.AdditionalHeaders
  1714. client.Config.AuthVersion = authVersion
  1715. client.Config.AdditionalHeaders = extraHeaders
  1716. // Create bucket
  1717. err = client.CreateBucket(bucketNameTest)
  1718. c.Assert(err, IsNil)
  1719. // Get bucket info
  1720. _, err = client.GetBucketInfo(bucketNameTest)
  1721. c.Assert(err, IsNil)
  1722. bucket, err := client.Bucket(bucketNameTest)
  1723. // Sign URL
  1724. str, err := bucket.SignURL(objectName, HTTPPut, 60)
  1725. c.Assert(err, IsNil)
  1726. if bucket.Client.Config.AuthVersion == AuthV1 {
  1727. c.Assert(strings.Contains(str, HTTPParamExpires+"="), Equals, true)
  1728. c.Assert(strings.Contains(str, HTTPParamAccessKeyID+"="), Equals, true)
  1729. c.Assert(strings.Contains(str, HTTPParamSignature+"="), Equals, true)
  1730. } else {
  1731. c.Assert(strings.Contains(str, HTTPParamSignatureVersion+"=OSS2"), Equals, true)
  1732. c.Assert(strings.Contains(str, HTTPParamExpiresV2+"="), Equals, true)
  1733. c.Assert(strings.Contains(str, HTTPParamAccessKeyIDV2+"="), Equals, true)
  1734. c.Assert(strings.Contains(str, HTTPParamSignatureV2+"="), Equals, true)
  1735. }
  1736. // Put object with URL
  1737. err = bucket.PutObjectWithURL(str, strings.NewReader(objectValue))
  1738. c.Assert(err, IsNil)
  1739. // Sign URL for get object
  1740. str, err = bucket.SignURL(objectName, HTTPGet, 60)
  1741. c.Assert(err, IsNil)
  1742. if bucket.Client.Config.AuthVersion == AuthV1 {
  1743. c.Assert(strings.Contains(str, HTTPParamExpires+"="), Equals, true)
  1744. c.Assert(strings.Contains(str, HTTPParamAccessKeyID+"="), Equals, true)
  1745. c.Assert(strings.Contains(str, HTTPParamSignature+"="), Equals, true)
  1746. } else {
  1747. c.Assert(strings.Contains(str, HTTPParamSignatureVersion+"=OSS2"), Equals, true)
  1748. c.Assert(strings.Contains(str, HTTPParamExpiresV2+"="), Equals, true)
  1749. c.Assert(strings.Contains(str, HTTPParamAccessKeyIDV2+"="), Equals, true)
  1750. c.Assert(strings.Contains(str, HTTPParamSignatureV2+"="), Equals, true)
  1751. }
  1752. // Get object with URL
  1753. body, err := bucket.GetObjectWithURL(str)
  1754. c.Assert(err, IsNil)
  1755. str, err = readBody(body)
  1756. c.Assert(err, IsNil)
  1757. c.Assert(str, Equals, objectValue)
  1758. // Put object
  1759. err = bucket.PutObject(objectName, strings.NewReader(objectValue))
  1760. c.Assert(err, IsNil)
  1761. // Get object
  1762. _, err = bucket.GetObject(objectName)
  1763. c.Assert(err, IsNil)
  1764. // List objects
  1765. _, err = bucket.ListObjects()
  1766. c.Assert(err, IsNil)
  1767. // Delete object
  1768. err = bucket.DeleteObject(objectName)
  1769. c.Assert(err, IsNil)
  1770. // Delete bucket
  1771. err = client.DeleteBucket(bucketNameTest)
  1772. c.Assert(err, IsNil)
  1773. client.Config.AuthVersion = oldType
  1774. client.Config.AdditionalHeaders = oldHeaders
  1775. }
  1776. func (s *OssClientSuite) TestProxy(c *C) {
  1777. s.ProxyTestFunc(c, AuthV1, []string{})
  1778. s.ProxyTestFunc(c, AuthV2, []string{})
  1779. s.ProxyTestFunc(c, AuthV2, []string{"host", "range", "user-agent"})
  1780. }
  1781. // TestProxy for https endpoint
  1782. func (s *OssClientSuite) TestHttpsEndpointProxy(c *C) {
  1783. bucketNameTest := bucketNamePrefix + RandLowStr(6)
  1784. objectName := objectNamePrefix + RandLowStr(6)
  1785. objectValue := RandLowStr(100)
  1786. httpsEndPoint := ""
  1787. if strings.HasPrefix(endpoint, "http://") {
  1788. httpsEndPoint = strings.Replace(endpoint, "http://", "https://", 1)
  1789. } else if !strings.HasPrefix(endpoint, "https://") {
  1790. httpsEndPoint = "https://" + endpoint
  1791. } else {
  1792. httpsEndPoint = endpoint
  1793. }
  1794. client, err := New(httpsEndPoint, accessID, accessKey, AuthProxy(proxyHost, proxyUser, proxyPasswd))
  1795. // Create bucket
  1796. err = client.CreateBucket(bucketNameTest)
  1797. c.Assert(err, IsNil)
  1798. bucket, err := client.Bucket(bucketNameTest)
  1799. // Put object
  1800. err = bucket.PutObject(objectName, strings.NewReader(objectValue))
  1801. c.Assert(err, IsNil)
  1802. // Get object
  1803. _, err = bucket.GetObject(objectName)
  1804. c.Assert(err, IsNil)
  1805. // List objects
  1806. _, err = bucket.ListObjects()
  1807. c.Assert(err, IsNil)
  1808. // Delete object
  1809. err = bucket.DeleteObject(objectName)
  1810. c.Assert(err, IsNil)
  1811. // Delete bucket
  1812. err = client.DeleteBucket(bucketNameTest)
  1813. c.Assert(err, IsNil)
  1814. }
  1815. // Private
  1816. func (s *OssClientSuite) checkBucket(buckets []BucketProperties, bucket string) bool {
  1817. for _, v := range buckets {
  1818. if v.Name == bucket {
  1819. return true
  1820. }
  1821. }
  1822. return false
  1823. }
  1824. func (s *OssClientSuite) getBucket(buckets []BucketProperties, bucket string) (bool, BucketProperties) {
  1825. for _, v := range buckets {
  1826. if v.Name == bucket {
  1827. return true, v
  1828. }
  1829. }
  1830. return false, BucketProperties{}
  1831. }
  1832. func (s *OssClientSuite) TestHttpLogNotSignUrl(c *C) {
  1833. logName := "." + string(os.PathSeparator) + "test-go-sdk-httpdebug.log" + RandStr(5)
  1834. f, err := os.OpenFile(logName, os.O_CREATE|os.O_TRUNC|os.O_RDWR, 0660)
  1835. c.Assert(err, IsNil)
  1836. client, err := New(endpoint, accessID, accessKey)
  1837. client.Config.LogLevel = Debug
  1838. client.Config.Logger = log.New(f, "", log.LstdFlags)
  1839. var testBucketName = bucketNamePrefix + RandLowStr(6)
  1840. // CreateBucket
  1841. err = client.CreateBucket(testBucketName)
  1842. f.Close()
  1843. // read log file,get http info
  1844. contents, err := ioutil.ReadFile(logName)
  1845. c.Assert(err, IsNil)
  1846. httpContent := string(contents)
  1847. //fmt.Println(httpContent)
  1848. c.Assert(strings.Contains(httpContent, "signStr"), Equals, true)
  1849. c.Assert(strings.Contains(httpContent, "Method:"), Equals, true)
  1850. // delete test bucket and log
  1851. os.Remove(logName)
  1852. client.DeleteBucket(testBucketName)
  1853. }
  1854. func (s *OssClientSuite) HttpLogSignUrlTestFunc(c *C, authVersion AuthVersionType, extraHeaders []string) {
  1855. logName := "." + string(os.PathSeparator) + "test-go-sdk-httpdebug-signurl.log" + RandStr(5)
  1856. f, err := os.OpenFile(logName, os.O_CREATE|os.O_TRUNC|os.O_RDWR, 0660)
  1857. c.Assert(err, IsNil)
  1858. client, err := New(endpoint, accessID, accessKey)
  1859. client.Config.LogLevel = Debug
  1860. client.Config.Logger = log.New(f, "", log.LstdFlags)
  1861. oldType := client.Config.AuthVersion
  1862. oldHeaders := client.Config.AdditionalHeaders
  1863. client.Config.AuthVersion = authVersion
  1864. client.Config.AdditionalHeaders = extraHeaders
  1865. var testBucketName = bucketNamePrefix + RandLowStr(6)
  1866. // CreateBucket
  1867. err = client.CreateBucket(testBucketName)
  1868. f.Close()
  1869. // clear log
  1870. f, err = os.OpenFile(logName, os.O_CREATE|os.O_TRUNC|os.O_RDWR, 0660)
  1871. client.Config.Logger = log.New(f, "", log.LstdFlags)
  1872. bucket, _ := client.Bucket(testBucketName)
  1873. objectName := objectNamePrefix + RandStr(8)
  1874. objectValue := RandStr(20)
  1875. // Sign URL for put
  1876. str, err := bucket.SignURL(objectName, HTTPPut, 60)
  1877. c.Assert(err, IsNil)
  1878. if bucket.Client.Config.AuthVersion == AuthV1 {
  1879. c.Assert(strings.Contains(str, HTTPParamExpires+"="), Equals, true)
  1880. c.Assert(strings.Contains(str, HTTPParamAccessKeyID+"="), Equals, true)
  1881. c.Assert(strings.Contains(str, HTTPParamSignature+"="), Equals, true)
  1882. } else {
  1883. c.Assert(strings.Contains(str, HTTPParamSignatureVersion+"=OSS2"), Equals, true)
  1884. c.Assert(strings.Contains(str, HTTPParamExpiresV2+"="), Equals, true)
  1885. c.Assert(strings.Contains(str, HTTPParamAccessKeyIDV2+"="), Equals, true)
  1886. c.Assert(strings.Contains(str, HTTPParamSignatureV2+"="), Equals, true)
  1887. }
  1888. // Error put object with URL
  1889. err = bucket.PutObjectWithURL(str, strings.NewReader(objectValue), ContentType("image/tiff"))
  1890. f.Close()
  1891. // read log file,get http info
  1892. contents, err := ioutil.ReadFile(logName)
  1893. c.Assert(err, IsNil)
  1894. httpContent := string(contents)
  1895. //fmt.Println(httpContent)
  1896. c.Assert(strings.Contains(httpContent, "signStr"), Equals, true)
  1897. c.Assert(strings.Contains(httpContent, "Method:"), Equals, true)
  1898. // delete test bucket and log
  1899. os.Remove(logName)
  1900. client.DeleteBucket(testBucketName)
  1901. client.Config.AuthVersion = oldType
  1902. client.Config.AdditionalHeaders = oldHeaders
  1903. }
  1904. func (s *OssClientSuite) TestHttpLogSignUrl(c *C) {
  1905. s.HttpLogSignUrlTestFunc(c, AuthV1, []string{})
  1906. s.HttpLogSignUrlTestFunc(c, AuthV2, []string{})
  1907. s.HttpLogSignUrlTestFunc(c, AuthV2, []string{"host", "range", "user-agent"})
  1908. }
  1909. func (s *OssClientSuite) TestSetLimitUploadSpeed(c *C) {
  1910. client, err := New(endpoint, accessID, accessKey)
  1911. c.Assert(err, IsNil)
  1912. err = client.LimitUploadSpeed(100)
  1913. goVersion := runtime.Version()
  1914. pSlice := strings.Split(strings.ToLower(goVersion), ".")
  1915. // compare with go1.7
  1916. if len(pSlice) >= 2 {
  1917. if pSlice[0] > "go1" {
  1918. c.Assert(err, IsNil)
  1919. } else if pSlice[0] == "go1" {
  1920. subVersion, _ := strconv.Atoi(pSlice[1])
  1921. if subVersion >= 7 {
  1922. c.Assert(err, IsNil)
  1923. } else {
  1924. c.Assert(err, NotNil)
  1925. }
  1926. } else {
  1927. c.Assert(err, NotNil)
  1928. }
  1929. }
  1930. }
  1931. func (s *OssClientSuite) TestBucketEncyptionError(c *C) {
  1932. client, err := New(endpoint, accessID, accessKey)
  1933. c.Assert(err, IsNil)
  1934. bucketName := bucketNamePrefix + RandLowStr(5)
  1935. err = client.CreateBucket(bucketName)
  1936. c.Assert(err, IsNil)
  1937. // SetBucketEncryption:AES256 ,"123"
  1938. encryptionRule := ServerEncryptionRule{}
  1939. encryptionRule.SSEDefault.SSEAlgorithm = string(AESAlgorithm)
  1940. encryptionRule.SSEDefault.KMSMasterKeyID = "123"
  1941. var responseHeader http.Header
  1942. err = client.SetBucketEncryption(bucketName, encryptionRule, GetResponseHeader(&responseHeader))
  1943. c.Assert(err, NotNil)
  1944. requestId := GetRequestId(responseHeader)
  1945. c.Assert(len(requestId) > 0, Equals, true)
  1946. // GetBucketEncryption
  1947. _, err = client.GetBucketEncryption(bucketName, GetResponseHeader(&responseHeader))
  1948. c.Assert(err, NotNil)
  1949. requestId = GetRequestId(responseHeader)
  1950. c.Assert(len(requestId) > 0, Equals, true)
  1951. // Get default bucket info
  1952. bucketResult, err := client.GetBucketInfo(bucketName)
  1953. c.Assert(err, IsNil)
  1954. c.Assert(bucketResult.BucketInfo.SseRule.SSEAlgorithm, Equals, "")
  1955. c.Assert(bucketResult.BucketInfo.SseRule.KMSMasterKeyID, Equals, "")
  1956. c.Assert(bucketResult.BucketInfo.Versioning, Equals, "")
  1957. err = client.DeleteBucket(bucketName)
  1958. c.Assert(err, IsNil)
  1959. }
  1960. func (s *OssClientSuite) TestBucketEncryptionPutAndGetAndDelete(c *C) {
  1961. client, err := New(endpoint, accessID, accessKey)
  1962. c.Assert(err, IsNil)
  1963. bucketName := bucketNamePrefix + RandLowStr(5)
  1964. err = client.CreateBucket(bucketName)
  1965. c.Assert(err, IsNil)
  1966. // SetBucketEncryption:KMS ,""
  1967. encryptionRule := ServerEncryptionRule{}
  1968. encryptionRule.SSEDefault.SSEAlgorithm = string(KMSAlgorithm)
  1969. var responseHeader http.Header
  1970. err = client.SetBucketEncryption(bucketName, encryptionRule, GetResponseHeader(&responseHeader))
  1971. c.Assert(err, IsNil)
  1972. requestId := GetRequestId(responseHeader)
  1973. c.Assert(len(requestId) > 0, Equals, true)
  1974. // GetBucketEncryption
  1975. getResult, err := client.GetBucketEncryption(bucketName, GetResponseHeader(&responseHeader))
  1976. c.Assert(err, IsNil)
  1977. requestId = GetRequestId(responseHeader)
  1978. c.Assert(len(requestId) > 0, Equals, true)
  1979. // check encryption value
  1980. c.Assert(encryptionRule.SSEDefault.SSEAlgorithm, Equals, getResult.SSEDefault.SSEAlgorithm)
  1981. c.Assert(encryptionRule.SSEDefault.KMSMasterKeyID, Equals, getResult.SSEDefault.KMSMasterKeyID)
  1982. // delete bucket encyption
  1983. err = client.DeleteBucketEncryption(bucketName, GetResponseHeader(&responseHeader))
  1984. c.Assert(err, IsNil)
  1985. requestId = GetRequestId(responseHeader)
  1986. c.Assert(len(requestId) > 0, Equals, true)
  1987. // GetBucketEncryption failure
  1988. _, err = client.GetBucketEncryption(bucketName, GetResponseHeader(&responseHeader))
  1989. c.Assert(err, NotNil)
  1990. requestId = GetRequestId(responseHeader)
  1991. c.Assert(len(requestId) > 0, Equals, true)
  1992. // Get default bucket info
  1993. bucketResult, err := client.GetBucketInfo(bucketName)
  1994. c.Assert(err, IsNil)
  1995. c.Assert(bucketResult.BucketInfo.SseRule.SSEAlgorithm, Equals, "")
  1996. c.Assert(bucketResult.BucketInfo.SseRule.KMSMasterKeyID, Equals, "")
  1997. c.Assert(bucketResult.BucketInfo.Versioning, Equals, "")
  1998. err = client.DeleteBucket(bucketName)
  1999. c.Assert(err, IsNil)
  2000. }
  2001. func (s *OssClientSuite) TestBucketEncryptionWithSm4(c *C) {
  2002. client, err := New(endpoint, accessID, accessKey)
  2003. c.Assert(err, IsNil)
  2004. bucketName := bucketNamePrefix + RandLowStr(5)
  2005. err = client.CreateBucket(bucketName)
  2006. c.Assert(err, IsNil)
  2007. // SetBucketEncryption:SM4 ,""
  2008. encryptionRule := ServerEncryptionRule{}
  2009. encryptionRule.SSEDefault.SSEAlgorithm = string(SM4Algorithm)
  2010. var responseHeader http.Header
  2011. err = client.SetBucketEncryption(bucketName, encryptionRule, GetResponseHeader(&responseHeader))
  2012. c.Assert(err, IsNil)
  2013. requestId := GetRequestId(responseHeader)
  2014. c.Assert(len(requestId) > 0, Equals, true)
  2015. // GetBucketEncryption
  2016. getResult, err := client.GetBucketEncryption(bucketName, GetResponseHeader(&responseHeader))
  2017. c.Assert(err, IsNil)
  2018. requestId = GetRequestId(responseHeader)
  2019. c.Assert(len(requestId) > 0, Equals, true)
  2020. // check encryption value
  2021. c.Assert(getResult.SSEDefault.SSEAlgorithm, Equals, string(SM4Algorithm))
  2022. c.Assert(getResult.SSEDefault.KMSMasterKeyID, Equals, "")
  2023. c.Assert(getResult.SSEDefault.KMSDataEncryption, Equals, "")
  2024. // Get default bucket info
  2025. bucketResult, err := client.GetBucketInfo(bucketName)
  2026. c.Assert(err, IsNil)
  2027. c.Assert(bucketResult.BucketInfo.SseRule.SSEAlgorithm, Equals, string(SM4Algorithm))
  2028. c.Assert(bucketResult.BucketInfo.SseRule.KMSMasterKeyID, Equals, "")
  2029. c.Assert(bucketResult.BucketInfo.SseRule.KMSDataEncryption, Equals, "")
  2030. err = client.DeleteBucket(bucketName)
  2031. c.Assert(err, IsNil)
  2032. }
  2033. func (s *OssClientSuite) TestBucketEncryptionWithKmsSm4(c *C) {
  2034. client, err := New(endpoint, accessID, accessKey)
  2035. c.Assert(err, IsNil)
  2036. bucketName := bucketNamePrefix + RandLowStr(5)
  2037. err = client.CreateBucket(bucketName)
  2038. c.Assert(err, IsNil)
  2039. // SetBucketEncryption:SM4 ,""
  2040. encryptionRule := ServerEncryptionRule{}
  2041. encryptionRule.SSEDefault.SSEAlgorithm = string(KMSAlgorithm)
  2042. encryptionRule.SSEDefault.KMSDataEncryption = string(SM4Algorithm)
  2043. var responseHeader http.Header
  2044. err = client.SetBucketEncryption(bucketName, encryptionRule, GetResponseHeader(&responseHeader))
  2045. c.Assert(err, IsNil)
  2046. requestId := GetRequestId(responseHeader)
  2047. c.Assert(len(requestId) > 0, Equals, true)
  2048. // GetBucketEncryption
  2049. getResult, err := client.GetBucketEncryption(bucketName, GetResponseHeader(&responseHeader))
  2050. c.Assert(err, IsNil)
  2051. requestId = GetRequestId(responseHeader)
  2052. c.Assert(len(requestId) > 0, Equals, true)
  2053. // check encryption value
  2054. c.Assert(getResult.SSEDefault.SSEAlgorithm, Equals, string(KMSAlgorithm))
  2055. c.Assert(getResult.SSEDefault.KMSMasterKeyID, Equals, "")
  2056. c.Assert(getResult.SSEDefault.KMSDataEncryption, Equals, string(SM4Algorithm))
  2057. // Get default bucket info
  2058. bucketResult, err := client.GetBucketInfo(bucketName)
  2059. c.Assert(err, IsNil)
  2060. c.Assert(bucketResult.BucketInfo.SseRule.SSEAlgorithm, Equals, string(KMSAlgorithm))
  2061. c.Assert(bucketResult.BucketInfo.SseRule.KMSMasterKeyID, Equals, "")
  2062. c.Assert(bucketResult.BucketInfo.SseRule.KMSDataEncryption, Equals, string(SM4Algorithm))
  2063. err = client.DeleteBucket(bucketName)
  2064. c.Assert(err, IsNil)
  2065. }
  2066. func (s *OssClientSuite) TestBucketEncyptionPutObjectSuccess(c *C) {
  2067. client, err := New(endpoint, accessID, accessKey)
  2068. c.Assert(err, IsNil)
  2069. bucketName := bucketNamePrefix + RandLowStr(5)
  2070. err = client.CreateBucket(bucketName)
  2071. c.Assert(err, IsNil)
  2072. // SetBucketEncryption:KMS ,""
  2073. encryptionRule := ServerEncryptionRule{}
  2074. encryptionRule.SSEDefault.SSEAlgorithm = string(KMSAlgorithm)
  2075. var responseHeader http.Header
  2076. err = client.SetBucketEncryption(bucketName, encryptionRule, GetResponseHeader(&responseHeader))
  2077. c.Assert(err, IsNil)
  2078. requestId := GetRequestId(responseHeader)
  2079. c.Assert(len(requestId) > 0, Equals, true)
  2080. // GetBucketEncryption
  2081. getResult, err := client.GetBucketEncryption(bucketName, GetResponseHeader(&responseHeader))
  2082. c.Assert(err, IsNil)
  2083. requestId = GetRequestId(responseHeader)
  2084. c.Assert(len(requestId) > 0, Equals, true)
  2085. // check encryption value
  2086. c.Assert(encryptionRule.SSEDefault.SSEAlgorithm, Equals, getResult.SSEDefault.SSEAlgorithm)
  2087. c.Assert(encryptionRule.SSEDefault.KMSMasterKeyID, Equals, getResult.SSEDefault.KMSMasterKeyID)
  2088. // Get default bucket info
  2089. bucketResult, err := client.GetBucketInfo(bucketName)
  2090. c.Assert(err, IsNil)
  2091. c.Assert(bucketResult.BucketInfo.SseRule.SSEAlgorithm, Equals, "KMS")
  2092. c.Assert(bucketResult.BucketInfo.SseRule.KMSMasterKeyID, Equals, "")
  2093. c.Assert(bucketResult.BucketInfo.Versioning, Equals, "")
  2094. err = client.DeleteBucket(bucketName)
  2095. c.Assert(err, IsNil)
  2096. }
  2097. func (s *OssClientSuite) TestBucketEncyptionPutObjectError(c *C) {
  2098. client, err := New(endpoint, accessID, accessKey)
  2099. c.Assert(err, IsNil)
  2100. bucketName := bucketNamePrefix + RandLowStr(5)
  2101. err = client.CreateBucket(bucketName)
  2102. c.Assert(err, IsNil)
  2103. // SetBucketEncryption:KMS ,""
  2104. encryptionRule := ServerEncryptionRule{}
  2105. encryptionRule.SSEDefault.SSEAlgorithm = string(KMSAlgorithm)
  2106. kmsId := "123"
  2107. encryptionRule.SSEDefault.KMSMasterKeyID = kmsId
  2108. var responseHeader http.Header
  2109. err = client.SetBucketEncryption(bucketName, encryptionRule, GetResponseHeader(&responseHeader))
  2110. c.Assert(err, IsNil)
  2111. requestId := GetRequestId(responseHeader)
  2112. c.Assert(len(requestId) > 0, Equals, true)
  2113. // GetBucketEncryption
  2114. getResult, err := client.GetBucketEncryption(bucketName, GetResponseHeader(&responseHeader))
  2115. c.Assert(err, IsNil)
  2116. requestId = GetRequestId(responseHeader)
  2117. c.Assert(len(requestId) > 0, Equals, true)
  2118. // check encryption value
  2119. c.Assert(encryptionRule.SSEDefault.SSEAlgorithm, Equals, getResult.SSEDefault.SSEAlgorithm)
  2120. c.Assert(encryptionRule.SSEDefault.KMSMasterKeyID, Equals, getResult.SSEDefault.KMSMasterKeyID)
  2121. // Get default bucket info
  2122. bucketResult, err := client.GetBucketInfo(bucketName)
  2123. c.Assert(err, IsNil)
  2124. c.Assert(bucketResult.BucketInfo.SseRule.SSEAlgorithm, Equals, "KMS")
  2125. c.Assert(bucketResult.BucketInfo.SseRule.KMSMasterKeyID, Equals, kmsId)
  2126. c.Assert(bucketResult.BucketInfo.Versioning, Equals, "")
  2127. // put and get object failure
  2128. bucket, err := client.Bucket(bucketName)
  2129. c.Assert(err, IsNil)
  2130. // put object failure
  2131. objectName := objectNamePrefix + RandStr(8)
  2132. context := RandStr(100)
  2133. err = bucket.PutObject(objectName, strings.NewReader(context))
  2134. c.Assert(err, NotNil)
  2135. err = client.DeleteBucket(bucketName)
  2136. c.Assert(err, IsNil)
  2137. }
  2138. func (s *OssClientSuite) TestBucketTaggingOperation(c *C) {
  2139. client, err := New(endpoint, accessID, accessKey)
  2140. c.Assert(err, IsNil)
  2141. bucketName := bucketNamePrefix + RandLowStr(5)
  2142. err = client.CreateBucket(bucketName)
  2143. c.Assert(err, IsNil)
  2144. var respHeader http.Header
  2145. // Bucket Tagging
  2146. var tagging Tagging
  2147. tagging.Tags = []Tag{Tag{Key: "testkey2", Value: "testvalue2"}}
  2148. err = client.SetBucketTagging(bucketName, tagging, GetResponseHeader(&respHeader))
  2149. c.Assert(err, IsNil)
  2150. c.Assert(GetRequestId(respHeader) != "", Equals, true)
  2151. getResult, err := client.GetBucketTagging(bucketName)
  2152. c.Assert(err, IsNil)
  2153. c.Assert(getResult.Tags[0].Key, Equals, tagging.Tags[0].Key)
  2154. c.Assert(getResult.Tags[0].Value, Equals, tagging.Tags[0].Value)
  2155. // delete BucketTagging
  2156. err = client.DeleteBucketTagging(bucketName, GetResponseHeader(&respHeader))
  2157. c.Assert(err, IsNil)
  2158. c.Assert(GetRequestId(respHeader) != "", Equals, true)
  2159. getResult, err = client.GetBucketTagging(bucketName, GetResponseHeader(&respHeader))
  2160. c.Assert(err, IsNil)
  2161. c.Assert(GetRequestId(respHeader) != "", Equals, true)
  2162. c.Assert(len(getResult.Tags), Equals, 0)
  2163. err = client.DeleteBucket(bucketName)
  2164. c.Assert(err, IsNil)
  2165. }
  2166. func (s *OssClientSuite) TestListBucketsTagging(c *C) {
  2167. client, err := New(endpoint, accessID, accessKey)
  2168. c.Assert(err, IsNil)
  2169. bucketName1 := bucketNamePrefix + RandLowStr(5)
  2170. err = client.CreateBucket(bucketName1)
  2171. c.Assert(err, IsNil)
  2172. bucketName2 := bucketNamePrefix + RandLowStr(5)
  2173. err = client.CreateBucket(bucketName2)
  2174. c.Assert(err, IsNil)
  2175. // Bucket Tagging
  2176. var tagging Tagging
  2177. tagging.Tags = []Tag{Tag{Key: "testkey", Value: "testvalue"}}
  2178. err = client.SetBucketTagging(bucketName1, tagging)
  2179. c.Assert(err, IsNil)
  2180. // list bucket
  2181. listResult, err := client.ListBuckets(TagKey("testkey"))
  2182. c.Assert(err, IsNil)
  2183. c.Assert(len(listResult.Buckets), Equals, 1)
  2184. c.Assert(listResult.Buckets[0].Name, Equals, bucketName1)
  2185. client.DeleteBucket(bucketName1)
  2186. client.DeleteBucket(bucketName2)
  2187. }
  2188. func (s *OssClientSuite) TestGetBucketStat(c *C) {
  2189. client, err := New(endpoint, accessID, accessKey)
  2190. c.Assert(err, IsNil)
  2191. bucketName := bucketNamePrefix + RandLowStr(5)
  2192. err = client.CreateBucket(bucketName)
  2193. c.Assert(err, IsNil)
  2194. bucket, err := client.Bucket(bucketName)
  2195. c.Assert(err, IsNil)
  2196. // put object
  2197. objectName := objectNamePrefix + RandLowStr(5)
  2198. err = bucket.PutObject(objectName, strings.NewReader(RandStr(10)))
  2199. c.Assert(err, IsNil)
  2200. bucket.DeleteObject(objectName)
  2201. err = bucket.PutObject(objectName, strings.NewReader(RandStr(10)))
  2202. c.Assert(err, IsNil)
  2203. bucket.DeleteObject(objectName)
  2204. _, err = client.GetBucketStat(bucketName)
  2205. c.Assert(err, IsNil)
  2206. client.DeleteBucket(bucketName)
  2207. }
  2208. func (s *OssBucketSuite) TestGetBucketVersioning(c *C) {
  2209. // create a bucket with default proprety
  2210. client, err := New(endpoint, accessID, accessKey)
  2211. c.Assert(err, IsNil)
  2212. bucketName := bucketNamePrefix + RandLowStr(6)
  2213. var respHeader http.Header
  2214. err = client.CreateBucket(bucketName, GetResponseHeader(&respHeader))
  2215. c.Assert(err, IsNil)
  2216. c.Assert(GetRequestId(respHeader) != "", Equals, true)
  2217. // put bucket version:enabled
  2218. var versioningConfig VersioningConfig
  2219. versioningConfig.Status = string(VersionEnabled)
  2220. err = client.SetBucketVersioning(bucketName, versioningConfig)
  2221. c.Assert(err, IsNil)
  2222. // get bucket version success
  2223. versioningResult, err := client.GetBucketVersioning(bucketName, GetResponseHeader(&respHeader))
  2224. c.Assert(versioningResult.Status, Equals, "Enabled")
  2225. c.Assert(GetRequestId(respHeader) != "", Equals, true)
  2226. ForceDeleteBucket(client, bucketName, c)
  2227. }
  2228. func (s *OssClientSuite) TestBucketPolicy(c *C) {
  2229. client, err := New(endpoint, accessID, accessKey)
  2230. c.Assert(err, IsNil)
  2231. bucketName := bucketNamePrefix + RandLowStr(5)
  2232. err = client.CreateBucket(bucketName)
  2233. c.Assert(err, IsNil)
  2234. var responseHeader http.Header
  2235. ret, err := client.GetBucketPolicy(bucketName, GetResponseHeader(&responseHeader))
  2236. c.Assert(err, NotNil)
  2237. requestId := GetRequestId(responseHeader)
  2238. c.Assert(len(requestId) > 0, Equals, true)
  2239. policyInfo := `
  2240. {
  2241. "Version":"1",
  2242. "Statement":[
  2243. {
  2244. "Action":[
  2245. "oss:GetObject",
  2246. "oss:PutObject"
  2247. ],
  2248. "Effect":"Deny",
  2249. "Principal":"[123456790]",
  2250. "Resource":["acs:oss:*:1234567890:*/*"]
  2251. }
  2252. ]
  2253. }`
  2254. err = client.SetBucketPolicy(bucketName, policyInfo, GetResponseHeader(&responseHeader))
  2255. c.Assert(err, IsNil)
  2256. requestId = GetRequestId(responseHeader)
  2257. c.Assert(len(requestId) > 0, Equals, true)
  2258. ret, err = client.GetBucketPolicy(bucketName, GetResponseHeader(&responseHeader))
  2259. c.Assert(err, IsNil)
  2260. testLogger.Println("policy:", ret)
  2261. c.Assert(ret, Equals, policyInfo)
  2262. requestId = GetRequestId(responseHeader)
  2263. c.Assert(len(requestId) > 0, Equals, true)
  2264. err = client.DeleteBucketPolicy(bucketName, GetResponseHeader(&responseHeader))
  2265. c.Assert(err, IsNil)
  2266. requestId = GetRequestId(responseHeader)
  2267. c.Assert(len(requestId) > 0, Equals, true)
  2268. client.DeleteBucket(bucketName)
  2269. }
  2270. func (s *OssClientSuite) TestBucketPolicyNegative(c *C) {
  2271. client, err := New(endpoint, accessID, accessKey)
  2272. c.Assert(err, IsNil)
  2273. bucketName := bucketNamePrefix + RandLowStr(5)
  2274. err = client.CreateBucket(bucketName)
  2275. c.Assert(err, IsNil)
  2276. var responseHeader http.Header
  2277. _, err = client.GetBucketPolicy(bucketName, GetResponseHeader(&responseHeader))
  2278. c.Assert(err, NotNil)
  2279. requestId := GetRequestId(responseHeader)
  2280. c.Assert(len(requestId) > 0, Equals, true)
  2281. // Setting the Version is 2, this is error policy
  2282. errPolicy := `
  2283. {
  2284. "Version":"2",
  2285. "Statement":[
  2286. {
  2287. "Action":[
  2288. "oss:GetObject",
  2289. "oss:PutObject"
  2290. ],
  2291. "Effect":"Deny",
  2292. "Principal":"[123456790]",
  2293. "Resource":["acs:oss:*:1234567890:*/*"]
  2294. }
  2295. ]
  2296. }`
  2297. err = client.SetBucketPolicy(bucketName, errPolicy, GetResponseHeader(&responseHeader))
  2298. c.Assert(err, NotNil)
  2299. testLogger.Println("err:", err)
  2300. requestId = GetRequestId(responseHeader)
  2301. c.Assert(len(requestId) > 0, Equals, true)
  2302. err = client.DeleteBucketPolicy(bucketName, GetResponseHeader(&responseHeader))
  2303. c.Assert(err, IsNil)
  2304. bucketNameEmpty := bucketNamePrefix + RandLowStr(5)
  2305. client.DeleteBucket(bucketNameEmpty)
  2306. err = client.DeleteBucketPolicy(bucketNameEmpty, GetResponseHeader(&responseHeader))
  2307. c.Assert(err, NotNil)
  2308. requestId = GetRequestId(responseHeader)
  2309. c.Assert(len(requestId) > 0, Equals, true)
  2310. client.DeleteBucket(bucketName)
  2311. }
  2312. func (s *OssClientSuite) TestSetBucketRequestPayment(c *C) {
  2313. client, err := New(endpoint, accessID, accessKey)
  2314. c.Assert(err, IsNil)
  2315. bucketName := bucketNamePrefix + RandLowStr(5)
  2316. err = client.CreateBucket(bucketName)
  2317. c.Assert(err, IsNil)
  2318. reqPayConf := RequestPaymentConfiguration{
  2319. Payer: "Requester",
  2320. }
  2321. err = client.SetBucketRequestPayment(bucketName, reqPayConf)
  2322. c.Assert(err, IsNil)
  2323. ret, err := client.GetBucketRequestPayment(bucketName)
  2324. c.Assert(err, IsNil)
  2325. c.Assert(ret.Payer, Equals, "Requester")
  2326. client.DeleteBucket(bucketName)
  2327. c.Assert(err, IsNil)
  2328. }
  2329. func (s *OssClientSuite) TestSetBucketRequestPaymentNegative(c *C) {
  2330. client, err := New(endpoint, accessID, accessKey)
  2331. c.Assert(err, IsNil)
  2332. bucketName := bucketNamePrefix + RandLowStr(5)
  2333. err = client.CreateBucket(bucketName)
  2334. c.Assert(err, IsNil)
  2335. reqPayConf := RequestPaymentConfiguration{
  2336. Payer: "Requesterttttt", // this is a error configuration
  2337. }
  2338. err = client.SetBucketRequestPayment(bucketName, reqPayConf)
  2339. c.Assert(err, NotNil)
  2340. ret, err := client.GetBucketRequestPayment(bucketName)
  2341. c.Assert(err, IsNil)
  2342. c.Assert(ret.Payer, Equals, "BucketOwner")
  2343. client.DeleteBucket(bucketName)
  2344. c.Assert(err, IsNil)
  2345. }
  2346. func (s *OssClientSuite) TestBucketQos(c *C) {
  2347. client, err := New(endpoint, accessID, accessKey)
  2348. c.Assert(err, IsNil)
  2349. ret, err := client.GetUserQoSInfo()
  2350. c.Assert(err, IsNil)
  2351. testLogger.Println("QosInfo:", ret)
  2352. bucketName := bucketNamePrefix + RandLowStr(5)
  2353. _ = client.DeleteBucket(bucketName)
  2354. err = client.CreateBucket(bucketName)
  2355. c.Assert(err, IsNil)
  2356. _, err = client.GetBucketQosInfo(bucketName)
  2357. c.Assert(err, NotNil)
  2358. // case 1 set BucketQoSConfiguration every member
  2359. five := 5
  2360. four := 4
  2361. totalQps := 200
  2362. qosConf := BucketQoSConfiguration{
  2363. TotalUploadBandwidth: &five,
  2364. IntranetUploadBandwidth: &four,
  2365. ExtranetUploadBandwidth: &four,
  2366. TotalDownloadBandwidth: &four,
  2367. IntranetDownloadBandwidth: &four,
  2368. ExtranetDownloadBandwidth: &four,
  2369. TotalQPS: &totalQps,
  2370. IntranetQPS: &totalQps,
  2371. ExtranetQPS: &totalQps,
  2372. }
  2373. var responseHeader http.Header
  2374. err = client.SetBucketQoSInfo(bucketName, qosConf, GetResponseHeader(&responseHeader))
  2375. c.Assert(err, IsNil)
  2376. requestId := GetRequestId(responseHeader)
  2377. c.Assert(len(requestId) > 0, Equals, true)
  2378. // wait a moment for configuration effect
  2379. time.Sleep(time.Second)
  2380. retQos, err := client.GetBucketQosInfo(bucketName)
  2381. c.Assert(err, IsNil)
  2382. // set qosConf default value
  2383. qosConf.XMLName.Local = "QoSConfiguration"
  2384. c.Assert(struct2string(retQos, c), Equals, struct2string(qosConf, c))
  2385. // case 2 set BucketQoSConfiguration not every member
  2386. qosConfNo := BucketQoSConfiguration{
  2387. TotalUploadBandwidth: &five,
  2388. IntranetUploadBandwidth: &four,
  2389. ExtranetUploadBandwidth: &four,
  2390. TotalDownloadBandwidth: &four,
  2391. IntranetDownloadBandwidth: &four,
  2392. ExtranetDownloadBandwidth: &four,
  2393. TotalQPS: &totalQps,
  2394. }
  2395. err = client.SetBucketQoSInfo(bucketName, qosConfNo)
  2396. c.Assert(err, IsNil)
  2397. // wait a moment for configuration effect
  2398. time.Sleep(time.Second)
  2399. retQos, err = client.GetBucketQosInfo(bucketName)
  2400. c.Assert(err, IsNil)
  2401. // set qosConfNo default value
  2402. qosConfNo.XMLName.Local = "QoSConfiguration"
  2403. defNum := -1
  2404. qosConfNo.IntranetQPS = &defNum
  2405. qosConfNo.ExtranetQPS = &defNum
  2406. c.Assert(struct2string(retQos, c), Equals, struct2string(qosConfNo, c))
  2407. err = client.DeleteBucketQosInfo(bucketName)
  2408. c.Assert(err, IsNil)
  2409. // wait a moment for configuration effect
  2410. time.Sleep(time.Second)
  2411. _, err = client.GetBucketQosInfo(bucketName)
  2412. c.Assert(err, NotNil)
  2413. // this is a error qos configuration
  2414. to := *ret.TotalUploadBandwidth + 2
  2415. qosErrConf := BucketQoSConfiguration{
  2416. TotalUploadBandwidth: &to, // this exceed user TotalUploadBandwidth
  2417. IntranetUploadBandwidth: &four,
  2418. ExtranetUploadBandwidth: &four,
  2419. TotalDownloadBandwidth: &four,
  2420. IntranetDownloadBandwidth: &four,
  2421. ExtranetDownloadBandwidth: &four,
  2422. TotalQPS: &totalQps,
  2423. IntranetQPS: &totalQps,
  2424. ExtranetQPS: &totalQps,
  2425. }
  2426. err = client.SetBucketQoSInfo(bucketName, qosErrConf)
  2427. c.Assert(err, NotNil)
  2428. err = client.DeleteBucketQosInfo(bucketName)
  2429. c.Assert(err, IsNil)
  2430. err = client.DeleteBucket(bucketName)
  2431. c.Assert(err, IsNil)
  2432. }
  2433. // struct to string
  2434. func struct2string(obj interface{}, c *C) string {
  2435. t := reflect.TypeOf(obj)
  2436. v := reflect.ValueOf(obj)
  2437. var data = make(map[string]interface{})
  2438. for i := 0; i < t.NumField(); i++ {
  2439. data[t.Field(i).Name] = v.Field(i).Interface()
  2440. }
  2441. str, err := json.Marshal(data)
  2442. c.Assert(err, IsNil)
  2443. return string(str)
  2444. }
  2445. type TestCredentials struct {
  2446. }
  2447. func (testCreInf *TestCredentials) GetAccessKeyID() string {
  2448. return os.Getenv("OSS_TEST_ACCESS_KEY_ID")
  2449. }
  2450. func (testCreInf *TestCredentials) GetAccessKeySecret() string {
  2451. return os.Getenv("OSS_TEST_ACCESS_KEY_SECRET")
  2452. }
  2453. func (testCreInf *TestCredentials) GetSecurityToken() string {
  2454. return ""
  2455. }
  2456. type TestCredentialsProvider struct {
  2457. }
  2458. func (testInfBuild *TestCredentialsProvider) GetCredentials() Credentials {
  2459. return &TestCredentials{}
  2460. }
  2461. func (s *OssClientSuite) TestClientCredentialInfBuild(c *C) {
  2462. var bucketNameTest = bucketNamePrefix + RandLowStr(6)
  2463. var defaultBuild TestCredentialsProvider
  2464. client, err := New(endpoint, "", "", SetCredentialsProvider(&defaultBuild))
  2465. c.Assert(err, IsNil)
  2466. err = client.CreateBucket(bucketNameTest)
  2467. c.Assert(err, IsNil)
  2468. err = client.DeleteBucket(bucketNameTest)
  2469. c.Assert(err, IsNil)
  2470. }
  2471. func (s *OssClientSuite) TestClientSetLocalIpError(c *C) {
  2472. // create client and bucket
  2473. ipAddr, err := net.ResolveIPAddr("ip", "127.0.0.1")
  2474. c.Assert(err, IsNil)
  2475. localTCPAddr := &(net.TCPAddr{IP: ipAddr.IP})
  2476. client, err := New(endpoint, accessID, accessKey, SetLocalAddr(localTCPAddr))
  2477. c.Assert(err, IsNil)
  2478. var bucketNameTest = bucketNamePrefix + RandLowStr(6)
  2479. err = client.CreateBucket(bucketNameTest)
  2480. c.Assert(err, NotNil)
  2481. }
  2482. func (s *OssClientSuite) TestClientSetLocalIpSuccess(c *C) {
  2483. //get local ip
  2484. conn, err := net.Dial("udp", "8.8.8.8:80")
  2485. c.Assert(err, IsNil)
  2486. localAddr := conn.LocalAddr().(*net.UDPAddr)
  2487. localIp := localAddr.IP.String()
  2488. conn.Close()
  2489. ipAddr, err := net.ResolveIPAddr("ip", localIp)
  2490. c.Assert(err, IsNil)
  2491. localTCPAddr := &(net.TCPAddr{IP: ipAddr.IP})
  2492. client, err := New(endpoint, accessID, accessKey, SetLocalAddr(localTCPAddr))
  2493. c.Assert(err, IsNil)
  2494. var bucketNameTest = bucketNamePrefix + RandLowStr(6)
  2495. err = client.CreateBucket(bucketNameTest)
  2496. c.Assert(err, IsNil)
  2497. err = client.DeleteBucket(bucketNameTest)
  2498. c.Assert(err, IsNil)
  2499. }
  2500. // TestCreateBucketInvalidName
  2501. func (s *OssClientSuite) TestCreateBucketInvalidName(c *C) {
  2502. var bucketNameTest = "-" + bucketNamePrefix + RandLowStr(6)
  2503. client, err := New(endpoint, accessID, accessKey)
  2504. c.Assert(err, IsNil)
  2505. // Create
  2506. err = client.CreateBucket(bucketNameTest)
  2507. c.Assert(err, NotNil)
  2508. }
  2509. // TestClientProcessEndpointSuccess
  2510. func (s *OssClientSuite) TestClientProcessEndpointSuccess(c *C) {
  2511. var bucketNameTest = bucketNamePrefix + RandLowStr(6)
  2512. testEndpoint := endpoint + "/" + "sina.com" + "?" + "para=abc"
  2513. client, err := New(testEndpoint, accessID, accessKey)
  2514. c.Assert(err, IsNil)
  2515. // Create
  2516. err = client.CreateBucket(bucketNameTest)
  2517. c.Assert(err, IsNil)
  2518. // delete
  2519. err = client.DeleteBucket(bucketNameTest)
  2520. c.Assert(err, IsNil)
  2521. }
  2522. // TestClientProcessEndpointSuccess
  2523. func (s *OssClientSuite) TestClientProcessEndpointError(c *C) {
  2524. var bucketNameTest = bucketNamePrefix + RandLowStr(6)
  2525. testEndpoint := "https://127.0.0.1/" + endpoint
  2526. client, err := New(testEndpoint, accessID, accessKey)
  2527. c.Assert(err, IsNil)
  2528. // Create
  2529. err = client.CreateBucket(bucketNameTest)
  2530. c.Assert(err, NotNil)
  2531. }
  2532. // TestClientBucketError
  2533. func (s *OssClientSuite) TestClientBucketError(c *C) {
  2534. client, err := New(endpoint, accessID, accessKey)
  2535. c.Assert(err, IsNil)
  2536. bucketName := "-" + RandLowStr(5)
  2537. _, err = client.Bucket(bucketName)
  2538. c.Assert(err, NotNil)
  2539. }
  2540. func (s *OssClientSuite) TestSetBucketInventory(c *C) {
  2541. client, err := New(endpoint, accessID, accessKey)
  2542. c.Assert(err, IsNil)
  2543. bucketName := bucketNamePrefix + RandLowStr(5)
  2544. err = client.CreateBucket(bucketName)
  2545. c.Assert(err, IsNil)
  2546. // encryption config
  2547. var invSseOss InvSseOss
  2548. invSseKms := InvSseKms{
  2549. KmsId: "keyId",
  2550. }
  2551. var invEncryption InvEncryption
  2552. bl := true
  2553. // not any encryption
  2554. invConfig := InventoryConfiguration{
  2555. Id: "report1",
  2556. IsEnabled: &bl,
  2557. Prefix: "filterPrefix/",
  2558. OSSBucketDestination: OSSBucketDestination{
  2559. Format: "CSV",
  2560. AccountId: accountID,
  2561. RoleArn: stsARN,
  2562. Bucket: "acs:oss:::" + bucketName,
  2563. Prefix: "prefix1",
  2564. },
  2565. Frequency: "Daily",
  2566. IncludedObjectVersions: "All",
  2567. OptionalFields: OptionalFields{
  2568. Field: []string{
  2569. "Size", "LastModifiedDate", "ETag", "StorageClass", "IsMultipartUploaded", "EncryptionStatus",
  2570. },
  2571. },
  2572. }
  2573. // case 1: not any encryption
  2574. err = client.SetBucketInventory(bucketName, invConfig)
  2575. c.Assert(err, IsNil)
  2576. // case 2: use kms encryption
  2577. invConfig.Id = "report2"
  2578. invEncryption.SseKms = &invSseKms
  2579. invEncryption.SseOss = nil
  2580. invConfig.OSSBucketDestination.Encryption = &invEncryption
  2581. err = client.SetBucketInventory(bucketName, invConfig)
  2582. c.Assert(err, IsNil)
  2583. // case 3: use SseOss encryption
  2584. invConfig.Id = "report3"
  2585. invEncryption.SseKms = nil
  2586. invEncryption.SseOss = &invSseOss
  2587. invConfig.OSSBucketDestination.Encryption = &invEncryption
  2588. err = client.SetBucketInventory(bucketName, invConfig)
  2589. c.Assert(err, IsNil)
  2590. //case 4: use two type encryption
  2591. invConfig.Id = "report4"
  2592. invEncryption.SseKms = &invSseKms
  2593. invEncryption.SseOss = &invSseOss
  2594. invConfig.OSSBucketDestination.Encryption = &invEncryption
  2595. err = client.SetBucketInventory(bucketName, invConfig)
  2596. c.Assert(err, NotNil)
  2597. err = client.DeleteBucket(bucketName)
  2598. c.Assert(err, IsNil)
  2599. }
  2600. func (s *OssClientSuite) TestBucketInventory(c *C) {
  2601. client, err := New(endpoint, accessID, accessKey)
  2602. c.Assert(err, IsNil)
  2603. bucketName := bucketNamePrefix + RandLowStr(5)
  2604. err = client.CreateBucket(bucketName)
  2605. c.Assert(err, IsNil)
  2606. bl := true
  2607. invConfig := InventoryConfiguration{
  2608. Id: "report1",
  2609. IsEnabled: &bl,
  2610. Prefix: "filterPrefix/",
  2611. OSSBucketDestination: OSSBucketDestination{
  2612. Format: "CSV",
  2613. AccountId: accountID,
  2614. RoleArn: stsARN,
  2615. Bucket: "acs:oss:::" + bucketName,
  2616. Prefix: "prefix1",
  2617. },
  2618. Frequency: "Daily",
  2619. IncludedObjectVersions: "All",
  2620. OptionalFields: OptionalFields{
  2621. Field: []string{
  2622. "Size", "LastModifiedDate", "ETag", "StorageClass", "IsMultipartUploaded", "EncryptionStatus",
  2623. },
  2624. },
  2625. }
  2626. // case 1: test SetBucketInventory
  2627. err = client.SetBucketInventory(bucketName, invConfig)
  2628. c.Assert(err, IsNil)
  2629. // case 2: test GetBucketInventory
  2630. out, err := client.GetBucketInventory(bucketName, "report1")
  2631. c.Assert(err, IsNil)
  2632. invConfig.XMLName.Local = "InventoryConfiguration"
  2633. invConfig.OSSBucketDestination.XMLName.Local = "OSSBucketDestination"
  2634. invConfig.OptionalFields.XMLName.Local = "OptionalFields"
  2635. c.Assert(struct2string(invConfig, c), Equals, struct2string(out, c))
  2636. // case 3: test ListBucketInventory
  2637. invConfig2 := InventoryConfiguration{
  2638. Id: "report2",
  2639. IsEnabled: &bl,
  2640. Prefix: "filterPrefix/",
  2641. OSSBucketDestination: OSSBucketDestination{
  2642. Format: "CSV",
  2643. AccountId: accountID,
  2644. RoleArn: stsARN,
  2645. Bucket: "acs:oss:::" + bucketName,
  2646. Prefix: "prefix1",
  2647. },
  2648. Frequency: "Daily",
  2649. IncludedObjectVersions: "All",
  2650. OptionalFields: OptionalFields{
  2651. Field: []string{
  2652. "Size", "LastModifiedDate", "ETag", "StorageClass", "IsMultipartUploaded", "EncryptionStatus",
  2653. },
  2654. },
  2655. }
  2656. invConfig2.XMLName.Local = "InventoryConfiguration"
  2657. invConfig2.OSSBucketDestination.XMLName.Local = "OSSBucketDestination"
  2658. invConfig2.OptionalFields.XMLName.Local = "OptionalFields"
  2659. err = client.SetBucketInventory(bucketName, invConfig2)
  2660. c.Assert(err, IsNil)
  2661. listInvConf, err := client.ListBucketInventory(bucketName, "", Marker("report1"), MaxKeys(2))
  2662. c.Assert(err, IsNil)
  2663. var listInvLocal ListInventoryConfigurationsResult
  2664. listInvLocal.InventoryConfiguration = []InventoryConfiguration{
  2665. invConfig,
  2666. invConfig2,
  2667. }
  2668. bo := false
  2669. listInvLocal.IsTruncated = &bo
  2670. listInvLocal.XMLName.Local = "ListInventoryConfigurationsResult"
  2671. c.Assert(struct2string(listInvLocal, c), Equals, struct2string(listInvConf, c))
  2672. for i := 3; i < 109; i++ {
  2673. invConfig2 := InventoryConfiguration{
  2674. Id: "report" + strconv.Itoa(i),
  2675. IsEnabled: &bl,
  2676. Prefix: "filterPrefix/",
  2677. OSSBucketDestination: OSSBucketDestination{
  2678. Format: "CSV",
  2679. AccountId: accountID,
  2680. RoleArn: stsARN,
  2681. Bucket: "acs:oss:::" + bucketName,
  2682. Prefix: "prefix1",
  2683. },
  2684. Frequency: "Daily",
  2685. IncludedObjectVersions: "All",
  2686. OptionalFields: OptionalFields{
  2687. Field: []string{
  2688. "Size", "LastModifiedDate", "ETag", "StorageClass", "IsMultipartUploaded", "EncryptionStatus",
  2689. },
  2690. },
  2691. }
  2692. err = client.SetBucketInventory(bucketName, invConfig2)
  2693. c.Assert(err, IsNil)
  2694. }
  2695. token := ""
  2696. for {
  2697. listInvConf1, err := client.ListBucketInventory(bucketName, token)
  2698. c.Assert(err, IsNil)
  2699. token = listInvConf1.NextContinuationToken
  2700. testLogger.Println(listInvConf1.NextContinuationToken, *listInvConf1.IsTruncated, token)
  2701. if *listInvConf1.IsTruncated == false {
  2702. break
  2703. } else {
  2704. c.Assert(listInvConf1.NextContinuationToken, Equals, "report91")
  2705. }
  2706. }
  2707. // case 4: test DeleteBucketInventory
  2708. for i := 1; i < 109; i++ {
  2709. err = client.DeleteBucketInventory(bucketName, "report"+strconv.Itoa(i))
  2710. c.Assert(err, IsNil)
  2711. }
  2712. err = client.DeleteBucket(bucketName)
  2713. c.Assert(err, IsNil)
  2714. }
  2715. func (s *OssClientSuite) TestBucketInventoryNegative(c *C) {
  2716. client, err := New(endpoint, accessID, accessKey)
  2717. c.Assert(err, IsNil)
  2718. bucketName := bucketNamePrefix + RandLowStr(5)
  2719. err = client.CreateBucket(bucketName)
  2720. c.Assert(err, IsNil)
  2721. bl := true
  2722. invConfigErr := InventoryConfiguration{
  2723. Id: "report1",
  2724. IsEnabled: &bl,
  2725. Prefix: "filterPrefix/",
  2726. OSSBucketDestination: OSSBucketDestination{
  2727. Format: "CSV",
  2728. AccountId: accountID,
  2729. RoleArn: stsARN,
  2730. Bucket: "test",
  2731. Prefix: "prefix1",
  2732. },
  2733. Frequency: "Daily",
  2734. IncludedObjectVersions: "All",
  2735. OptionalFields: OptionalFields{
  2736. Field: []string{
  2737. "Size", "LastModifiedDate", "ETag", "StorageClass", "IsMultipartUploaded", "EncryptionStatus",
  2738. },
  2739. },
  2740. }
  2741. // case 1: test SetBucketInventory
  2742. err = client.SetBucketInventory(bucketName, invConfigErr)
  2743. c.Assert(err, NotNil)
  2744. // case 2: test GetBucketInventory
  2745. _, err = client.GetBucketInventory(bucketName, "report1")
  2746. c.Assert(err, NotNil)
  2747. // case 3: test ListBucketInventory
  2748. _, err = client.ListBucketInventory(bucketName, "", Marker("report1"), MaxKeys(2))
  2749. c.Assert(err, NotNil)
  2750. // case 4: test DeleteBucketInventory
  2751. err = client.DeleteBucketInventory(bucketName, "report1")
  2752. c.Assert(err, IsNil)
  2753. err = client.DeleteBucket(bucketName)
  2754. c.Assert(err, IsNil)
  2755. }
  2756. func (s *OssClientSuite) TestBucketAsyncTask(c *C) {
  2757. client, err := New(endpoint, accessID, accessKey)
  2758. c.Assert(err, IsNil)
  2759. bucketName := bucketNamePrefix + RandLowStr(5)
  2760. err = client.CreateBucket(bucketName)
  2761. c.Assert(err, IsNil)
  2762. objectName := objectNamePrefix + RandLowStr(6)
  2763. // set asyn task,IgnoreSameKey is false
  2764. asynConf := AsyncFetchTaskConfiguration{
  2765. Url: "http://www.baidu.com",
  2766. Object: objectName,
  2767. Host: "",
  2768. ContentMD5: "",
  2769. Callback: "",
  2770. StorageClass: "",
  2771. IgnoreSameKey: false,
  2772. }
  2773. asynResult, err := client.SetBucketAsyncTask(bucketName, asynConf)
  2774. c.Assert(err, IsNil)
  2775. c.Assert(len(asynResult.TaskId) > 0, Equals, true)
  2776. // get asyn task
  2777. asynTask, err := client.GetBucketAsyncTask(bucketName, asynResult.TaskId)
  2778. c.Assert(err, IsNil)
  2779. c.Assert(asynResult.TaskId, Equals, asynTask.TaskId)
  2780. c.Assert(len(asynTask.State) > 0, Equals, true)
  2781. c.Assert(asynConf.Url, Equals, asynTask.TaskInfo.Url)
  2782. c.Assert(asynConf.Object, Equals, asynTask.TaskInfo.Object)
  2783. c.Assert(asynConf.Callback, Equals, asynTask.TaskInfo.Callback)
  2784. c.Assert(asynConf.IgnoreSameKey, Equals, asynTask.TaskInfo.IgnoreSameKey)
  2785. // test again,IgnoreSameKey is true
  2786. asynConf.IgnoreSameKey = true
  2787. asynResult, err = client.SetBucketAsyncTask(bucketName, asynConf)
  2788. c.Assert(err, IsNil)
  2789. c.Assert(len(asynResult.TaskId) > 0, Equals, true)
  2790. asynTask, err = client.GetBucketAsyncTask(bucketName, asynResult.TaskId)
  2791. c.Assert(asynConf.IgnoreSameKey, Equals, asynTask.TaskInfo.IgnoreSameKey)
  2792. err = client.DeleteBucket(bucketName)
  2793. c.Assert(err, IsNil)
  2794. }
  2795. func (s *OssClientSuite) TestClientOptionHeader(c *C) {
  2796. // create a bucket with default proprety
  2797. client, err := New(endpoint, accessID, accessKey)
  2798. c.Assert(err, IsNil)
  2799. bucketName := bucketNamePrefix + RandLowStr(6)
  2800. var respHeader http.Header
  2801. err = client.CreateBucket(bucketName, GetResponseHeader(&respHeader))
  2802. c.Assert(err, IsNil)
  2803. c.Assert(GetRequestId(respHeader) != "", Equals, true)
  2804. // put bucket version:enabled
  2805. var versioningConfig VersioningConfig
  2806. versioningConfig.Status = string(VersionEnabled)
  2807. err = client.SetBucketVersioning(bucketName, versioningConfig)
  2808. c.Assert(err, IsNil)
  2809. // get bucket version success,use payer
  2810. options := []Option{RequestPayer(BucketOwner), GetResponseHeader(&respHeader)}
  2811. versioningResult, err := client.GetBucketVersioning(bucketName, options...)
  2812. c.Assert(versioningResult.Status, Equals, "Enabled")
  2813. c.Assert(GetRequestId(respHeader) != "", Equals, true)
  2814. //list buckets,use payer
  2815. _, err = client.ListBuckets(options...)
  2816. c.Assert(err, IsNil)
  2817. ForceDeleteBucket(client, bucketName, c)
  2818. }
  2819. // compare with go1.7
  2820. func compareVersion(goVersion string) bool {
  2821. nowVersion := runtime.Version()
  2822. nowVersion = strings.Replace(nowVersion, "go", "", -1)
  2823. pSlice1 := strings.Split(goVersion, ".")
  2824. pSlice2 := strings.Split(nowVersion, ".")
  2825. for k, v := range pSlice2 {
  2826. n2, _ := strconv.Atoi(string(v))
  2827. n1, _ := strconv.Atoi(string(pSlice1[k]))
  2828. if n2 > n1 {
  2829. return true
  2830. }
  2831. if n2 < n1 {
  2832. return false
  2833. }
  2834. }
  2835. return true
  2836. }
  2837. func homeHandler(w http.ResponseWriter, r *http.Request) {
  2838. http.Redirect(w, r, "/redirectTo", http.StatusFound)
  2839. }
  2840. func targetHandler(w http.ResponseWriter, r *http.Request) {
  2841. fmt.Fprintf(w, "You have been redirected here!")
  2842. }
  2843. func (s *OssClientSuite) TestClientRedirect(c *C) {
  2844. // must go1.7.0 onward
  2845. if !compareVersion("1.7.0") {
  2846. return
  2847. }
  2848. // get port
  2849. rand.Seed(time.Now().Unix())
  2850. port := 10000 + rand.Intn(10000)
  2851. // start http server
  2852. httpAddr := fmt.Sprintf("127.0.0.1:%d", port)
  2853. mux := http.NewServeMux()
  2854. mux.HandleFunc("/redirectTo", targetHandler)
  2855. mux.HandleFunc("/", homeHandler)
  2856. svr := &http.Server{
  2857. Addr: httpAddr,
  2858. ReadTimeout: 10 * time.Second,
  2859. WriteTimeout: 10 * time.Second,
  2860. MaxHeaderBytes: 1 << 20,
  2861. Handler: mux,
  2862. }
  2863. go func() {
  2864. svr.ListenAndServe()
  2865. }()
  2866. url := "http://" + httpAddr
  2867. // create client 1,redirect disable
  2868. client1, err := New(endpoint, accessID, accessKey, RedirectEnabled(false))
  2869. resp, err := client1.Conn.client.Get(url)
  2870. c.Assert(err, IsNil)
  2871. c.Assert(resp.StatusCode, Equals, http.StatusFound)
  2872. resp.Body.Close()
  2873. // create client2, redirect enabled
  2874. client2, err := New(endpoint, accessID, accessKey, RedirectEnabled(true))
  2875. resp, err = client2.Conn.client.Get(url)
  2876. c.Assert(err, IsNil)
  2877. c.Assert(resp.StatusCode, Equals, 200)
  2878. data, err := ioutil.ReadAll(resp.Body)
  2879. c.Assert(string(data), Equals, "You have been redirected here!")
  2880. resp.Body.Close()
  2881. }