bucket_test.go 76 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293129412951296129712981299130013011302130313041305130613071308130913101311131213131314131513161317131813191320132113221323132413251326132713281329133013311332133313341335133613371338133913401341134213431344134513461347134813491350135113521353135413551356135713581359136013611362136313641365136613671368136913701371137213731374137513761377137813791380138113821383138413851386138713881389139013911392139313941395139613971398139914001401140214031404140514061407140814091410141114121413141414151416141714181419142014211422142314241425142614271428142914301431143214331434143514361437143814391440144114421443144414451446144714481449145014511452145314541455145614571458145914601461146214631464146514661467146814691470147114721473147414751476147714781479148014811482148314841485148614871488148914901491149214931494149514961497149814991500150115021503150415051506150715081509151015111512151315141515151615171518151915201521152215231524152515261527152815291530153115321533153415351536153715381539154015411542154315441545154615471548154915501551155215531554155515561557155815591560156115621563156415651566156715681569157015711572157315741575157615771578157915801581158215831584158515861587158815891590159115921593159415951596159715981599160016011602160316041605160616071608160916101611161216131614161516161617161816191620162116221623162416251626162716281629163016311632163316341635163616371638163916401641164216431644164516461647164816491650165116521653165416551656165716581659166016611662166316641665166616671668166916701671167216731674167516761677167816791680168116821683168416851686168716881689169016911692169316941695169616971698169917001701170217031704170517061707170817091710171117121713171417151716171717181719172017211722172317241725172617271728172917301731173217331734173517361737173817391740174117421743174417451746174717481749175017511752175317541755175617571758175917601761176217631764176517661767176817691770177117721773177417751776177717781779178017811782178317841785178617871788178917901791179217931794179517961797179817991800180118021803180418051806180718081809181018111812181318141815181618171818181918201821182218231824182518261827182818291830183118321833183418351836183718381839184018411842184318441845184618471848184918501851185218531854185518561857185818591860186118621863186418651866186718681869187018711872187318741875187618771878187918801881188218831884188518861887188818891890189118921893189418951896189718981899190019011902190319041905190619071908190919101911191219131914191519161917191819191920192119221923192419251926192719281929193019311932193319341935193619371938193919401941194219431944194519461947194819491950195119521953195419551956195719581959196019611962196319641965196619671968196919701971197219731974197519761977197819791980198119821983198419851986198719881989199019911992199319941995199619971998199920002001200220032004200520062007200820092010201120122013201420152016201720182019202020212022202320242025202620272028202920302031203220332034203520362037203820392040204120422043204420452046204720482049205020512052205320542055205620572058205920602061206220632064206520662067206820692070207120722073207420752076207720782079208020812082208320842085208620872088208920902091209220932094209520962097209820992100210121022103210421052106210721082109211021112112211321142115211621172118211921202121212221232124212521262127212821292130213121322133213421352136213721382139214021412142214321442145214621472148214921502151215221532154215521562157215821592160216121622163216421652166216721682169217021712172217321742175217621772178217921802181218221832184218521862187218821892190219121922193219421952196219721982199220022012202220322042205220622072208220922102211221222132214221522162217221822192220222122222223222422252226222722282229223022312232223322342235223622372238223922402241224222432244224522462247224822492250225122522253225422552256225722582259226022612262226322642265226622672268226922702271227222732274227522762277227822792280228122822283228422852286228722882289229022912292229322942295229622972298229923002301230223032304230523062307230823092310231123122313231423152316231723182319232023212322232323242325232623272328232923302331233223332334233523362337233823392340234123422343234423452346234723482349235023512352235323542355235623572358235923602361236223632364236523662367236823692370237123722373237423752376237723782379238023812382238323842385238623872388238923902391239223932394239523962397239823992400240124022403240424052406240724082409241024112412241324142415241624172418241924202421242224232424242524262427242824292430243124322433243424352436243724382439244024412442244324442445244624472448244924502451245224532454245524562457245824592460246124622463246424652466246724682469247024712472247324742475247624772478247924802481248224832484248524862487248824892490249124922493249424952496249724982499250025012502250325042505250625072508250925102511251225132514251525162517251825192520252125222523252425252526252725282529253025312532253325342535253625372538253925402541254225432544254525462547254825492550255125522553255425552556255725582559256025612562256325642565256625672568256925702571257225732574257525762577257825792580258125822583258425852586258725882589259025912592259325942595259625972598259926002601260226032604260526062607260826092610261126122613261426152616261726182619262026212622262326242625262626272628262926302631263226332634263526362637263826392640264126422643264426452646264726482649265026512652265326542655265626572658265926602661266226632664266526662667266826692670267126722673
  1. // Bucket test
  2. package oss
  3. import (
  4. "bytes"
  5. "encoding/base64"
  6. "errors"
  7. "fmt"
  8. "io"
  9. "io/ioutil"
  10. "math/rand"
  11. "net/http"
  12. "os"
  13. "path/filepath"
  14. "strconv"
  15. "strings"
  16. "time"
  17. "github.com/baiyubin/aliyun-sts-go-sdk/sts"
  18. . "gopkg.in/check.v1"
  19. )
  20. type OssBucketSuite struct {
  21. client *Client
  22. bucket *Bucket
  23. archiveBucket *Bucket
  24. }
  25. var _ = Suite(&OssBucketSuite{})
  26. var (
  27. pastDate = time.Date(2009, time.November, 10, 23, 0, 0, 0, time.UTC)
  28. futureDate = time.Date(2049, time.January, 10, 23, 0, 0, 0, time.UTC)
  29. )
  30. // SetUpSuite runs once when the suite starts running.
  31. func (s *OssBucketSuite) SetUpSuite(c *C) {
  32. client, err := New(endpoint, accessID, accessKey)
  33. c.Assert(err, IsNil)
  34. s.client = client
  35. s.client.CreateBucket(bucketName)
  36. archiveBucketName := bucketName + "-archive"
  37. err = s.client.CreateBucket(archiveBucketName, StorageClass(StorageArchive))
  38. c.Assert(err, IsNil)
  39. time.Sleep(5 * time.Second)
  40. bucket, err := s.client.Bucket(bucketName)
  41. c.Assert(err, IsNil)
  42. s.bucket = bucket
  43. archiveBucket, err := s.client.Bucket(archiveBucketName)
  44. c.Assert(err, IsNil)
  45. s.archiveBucket = archiveBucket
  46. testLogger.Println("test bucket started")
  47. }
  48. // TearDownSuite runs before each test or benchmark starts running.
  49. func (s *OssBucketSuite) TearDownSuite(c *C) {
  50. for _, bucket := range []*Bucket{s.bucket, s.archiveBucket} {
  51. // Delete multipart
  52. lmu, err := bucket.ListMultipartUploads()
  53. c.Assert(err, IsNil)
  54. for _, upload := range lmu.Uploads {
  55. imur := InitiateMultipartUploadResult{Bucket: bucketName, Key: upload.Key, UploadID: upload.UploadID}
  56. err = bucket.AbortMultipartUpload(imur)
  57. c.Assert(err, IsNil)
  58. }
  59. // Delete objects
  60. lor, err := bucket.ListObjects()
  61. c.Assert(err, IsNil)
  62. for _, object := range lor.Objects {
  63. err = bucket.DeleteObject(object.Key)
  64. c.Assert(err, IsNil)
  65. }
  66. }
  67. testLogger.Println("test bucket completed")
  68. }
  69. // SetUpTest runs after each test or benchmark runs.
  70. func (s *OssBucketSuite) SetUpTest(c *C) {
  71. err := removeTempFiles("../oss", ".jpg")
  72. c.Assert(err, IsNil)
  73. }
  74. // TearDownTest runs once after all tests or benchmarks have finished running.
  75. func (s *OssBucketSuite) TearDownTest(c *C) {
  76. err := removeTempFiles("../oss", ".jpg")
  77. c.Assert(err, IsNil)
  78. err = removeTempFiles("../oss", ".txt")
  79. c.Assert(err, IsNil)
  80. err = removeTempFiles("../oss", ".temp")
  81. c.Assert(err, IsNil)
  82. err = removeTempFiles("../oss", ".txt1")
  83. c.Assert(err, IsNil)
  84. err = removeTempFiles("../oss", ".txt2")
  85. c.Assert(err, IsNil)
  86. }
  87. // TestPutObject
  88. func (s *OssBucketSuite) TestPutObject(c *C) {
  89. objectName := objectNamePrefix + "tpo"
  90. objectValue := "大江东去,浪淘尽,千古风流人物。 故垒西边,人道是、三国周郎赤壁。 乱石穿空,惊涛拍岸,卷起千堆雪。 江山如画,一时多少豪杰。" +
  91. "遥想公谨当年,小乔初嫁了,雄姿英发。 羽扇纶巾,谈笑间、樯橹灰飞烟灭。故国神游,多情应笑我,早生华发,人生如梦,一尊还酹江月。"
  92. // Put string
  93. err := s.bucket.PutObject(objectName, strings.NewReader(objectValue))
  94. c.Assert(err, IsNil)
  95. // Check
  96. body, err := s.bucket.GetObject(objectName)
  97. c.Assert(err, IsNil)
  98. str, err := readBody(body)
  99. c.Assert(err, IsNil)
  100. c.Assert(str, Equals, objectValue)
  101. acl, err := s.bucket.GetObjectACL(objectName)
  102. c.Assert(err, IsNil)
  103. testLogger.Println("aclRes:", acl)
  104. c.Assert(acl.ACL, Equals, "default")
  105. err = s.bucket.DeleteObject(objectName)
  106. c.Assert(err, IsNil)
  107. // Put bytes
  108. err = s.bucket.PutObject(objectName, bytes.NewReader([]byte(objectValue)))
  109. c.Assert(err, IsNil)
  110. // Check
  111. body, err = s.bucket.GetObject(objectName)
  112. c.Assert(err, IsNil)
  113. str, err = readBody(body)
  114. c.Assert(err, IsNil)
  115. c.Assert(str, Equals, objectValue)
  116. err = s.bucket.DeleteObject(objectName)
  117. c.Assert(err, IsNil)
  118. // Put file
  119. err = createFileAndWrite(objectName+".txt", []byte(objectValue))
  120. c.Assert(err, IsNil)
  121. fd, err := os.Open(objectName + ".txt")
  122. c.Assert(err, IsNil)
  123. err = s.bucket.PutObject(objectName, fd)
  124. c.Assert(err, IsNil)
  125. os.Remove(objectName + ".txt")
  126. // Check
  127. body, err = s.bucket.GetObject(objectName)
  128. c.Assert(err, IsNil)
  129. str, err = readBody(body)
  130. c.Assert(err, IsNil)
  131. c.Assert(str, Equals, objectValue)
  132. err = s.bucket.DeleteObject(objectName)
  133. c.Assert(err, IsNil)
  134. // Put with properties
  135. objectName = objectNamePrefix + "tpox"
  136. options := []Option{
  137. Expires(futureDate),
  138. ObjectACL(ACLPublicRead),
  139. Meta("myprop", "mypropval"),
  140. }
  141. err = s.bucket.PutObject(objectName, strings.NewReader(objectValue), options...)
  142. c.Assert(err, IsNil)
  143. // Check
  144. body, err = s.bucket.GetObject(objectName)
  145. c.Assert(err, IsNil)
  146. str, err = readBody(body)
  147. c.Assert(err, IsNil)
  148. c.Assert(str, Equals, objectValue)
  149. acl, err = s.bucket.GetObjectACL(objectName)
  150. c.Assert(err, IsNil)
  151. testLogger.Println("GetObjectACL:", acl)
  152. c.Assert(acl.ACL, Equals, string(ACLPublicRead))
  153. meta, err := s.bucket.GetObjectDetailedMeta(objectName)
  154. c.Assert(err, IsNil)
  155. testLogger.Println("GetObjectDetailedMeta:", meta)
  156. c.Assert(meta.Get("X-Oss-Meta-Myprop"), Equals, "mypropval")
  157. err = s.bucket.DeleteObject(objectName)
  158. c.Assert(err, IsNil)
  159. }
  160. func (s *OssBucketSuite) TestSignURL(c *C) {
  161. objectName := objectNamePrefix + randStr(5)
  162. objectValue := randStr(20)
  163. filePath := randLowStr(10)
  164. content := "复写object"
  165. createFile(filePath, content, c)
  166. notExistfilePath := randLowStr(10)
  167. os.Remove(notExistfilePath)
  168. // Sign URL for put
  169. str, err := s.bucket.SignURL(objectName, HTTPPut, 60)
  170. c.Assert(err, IsNil)
  171. c.Assert(strings.Contains(str, HTTPParamExpires+"="), Equals, true)
  172. c.Assert(strings.Contains(str, HTTPParamAccessKeyID+"="), Equals, true)
  173. c.Assert(strings.Contains(str, HTTPParamSignature+"="), Equals, true)
  174. // Error put object with URL
  175. err = s.bucket.PutObjectWithURL(str, strings.NewReader(objectValue), ContentType("image/tiff"))
  176. c.Assert(err, NotNil)
  177. c.Assert(err.(ServiceError).Code, Equals, "SignatureDoesNotMatch")
  178. err = s.bucket.PutObjectFromFileWithURL(str, filePath, ContentType("image/tiff"))
  179. c.Assert(err, NotNil)
  180. c.Assert(err.(ServiceError).Code, Equals, "SignatureDoesNotMatch")
  181. // Put object with URL
  182. err = s.bucket.PutObjectWithURL(str, strings.NewReader(objectValue))
  183. c.Assert(err, IsNil)
  184. acl, err := s.bucket.GetObjectACL(objectName)
  185. c.Assert(err, IsNil)
  186. c.Assert(acl.ACL, Equals, "default")
  187. // Get object meta
  188. meta, err := s.bucket.GetObjectDetailedMeta(objectName)
  189. c.Assert(err, IsNil)
  190. c.Assert(meta.Get(HTTPHeaderContentType), Equals, "application/octet-stream")
  191. c.Assert(meta.Get("X-Oss-Meta-Myprop"), Equals, "")
  192. // Sign URL for function GetObjectWithURL
  193. str, err = s.bucket.SignURL(objectName, HTTPGet, 60)
  194. c.Assert(err, IsNil)
  195. c.Assert(strings.Contains(str, HTTPParamExpires+"="), Equals, true)
  196. c.Assert(strings.Contains(str, HTTPParamAccessKeyID+"="), Equals, true)
  197. c.Assert(strings.Contains(str, HTTPParamSignature+"="), Equals, true)
  198. // Get object with URL
  199. body, err := s.bucket.GetObjectWithURL(str)
  200. c.Assert(err, IsNil)
  201. str, err = readBody(body)
  202. c.Assert(err, IsNil)
  203. c.Assert(str, Equals, objectValue)
  204. // Sign URL for function PutObjectWithURL
  205. options := []Option{
  206. ObjectACL(ACLPublicRead),
  207. Meta("myprop", "mypropval"),
  208. ContentType("image/tiff"),
  209. ResponseContentEncoding("deflate"),
  210. }
  211. str, err = s.bucket.SignURL(objectName, HTTPPut, 60, options...)
  212. c.Assert(err, IsNil)
  213. c.Assert(strings.Contains(str, HTTPParamExpires+"="), Equals, true)
  214. c.Assert(strings.Contains(str, HTTPParamAccessKeyID+"="), Equals, true)
  215. c.Assert(strings.Contains(str, HTTPParamSignature+"="), Equals, true)
  216. // Put object with URL from file
  217. // Without option, error
  218. err = s.bucket.PutObjectWithURL(str, strings.NewReader(objectValue))
  219. c.Assert(err, NotNil)
  220. c.Assert(err.(ServiceError).Code, Equals, "SignatureDoesNotMatch")
  221. err = s.bucket.PutObjectFromFileWithURL(str, filePath)
  222. c.Assert(err, NotNil)
  223. c.Assert(err.(ServiceError).Code, Equals, "SignatureDoesNotMatch")
  224. // With option, error file
  225. err = s.bucket.PutObjectFromFileWithURL(str, notExistfilePath, options...)
  226. c.Assert(err, NotNil)
  227. // With option
  228. err = s.bucket.PutObjectFromFileWithURL(str, filePath, options...)
  229. c.Assert(err, IsNil)
  230. // Get object meta
  231. meta, err = s.bucket.GetObjectDetailedMeta(objectName)
  232. c.Assert(err, IsNil)
  233. c.Assert(meta.Get("X-Oss-Meta-Myprop"), Equals, "mypropval")
  234. c.Assert(meta.Get(HTTPHeaderContentType), Equals, "image/tiff")
  235. acl, err = s.bucket.GetObjectACL(objectName)
  236. c.Assert(err, IsNil)
  237. c.Assert(acl.ACL, Equals, string(ACLPublicRead))
  238. // Sign URL for function GetObjectToFileWithURL
  239. str, err = s.bucket.SignURL(objectName, HTTPGet, 60)
  240. c.Assert(err, IsNil)
  241. // Get object to file with URL
  242. newFile := randStr(10)
  243. err = s.bucket.GetObjectToFileWithURL(str, newFile)
  244. c.Assert(err, IsNil)
  245. eq, err := compareFiles(filePath, newFile)
  246. c.Assert(err, IsNil)
  247. c.Assert(eq, Equals, true)
  248. os.Remove(newFile)
  249. // Get object to file error
  250. err = s.bucket.GetObjectToFileWithURL(str, newFile, options...)
  251. c.Assert(err, NotNil)
  252. c.Assert(err.(ServiceError).Code, Equals, "SignatureDoesNotMatch")
  253. _, err = os.Stat(newFile)
  254. c.Assert(err, NotNil)
  255. // Get object error
  256. body, err = s.bucket.GetObjectWithURL(str, options...)
  257. c.Assert(err, NotNil)
  258. c.Assert(err.(ServiceError).Code, Equals, "SignatureDoesNotMatch")
  259. c.Assert(body, IsNil)
  260. // Sign URL for function GetObjectToFileWithURL
  261. options = []Option{
  262. Expires(futureDate),
  263. ObjectACL(ACLPublicRead),
  264. Meta("myprop", "mypropval"),
  265. ContentType("image/tiff"),
  266. ResponseContentEncoding("deflate"),
  267. }
  268. str, err = s.bucket.SignURL(objectName, HTTPGet, 60, options...)
  269. c.Assert(err, IsNil)
  270. // Get object to file with URL and options
  271. err = s.bucket.GetObjectToFileWithURL(str, newFile, options...)
  272. c.Assert(err, IsNil)
  273. eq, err = compareFiles(filePath, newFile)
  274. c.Assert(err, IsNil)
  275. c.Assert(eq, Equals, true)
  276. os.Remove(newFile)
  277. // Get object to file error
  278. err = s.bucket.GetObjectToFileWithURL(str, newFile)
  279. c.Assert(err, NotNil)
  280. c.Assert(err.(ServiceError).Code, Equals, "SignatureDoesNotMatch")
  281. _, err = os.Stat(newFile)
  282. c.Assert(err, NotNil)
  283. // Get object error
  284. body, err = s.bucket.GetObjectWithURL(str)
  285. c.Assert(err, NotNil)
  286. c.Assert(err.(ServiceError).Code, Equals, "SignatureDoesNotMatch")
  287. c.Assert(body, IsNil)
  288. err = s.bucket.PutObjectFromFile(objectName, "../sample/The Go Programming Language.html")
  289. c.Assert(err, IsNil)
  290. str, err = s.bucket.SignURL(objectName, HTTPGet, 3600, AcceptEncoding("gzip"))
  291. c.Assert(err, IsNil)
  292. s.bucket.GetObjectToFileWithURL(str, newFile)
  293. c.Assert(err, IsNil)
  294. os.Remove(filePath)
  295. os.Remove(newFile)
  296. // Sign URL error
  297. str, err = s.bucket.SignURL(objectName, HTTPGet, -1)
  298. c.Assert(err, NotNil)
  299. err = s.bucket.DeleteObject(objectName)
  300. c.Assert(err, IsNil)
  301. // Invalid URL parse
  302. str = randStr(20)
  303. err = s.bucket.PutObjectWithURL(str, strings.NewReader(objectValue))
  304. c.Assert(err, NotNil)
  305. err = s.bucket.GetObjectToFileWithURL(str, newFile)
  306. c.Assert(err, NotNil)
  307. }
  308. func (s *OssBucketSuite) TestSignURLWithEscapedKey(c *C) {
  309. // Key with '/'
  310. objectName := "zyimg/86/e8/653b5dc97bb0022051a84c632bc4"
  311. objectValue := "弃我去者,昨日之日不可留;乱我心者,今日之日多烦忧。长风万里送秋雁,对此可以酣高楼。蓬莱文章建安骨,中间小谢又清发。" +
  312. "俱怀逸兴壮思飞,欲上青天揽明月。抽刀断水水更流,举杯销愁愁更愁。人生在世不称意,明朝散发弄扁舟。"
  313. // Sign URL for function PutObjectWithURL
  314. str, err := s.bucket.SignURL(objectName, HTTPPut, 60)
  315. c.Assert(err, IsNil)
  316. c.Assert(strings.Contains(str, HTTPParamExpires+"="), Equals, true)
  317. c.Assert(strings.Contains(str, HTTPParamAccessKeyID+"="), Equals, true)
  318. c.Assert(strings.Contains(str, HTTPParamSignature+"="), Equals, true)
  319. // Put object with URL
  320. err = s.bucket.PutObjectWithURL(str, strings.NewReader(objectValue))
  321. c.Assert(err, IsNil)
  322. // Sign URL for function GetObjectWithURL
  323. str, err = s.bucket.SignURL(objectName, HTTPGet, 60)
  324. c.Assert(err, IsNil)
  325. c.Assert(strings.Contains(str, HTTPParamExpires+"="), Equals, true)
  326. c.Assert(strings.Contains(str, HTTPParamAccessKeyID+"="), Equals, true)
  327. c.Assert(strings.Contains(str, HTTPParamSignature+"="), Equals, true)
  328. // Get object with URL
  329. body, err := s.bucket.GetObjectWithURL(str)
  330. c.Assert(err, IsNil)
  331. str, err = readBody(body)
  332. c.Assert(err, IsNil)
  333. c.Assert(str, Equals, objectValue)
  334. // Key with escaped chars
  335. objectName = "<>[]()`?.,!@#$%^&'/*-_=+~:;"
  336. // Sign URL for funciton PutObjectWithURL
  337. str, err = s.bucket.SignURL(objectName, HTTPPut, 60)
  338. c.Assert(err, IsNil)
  339. c.Assert(strings.Contains(str, HTTPParamExpires+"="), Equals, true)
  340. c.Assert(strings.Contains(str, HTTPParamAccessKeyID+"="), Equals, true)
  341. c.Assert(strings.Contains(str, HTTPParamSignature+"="), Equals, true)
  342. // Put object with URL
  343. err = s.bucket.PutObjectWithURL(str, strings.NewReader(objectValue))
  344. c.Assert(err, IsNil)
  345. // Sign URL for function GetObjectWithURL
  346. str, err = s.bucket.SignURL(objectName, HTTPGet, 60)
  347. c.Assert(err, IsNil)
  348. c.Assert(strings.Contains(str, HTTPParamExpires+"="), Equals, true)
  349. c.Assert(strings.Contains(str, HTTPParamAccessKeyID+"="), Equals, true)
  350. c.Assert(strings.Contains(str, HTTPParamSignature+"="), Equals, true)
  351. // Get object with URL
  352. body, err = s.bucket.GetObjectWithURL(str)
  353. c.Assert(err, IsNil)
  354. str, err = readBody(body)
  355. c.Assert(err, IsNil)
  356. c.Assert(str, Equals, objectValue)
  357. // Key with Chinese chars
  358. objectName = "风吹柳花满店香,吴姬压酒劝客尝。金陵子弟来相送,欲行不行各尽觞。请君试问东流水,别意与之谁短长。"
  359. // Sign URL for function PutObjectWithURL
  360. str, err = s.bucket.SignURL(objectName, HTTPPut, 60)
  361. c.Assert(err, IsNil)
  362. c.Assert(strings.Contains(str, HTTPParamExpires+"="), Equals, true)
  363. c.Assert(strings.Contains(str, HTTPParamAccessKeyID+"="), Equals, true)
  364. c.Assert(strings.Contains(str, HTTPParamSignature+"="), Equals, true)
  365. // Put object with URL
  366. err = s.bucket.PutObjectWithURL(str, strings.NewReader(objectValue))
  367. c.Assert(err, IsNil)
  368. // Sign URL for get function GetObjectWithURL
  369. str, err = s.bucket.SignURL(objectName, HTTPGet, 60)
  370. c.Assert(err, IsNil)
  371. c.Assert(strings.Contains(str, HTTPParamExpires+"="), Equals, true)
  372. c.Assert(strings.Contains(str, HTTPParamAccessKeyID+"="), Equals, true)
  373. c.Assert(strings.Contains(str, HTTPParamSignature+"="), Equals, true)
  374. // Get object with URL
  375. body, err = s.bucket.GetObjectWithURL(str)
  376. c.Assert(err, IsNil)
  377. str, err = readBody(body)
  378. c.Assert(err, IsNil)
  379. c.Assert(str, Equals, objectValue)
  380. // Key
  381. objectName = "test/此情无计可消除/才下眉头/却上 心头/。,;:‘’“”?()『』【】《》!@#¥%……&×/test+ =-_*&^%$#@!`~[]{}()<>|\\/?.,;.txt"
  382. // Sign URL for function PutObjectWithURL
  383. str, err = s.bucket.SignURL(objectName, HTTPPut, 60)
  384. c.Assert(err, IsNil)
  385. // Put object with URL
  386. err = s.bucket.PutObjectWithURL(str, strings.NewReader(objectValue))
  387. c.Assert(err, IsNil)
  388. // Sign URL for function GetObjectWithURL
  389. str, err = s.bucket.SignURL(objectName, HTTPGet, 60)
  390. c.Assert(err, IsNil)
  391. // Get object with URL
  392. body, err = s.bucket.GetObjectWithURL(str)
  393. c.Assert(err, IsNil)
  394. str, err = readBody(body)
  395. c.Assert(err, IsNil)
  396. c.Assert(str, Equals, objectValue)
  397. // Put object
  398. err = s.bucket.PutObject(objectName, bytes.NewReader([]byte(objectValue)))
  399. c.Assert(err, IsNil)
  400. // Get object
  401. body, err = s.bucket.GetObject(objectName)
  402. c.Assert(err, IsNil)
  403. str, err = readBody(body)
  404. c.Assert(err, IsNil)
  405. c.Assert(str, Equals, objectValue)
  406. // Delete object
  407. err = s.bucket.DeleteObject(objectName)
  408. c.Assert(err, IsNil)
  409. }
  410. func (s *OssBucketSuite) TestSignURLWithEscapedKeyAndPorxy(c *C) {
  411. // Key with '/'
  412. objectName := "zyimg/86/e8/653b5dc97bb0022051a84c632bc4"
  413. objectValue := "弃我去者,昨日之日不可留;乱我心者,今日之日多烦忧。长风万里送秋雁,对此可以酣高楼。蓬莱文章建安骨,中间小谢又清发。" +
  414. "俱怀逸兴壮思飞,欲上青天揽明月。抽刀断水水更流,举杯销愁愁更愁。人生在世不称意,明朝散发弄扁舟。"
  415. client, err := New(endpoint, accessID, accessKey, AuthProxy(proxyHost, proxyUser, proxyPasswd))
  416. bucket, err := client.Bucket(bucketName)
  417. // Sign URL for put
  418. str, err := bucket.SignURL(objectName, HTTPPut, 60)
  419. c.Assert(err, IsNil)
  420. c.Assert(strings.Contains(str, HTTPParamExpires+"="), Equals, true)
  421. c.Assert(strings.Contains(str, HTTPParamAccessKeyID+"="), Equals, true)
  422. c.Assert(strings.Contains(str, HTTPParamSignature+"="), Equals, true)
  423. // Put object with URL
  424. err = bucket.PutObjectWithURL(str, strings.NewReader(objectValue))
  425. c.Assert(err, IsNil)
  426. // Sign URL for function GetObjectWithURL
  427. str, err = bucket.SignURL(objectName, HTTPGet, 60)
  428. c.Assert(err, IsNil)
  429. c.Assert(strings.Contains(str, HTTPParamExpires+"="), Equals, true)
  430. c.Assert(strings.Contains(str, HTTPParamAccessKeyID+"="), Equals, true)
  431. c.Assert(strings.Contains(str, HTTPParamSignature+"="), Equals, true)
  432. // Get object with URL
  433. body, err := bucket.GetObjectWithURL(str)
  434. c.Assert(err, IsNil)
  435. str, err = readBody(body)
  436. c.Assert(err, IsNil)
  437. c.Assert(str, Equals, objectValue)
  438. // Key with Chinese chars
  439. objectName = "test/此情无计可消除/才下眉头/却上 心头/。,;:‘’“”?()『』【】《》!@#¥%……&×/test+ =-_*&^%$#@!`~[]{}()<>|\\/?.,;.txt"
  440. // Sign URL for function PutObjectWithURL
  441. str, err = bucket.SignURL(objectName, HTTPPut, 60)
  442. c.Assert(err, IsNil)
  443. // Put object with URL
  444. err = bucket.PutObjectWithURL(str, strings.NewReader(objectValue))
  445. c.Assert(err, IsNil)
  446. // Sign URL for function GetObjectWithURL
  447. str, err = bucket.SignURL(objectName, HTTPGet, 60)
  448. c.Assert(err, IsNil)
  449. // Get object with URL
  450. body, err = bucket.GetObjectWithURL(str)
  451. c.Assert(err, IsNil)
  452. str, err = readBody(body)
  453. c.Assert(err, IsNil)
  454. c.Assert(str, Equals, objectValue)
  455. // Put object
  456. err = bucket.PutObject(objectName, bytes.NewReader([]byte(objectValue)))
  457. c.Assert(err, IsNil)
  458. // Get object
  459. body, err = bucket.GetObject(objectName)
  460. c.Assert(err, IsNil)
  461. str, err = readBody(body)
  462. c.Assert(err, IsNil)
  463. c.Assert(str, Equals, objectValue)
  464. // Delete object
  465. err = bucket.DeleteObject(objectName)
  466. c.Assert(err, IsNil)
  467. }
  468. // TestPutObjectType
  469. func (s *OssBucketSuite) TestPutObjectType(c *C) {
  470. objectName := objectNamePrefix + "tptt"
  471. objectValue := "乱石穿空,惊涛拍岸,卷起千堆雪。 江山如画,一时多少豪杰。"
  472. // Put
  473. err := s.bucket.PutObject(objectName, strings.NewReader(objectValue))
  474. c.Assert(err, IsNil)
  475. // Check
  476. time.Sleep(time.Second)
  477. body, err := s.bucket.GetObject(objectName)
  478. c.Assert(err, IsNil)
  479. str, err := readBody(body)
  480. c.Assert(err, IsNil)
  481. c.Assert(str, Equals, objectValue)
  482. meta, err := s.bucket.GetObjectDetailedMeta(objectName)
  483. c.Assert(err, IsNil)
  484. c.Assert(meta.Get("Content-Type"), Equals, "application/octet-stream")
  485. err = s.bucket.DeleteObject(objectName)
  486. c.Assert(err, IsNil)
  487. // Put
  488. err = s.bucket.PutObject(objectName+".txt", strings.NewReader(objectValue))
  489. c.Assert(err, IsNil)
  490. meta, err = s.bucket.GetObjectDetailedMeta(objectName + ".txt")
  491. c.Assert(err, IsNil)
  492. c.Assert(meta.Get("Content-Type"), Equals, "text/plain; charset=utf-8")
  493. err = s.bucket.DeleteObject(objectName + ".txt")
  494. c.Assert(err, IsNil)
  495. // Put
  496. err = s.bucket.PutObject(objectName+".apk", strings.NewReader(objectValue))
  497. c.Assert(err, IsNil)
  498. meta, err = s.bucket.GetObjectDetailedMeta(objectName + ".apk")
  499. c.Assert(err, IsNil)
  500. c.Assert(meta.Get("Content-Type"), Equals, "application/vnd.android.package-archive")
  501. err = s.bucket.DeleteObject(objectName + ".txt")
  502. c.Assert(err, IsNil)
  503. }
  504. // TestPutObject
  505. func (s *OssBucketSuite) TestPutObjectKeyChars(c *C) {
  506. objectName := objectNamePrefix + "tpokc"
  507. objectValue := "白日依山尽,黄河入海流。欲穷千里目,更上一层楼。"
  508. // Put
  509. objectKey := objectName + "十步杀一人,千里不留行。事了拂衣去,深藏身与名"
  510. err := s.bucket.PutObject(objectKey, strings.NewReader(objectValue))
  511. c.Assert(err, IsNil)
  512. // Check
  513. body, err := s.bucket.GetObject(objectKey)
  514. c.Assert(err, IsNil)
  515. str, err := readBody(body)
  516. c.Assert(err, IsNil)
  517. c.Assert(str, Equals, objectValue)
  518. err = s.bucket.DeleteObject(objectKey)
  519. c.Assert(err, IsNil)
  520. // Put
  521. objectKey = objectName + "ごきげん如何ですかおれの顔をよく拝んでおけ"
  522. err = s.bucket.PutObject(objectKey, strings.NewReader(objectValue))
  523. c.Assert(err, IsNil)
  524. // Check
  525. body, err = s.bucket.GetObject(objectKey)
  526. c.Assert(err, IsNil)
  527. str, err = readBody(body)
  528. c.Assert(err, IsNil)
  529. c.Assert(str, Equals, objectValue)
  530. err = s.bucket.DeleteObject(objectKey)
  531. c.Assert(err, IsNil)
  532. // Put
  533. objectKey = objectName + "~!@#$%^&*()_-+=|\\[]{}<>,./?"
  534. err = s.bucket.PutObject(objectKey, strings.NewReader(objectValue))
  535. c.Assert(err, IsNil)
  536. // Check
  537. body, err = s.bucket.GetObject(objectKey)
  538. c.Assert(err, IsNil)
  539. str, err = readBody(body)
  540. c.Assert(err, IsNil)
  541. c.Assert(str, Equals, objectValue)
  542. err = s.bucket.DeleteObject(objectKey)
  543. c.Assert(err, IsNil)
  544. // Put
  545. objectKey = "go/中国 日本 +-#&=*"
  546. err = s.bucket.PutObject(objectKey, strings.NewReader(objectValue))
  547. c.Assert(err, IsNil)
  548. // Check
  549. body, err = s.bucket.GetObject(objectKey)
  550. c.Assert(err, IsNil)
  551. str, err = readBody(body)
  552. c.Assert(err, IsNil)
  553. c.Assert(str, Equals, objectValue)
  554. err = s.bucket.DeleteObject(objectKey)
  555. c.Assert(err, IsNil)
  556. }
  557. // TestPutObjectNegative
  558. func (s *OssBucketSuite) TestPutObjectNegative(c *C) {
  559. objectName := objectNamePrefix + "tpon"
  560. objectValue := "大江东去,浪淘尽,千古风流人物。 "
  561. // Put
  562. objectName = objectNamePrefix + "tpox"
  563. err := s.bucket.PutObject(objectName, strings.NewReader(objectValue),
  564. Meta("meta-my", "myprop"))
  565. c.Assert(err, IsNil)
  566. // Check meta
  567. body, err := s.bucket.GetObject(objectName)
  568. c.Assert(err, IsNil)
  569. str, err := readBody(body)
  570. c.Assert(err, IsNil)
  571. c.Assert(str, Equals, objectValue)
  572. meta, err := s.bucket.GetObjectDetailedMeta(objectName)
  573. c.Assert(err, IsNil)
  574. c.Assert(meta.Get("X-Oss-Meta-My"), Not(Equals), "myprop")
  575. c.Assert(meta.Get("X-Oss-Meta-My"), Equals, "")
  576. err = s.bucket.DeleteObject(objectName)
  577. c.Assert(err, IsNil)
  578. // Invalid option
  579. err = s.bucket.PutObject(objectName, strings.NewReader(objectValue),
  580. IfModifiedSince(pastDate))
  581. c.Assert(err, NotNil)
  582. err = s.bucket.PutObjectFromFile(objectName, "bucket.go", IfModifiedSince(pastDate))
  583. c.Assert(err, NotNil)
  584. err = s.bucket.PutObjectFromFile(objectName, "/tmp/xxx")
  585. c.Assert(err, NotNil)
  586. }
  587. // TestPutObjectFromFile
  588. func (s *OssBucketSuite) TestPutObjectFromFile(c *C) {
  589. objectName := objectNamePrefix + "tpoff"
  590. localFile := "../sample/BingWallpaper-2015-11-07.jpg"
  591. newFile := "newpic11.jpg"
  592. // Put
  593. err := s.bucket.PutObjectFromFile(objectName, localFile)
  594. c.Assert(err, IsNil)
  595. // Check
  596. err = s.bucket.GetObjectToFile(objectName, newFile)
  597. c.Assert(err, IsNil)
  598. eq, err := compareFiles(localFile, newFile)
  599. c.Assert(err, IsNil)
  600. c.Assert(eq, Equals, true)
  601. acl, err := s.bucket.GetObjectACL(objectName)
  602. c.Assert(err, IsNil)
  603. testLogger.Println("aclRes:", acl)
  604. c.Assert(acl.ACL, Equals, "default")
  605. err = s.bucket.DeleteObject(objectName)
  606. c.Assert(err, IsNil)
  607. // Put with properties
  608. options := []Option{
  609. Expires(futureDate),
  610. ObjectACL(ACLPublicRead),
  611. Meta("myprop", "mypropval"),
  612. }
  613. os.Remove(newFile)
  614. err = s.bucket.PutObjectFromFile(objectName, localFile, options...)
  615. c.Assert(err, IsNil)
  616. // Check
  617. err = s.bucket.GetObjectToFile(objectName, newFile)
  618. c.Assert(err, IsNil)
  619. eq, err = compareFiles(localFile, newFile)
  620. c.Assert(err, IsNil)
  621. c.Assert(eq, Equals, true)
  622. acl, err = s.bucket.GetObjectACL(objectName)
  623. c.Assert(err, IsNil)
  624. testLogger.Println("GetObjectACL:", acl)
  625. c.Assert(acl.ACL, Equals, string(ACLPublicRead))
  626. meta, err := s.bucket.GetObjectDetailedMeta(objectName)
  627. c.Assert(err, IsNil)
  628. testLogger.Println("GetObjectDetailedMeta:", meta)
  629. c.Assert(meta.Get("X-Oss-Meta-Myprop"), Equals, "mypropval")
  630. err = s.bucket.DeleteObject(objectName)
  631. c.Assert(err, IsNil)
  632. os.Remove(newFile)
  633. }
  634. // TestPutObjectFromFile
  635. func (s *OssBucketSuite) TestPutObjectFromFileType(c *C) {
  636. objectName := objectNamePrefix + "tpoffwm"
  637. localFile := "../sample/BingWallpaper-2015-11-07.jpg"
  638. newFile := "newpic11.jpg"
  639. // Put
  640. err := s.bucket.PutObjectFromFile(objectName, localFile)
  641. c.Assert(err, IsNil)
  642. // Check
  643. err = s.bucket.GetObjectToFile(objectName, newFile)
  644. c.Assert(err, IsNil)
  645. eq, err := compareFiles(localFile, newFile)
  646. c.Assert(err, IsNil)
  647. c.Assert(eq, Equals, true)
  648. meta, err := s.bucket.GetObjectDetailedMeta(objectName)
  649. c.Assert(err, IsNil)
  650. c.Assert(meta.Get("Content-Type"), Equals, "image/jpeg")
  651. err = s.bucket.DeleteObject(objectName)
  652. c.Assert(err, IsNil)
  653. os.Remove(newFile)
  654. }
  655. // TestGetObject
  656. func (s *OssBucketSuite) TestGetObject(c *C) {
  657. objectName := objectNamePrefix + "tgo"
  658. objectValue := "长忆观潮,满郭人争江上望。来疑沧海尽成空,万面鼓声中。弄潮儿向涛头立,手把红旗旗不湿。别来几向梦中看,梦觉尚心寒。"
  659. // Put
  660. err := s.bucket.PutObject(objectName, strings.NewReader(objectValue))
  661. c.Assert(err, IsNil)
  662. // Check
  663. body, err := s.bucket.GetObject(objectName)
  664. c.Assert(err, IsNil)
  665. data, err := ioutil.ReadAll(body)
  666. body.Close()
  667. str := string(data)
  668. c.Assert(str, Equals, objectValue)
  669. testLogger.Println("GetObjec:", str)
  670. // Range
  671. var subObjectValue = string(([]byte(objectValue))[15:36])
  672. body, err = s.bucket.GetObject(objectName, Range(15, 35))
  673. c.Assert(err, IsNil)
  674. data, err = ioutil.ReadAll(body)
  675. body.Close()
  676. str = string(data)
  677. c.Assert(str, Equals, subObjectValue)
  678. testLogger.Println("GetObject:", str, ",", subObjectValue)
  679. // If-Modified-Since
  680. _, err = s.bucket.GetObject(objectName, IfModifiedSince(futureDate))
  681. c.Assert(err, NotNil)
  682. // If-Unmodified-Since
  683. body, err = s.bucket.GetObject(objectName, IfUnmodifiedSince(futureDate))
  684. c.Assert(err, IsNil)
  685. data, err = ioutil.ReadAll(body)
  686. body.Close()
  687. c.Assert(string(data), Equals, objectValue)
  688. meta, err := s.bucket.GetObjectDetailedMeta(objectName)
  689. c.Assert(err, IsNil)
  690. // If-Match
  691. body, err = s.bucket.GetObject(objectName, IfMatch(meta.Get("Etag")))
  692. c.Assert(err, IsNil)
  693. data, err = ioutil.ReadAll(body)
  694. body.Close()
  695. c.Assert(string(data), Equals, objectValue)
  696. // If-None-Match
  697. _, err = s.bucket.GetObject(objectName, IfNoneMatch(meta.Get("Etag")))
  698. c.Assert(err, NotNil)
  699. // process
  700. err = s.bucket.PutObjectFromFile(objectName, "../sample/BingWallpaper-2015-11-07.jpg")
  701. c.Assert(err, IsNil)
  702. _, err = s.bucket.GetObject(objectName, Process("image/format,png"))
  703. c.Assert(err, IsNil)
  704. err = s.bucket.DeleteObject(objectName)
  705. c.Assert(err, IsNil)
  706. }
  707. // TestGetObjectNegative
  708. func (s *OssBucketSuite) TestGetObjectToWriterNegative(c *C) {
  709. objectName := objectNamePrefix + "tgotwn"
  710. objectValue := "长忆观潮,满郭人争江上望。"
  711. // Object not exist
  712. _, err := s.bucket.GetObject("NotExist")
  713. c.Assert(err, NotNil)
  714. // Constraint invalid
  715. err = s.bucket.PutObject(objectName, strings.NewReader(objectValue))
  716. c.Assert(err, IsNil)
  717. // Out of range
  718. _, err = s.bucket.GetObject(objectName, Range(15, 1000))
  719. c.Assert(err, IsNil)
  720. // Not exist
  721. err = s.bucket.GetObjectToFile(objectName, "/root/123abc9874")
  722. c.Assert(err, NotNil)
  723. // Invalid option
  724. _, err = s.bucket.GetObject(objectName, ACL(ACLPublicRead))
  725. c.Assert(err, IsNil)
  726. err = s.bucket.GetObjectToFile(objectName, "newpic15.jpg", ACL(ACLPublicRead))
  727. c.Assert(err, IsNil)
  728. err = s.bucket.DeleteObject(objectName)
  729. c.Assert(err, IsNil)
  730. }
  731. // TestGetObjectToFile
  732. func (s *OssBucketSuite) TestGetObjectToFile(c *C) {
  733. objectName := objectNamePrefix + "tgotf"
  734. objectValue := "江南好,风景旧曾谙;日出江花红胜火,春来江水绿如蓝。能不忆江南?江南忆,最忆是杭州;山寺月中寻桂子,郡亭枕上看潮头。何日更重游!"
  735. newFile := "newpic15.jpg"
  736. // Put
  737. var val = []byte(objectValue)
  738. err := s.bucket.PutObject(objectName, strings.NewReader(objectValue))
  739. c.Assert(err, IsNil)
  740. // Check
  741. err = s.bucket.GetObjectToFile(objectName, newFile)
  742. c.Assert(err, IsNil)
  743. eq, err := compareFileData(newFile, val)
  744. c.Assert(err, IsNil)
  745. c.Assert(eq, Equals, true)
  746. os.Remove(newFile)
  747. // Range
  748. err = s.bucket.GetObjectToFile(objectName, newFile, Range(15, 35))
  749. c.Assert(err, IsNil)
  750. eq, err = compareFileData(newFile, val[15:36])
  751. c.Assert(err, IsNil)
  752. c.Assert(eq, Equals, true)
  753. os.Remove(newFile)
  754. err = s.bucket.GetObjectToFile(objectName, newFile, NormalizedRange("15-35"))
  755. c.Assert(err, IsNil)
  756. eq, err = compareFileData(newFile, val[15:36])
  757. c.Assert(err, IsNil)
  758. c.Assert(eq, Equals, true)
  759. os.Remove(newFile)
  760. err = s.bucket.GetObjectToFile(objectName, newFile, NormalizedRange("15-"))
  761. c.Assert(err, IsNil)
  762. eq, err = compareFileData(newFile, val[15:])
  763. c.Assert(err, IsNil)
  764. c.Assert(eq, Equals, true)
  765. os.Remove(newFile)
  766. err = s.bucket.GetObjectToFile(objectName, newFile, NormalizedRange("-10"))
  767. c.Assert(err, IsNil)
  768. eq, err = compareFileData(newFile, val[(len(val)-10):len(val)])
  769. c.Assert(err, IsNil)
  770. c.Assert(eq, Equals, true)
  771. os.Remove(newFile)
  772. // If-Modified-Since
  773. err = s.bucket.GetObjectToFile(objectName, newFile, IfModifiedSince(futureDate))
  774. c.Assert(err, NotNil)
  775. // If-Unmodified-Since
  776. err = s.bucket.GetObjectToFile(objectName, newFile, IfUnmodifiedSince(futureDate))
  777. c.Assert(err, IsNil)
  778. eq, err = compareFileData(newFile, val)
  779. c.Assert(err, IsNil)
  780. c.Assert(eq, Equals, true)
  781. os.Remove(newFile)
  782. meta, err := s.bucket.GetObjectDetailedMeta(objectName)
  783. c.Assert(err, IsNil)
  784. testLogger.Println("GetObjectDetailedMeta:", meta)
  785. // If-Match
  786. err = s.bucket.GetObjectToFile(objectName, newFile, IfMatch(meta.Get("Etag")))
  787. c.Assert(err, IsNil)
  788. eq, err = compareFileData(newFile, val)
  789. c.Assert(err, IsNil)
  790. c.Assert(eq, Equals, true)
  791. // If-None-Match
  792. err = s.bucket.GetObjectToFile(objectName, newFile, IfNoneMatch(meta.Get("Etag")))
  793. c.Assert(err, NotNil)
  794. // Accept-Encoding:gzip
  795. err = s.bucket.PutObjectFromFile(objectName, "../sample/The Go Programming Language.html")
  796. c.Assert(err, IsNil)
  797. err = s.bucket.GetObjectToFile(objectName, newFile, AcceptEncoding("gzip"))
  798. c.Assert(err, IsNil)
  799. os.Remove(newFile)
  800. err = s.bucket.DeleteObject(objectName)
  801. c.Assert(err, IsNil)
  802. }
  803. // TestListObjects
  804. func (s *OssBucketSuite) TestListObjects(c *C) {
  805. objectName := objectNamePrefix + "tlo"
  806. // List empty bucket
  807. lor, err := s.bucket.ListObjects()
  808. c.Assert(err, IsNil)
  809. left := len(lor.Objects)
  810. // Put three objects
  811. err = s.bucket.PutObject(objectName+"1", strings.NewReader(""))
  812. c.Assert(err, IsNil)
  813. err = s.bucket.PutObject(objectName+"2", strings.NewReader(""))
  814. c.Assert(err, IsNil)
  815. err = s.bucket.PutObject(objectName+"3", strings.NewReader(""))
  816. c.Assert(err, IsNil)
  817. // List
  818. lor, err = s.bucket.ListObjects()
  819. c.Assert(err, IsNil)
  820. c.Assert(len(lor.Objects), Equals, left+3)
  821. // List with prefix
  822. lor, err = s.bucket.ListObjects(Prefix(objectName + "2"))
  823. c.Assert(err, IsNil)
  824. c.Assert(len(lor.Objects), Equals, 1)
  825. lor, err = s.bucket.ListObjects(Prefix(objectName + "22"))
  826. c.Assert(err, IsNil)
  827. c.Assert(len(lor.Objects), Equals, 0)
  828. // List with max keys
  829. lor, err = s.bucket.ListObjects(Prefix(objectName), MaxKeys(2))
  830. c.Assert(err, IsNil)
  831. c.Assert(len(lor.Objects), Equals, 2)
  832. // List with marker
  833. lor, err = s.bucket.ListObjects(Marker(objectName+"1"), MaxKeys(1))
  834. c.Assert(err, IsNil)
  835. c.Assert(len(lor.Objects), Equals, 1)
  836. err = s.bucket.DeleteObject(objectName + "1")
  837. c.Assert(err, IsNil)
  838. err = s.bucket.DeleteObject(objectName + "2")
  839. c.Assert(err, IsNil)
  840. err = s.bucket.DeleteObject(objectName + "3")
  841. c.Assert(err, IsNil)
  842. }
  843. // TestListObjects
  844. func (s *OssBucketSuite) TestListObjectsEncodingType(c *C) {
  845. objectName := objectNamePrefix + "床前明月光,疑是地上霜。举头望明月,低头思故乡。" + "tloet"
  846. for i := 0; i < 10; i++ {
  847. err := s.bucket.PutObject(objectName+strconv.Itoa(i), strings.NewReader(""))
  848. c.Assert(err, IsNil)
  849. }
  850. lor, err := s.bucket.ListObjects(Prefix(objectNamePrefix + "床前明月光,"))
  851. c.Assert(err, IsNil)
  852. c.Assert(len(lor.Objects), Equals, 10)
  853. lor, err = s.bucket.ListObjects(Prefix(objectNamePrefix + "床前明月光,"))
  854. c.Assert(err, IsNil)
  855. c.Assert(len(lor.Objects), Equals, 10)
  856. lor, err = s.bucket.ListObjects(Marker(objectNamePrefix + "床前明月光,疑是地上霜。举头望明月,低头思故乡。"))
  857. c.Assert(err, IsNil)
  858. c.Assert(len(lor.Objects), Equals, 10)
  859. lor, err = s.bucket.ListObjects(Prefix(objectNamePrefix + "床前明月光"))
  860. c.Assert(err, IsNil)
  861. for i, obj := range lor.Objects {
  862. c.Assert(obj.Key, Equals, objectNamePrefix+"床前明月光,疑是地上霜。举头望明月,低头思故乡。tloet"+strconv.Itoa(i))
  863. }
  864. for i := 0; i < 10; i++ {
  865. err = s.bucket.DeleteObject(objectName + strconv.Itoa(i))
  866. c.Assert(err, IsNil)
  867. }
  868. // Special characters
  869. objectName = "go go ` ~ ! @ # $ % ^ & * () - _ + =[] {} \\ | < > , . ? / 0"
  870. err = s.bucket.PutObject(objectName, strings.NewReader("明月几时有,把酒问青天"))
  871. c.Assert(err, IsNil)
  872. lor, err = s.bucket.ListObjects(Prefix(objectName))
  873. c.Assert(err, IsNil)
  874. c.Assert(len(lor.Objects), Equals, 1)
  875. err = s.bucket.DeleteObject(objectName)
  876. c.Assert(err, IsNil)
  877. objectName = "go/中国 日本 +-#&=*"
  878. err = s.bucket.PutObject(objectName, strings.NewReader("明月几时有,把酒问青天"))
  879. c.Assert(err, IsNil)
  880. lor, err = s.bucket.ListObjects(Prefix(objectName))
  881. c.Assert(err, IsNil)
  882. c.Assert(len(lor.Objects), Equals, 1)
  883. err = s.bucket.DeleteObject(objectName)
  884. c.Assert(err, IsNil)
  885. }
  886. // TestIsBucketExist
  887. func (s *OssBucketSuite) TestIsObjectExist(c *C) {
  888. objectName := objectNamePrefix + "tibe"
  889. // Put three objects
  890. err := s.bucket.PutObject(objectName+"1", strings.NewReader(""))
  891. c.Assert(err, IsNil)
  892. err = s.bucket.PutObject(objectName+"11", strings.NewReader(""))
  893. c.Assert(err, IsNil)
  894. err = s.bucket.PutObject(objectName+"111", strings.NewReader(""))
  895. c.Assert(err, IsNil)
  896. // Exist
  897. exist, err := s.bucket.IsObjectExist(objectName + "11")
  898. c.Assert(err, IsNil)
  899. c.Assert(exist, Equals, true)
  900. exist, err = s.bucket.IsObjectExist(objectName + "1")
  901. c.Assert(err, IsNil)
  902. c.Assert(exist, Equals, true)
  903. exist, err = s.bucket.IsObjectExist(objectName + "111")
  904. c.Assert(err, IsNil)
  905. c.Assert(exist, Equals, true)
  906. // Not exist
  907. exist, err = s.bucket.IsObjectExist(objectName + "1111")
  908. c.Assert(err, IsNil)
  909. c.Assert(exist, Equals, false)
  910. exist, err = s.bucket.IsObjectExist(objectName)
  911. c.Assert(err, IsNil)
  912. c.Assert(exist, Equals, false)
  913. err = s.bucket.DeleteObject(objectName + "1")
  914. c.Assert(err, IsNil)
  915. err = s.bucket.DeleteObject(objectName + "11")
  916. c.Assert(err, IsNil)
  917. err = s.bucket.DeleteObject(objectName + "111")
  918. c.Assert(err, IsNil)
  919. }
  920. // TestDeleteObject
  921. func (s *OssBucketSuite) TestDeleteObject(c *C) {
  922. objectName := objectNamePrefix + "tdo"
  923. err := s.bucket.PutObject(objectName, strings.NewReader(""))
  924. c.Assert(err, IsNil)
  925. lor, err := s.bucket.ListObjects(Prefix(objectName))
  926. c.Assert(err, IsNil)
  927. c.Assert(len(lor.Objects), Equals, 1)
  928. // Delete
  929. err = s.bucket.DeleteObject(objectName)
  930. c.Assert(err, IsNil)
  931. // Duplicate delete
  932. err = s.bucket.DeleteObject(objectName)
  933. c.Assert(err, IsNil)
  934. lor, err = s.bucket.ListObjects(Prefix(objectName))
  935. c.Assert(err, IsNil)
  936. c.Assert(len(lor.Objects), Equals, 0)
  937. }
  938. // TestDeleteObjects
  939. func (s *OssBucketSuite) TestDeleteObjects(c *C) {
  940. objectName := objectNamePrefix + "tdos"
  941. // Delete objects
  942. err := s.bucket.PutObject(objectName, strings.NewReader(""))
  943. c.Assert(err, IsNil)
  944. res, err := s.bucket.DeleteObjects([]string{objectName})
  945. c.Assert(err, IsNil)
  946. c.Assert(len(res.DeletedObjects), Equals, 1)
  947. lor, err := s.bucket.ListObjects(Prefix(objectName))
  948. c.Assert(err, IsNil)
  949. c.Assert(len(lor.Objects), Equals, 0)
  950. // Delete objects
  951. err = s.bucket.PutObject(objectName+"1", strings.NewReader(""))
  952. c.Assert(err, IsNil)
  953. err = s.bucket.PutObject(objectName+"2", strings.NewReader(""))
  954. c.Assert(err, IsNil)
  955. res, err = s.bucket.DeleteObjects([]string{objectName + "1", objectName + "2"})
  956. c.Assert(err, IsNil)
  957. c.Assert(len(res.DeletedObjects), Equals, 2)
  958. lor, err = s.bucket.ListObjects(Prefix(objectName))
  959. c.Assert(err, IsNil)
  960. c.Assert(len(lor.Objects), Equals, 0)
  961. // Delete 0
  962. _, err = s.bucket.DeleteObjects([]string{})
  963. c.Assert(err, NotNil)
  964. // DeleteObjectsQuiet
  965. err = s.bucket.PutObject(objectName+"1", strings.NewReader(""))
  966. c.Assert(err, IsNil)
  967. err = s.bucket.PutObject(objectName+"2", strings.NewReader(""))
  968. c.Assert(err, IsNil)
  969. res, err = s.bucket.DeleteObjects([]string{objectName + "1", objectName + "2"},
  970. DeleteObjectsQuiet(false))
  971. c.Assert(err, IsNil)
  972. c.Assert(len(res.DeletedObjects), Equals, 2)
  973. lor, err = s.bucket.ListObjects(Prefix(objectName))
  974. c.Assert(err, IsNil)
  975. c.Assert(len(lor.Objects), Equals, 0)
  976. // DeleteObjectsQuiet
  977. err = s.bucket.PutObject(objectName+"1", strings.NewReader(""))
  978. c.Assert(err, IsNil)
  979. err = s.bucket.PutObject(objectName+"2", strings.NewReader(""))
  980. c.Assert(err, IsNil)
  981. res, err = s.bucket.DeleteObjects([]string{objectName + "1", objectName + "2"},
  982. DeleteObjectsQuiet(true))
  983. c.Assert(err, IsNil)
  984. c.Assert(len(res.DeletedObjects), Equals, 0)
  985. lor, err = s.bucket.ListObjects(Prefix(objectName))
  986. c.Assert(err, IsNil)
  987. c.Assert(len(lor.Objects), Equals, 0)
  988. // EncodingType
  989. err = s.bucket.PutObject("中国人", strings.NewReader(""))
  990. c.Assert(err, IsNil)
  991. res, err = s.bucket.DeleteObjects([]string{"中国人"})
  992. c.Assert(err, IsNil)
  993. c.Assert(len(res.DeletedObjects), Equals, 1)
  994. c.Assert(res.DeletedObjects[0], Equals, "中国人")
  995. // EncodingType
  996. err = s.bucket.PutObject("中国人", strings.NewReader(""))
  997. c.Assert(err, IsNil)
  998. res, err = s.bucket.DeleteObjects([]string{"中国人"}, DeleteObjectsQuiet(false))
  999. c.Assert(err, IsNil)
  1000. c.Assert(len(res.DeletedObjects), Equals, 1)
  1001. c.Assert(res.DeletedObjects[0], Equals, "中国人")
  1002. // EncodingType
  1003. err = s.bucket.PutObject("中国人", strings.NewReader(""))
  1004. c.Assert(err, IsNil)
  1005. res, err = s.bucket.DeleteObjects([]string{"中国人"}, DeleteObjectsQuiet(true))
  1006. c.Assert(err, IsNil)
  1007. c.Assert(len(res.DeletedObjects), Equals, 0)
  1008. // Special characters
  1009. key := "A ' < > \" & ~ ` ! @ # $ % ^ & * ( ) [] {} - _ + = / | \\ ? . , : ; A"
  1010. err = s.bucket.PutObject(key, strings.NewReader("value"))
  1011. c.Assert(err, IsNil)
  1012. _, err = s.bucket.DeleteObjects([]string{key})
  1013. c.Assert(err, IsNil)
  1014. ress, err := s.bucket.ListObjects(Prefix(key))
  1015. c.Assert(err, IsNil)
  1016. c.Assert(len(ress.Objects), Equals, 0)
  1017. // Not exist
  1018. _, err = s.bucket.DeleteObjects([]string{"NotExistObject"})
  1019. c.Assert(err, IsNil)
  1020. }
  1021. // TestSetObjectMeta
  1022. func (s *OssBucketSuite) TestSetObjectMeta(c *C) {
  1023. objectName := objectNamePrefix + "tsom"
  1024. err := s.bucket.PutObject(objectName, strings.NewReader(""))
  1025. c.Assert(err, IsNil)
  1026. err = s.bucket.SetObjectMeta(objectName,
  1027. Expires(futureDate),
  1028. Meta("myprop", "mypropval"))
  1029. c.Assert(err, IsNil)
  1030. meta, err := s.bucket.GetObjectDetailedMeta(objectName)
  1031. c.Assert(err, IsNil)
  1032. testLogger.Println("Meta:", meta)
  1033. c.Assert(meta.Get("Expires"), Equals, futureDate.Format(http.TimeFormat))
  1034. c.Assert(meta.Get("X-Oss-Meta-Myprop"), Equals, "mypropval")
  1035. acl, err := s.bucket.GetObjectACL(objectName)
  1036. c.Assert(err, IsNil)
  1037. c.Assert(acl.ACL, Equals, "default")
  1038. // Invalid option
  1039. err = s.bucket.SetObjectMeta(objectName, AcceptEncoding("url"))
  1040. c.Assert(err, IsNil)
  1041. // Invalid option value
  1042. err = s.bucket.SetObjectMeta(objectName, ServerSideEncryption("invalid"))
  1043. c.Assert(err, NotNil)
  1044. err = s.bucket.DeleteObject(objectName)
  1045. c.Assert(err, IsNil)
  1046. // Not exist
  1047. err = s.bucket.SetObjectMeta(objectName, Expires(futureDate))
  1048. c.Assert(err, NotNil)
  1049. }
  1050. // TestGetObjectMeta
  1051. func (s *OssBucketSuite) TestGetObjectMeta(c *C) {
  1052. objectName := objectNamePrefix + "tgom"
  1053. // Put
  1054. err := s.bucket.PutObject(objectName, strings.NewReader(""))
  1055. c.Assert(err, IsNil)
  1056. meta, err := s.bucket.GetObjectMeta(objectName)
  1057. c.Assert(err, IsNil)
  1058. c.Assert(len(meta) > 0, Equals, true)
  1059. err = s.bucket.DeleteObject(objectName)
  1060. c.Assert(err, IsNil)
  1061. _, err = s.bucket.GetObjectMeta("NotExistObject")
  1062. c.Assert(err, NotNil)
  1063. }
  1064. // TestGetObjectDetailedMeta
  1065. func (s *OssBucketSuite) TestGetObjectDetailedMeta(c *C) {
  1066. objectName := objectNamePrefix + "tgodm"
  1067. // Put
  1068. err := s.bucket.PutObject(objectName, strings.NewReader(""),
  1069. Expires(futureDate), Meta("myprop", "mypropval"))
  1070. c.Assert(err, IsNil)
  1071. // Check
  1072. meta, err := s.bucket.GetObjectDetailedMeta(objectName)
  1073. c.Assert(err, IsNil)
  1074. testLogger.Println("GetObjectDetailedMeta:", meta)
  1075. c.Assert(meta.Get("Expires"), Equals, futureDate.Format(http.TimeFormat))
  1076. c.Assert(meta.Get("X-Oss-Meta-Myprop"), Equals, "mypropval")
  1077. c.Assert(meta.Get("Content-Length"), Equals, "0")
  1078. c.Assert(len(meta.Get("Date")) > 0, Equals, true)
  1079. c.Assert(len(meta.Get("X-Oss-Request-Id")) > 0, Equals, true)
  1080. c.Assert(len(meta.Get("Last-Modified")) > 0, Equals, true)
  1081. // IfModifiedSince/IfModifiedSince
  1082. _, err = s.bucket.GetObjectDetailedMeta(objectName, IfModifiedSince(futureDate))
  1083. c.Assert(err, NotNil)
  1084. meta, err = s.bucket.GetObjectDetailedMeta(objectName, IfUnmodifiedSince(futureDate))
  1085. c.Assert(err, IsNil)
  1086. c.Assert(meta.Get("Expires"), Equals, futureDate.Format(http.TimeFormat))
  1087. c.Assert(meta.Get("X-Oss-Meta-Myprop"), Equals, "mypropval")
  1088. // IfMatch/IfNoneMatch
  1089. _, err = s.bucket.GetObjectDetailedMeta(objectName, IfNoneMatch(meta.Get("Etag")))
  1090. c.Assert(err, NotNil)
  1091. meta, err = s.bucket.GetObjectDetailedMeta(objectName, IfMatch(meta.Get("Etag")))
  1092. c.Assert(err, IsNil)
  1093. c.Assert(meta.Get("Expires"), Equals, futureDate.Format(http.TimeFormat))
  1094. c.Assert(meta.Get("X-Oss-Meta-Myprop"), Equals, "mypropval")
  1095. err = s.bucket.DeleteObject(objectName)
  1096. c.Assert(err, IsNil)
  1097. _, err = s.bucket.GetObjectDetailedMeta("NotExistObject")
  1098. c.Assert(err, NotNil)
  1099. }
  1100. // TestSetAndGetObjectAcl
  1101. func (s *OssBucketSuite) TestSetAndGetObjectAcl(c *C) {
  1102. objectName := objectNamePrefix + "tsgba"
  1103. err := s.bucket.PutObject(objectName, strings.NewReader(""))
  1104. c.Assert(err, IsNil)
  1105. // Default
  1106. acl, err := s.bucket.GetObjectACL(objectName)
  1107. c.Assert(err, IsNil)
  1108. c.Assert(acl.ACL, Equals, "default")
  1109. // Set ACL_PUBLIC_RW
  1110. err = s.bucket.SetObjectACL(objectName, ACLPublicReadWrite)
  1111. c.Assert(err, IsNil)
  1112. acl, err = s.bucket.GetObjectACL(objectName)
  1113. c.Assert(err, IsNil)
  1114. c.Assert(acl.ACL, Equals, string(ACLPublicReadWrite))
  1115. // Set ACL_PRIVATE
  1116. err = s.bucket.SetObjectACL(objectName, ACLPrivate)
  1117. c.Assert(err, IsNil)
  1118. acl, err = s.bucket.GetObjectACL(objectName)
  1119. c.Assert(err, IsNil)
  1120. c.Assert(acl.ACL, Equals, string(ACLPrivate))
  1121. // Set ACL_PUBLIC_R
  1122. err = s.bucket.SetObjectACL(objectName, ACLPublicRead)
  1123. c.Assert(err, IsNil)
  1124. acl, err = s.bucket.GetObjectACL(objectName)
  1125. c.Assert(err, IsNil)
  1126. c.Assert(acl.ACL, Equals, string(ACLPublicRead))
  1127. err = s.bucket.DeleteObject(objectName)
  1128. c.Assert(err, IsNil)
  1129. }
  1130. // TestSetAndGetObjectAclNegative
  1131. func (s *OssBucketSuite) TestSetAndGetObjectAclNegative(c *C) {
  1132. objectName := objectNamePrefix + "tsgban"
  1133. // Object not exist
  1134. err := s.bucket.SetObjectACL(objectName, ACLPublicRead)
  1135. c.Assert(err, NotNil)
  1136. }
  1137. // TestCopyObject
  1138. func (s *OssBucketSuite) TestCopyObject(c *C) {
  1139. objectName := objectNamePrefix + "tco"
  1140. objectValue := "男儿何不带吴钩,收取关山五十州。请君暂上凌烟阁,若个书生万户侯?"
  1141. err := s.bucket.PutObject(objectName, strings.NewReader(objectValue),
  1142. ACL(ACLPublicRead), Meta("my", "myprop"))
  1143. c.Assert(err, IsNil)
  1144. // Copy
  1145. var objectNameDest = objectName + "dest"
  1146. _, err = s.bucket.CopyObject(objectName, objectNameDest)
  1147. c.Assert(err, IsNil)
  1148. // Check
  1149. lor, err := s.bucket.ListObjects(Prefix(objectName))
  1150. c.Assert(err, IsNil)
  1151. testLogger.Println("objects:", lor.Objects)
  1152. c.Assert(len(lor.Objects), Equals, 2)
  1153. body, err := s.bucket.GetObject(objectName)
  1154. c.Assert(err, IsNil)
  1155. str, err := readBody(body)
  1156. c.Assert(err, IsNil)
  1157. c.Assert(str, Equals, objectValue)
  1158. err = s.bucket.DeleteObject(objectNameDest)
  1159. c.Assert(err, IsNil)
  1160. // Copy with constraints x-oss-copy-source-if-modified-since
  1161. _, err = s.bucket.CopyObject(objectName, objectNameDest, CopySourceIfModifiedSince(futureDate))
  1162. c.Assert(err, NotNil)
  1163. testLogger.Println("CopyObject:", err)
  1164. // Copy with constraints x-oss-copy-source-if-unmodified-since
  1165. _, err = s.bucket.CopyObject(objectName, objectNameDest, CopySourceIfUnmodifiedSince(futureDate))
  1166. c.Assert(err, IsNil)
  1167. // Check
  1168. lor, err = s.bucket.ListObjects(Prefix(objectName))
  1169. c.Assert(err, IsNil)
  1170. testLogger.Println("objects:", lor.Objects)
  1171. c.Assert(len(lor.Objects), Equals, 2)
  1172. body, err = s.bucket.GetObject(objectName)
  1173. c.Assert(err, IsNil)
  1174. str, err = readBody(body)
  1175. c.Assert(err, IsNil)
  1176. c.Assert(str, Equals, objectValue)
  1177. err = s.bucket.DeleteObject(objectNameDest)
  1178. c.Assert(err, IsNil)
  1179. // Copy with constraints x-oss-copy-source-if-match
  1180. meta, err := s.bucket.GetObjectDetailedMeta(objectName)
  1181. c.Assert(err, IsNil)
  1182. testLogger.Println("GetObjectDetailedMeta:", meta)
  1183. _, err = s.bucket.CopyObject(objectName, objectNameDest, CopySourceIfMatch(meta.Get("Etag")))
  1184. c.Assert(err, IsNil)
  1185. // Check
  1186. body, err = s.bucket.GetObject(objectName)
  1187. c.Assert(err, IsNil)
  1188. str, err = readBody(body)
  1189. c.Assert(err, IsNil)
  1190. c.Assert(str, Equals, objectValue)
  1191. err = s.bucket.DeleteObject(objectNameDest)
  1192. c.Assert(err, IsNil)
  1193. // Copy with constraints x-oss-copy-source-if-none-match
  1194. _, err = s.bucket.CopyObject(objectName, objectNameDest, CopySourceIfNoneMatch(meta.Get("Etag")))
  1195. c.Assert(err, NotNil)
  1196. // Copy with constraints x-oss-metadata-directive
  1197. _, err = s.bucket.CopyObject(objectName, objectNameDest, Meta("my", "mydestprop"),
  1198. MetadataDirective(MetaCopy))
  1199. c.Assert(err, IsNil)
  1200. // Check
  1201. body, err = s.bucket.GetObject(objectName)
  1202. c.Assert(err, IsNil)
  1203. str, err = readBody(body)
  1204. c.Assert(err, IsNil)
  1205. c.Assert(str, Equals, objectValue)
  1206. destMeta, err := s.bucket.GetObjectDetailedMeta(objectNameDest)
  1207. c.Assert(err, IsNil)
  1208. c.Assert(meta.Get("X-Oss-Meta-My"), Equals, "myprop")
  1209. acl, err := s.bucket.GetObjectACL(objectNameDest)
  1210. c.Assert(err, IsNil)
  1211. c.Assert(acl.ACL, Equals, "default")
  1212. err = s.bucket.DeleteObject(objectNameDest)
  1213. c.Assert(err, IsNil)
  1214. // Copy with constraints x-oss-metadata-directive and self defined dest object meta
  1215. options := []Option{
  1216. ObjectACL(ACLPublicReadWrite),
  1217. Meta("my", "mydestprop"),
  1218. MetadataDirective(MetaReplace),
  1219. }
  1220. _, err = s.bucket.CopyObject(objectName, objectNameDest, options...)
  1221. c.Assert(err, IsNil)
  1222. // Check
  1223. body, err = s.bucket.GetObject(objectName)
  1224. c.Assert(err, IsNil)
  1225. str, err = readBody(body)
  1226. c.Assert(err, IsNil)
  1227. c.Assert(str, Equals, objectValue)
  1228. destMeta, err = s.bucket.GetObjectDetailedMeta(objectNameDest)
  1229. c.Assert(err, IsNil)
  1230. c.Assert(destMeta.Get("X-Oss-Meta-My"), Equals, "mydestprop")
  1231. acl, err = s.bucket.GetObjectACL(objectNameDest)
  1232. c.Assert(err, IsNil)
  1233. c.Assert(acl.ACL, Equals, string(ACLPublicReadWrite))
  1234. err = s.bucket.DeleteObject(objectNameDest)
  1235. c.Assert(err, IsNil)
  1236. err = s.bucket.DeleteObject(objectName)
  1237. c.Assert(err, IsNil)
  1238. }
  1239. // TestCopyObjectToOrFrom
  1240. func (s *OssBucketSuite) TestCopyObjectToOrFrom(c *C) {
  1241. objectName := objectNamePrefix + "tcotof" + randLowStr(5)
  1242. objectValue := "男儿何不带吴钩,收取关山五十州。请君暂上凌烟阁,若个书生万户侯?"
  1243. destBucketName := bucketName + "-dest" + randLowStr(5)
  1244. objectNameDest := objectName + "dest"
  1245. err := s.client.CreateBucket(destBucketName)
  1246. c.Assert(err, IsNil)
  1247. destBucket, err := s.client.Bucket(destBucketName)
  1248. c.Assert(err, IsNil)
  1249. err = s.bucket.PutObject(objectName, strings.NewReader(objectValue))
  1250. c.Assert(err, IsNil)
  1251. // Copy from
  1252. _, err = destBucket.CopyObjectFrom(bucketName, objectName, objectNameDest)
  1253. c.Assert(err, IsNil)
  1254. // Check
  1255. body, err := destBucket.GetObject(objectNameDest)
  1256. c.Assert(err, IsNil)
  1257. str, err := readBody(body)
  1258. c.Assert(err, IsNil)
  1259. c.Assert(str, Equals, objectValue)
  1260. err = s.bucket.DeleteObject(objectName)
  1261. c.Assert(err, IsNil)
  1262. // Copy to
  1263. _, err = destBucket.CopyObjectTo(bucketName, objectName, objectNameDest)
  1264. c.Assert(err, IsNil)
  1265. // Check
  1266. body, err = s.bucket.GetObject(objectName)
  1267. c.Assert(err, IsNil)
  1268. str, err = readBody(body)
  1269. c.Assert(err, IsNil)
  1270. c.Assert(str, Equals, objectValue)
  1271. // Clean
  1272. err = destBucket.DeleteObject(objectNameDest)
  1273. c.Assert(err, IsNil)
  1274. err = s.bucket.DeleteObject(objectName)
  1275. c.Assert(err, IsNil)
  1276. err = s.client.DeleteBucket(destBucketName)
  1277. c.Assert(err, IsNil)
  1278. }
  1279. // TestCopyObjectToOrFromNegative
  1280. func (s *OssBucketSuite) TestCopyObjectToOrFromNegative(c *C) {
  1281. objectName := objectNamePrefix + "tcotofn"
  1282. destBucket := bucketName + "-destn"
  1283. objectNameDest := objectName + "destn"
  1284. // Object not exist
  1285. _, err := s.bucket.CopyObjectTo(bucketName, objectName, objectNameDest)
  1286. c.Assert(err, NotNil)
  1287. // Bucket not exist
  1288. _, err = s.bucket.CopyObjectFrom(destBucket, objectNameDest, objectName)
  1289. c.Assert(err, NotNil)
  1290. }
  1291. // TestAppendObject
  1292. func (s *OssBucketSuite) TestAppendObject(c *C) {
  1293. objectName := objectNamePrefix + "tao"
  1294. objectValue := "昨夜雨疏风骤,浓睡不消残酒。试问卷帘人,却道海棠依旧。知否?知否?应是绿肥红瘦。"
  1295. var val = []byte(objectValue)
  1296. var localFile = "testx.txt"
  1297. var nextPos int64
  1298. var midPos = 1 + rand.Intn(len(val)-1)
  1299. var err = createFileAndWrite(localFile+"1", val[0:midPos])
  1300. c.Assert(err, IsNil)
  1301. err = createFileAndWrite(localFile+"2", val[midPos:])
  1302. c.Assert(err, IsNil)
  1303. // String append
  1304. nextPos, err = s.bucket.AppendObject(objectName, strings.NewReader("昨夜雨疏风骤,浓睡不消残酒。试问卷帘人,"), nextPos)
  1305. c.Assert(err, IsNil)
  1306. nextPos, err = s.bucket.AppendObject(objectName, strings.NewReader("却道海棠依旧。知否?知否?应是绿肥红瘦。"), nextPos)
  1307. c.Assert(err, IsNil)
  1308. body, err := s.bucket.GetObject(objectName)
  1309. c.Assert(err, IsNil)
  1310. str, err := readBody(body)
  1311. c.Assert(err, IsNil)
  1312. c.Assert(str, Equals, objectValue)
  1313. err = s.bucket.DeleteObject(objectName)
  1314. c.Assert(err, IsNil)
  1315. // Byte append
  1316. nextPos = 0
  1317. nextPos, err = s.bucket.AppendObject(objectName, bytes.NewReader(val[0:midPos]), nextPos)
  1318. c.Assert(err, IsNil)
  1319. nextPos, err = s.bucket.AppendObject(objectName, bytes.NewReader(val[midPos:]), nextPos)
  1320. c.Assert(err, IsNil)
  1321. body, err = s.bucket.GetObject(objectName)
  1322. c.Assert(err, IsNil)
  1323. str, err = readBody(body)
  1324. c.Assert(err, IsNil)
  1325. c.Assert(str, Equals, objectValue)
  1326. err = s.bucket.DeleteObject(objectName)
  1327. c.Assert(err, IsNil)
  1328. // File append
  1329. options := []Option{
  1330. ObjectACL(ACLPublicReadWrite),
  1331. Meta("my", "myprop"),
  1332. }
  1333. fd, err := os.Open(localFile + "1")
  1334. c.Assert(err, IsNil)
  1335. defer fd.Close()
  1336. nextPos = 0
  1337. nextPos, err = s.bucket.AppendObject(objectName, fd, nextPos, options...)
  1338. c.Assert(err, IsNil)
  1339. meta, err := s.bucket.GetObjectDetailedMeta(objectName)
  1340. c.Assert(err, IsNil)
  1341. testLogger.Println("GetObjectDetailedMeta:", meta, ",", nextPos)
  1342. c.Assert(meta.Get("X-Oss-Object-Type"), Equals, "Appendable")
  1343. c.Assert(meta.Get("X-Oss-Meta-My"), Equals, "myprop")
  1344. c.Assert(meta.Get("x-oss-Meta-Mine"), Equals, "")
  1345. c.Assert(meta.Get("X-Oss-Next-Append-Position"), Equals, strconv.FormatInt(nextPos, 10))
  1346. acl, err := s.bucket.GetObjectACL(objectName)
  1347. c.Assert(err, IsNil)
  1348. testLogger.Println("GetObjectACL:", acl)
  1349. c.Assert(acl.ACL, Equals, string(ACLPublicReadWrite))
  1350. // Second append
  1351. options = []Option{
  1352. ObjectACL(ACLPublicRead),
  1353. Meta("my", "myproptwo"),
  1354. Meta("mine", "mypropmine"),
  1355. }
  1356. fd, err = os.Open(localFile + "2")
  1357. c.Assert(err, IsNil)
  1358. defer fd.Close()
  1359. nextPos, err = s.bucket.AppendObject(objectName, fd, nextPos, options...)
  1360. c.Assert(err, IsNil)
  1361. body, err = s.bucket.GetObject(objectName)
  1362. c.Assert(err, IsNil)
  1363. str, err = readBody(body)
  1364. c.Assert(err, IsNil)
  1365. c.Assert(str, Equals, objectValue)
  1366. meta, err = s.bucket.GetObjectDetailedMeta(objectName)
  1367. c.Assert(err, IsNil)
  1368. testLogger.Println("GetObjectDetailedMeta xxx:", meta)
  1369. c.Assert(meta.Get("X-Oss-Object-Type"), Equals, "Appendable")
  1370. c.Assert(meta.Get("X-Oss-Meta-My"), Equals, "myprop")
  1371. c.Assert(meta.Get("x-Oss-Meta-Mine"), Equals, "")
  1372. c.Assert(meta.Get("X-Oss-Next-Append-Position"), Equals, strconv.FormatInt(nextPos, 10))
  1373. acl, err = s.bucket.GetObjectACL(objectName)
  1374. c.Assert(err, IsNil)
  1375. c.Assert(acl.ACL, Equals, string(ACLPublicRead))
  1376. err = s.bucket.DeleteObject(objectName)
  1377. c.Assert(err, IsNil)
  1378. }
  1379. // TestAppendObjectNegative
  1380. func (s *OssBucketSuite) TestAppendObjectNegative(c *C) {
  1381. objectName := objectNamePrefix + "taon"
  1382. nextPos := int64(0)
  1383. nextPos, err := s.bucket.AppendObject(objectName, strings.NewReader("ObjectValue"), nextPos)
  1384. c.Assert(err, IsNil)
  1385. nextPos, err = s.bucket.AppendObject(objectName, strings.NewReader("ObjectValue"), 0)
  1386. c.Assert(err, NotNil)
  1387. err = s.bucket.DeleteObject(objectName)
  1388. c.Assert(err, IsNil)
  1389. }
  1390. // TestContentType
  1391. func (s *OssBucketSuite) TestAddContentType(c *C) {
  1392. opts := addContentType(nil, "abc.txt")
  1393. typ, err := findOption(opts, HTTPHeaderContentType, "")
  1394. c.Assert(err, IsNil)
  1395. c.Assert(typ, Equals, "text/plain; charset=utf-8")
  1396. opts = addContentType(nil)
  1397. typ, err = findOption(opts, HTTPHeaderContentType, "")
  1398. c.Assert(err, IsNil)
  1399. c.Assert(len(opts), Equals, 1)
  1400. c.Assert(typ, Equals, "application/octet-stream")
  1401. opts = addContentType(nil, "abc.txt", "abc.pdf")
  1402. typ, err = findOption(opts, HTTPHeaderContentType, "")
  1403. c.Assert(err, IsNil)
  1404. c.Assert(typ, Equals, "text/plain; charset=utf-8")
  1405. opts = addContentType(nil, "abc", "abc.txt", "abc.pdf")
  1406. typ, err = findOption(opts, HTTPHeaderContentType, "")
  1407. c.Assert(err, IsNil)
  1408. c.Assert(typ, Equals, "text/plain; charset=utf-8")
  1409. opts = addContentType(nil, "abc", "abc", "edf")
  1410. typ, err = findOption(opts, HTTPHeaderContentType, "")
  1411. c.Assert(err, IsNil)
  1412. c.Assert(typ, Equals, "application/octet-stream")
  1413. opts = addContentType([]Option{Meta("meta", "my")}, "abc", "abc.txt", "abc.pdf")
  1414. typ, err = findOption(opts, HTTPHeaderContentType, "")
  1415. c.Assert(err, IsNil)
  1416. c.Assert(len(opts), Equals, 2)
  1417. c.Assert(typ, Equals, "text/plain; charset=utf-8")
  1418. }
  1419. func (s *OssBucketSuite) TestGetConfig(c *C) {
  1420. client, err := New(endpoint, accessID, accessKey, UseCname(true),
  1421. Timeout(11, 12), SecurityToken("token"), EnableMD5(false))
  1422. c.Assert(err, IsNil)
  1423. bucket, err := client.Bucket(bucketName)
  1424. c.Assert(err, IsNil)
  1425. c.Assert(bucket.getConfig().HTTPTimeout.ConnectTimeout, Equals, time.Second*11)
  1426. c.Assert(bucket.getConfig().HTTPTimeout.ReadWriteTimeout, Equals, time.Second*12)
  1427. c.Assert(bucket.getConfig().HTTPTimeout.HeaderTimeout, Equals, time.Second*12)
  1428. c.Assert(bucket.getConfig().HTTPTimeout.IdleConnTimeout, Equals, time.Second*12)
  1429. c.Assert(bucket.getConfig().HTTPTimeout.LongTimeout, Equals, time.Second*12*10)
  1430. c.Assert(bucket.getConfig().SecurityToken, Equals, "token")
  1431. c.Assert(bucket.getConfig().IsCname, Equals, true)
  1432. c.Assert(bucket.getConfig().IsEnableMD5, Equals, false)
  1433. }
  1434. func (s *OssBucketSuite) TestSTSToken(c *C) {
  1435. objectName := objectNamePrefix + "tst"
  1436. objectValue := "红藕香残玉簟秋。轻解罗裳,独上兰舟。云中谁寄锦书来?雁字回时,月满西楼。"
  1437. stsClient := sts.NewClient(stsaccessID, stsaccessKey, stsARN, "oss_test_sess")
  1438. resp, err := stsClient.AssumeRole(1800)
  1439. c.Assert(err, IsNil)
  1440. client, err := New(endpoint, resp.Credentials.AccessKeyId, resp.Credentials.AccessKeySecret,
  1441. SecurityToken(resp.Credentials.SecurityToken))
  1442. c.Assert(err, IsNil)
  1443. bucket, err := client.Bucket(bucketName)
  1444. c.Assert(err, IsNil)
  1445. // Put
  1446. err = bucket.PutObject(objectName, strings.NewReader(objectValue))
  1447. c.Assert(err, IsNil)
  1448. // Get
  1449. body, err := bucket.GetObject(objectName)
  1450. c.Assert(err, IsNil)
  1451. str, err := readBody(body)
  1452. c.Assert(err, IsNil)
  1453. c.Assert(str, Equals, objectValue)
  1454. // List
  1455. lor, err := bucket.ListObjects()
  1456. c.Assert(err, IsNil)
  1457. testLogger.Println("Objects:", lor.Objects)
  1458. // Put with URL
  1459. signedURL, err := bucket.SignURL(objectName, HTTPPut, 3600)
  1460. c.Assert(err, IsNil)
  1461. err = bucket.PutObjectWithURL(signedURL, strings.NewReader(objectValue))
  1462. c.Assert(err, IsNil)
  1463. // Get with URL
  1464. signedURL, err = bucket.SignURL(objectName, HTTPGet, 3600)
  1465. c.Assert(err, IsNil)
  1466. body, err = bucket.GetObjectWithURL(signedURL)
  1467. c.Assert(err, IsNil)
  1468. str, err = readBody(body)
  1469. c.Assert(err, IsNil)
  1470. c.Assert(str, Equals, objectValue)
  1471. // Delete
  1472. err = bucket.DeleteObject(objectName)
  1473. c.Assert(err, IsNil)
  1474. }
  1475. func (s *OssBucketSuite) TestSTSTonekNegative(c *C) {
  1476. objectName := objectNamePrefix + "tstg"
  1477. localFile := objectName + ".jpg"
  1478. client, err := New(endpoint, accessID, accessKey, SecurityToken("Invalid"))
  1479. c.Assert(err, IsNil)
  1480. _, err = client.ListBuckets()
  1481. c.Assert(err, NotNil)
  1482. bucket, err := client.Bucket(bucketName)
  1483. c.Assert(err, IsNil)
  1484. err = bucket.PutObject(objectName, strings.NewReader(""))
  1485. c.Assert(err, NotNil)
  1486. err = bucket.PutObjectFromFile(objectName, "")
  1487. c.Assert(err, NotNil)
  1488. _, err = bucket.GetObject(objectName)
  1489. c.Assert(err, NotNil)
  1490. err = bucket.GetObjectToFile(objectName, "")
  1491. c.Assert(err, NotNil)
  1492. _, err = bucket.ListObjects()
  1493. c.Assert(err, NotNil)
  1494. err = bucket.SetObjectACL(objectName, ACLPublicRead)
  1495. c.Assert(err, NotNil)
  1496. _, err = bucket.GetObjectACL(objectName)
  1497. c.Assert(err, NotNil)
  1498. err = bucket.UploadFile(objectName, localFile, MinPartSize)
  1499. c.Assert(err, NotNil)
  1500. err = bucket.DownloadFile(objectName, localFile, MinPartSize)
  1501. c.Assert(err, NotNil)
  1502. _, err = bucket.IsObjectExist(objectName)
  1503. c.Assert(err, NotNil)
  1504. _, err = bucket.ListMultipartUploads()
  1505. c.Assert(err, NotNil)
  1506. err = bucket.DeleteObject(objectName)
  1507. c.Assert(err, NotNil)
  1508. _, err = bucket.DeleteObjects([]string{objectName})
  1509. c.Assert(err, NotNil)
  1510. err = client.DeleteBucket(bucketName)
  1511. c.Assert(err, NotNil)
  1512. }
  1513. func (s *OssBucketSuite) TestUploadBigFile(c *C) {
  1514. objectName := objectNamePrefix + "tubf"
  1515. bigFile := "D:\\tmp\\bigfile.zip"
  1516. newFile := "D:\\tmp\\newbigfile.zip"
  1517. exist, err := isFileExist(bigFile)
  1518. c.Assert(err, IsNil)
  1519. if !exist {
  1520. return
  1521. }
  1522. // Put
  1523. start := GetNowSec()
  1524. err = s.bucket.PutObjectFromFile(objectName, bigFile)
  1525. c.Assert(err, IsNil)
  1526. end := GetNowSec()
  1527. testLogger.Println("Put big file:", bigFile, "use sec:", end-start)
  1528. // Check
  1529. start = GetNowSec()
  1530. err = s.bucket.GetObjectToFile(objectName, newFile)
  1531. c.Assert(err, IsNil)
  1532. end = GetNowSec()
  1533. testLogger.Println("Get big file:", bigFile, "use sec:", end-start)
  1534. start = GetNowSec()
  1535. eq, err := compareFiles(bigFile, newFile)
  1536. c.Assert(err, IsNil)
  1537. c.Assert(eq, Equals, true)
  1538. end = GetNowSec()
  1539. testLogger.Println("Compare big file:", bigFile, "use sec:", end-start)
  1540. err = s.bucket.DeleteObject(objectName)
  1541. c.Assert(err, IsNil)
  1542. }
  1543. func (s *OssBucketSuite) TestSymlink(c *C) {
  1544. objectName := objectNamePrefix + "符号链接"
  1545. targetObjectName := objectNamePrefix + "符号链接目标文件"
  1546. err := s.bucket.DeleteObject(objectName)
  1547. c.Assert(err, IsNil)
  1548. err = s.bucket.DeleteObject(targetObjectName)
  1549. c.Assert(err, IsNil)
  1550. meta, err := s.bucket.GetSymlink(objectName)
  1551. c.Assert(err, NotNil)
  1552. // Put symlink
  1553. err = s.bucket.PutSymlink(objectName, targetObjectName)
  1554. c.Assert(err, IsNil)
  1555. err = s.bucket.PutObject(targetObjectName, strings.NewReader("target"))
  1556. c.Assert(err, IsNil)
  1557. err = s.bucket.PutSymlink(objectName, targetObjectName)
  1558. c.Assert(err, IsNil)
  1559. meta, err = s.bucket.GetSymlink(objectName)
  1560. c.Assert(err, IsNil)
  1561. c.Assert(meta.Get(HTTPHeaderOssSymlinkTarget), Equals, targetObjectName)
  1562. // List object
  1563. lor, err := s.bucket.ListObjects()
  1564. c.Assert(err, IsNil)
  1565. exist, v := s.getObject(lor.Objects, objectName)
  1566. c.Assert(exist, Equals, true)
  1567. c.Assert(v.Type, Equals, "Symlink")
  1568. body, err := s.bucket.GetObject(objectName)
  1569. c.Assert(err, IsNil)
  1570. str, err := readBody(body)
  1571. c.Assert(err, IsNil)
  1572. c.Assert(str, Equals, "target")
  1573. meta, err = s.bucket.GetSymlink(targetObjectName)
  1574. c.Assert(err, NotNil)
  1575. err = s.bucket.PutObject(objectName, strings.NewReader("src"))
  1576. c.Assert(err, IsNil)
  1577. body, err = s.bucket.GetObject(objectName)
  1578. c.Assert(err, IsNil)
  1579. str, err = readBody(body)
  1580. c.Assert(err, IsNil)
  1581. c.Assert(str, Equals, "src")
  1582. err = s.bucket.DeleteObject(objectName)
  1583. c.Assert(err, IsNil)
  1584. err = s.bucket.DeleteObject(targetObjectName)
  1585. c.Assert(err, IsNil)
  1586. // Put symlink again
  1587. objectName = objectNamePrefix + "symlink"
  1588. targetObjectName = objectNamePrefix + "symlink-target"
  1589. err = s.bucket.PutSymlink(objectName, targetObjectName)
  1590. c.Assert(err, IsNil)
  1591. err = s.bucket.PutObject(targetObjectName, strings.NewReader("target1"))
  1592. c.Assert(err, IsNil)
  1593. meta, err = s.bucket.GetSymlink(objectName)
  1594. c.Assert(err, IsNil)
  1595. c.Assert(meta.Get(HTTPHeaderOssSymlinkTarget), Equals, targetObjectName)
  1596. body, err = s.bucket.GetObject(objectName)
  1597. c.Assert(err, IsNil)
  1598. str, err = readBody(body)
  1599. c.Assert(err, IsNil)
  1600. c.Assert(str, Equals, "target1")
  1601. err = s.bucket.DeleteObject(objectName)
  1602. c.Assert(err, IsNil)
  1603. err = s.bucket.DeleteObject(targetObjectName)
  1604. c.Assert(err, IsNil)
  1605. }
  1606. // TestRestoreObject
  1607. func (s *OssBucketSuite) TestRestoreObject(c *C) {
  1608. objectName := objectNamePrefix + "restore"
  1609. // List objects
  1610. lor, err := s.archiveBucket.ListObjects()
  1611. c.Assert(err, IsNil)
  1612. left := len(lor.Objects)
  1613. // Put object
  1614. err = s.archiveBucket.PutObject(objectName, strings.NewReader(""))
  1615. c.Assert(err, IsNil)
  1616. // List
  1617. lor, err = s.archiveBucket.ListObjects()
  1618. c.Assert(err, IsNil)
  1619. c.Assert(len(lor.Objects), Equals, left+1)
  1620. for _, object := range lor.Objects {
  1621. c.Assert(object.StorageClass, Equals, string(StorageArchive))
  1622. c.Assert(object.Type, Equals, "Normal")
  1623. }
  1624. // Head object
  1625. meta, err := s.archiveBucket.GetObjectDetailedMeta(objectName)
  1626. c.Assert(err, IsNil)
  1627. _, ok := meta["X-Oss-Restore"]
  1628. c.Assert(ok, Equals, false)
  1629. c.Assert(meta.Get("X-Oss-Storage-Class"), Equals, "Archive")
  1630. // Error restore object
  1631. err = s.archiveBucket.RestoreObject("notexistobject")
  1632. c.Assert(err, NotNil)
  1633. // Restore object
  1634. err = s.archiveBucket.RestoreObject(objectName)
  1635. c.Assert(err, IsNil)
  1636. // Head object
  1637. meta, err = s.archiveBucket.GetObjectDetailedMeta(objectName)
  1638. c.Assert(err, IsNil)
  1639. c.Assert(meta.Get("X-Oss-Restore"), Equals, "ongoing-request=\"true\"")
  1640. c.Assert(meta.Get("X-Oss-Storage-Class"), Equals, "Archive")
  1641. }
  1642. // TestProcessObject
  1643. func (s *OssBucketSuite) TestProcessObject(c *C) {
  1644. objectName := objectNamePrefix + "_process_src.jpg"
  1645. err := s.bucket.PutObjectFromFile(objectName, "../sample/BingWallpaper-2015-11-07.jpg")
  1646. c.Assert(err, IsNil)
  1647. // If bucket-name not specified, it is saved to the current bucket by default.
  1648. destObjName := objectNamePrefix + "_process_dest_1.jpg"
  1649. process := fmt.Sprintf("image/resize,w_100|sys/saveas,o_%v", base64.URLEncoding.EncodeToString([]byte(destObjName)))
  1650. result, err := s.bucket.ProcessObject(objectName, process)
  1651. c.Assert(err, IsNil)
  1652. exist, _ := s.bucket.IsObjectExist(destObjName)
  1653. c.Assert(exist, Equals, true)
  1654. c.Assert(result.Bucket, Equals, "")
  1655. c.Assert(result.Object, Equals, destObjName)
  1656. destObjName = objectNamePrefix + "_process_dest_1.jpg"
  1657. process = fmt.Sprintf("image/resize,w_100|sys/saveas,o_%v,b_%v", base64.URLEncoding.EncodeToString([]byte(destObjName)), base64.URLEncoding.EncodeToString([]byte(s.bucket.BucketName)))
  1658. result, err = s.bucket.ProcessObject(objectName, process)
  1659. c.Assert(err, IsNil)
  1660. exist, _ = s.bucket.IsObjectExist(destObjName)
  1661. c.Assert(exist, Equals, true)
  1662. c.Assert(result.Bucket, Equals, s.bucket.BucketName)
  1663. c.Assert(result.Object, Equals, destObjName)
  1664. //no support process
  1665. process = fmt.Sprintf("image/resize,w_100|saveas,o_%v,b_%v", base64.URLEncoding.EncodeToString([]byte(destObjName)), base64.URLEncoding.EncodeToString([]byte(s.bucket.BucketName)))
  1666. result, err = s.bucket.ProcessObject(objectName, process)
  1667. c.Assert(err, NotNil)
  1668. }
  1669. // Private
  1670. func createFileAndWrite(fileName string, data []byte) error {
  1671. os.Remove(fileName)
  1672. fo, err := os.Create(fileName)
  1673. if err != nil {
  1674. return err
  1675. }
  1676. defer fo.Close()
  1677. bytes, err := fo.Write(data)
  1678. if err != nil {
  1679. return err
  1680. }
  1681. if bytes != len(data) {
  1682. return fmt.Errorf(fmt.Sprintf("write %d bytes not equal data length %d", bytes, len(data)))
  1683. }
  1684. return nil
  1685. }
  1686. // Compare the content between fileL and fileR
  1687. func compareFiles(fileL string, fileR string) (bool, error) {
  1688. finL, err := os.Open(fileL)
  1689. if err != nil {
  1690. return false, err
  1691. }
  1692. defer finL.Close()
  1693. finR, err := os.Open(fileR)
  1694. if err != nil {
  1695. return false, err
  1696. }
  1697. defer finR.Close()
  1698. statL, err := finL.Stat()
  1699. if err != nil {
  1700. return false, err
  1701. }
  1702. statR, err := finR.Stat()
  1703. if err != nil {
  1704. return false, err
  1705. }
  1706. if statL.Size() != statR.Size() {
  1707. return false, nil
  1708. }
  1709. size := statL.Size()
  1710. if size > 102400 {
  1711. size = 102400
  1712. }
  1713. bufL := make([]byte, size)
  1714. bufR := make([]byte, size)
  1715. for {
  1716. n, _ := finL.Read(bufL)
  1717. if 0 == n {
  1718. break
  1719. }
  1720. n, _ = finR.Read(bufR)
  1721. if 0 == n {
  1722. break
  1723. }
  1724. if !bytes.Equal(bufL, bufR) {
  1725. return false, nil
  1726. }
  1727. }
  1728. return true, nil
  1729. }
  1730. // Compare the content of file and data
  1731. func compareFileData(file string, data []byte) (bool, error) {
  1732. fin, err := os.Open(file)
  1733. if err != nil {
  1734. return false, err
  1735. }
  1736. defer fin.Close()
  1737. stat, err := fin.Stat()
  1738. if err != nil {
  1739. return false, err
  1740. }
  1741. if stat.Size() != (int64)(len(data)) {
  1742. return false, nil
  1743. }
  1744. buf := make([]byte, stat.Size())
  1745. n, err := fin.Read(buf)
  1746. if err != nil {
  1747. return false, err
  1748. }
  1749. if stat.Size() != (int64)(n) {
  1750. return false, errors.New("read error")
  1751. }
  1752. if !bytes.Equal(buf, data) {
  1753. return false, nil
  1754. }
  1755. return true, nil
  1756. }
  1757. func walkDir(dirPth, suffix string) ([]string, error) {
  1758. var files = []string{}
  1759. suffix = strings.ToUpper(suffix)
  1760. err := filepath.Walk(dirPth,
  1761. func(filename string, fi os.FileInfo, err error) error {
  1762. if err != nil {
  1763. return err
  1764. }
  1765. if fi.IsDir() {
  1766. return nil
  1767. }
  1768. if strings.HasSuffix(strings.ToUpper(fi.Name()), suffix) {
  1769. files = append(files, filename)
  1770. }
  1771. return nil
  1772. })
  1773. return files, err
  1774. }
  1775. func removeTempFiles(path string, prefix string) error {
  1776. files, err := walkDir(path, prefix)
  1777. if err != nil {
  1778. return nil
  1779. }
  1780. for _, file := range files {
  1781. os.Remove(file)
  1782. }
  1783. return nil
  1784. }
  1785. func isFileExist(filename string) (bool, error) {
  1786. _, err := os.Stat(filename)
  1787. if err != nil && os.IsNotExist(err) {
  1788. return false, nil
  1789. } else if err != nil {
  1790. return false, err
  1791. } else {
  1792. return true, nil
  1793. }
  1794. }
  1795. func readBody(body io.ReadCloser) (string, error) {
  1796. data, err := ioutil.ReadAll(body)
  1797. body.Close()
  1798. if err != nil {
  1799. return "", err
  1800. }
  1801. return string(data), nil
  1802. }
  1803. func (s *OssBucketSuite) getObject(objects []ObjectProperties, object string) (bool, ObjectProperties) {
  1804. for _, v := range objects {
  1805. if v.Key == object {
  1806. return true, v
  1807. }
  1808. }
  1809. return false, ObjectProperties{}
  1810. }
  1811. func (s *OssBucketSuite) detectUploadSpeed(bucket *Bucket, c *C) (upSpeed int) {
  1812. objectName := objectNamePrefix + getUuid()
  1813. // 1M byte
  1814. textBuffer := randStr(1024 * 1024)
  1815. // Put string
  1816. startT := time.Now()
  1817. err := bucket.PutObject(objectName, strings.NewReader(textBuffer))
  1818. endT := time.Now()
  1819. c.Assert(err, IsNil)
  1820. err = bucket.DeleteObject(objectName)
  1821. c.Assert(err, IsNil)
  1822. // byte/s
  1823. upSpeed = len(textBuffer) * 1000 / int(endT.UnixNano()/1000/1000-startT.UnixNano()/1000/1000)
  1824. return
  1825. }
  1826. func (s *OssBucketSuite) TestPutSingleObjectLimitSpeed(c *C) {
  1827. // create client and bucket
  1828. client, err := New(endpoint, accessID, accessKey)
  1829. c.Assert(err, IsNil)
  1830. err = client.LimitUploadSpeed(1)
  1831. if err != nil {
  1832. // go version is less than go1.7,not support limit upload speed
  1833. // doesn't run this test
  1834. return
  1835. } else {
  1836. // set unlimited again
  1837. client.LimitUploadSpeed(0)
  1838. }
  1839. bucketName := bucketNamePrefix + randLowStr(5)
  1840. err = client.CreateBucket(bucketName)
  1841. c.Assert(err, IsNil)
  1842. bucket, err := client.Bucket(bucketName)
  1843. c.Assert(err, IsNil)
  1844. //detect speed:byte/s
  1845. detectSpeed := s.detectUploadSpeed(bucket, c)
  1846. var limitSpeed = 0
  1847. if detectSpeed <= perTokenBandSize*2 {
  1848. limitSpeed = perTokenBandSize
  1849. } else {
  1850. //this situation, the test works better
  1851. limitSpeed = detectSpeed / 2
  1852. }
  1853. // KB/s
  1854. err = client.LimitUploadSpeed(limitSpeed / perTokenBandSize)
  1855. c.Assert(err, IsNil)
  1856. objectName := objectNamePrefix + getUuid()
  1857. // 1M byte
  1858. textBuffer := randStr(1024 * 1024)
  1859. // Put body
  1860. startT := time.Now()
  1861. err = bucket.PutObject(objectName, strings.NewReader(textBuffer))
  1862. endT := time.Now()
  1863. realSpeed := int64(len(textBuffer)) * 1000 / (endT.UnixNano()/1000/1000 - startT.UnixNano()/1000/1000)
  1864. fmt.Printf("detect speed:%d,limit speed:%d,real speed:%d.\n", detectSpeed, limitSpeed, realSpeed)
  1865. c.Assert(float64(realSpeed) < float64(limitSpeed)*1.1, Equals, true)
  1866. if detectSpeed > perTokenBandSize {
  1867. // the minimum uploas limit speed is perTokenBandSize(1024 byte/s)
  1868. c.Assert(float64(realSpeed) > float64(limitSpeed)*0.9, Equals, true)
  1869. }
  1870. // Get object and compare content
  1871. body, err := bucket.GetObject(objectName)
  1872. c.Assert(err, IsNil)
  1873. str, err := readBody(body)
  1874. c.Assert(err, IsNil)
  1875. c.Assert(str, Equals, textBuffer)
  1876. bucket.DeleteObject(objectName)
  1877. client.DeleteBucket(bucketName)
  1878. c.Assert(err, IsNil)
  1879. return
  1880. }
  1881. func putObjectRoutin(bucket *Bucket, object string, textBuffer *string, notifyChan chan int) error {
  1882. err := bucket.PutObject(object, strings.NewReader(*textBuffer))
  1883. if err == nil {
  1884. notifyChan <- 1
  1885. } else {
  1886. notifyChan <- 0
  1887. }
  1888. return err
  1889. }
  1890. func (s *OssBucketSuite) TestPutManyObjectLimitSpeed(c *C) {
  1891. // create client and bucket
  1892. client, err := New(endpoint, accessID, accessKey)
  1893. c.Assert(err, IsNil)
  1894. err = client.LimitUploadSpeed(1)
  1895. if err != nil {
  1896. // go version is less than go1.7,not support limit upload speed
  1897. // doesn't run this test
  1898. return
  1899. } else {
  1900. // set unlimited
  1901. client.LimitUploadSpeed(0)
  1902. }
  1903. bucketName := bucketNamePrefix + randLowStr(5)
  1904. err = client.CreateBucket(bucketName)
  1905. c.Assert(err, IsNil)
  1906. bucket, err := client.Bucket(bucketName)
  1907. c.Assert(err, IsNil)
  1908. //detect speed:byte/s
  1909. detectSpeed := s.detectUploadSpeed(bucket, c)
  1910. var limitSpeed = 0
  1911. if detectSpeed <= perTokenBandSize*2 {
  1912. limitSpeed = perTokenBandSize
  1913. } else {
  1914. limitSpeed = detectSpeed / 2
  1915. }
  1916. // KB/s
  1917. err = client.LimitUploadSpeed(limitSpeed / perTokenBandSize)
  1918. c.Assert(err, IsNil)
  1919. // object1
  1920. objectNameFirst := objectNamePrefix + getUuid()
  1921. objectNameSecond := objectNamePrefix + getUuid()
  1922. // 1M byte
  1923. textBuffer := randStr(1024 * 1024)
  1924. objectCount := 2
  1925. notifyChan := make(chan int, objectCount)
  1926. //start routin
  1927. startT := time.Now()
  1928. go putObjectRoutin(bucket, objectNameFirst, &textBuffer, notifyChan)
  1929. go putObjectRoutin(bucket, objectNameSecond, &textBuffer, notifyChan)
  1930. // wait routin end
  1931. sum := int(0)
  1932. for j := 0; j < objectCount; j++ {
  1933. result := <-notifyChan
  1934. sum += result
  1935. }
  1936. endT := time.Now()
  1937. realSpeed := len(textBuffer) * 2 * 1000 / int(endT.UnixNano()/1000/1000-startT.UnixNano()/1000/1000)
  1938. c.Assert(float64(realSpeed) < float64(limitSpeed)*1.1, Equals, true)
  1939. if detectSpeed > perTokenBandSize {
  1940. // the minimum uploas limit speed is perTokenBandSize(1024 byte/s)
  1941. c.Assert(float64(realSpeed) > float64(limitSpeed)*0.9, Equals, true)
  1942. }
  1943. c.Assert(sum, Equals, 2)
  1944. // Get object and compare content
  1945. body, err := bucket.GetObject(objectNameFirst)
  1946. c.Assert(err, IsNil)
  1947. str, err := readBody(body)
  1948. c.Assert(err, IsNil)
  1949. c.Assert(str, Equals, textBuffer)
  1950. body, err = bucket.GetObject(objectNameSecond)
  1951. c.Assert(err, IsNil)
  1952. str, err = readBody(body)
  1953. c.Assert(err, IsNil)
  1954. c.Assert(str, Equals, textBuffer)
  1955. // clear bucket and object
  1956. bucket.DeleteObject(objectNameFirst)
  1957. bucket.DeleteObject(objectNameSecond)
  1958. client.DeleteBucket(bucketName)
  1959. fmt.Printf("detect speed:%d,limit speed:%d,real speed:%d.\n", detectSpeed, limitSpeed, realSpeed)
  1960. return
  1961. }
  1962. func (s *OssBucketSuite) TestPutMultipartObjectLimitSpeed(c *C) {
  1963. // create client and bucket
  1964. client, err := New(endpoint, accessID, accessKey)
  1965. c.Assert(err, IsNil)
  1966. err = client.LimitUploadSpeed(1)
  1967. if err != nil {
  1968. // go version is less than go1.7,not support limit upload speed
  1969. // doesn't run this test
  1970. return
  1971. } else {
  1972. // set unlimited
  1973. client.LimitUploadSpeed(0)
  1974. }
  1975. bucketName := bucketNamePrefix + randLowStr(5)
  1976. err = client.CreateBucket(bucketName)
  1977. c.Assert(err, IsNil)
  1978. bucket, err := client.Bucket(bucketName)
  1979. c.Assert(err, IsNil)
  1980. //detect speed:byte/s
  1981. detectSpeed := s.detectUploadSpeed(bucket, c)
  1982. var limitSpeed = 0
  1983. if detectSpeed <= perTokenBandSize*2 {
  1984. limitSpeed = perTokenBandSize
  1985. } else {
  1986. //this situation, the test works better
  1987. limitSpeed = detectSpeed / 2
  1988. }
  1989. // KB/s
  1990. err = client.LimitUploadSpeed(limitSpeed / perTokenBandSize)
  1991. c.Assert(err, IsNil)
  1992. objectName := objectNamePrefix + getUuid()
  1993. fileName := "." + string(os.PathSeparator) + objectName
  1994. // 1M byte
  1995. fileSize := 0
  1996. textBuffer := randStr(1024 * 1024)
  1997. if detectSpeed < perTokenBandSize {
  1998. ioutil.WriteFile(fileName, []byte(textBuffer), 0644)
  1999. f, err := os.Stat(fileName)
  2000. c.Assert(err, IsNil)
  2001. fileSize = int(f.Size())
  2002. c.Assert(fileSize, Equals, len(textBuffer))
  2003. } else {
  2004. loopCount := 5
  2005. f, err := os.OpenFile(fileName, os.O_CREATE|os.O_APPEND|os.O_RDWR, 0660)
  2006. c.Assert(err, IsNil)
  2007. for i := 0; i < loopCount; i++ {
  2008. f.Write([]byte(textBuffer))
  2009. }
  2010. fileInfo, err := f.Stat()
  2011. c.Assert(err, IsNil)
  2012. fileSize = int(fileInfo.Size())
  2013. c.Assert(fileSize, Equals, len(textBuffer)*loopCount)
  2014. f.Close()
  2015. }
  2016. // Put body
  2017. startT := time.Now()
  2018. err = bucket.UploadFile(objectName, fileName, 100*1024, Routines(3), Checkpoint(true, ""))
  2019. endT := time.Now()
  2020. c.Assert(err, IsNil)
  2021. realSpeed := fileSize * 1000 / int(endT.UnixNano()/1000/1000-startT.UnixNano()/1000/1000)
  2022. c.Assert(float64(realSpeed) < float64(limitSpeed)*1.1, Equals, true)
  2023. if detectSpeed > perTokenBandSize {
  2024. // the minimum uploas limit speed is perTokenBandSize(1024 byte/s)
  2025. c.Assert(float64(realSpeed) > float64(limitSpeed)*0.9, Equals, true)
  2026. }
  2027. // Get object and compare content
  2028. body, err := bucket.GetObject(objectName)
  2029. c.Assert(err, IsNil)
  2030. str, err := readBody(body)
  2031. c.Assert(err, IsNil)
  2032. fileBody, err := ioutil.ReadFile(fileName)
  2033. c.Assert(err, IsNil)
  2034. c.Assert(str, Equals, string(fileBody))
  2035. // delete bucket、object、file
  2036. bucket.DeleteObject(objectName)
  2037. client.DeleteBucket(bucketName)
  2038. os.Remove(fileName)
  2039. fmt.Printf("detect speed:%d,limit speed:%d,real speed:%d.\n", detectSpeed, limitSpeed, realSpeed)
  2040. return
  2041. }
  2042. func (s *OssBucketSuite) TestPutObjectFromFileLimitSpeed(c *C) {
  2043. // create client and bucket
  2044. client, err := New(endpoint, accessID, accessKey)
  2045. c.Assert(err, IsNil)
  2046. err = client.LimitUploadSpeed(1)
  2047. if err != nil {
  2048. // go version is less than go1.7,not support limit upload speed
  2049. // doesn't run this test
  2050. return
  2051. } else {
  2052. // set unlimited
  2053. client.LimitUploadSpeed(0)
  2054. }
  2055. bucketName := bucketNamePrefix + randLowStr(5)
  2056. err = client.CreateBucket(bucketName)
  2057. c.Assert(err, IsNil)
  2058. bucket, err := client.Bucket(bucketName)
  2059. c.Assert(err, IsNil)
  2060. //detect speed:byte/s
  2061. detectSpeed := s.detectUploadSpeed(bucket, c)
  2062. var limitSpeed = 0
  2063. if detectSpeed <= perTokenBandSize*2 {
  2064. limitSpeed = perTokenBandSize
  2065. } else {
  2066. //this situation, the test works better
  2067. limitSpeed = detectSpeed / 2
  2068. }
  2069. // KB/s
  2070. err = client.LimitUploadSpeed(limitSpeed / perTokenBandSize)
  2071. c.Assert(err, IsNil)
  2072. objectName := objectNamePrefix + getUuid()
  2073. fileName := "." + string(os.PathSeparator) + objectName
  2074. // 1M byte
  2075. fileSize := 0
  2076. textBuffer := randStr(1024 * 1024)
  2077. if detectSpeed < perTokenBandSize {
  2078. ioutil.WriteFile(fileName, []byte(textBuffer), 0644)
  2079. f, err := os.Stat(fileName)
  2080. c.Assert(err, IsNil)
  2081. fileSize = int(f.Size())
  2082. c.Assert(fileSize, Equals, len(textBuffer))
  2083. } else {
  2084. loopCount := 2
  2085. f, err := os.OpenFile(fileName, os.O_CREATE|os.O_APPEND|os.O_RDWR, 0660)
  2086. c.Assert(err, IsNil)
  2087. for i := 0; i < loopCount; i++ {
  2088. f.Write([]byte(textBuffer))
  2089. }
  2090. fileInfo, err := f.Stat()
  2091. c.Assert(err, IsNil)
  2092. fileSize = int(fileInfo.Size())
  2093. c.Assert(fileSize, Equals, len(textBuffer)*loopCount)
  2094. f.Close()
  2095. }
  2096. // Put body
  2097. startT := time.Now()
  2098. err = bucket.PutObjectFromFile(objectName, fileName)
  2099. endT := time.Now()
  2100. c.Assert(err, IsNil)
  2101. realSpeed := fileSize * 1000 / int(endT.UnixNano()/1000/1000-startT.UnixNano()/1000/1000)
  2102. c.Assert(float64(realSpeed) < float64(limitSpeed)*1.1, Equals, true)
  2103. if detectSpeed > perTokenBandSize {
  2104. // the minimum uploas limit speed is perTokenBandSize(1024 byte/s)
  2105. c.Assert(float64(realSpeed) > float64(limitSpeed)*0.9, Equals, true)
  2106. }
  2107. // Get object and compare content
  2108. body, err := bucket.GetObject(objectName)
  2109. c.Assert(err, IsNil)
  2110. str, err := readBody(body)
  2111. c.Assert(err, IsNil)
  2112. fileBody, err := ioutil.ReadFile(fileName)
  2113. c.Assert(err, IsNil)
  2114. c.Assert(str, Equals, string(fileBody))
  2115. // delete bucket、file、object
  2116. bucket.DeleteObject(objectName)
  2117. client.DeleteBucket(bucketName)
  2118. os.Remove(fileName)
  2119. fmt.Printf("detect speed:%d,limit speed:%d,real speed:%d.\n", detectSpeed, limitSpeed, realSpeed)
  2120. return
  2121. }