bucket_test.go 57 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293129412951296129712981299130013011302130313041305130613071308130913101311131213131314131513161317131813191320132113221323132413251326132713281329133013311332133313341335133613371338133913401341134213431344134513461347134813491350135113521353135413551356135713581359136013611362136313641365136613671368136913701371137213731374137513761377137813791380138113821383138413851386138713881389139013911392139313941395139613971398139914001401140214031404140514061407140814091410141114121413141414151416141714181419142014211422142314241425142614271428142914301431143214331434143514361437143814391440144114421443144414451446144714481449145014511452145314541455145614571458145914601461146214631464146514661467146814691470147114721473147414751476147714781479148014811482148314841485148614871488148914901491149214931494149514961497149814991500150115021503150415051506150715081509151015111512151315141515151615171518151915201521152215231524152515261527152815291530153115321533153415351536153715381539154015411542154315441545154615471548154915501551155215531554155515561557155815591560156115621563156415651566156715681569157015711572157315741575157615771578157915801581158215831584158515861587158815891590159115921593159415951596159715981599160016011602160316041605160616071608160916101611161216131614161516161617161816191620162116221623162416251626162716281629163016311632163316341635163616371638163916401641164216431644164516461647164816491650165116521653165416551656165716581659166016611662166316641665166616671668166916701671167216731674167516761677167816791680168116821683168416851686168716881689169016911692169316941695169616971698169917001701170217031704170517061707170817091710171117121713171417151716171717181719172017211722172317241725172617271728172917301731173217331734173517361737173817391740174117421743174417451746174717481749175017511752175317541755175617571758175917601761176217631764176517661767176817691770177117721773177417751776177717781779178017811782178317841785178617871788178917901791179217931794179517961797179817991800180118021803180418051806180718081809181018111812181318141815181618171818181918201821182218231824182518261827182818291830183118321833183418351836183718381839184018411842184318441845184618471848184918501851185218531854185518561857185818591860186118621863186418651866186718681869187018711872187318741875187618771878187918801881188218831884188518861887188818891890189118921893189418951896189718981899190019011902190319041905190619071908190919101911191219131914191519161917191819191920192119221923192419251926192719281929193019311932193319341935193619371938193919401941194219431944194519461947194819491950195119521953195419551956195719581959196019611962196319641965196619671968196919701971197219731974197519761977197819791980198119821983198419851986198719881989199019911992199319941995199619971998199920002001200220032004200520062007200820092010201120122013201420152016201720182019202020212022202320242025202620272028202920302031203220332034203520362037203820392040204120422043204420452046204720482049
  1. // bucket test
  2. package oss
  3. import (
  4. "bytes"
  5. "encoding/json"
  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. . "gopkg.in/check.v1"
  18. )
  19. type OssBucketSuite struct {
  20. client *Client
  21. bucket *Bucket
  22. archiveBucket *Bucket
  23. }
  24. var _ = Suite(&OssBucketSuite{})
  25. var (
  26. pastDate = time.Date(2009, time.November, 10, 23, 0, 0, 0, time.UTC)
  27. futureDate = time.Date(2049, time.January, 10, 23, 0, 0, 0, time.UTC)
  28. )
  29. // Run once when the suite starts running
  30. func (s *OssBucketSuite) SetUpSuite(c *C) {
  31. client, err := New(endpoint, accessID, accessKey)
  32. c.Assert(err, IsNil)
  33. s.client = client
  34. s.client.CreateBucket(bucketName)
  35. err = s.client.CreateBucket(archiveBucketName, StorageClass(StorageArchive))
  36. c.Assert(err, IsNil)
  37. time.Sleep(5 * time.Second)
  38. bucket, err := s.client.Bucket(bucketName)
  39. c.Assert(err, IsNil)
  40. s.bucket = bucket
  41. archiveBucket, err := s.client.Bucket(archiveBucketName)
  42. c.Assert(err, IsNil)
  43. s.archiveBucket = archiveBucket
  44. testLogger.Println("test bucket started")
  45. }
  46. // Run before each test or benchmark starts running
  47. func (s *OssBucketSuite) TearDownSuite(c *C) {
  48. for _, bucket := range []*Bucket{s.bucket, s.archiveBucket} {
  49. // Delete Multipart
  50. lmu, err := bucket.ListMultipartUploads()
  51. c.Assert(err, IsNil)
  52. for _, upload := range lmu.Uploads {
  53. imur := InitiateMultipartUploadResult{Bucket: bucketName, Key: upload.Key, UploadID: upload.UploadID}
  54. err = bucket.AbortMultipartUpload(imur)
  55. c.Assert(err, IsNil)
  56. }
  57. // Delete Objects
  58. lor, err := bucket.ListObjects()
  59. c.Assert(err, IsNil)
  60. for _, object := range lor.Objects {
  61. err = bucket.DeleteObject(object.Key)
  62. c.Assert(err, IsNil)
  63. }
  64. }
  65. testLogger.Println("test bucket completed")
  66. }
  67. // Run after each test or benchmark runs
  68. func (s *OssBucketSuite) SetUpTest(c *C) {
  69. err := removeTempFiles("../oss", ".jpg")
  70. c.Assert(err, IsNil)
  71. }
  72. // Run once after all tests or benchmarks have finished running
  73. func (s *OssBucketSuite) TearDownTest(c *C) {
  74. err := removeTempFiles("../oss", ".jpg")
  75. c.Assert(err, IsNil)
  76. err = removeTempFiles("../oss", ".txt")
  77. c.Assert(err, IsNil)
  78. err = removeTempFiles("../oss", ".temp")
  79. c.Assert(err, IsNil)
  80. err = removeTempFiles("../oss", ".txt1")
  81. c.Assert(err, IsNil)
  82. err = removeTempFiles("../oss", ".txt2")
  83. c.Assert(err, IsNil)
  84. }
  85. // TestPutObject
  86. func (s *OssBucketSuite) TestPutObject(c *C) {
  87. objectName := objectNamePrefix + "tpo"
  88. objectValue := "大江东去,浪淘尽,千古风流人物。 故垒西边,人道是、三国周郎赤壁。 乱石穿空,惊涛拍岸,卷起千堆雪。 江山如画,一时多少豪杰。" +
  89. "遥想公谨当年,小乔初嫁了,雄姿英发。 羽扇纶巾,谈笑间、樯橹灰飞烟灭。故国神游,多情应笑我,早生华发,人生如梦,一尊还酹江月。"
  90. // string put
  91. err := s.bucket.PutObject(objectName, strings.NewReader(objectValue))
  92. c.Assert(err, IsNil)
  93. // Check
  94. body, err := s.bucket.GetObject(objectName)
  95. c.Assert(err, IsNil)
  96. str, err := readBody(body)
  97. c.Assert(err, IsNil)
  98. c.Assert(str, Equals, objectValue)
  99. acl, err := s.bucket.GetObjectACL(objectName)
  100. c.Assert(err, IsNil)
  101. testLogger.Println("aclRes:", acl)
  102. c.Assert(acl.ACL, Equals, "default")
  103. err = s.bucket.DeleteObject(objectName)
  104. c.Assert(err, IsNil)
  105. // bytes put
  106. err = s.bucket.PutObject(objectName, bytes.NewReader([]byte(objectValue)))
  107. c.Assert(err, IsNil)
  108. // Check
  109. body, err = s.bucket.GetObject(objectName)
  110. c.Assert(err, IsNil)
  111. str, err = readBody(body)
  112. c.Assert(err, IsNil)
  113. c.Assert(str, Equals, objectValue)
  114. err = s.bucket.DeleteObject(objectName)
  115. c.Assert(err, IsNil)
  116. // file put
  117. err = createFileAndWrite(objectName+".txt", []byte(objectValue))
  118. c.Assert(err, IsNil)
  119. fd, err := os.Open(objectName + ".txt")
  120. c.Assert(err, IsNil)
  121. err = s.bucket.PutObject(objectName, fd)
  122. c.Assert(err, IsNil)
  123. os.Remove(objectName + ".txt")
  124. // Check
  125. body, err = s.bucket.GetObject(objectName)
  126. c.Assert(err, IsNil)
  127. str, err = readBody(body)
  128. c.Assert(err, IsNil)
  129. c.Assert(str, Equals, objectValue)
  130. err = s.bucket.DeleteObject(objectName)
  131. c.Assert(err, IsNil)
  132. // Put with properties
  133. objectName = objectNamePrefix + "tpox"
  134. options := []Option{
  135. Expires(futureDate),
  136. ObjectACL(ACLPublicRead),
  137. Meta("myprop", "mypropval"),
  138. }
  139. err = s.bucket.PutObject(objectName, strings.NewReader(objectValue), options...)
  140. c.Assert(err, IsNil)
  141. // Check
  142. body, err = s.bucket.GetObject(objectName)
  143. c.Assert(err, IsNil)
  144. str, err = readBody(body)
  145. c.Assert(err, IsNil)
  146. c.Assert(str, Equals, objectValue)
  147. acl, err = s.bucket.GetObjectACL(objectName)
  148. c.Assert(err, IsNil)
  149. testLogger.Println("GetObjectACL:", acl)
  150. c.Assert(acl.ACL, Equals, string(ACLPublicRead))
  151. meta, err := s.bucket.GetObjectDetailedMeta(objectName)
  152. c.Assert(err, IsNil)
  153. testLogger.Println("GetObjectDetailedMeta:", meta)
  154. c.Assert(meta.Get("X-Oss-Meta-Myprop"), Equals, "mypropval")
  155. err = s.bucket.DeleteObject(objectName)
  156. c.Assert(err, IsNil)
  157. }
  158. func (s *OssBucketSuite) TestSignURL(c *C) {
  159. objectName := objectNamePrefix + randStr(5)
  160. objectValue := randStr(20)
  161. filePath := randLowStr(10)
  162. content := "复写object"
  163. createFile(filePath, content, c)
  164. notExistfilePath := randLowStr(10)
  165. os.Remove(notExistfilePath)
  166. // sign url for put
  167. str, err := s.bucket.SignURL(objectName, HTTPPut, 60)
  168. c.Assert(err, IsNil)
  169. c.Assert(strings.Contains(str, HTTPParamExpires+"="), Equals, true)
  170. c.Assert(strings.Contains(str, HTTPParamAccessKeyID+"="), Equals, true)
  171. c.Assert(strings.Contains(str, HTTPParamSignature+"="), Equals, true)
  172. // error put object with url
  173. err = s.bucket.PutObjectWithURL(str, strings.NewReader(objectValue), ContentType("image/tiff"))
  174. c.Assert(err, NotNil)
  175. c.Assert(err.(ServiceError).Code, Equals, "SignatureDoesNotMatch")
  176. err = s.bucket.PutObjectFromFileWithURL(str, filePath, ContentType("image/tiff"))
  177. c.Assert(err, NotNil)
  178. c.Assert(err.(ServiceError).Code, Equals, "SignatureDoesNotMatch")
  179. // put object with url
  180. err = s.bucket.PutObjectWithURL(str, strings.NewReader(objectValue))
  181. c.Assert(err, IsNil)
  182. acl, err := s.bucket.GetObjectACL(objectName)
  183. c.Assert(err, IsNil)
  184. c.Assert(acl.ACL, Equals, "default")
  185. // get object meta
  186. meta, err := s.bucket.GetObjectDetailedMeta(objectName)
  187. c.Assert(err, IsNil)
  188. c.Assert(meta.Get(HTTPHeaderContentType), Equals, "application/octet-stream")
  189. c.Assert(meta.Get("X-Oss-Meta-Myprop"), Equals, "")
  190. // sign url for get object
  191. str, err = s.bucket.SignURL(objectName, HTTPGet, 60)
  192. c.Assert(err, IsNil)
  193. c.Assert(strings.Contains(str, HTTPParamExpires+"="), Equals, true)
  194. c.Assert(strings.Contains(str, HTTPParamAccessKeyID+"="), Equals, true)
  195. c.Assert(strings.Contains(str, HTTPParamSignature+"="), Equals, true)
  196. // get object with url
  197. body, err := s.bucket.GetObjectWithURL(str)
  198. c.Assert(err, IsNil)
  199. str, err = readBody(body)
  200. c.Assert(err, IsNil)
  201. c.Assert(str, Equals, objectValue)
  202. // sign url for put with options
  203. options := []Option{
  204. ObjectACL(ACLPublicRead),
  205. Meta("myprop", "mypropval"),
  206. ContentType("image/tiff"),
  207. ResponseContentEncoding("deflate"),
  208. }
  209. str, err = s.bucket.SignURL(objectName, HTTPPut, 60, options...)
  210. c.Assert(err, IsNil)
  211. c.Assert(strings.Contains(str, HTTPParamExpires+"="), Equals, true)
  212. c.Assert(strings.Contains(str, HTTPParamAccessKeyID+"="), Equals, true)
  213. c.Assert(strings.Contains(str, HTTPParamSignature+"="), Equals, true)
  214. // put object with url from file
  215. // without option, error
  216. err = s.bucket.PutObjectWithURL(str, strings.NewReader(objectValue))
  217. c.Assert(err, NotNil)
  218. c.Assert(err.(ServiceError).Code, Equals, "SignatureDoesNotMatch")
  219. err = s.bucket.PutObjectFromFileWithURL(str, filePath)
  220. c.Assert(err, NotNil)
  221. c.Assert(err.(ServiceError).Code, Equals, "SignatureDoesNotMatch")
  222. // with option, error file
  223. err = s.bucket.PutObjectFromFileWithURL(str, notExistfilePath, options...)
  224. c.Assert(err, NotNil)
  225. // with option
  226. err = s.bucket.PutObjectFromFileWithURL(str, filePath, options...)
  227. c.Assert(err, IsNil)
  228. // get object meta
  229. meta, err = s.bucket.GetObjectDetailedMeta(objectName)
  230. c.Assert(err, IsNil)
  231. c.Assert(meta.Get("X-Oss-Meta-Myprop"), Equals, "mypropval")
  232. c.Assert(meta.Get(HTTPHeaderContentType), Equals, "image/tiff")
  233. acl, err = s.bucket.GetObjectACL(objectName)
  234. c.Assert(err, IsNil)
  235. c.Assert(acl.ACL, Equals, string(ACLPublicRead))
  236. // sign url for get object
  237. str, err = s.bucket.SignURL(objectName, HTTPGet, 60)
  238. c.Assert(err, IsNil)
  239. // get object to file with url
  240. newFile := randStr(10)
  241. err = s.bucket.GetObjectToFileWithURL(str, newFile)
  242. c.Assert(err, IsNil)
  243. eq, err := compareFiles(filePath, newFile)
  244. c.Assert(err, IsNil)
  245. c.Assert(eq, Equals, true)
  246. os.Remove(newFile)
  247. // get object to file error
  248. err = s.bucket.GetObjectToFileWithURL(str, newFile, options...)
  249. c.Assert(err, NotNil)
  250. c.Assert(err.(ServiceError).Code, Equals, "SignatureDoesNotMatch")
  251. _, err = os.Stat(newFile)
  252. c.Assert(err, NotNil)
  253. // get object error
  254. body, err = s.bucket.GetObjectWithURL(str, options...)
  255. c.Assert(err, NotNil)
  256. c.Assert(err.(ServiceError).Code, Equals, "SignatureDoesNotMatch")
  257. c.Assert(body, IsNil)
  258. // sign url for get object with options
  259. options = []Option{
  260. Expires(futureDate),
  261. ObjectACL(ACLPublicRead),
  262. Meta("myprop", "mypropval"),
  263. ContentType("image/tiff"),
  264. ResponseContentEncoding("deflate"),
  265. }
  266. str, err = s.bucket.SignURL(objectName, HTTPGet, 60, options...)
  267. c.Assert(err, IsNil)
  268. // get object to file with url and options
  269. err = s.bucket.GetObjectToFileWithURL(str, newFile, options...)
  270. c.Assert(err, IsNil)
  271. eq, err = compareFiles(filePath, newFile)
  272. c.Assert(err, IsNil)
  273. c.Assert(eq, Equals, true)
  274. os.Remove(newFile)
  275. // get object to file error
  276. err = s.bucket.GetObjectToFileWithURL(str, newFile)
  277. c.Assert(err, NotNil)
  278. c.Assert(err.(ServiceError).Code, Equals, "SignatureDoesNotMatch")
  279. _, err = os.Stat(newFile)
  280. c.Assert(err, NotNil)
  281. // get object error
  282. body, err = s.bucket.GetObjectWithURL(str)
  283. c.Assert(err, NotNil)
  284. c.Assert(err.(ServiceError).Code, Equals, "SignatureDoesNotMatch")
  285. c.Assert(body, IsNil)
  286. os.Remove(filePath)
  287. os.Remove(newFile)
  288. // sign url error
  289. str, err = s.bucket.SignURL(objectName, HTTPGet, -1)
  290. c.Assert(err, NotNil)
  291. err = s.bucket.DeleteObject(objectName)
  292. c.Assert(err, IsNil)
  293. // invalid url parse
  294. str = randStr(20)
  295. err = s.bucket.PutObjectWithURL(str, strings.NewReader(objectValue))
  296. c.Assert(err, NotNil)
  297. err = s.bucket.GetObjectToFileWithURL(str, newFile)
  298. c.Assert(err, NotNil)
  299. }
  300. // TestPutObjectType
  301. func (s *OssBucketSuite) TestPutObjectType(c *C) {
  302. objectName := objectNamePrefix + "tptt"
  303. objectValue := "乱石穿空,惊涛拍岸,卷起千堆雪。 江山如画,一时多少豪杰。"
  304. // Put
  305. err := s.bucket.PutObject(objectName, strings.NewReader(objectValue))
  306. c.Assert(err, IsNil)
  307. // Check
  308. time.Sleep(time.Second)
  309. body, err := s.bucket.GetObject(objectName)
  310. c.Assert(err, IsNil)
  311. str, err := readBody(body)
  312. c.Assert(err, IsNil)
  313. c.Assert(str, Equals, objectValue)
  314. meta, err := s.bucket.GetObjectDetailedMeta(objectName)
  315. c.Assert(err, IsNil)
  316. c.Assert(meta.Get("Content-Type"), Equals, "application/octet-stream")
  317. err = s.bucket.DeleteObject(objectName)
  318. c.Assert(err, IsNil)
  319. // Put
  320. err = s.bucket.PutObject(objectName+".txt", strings.NewReader(objectValue))
  321. c.Assert(err, IsNil)
  322. meta, err = s.bucket.GetObjectDetailedMeta(objectName + ".txt")
  323. c.Assert(err, IsNil)
  324. c.Assert(meta.Get("Content-Type"), Equals, "text/plain; charset=utf-8")
  325. err = s.bucket.DeleteObject(objectName + ".txt")
  326. c.Assert(err, IsNil)
  327. // Put
  328. err = s.bucket.PutObject(objectName+".apk", strings.NewReader(objectValue))
  329. c.Assert(err, IsNil)
  330. meta, err = s.bucket.GetObjectDetailedMeta(objectName + ".apk")
  331. c.Assert(err, IsNil)
  332. c.Assert(meta.Get("Content-Type"), Equals, "application/vnd.android.package-archive")
  333. err = s.bucket.DeleteObject(objectName + ".txt")
  334. c.Assert(err, IsNil)
  335. }
  336. // TestPutObject
  337. func (s *OssBucketSuite) TestPutObjectKeyChars(c *C) {
  338. objectName := objectNamePrefix + "tpokc"
  339. objectValue := "白日依山尽,黄河入海流。欲穷千里目,更上一层楼。"
  340. // Put
  341. objectKey := objectName + "十步杀一人,千里不留行。事了拂衣去,深藏身与名"
  342. err := s.bucket.PutObject(objectKey, strings.NewReader(objectValue))
  343. c.Assert(err, IsNil)
  344. // Check
  345. body, err := s.bucket.GetObject(objectKey)
  346. c.Assert(err, IsNil)
  347. str, err := readBody(body)
  348. c.Assert(err, IsNil)
  349. c.Assert(str, Equals, objectValue)
  350. err = s.bucket.DeleteObject(objectKey)
  351. c.Assert(err, IsNil)
  352. // Put
  353. objectKey = objectName + "ごきげん如何ですかおれの顔をよく拝んでおけ"
  354. err = s.bucket.PutObject(objectKey, strings.NewReader(objectValue))
  355. c.Assert(err, IsNil)
  356. // Check
  357. body, err = s.bucket.GetObject(objectKey)
  358. c.Assert(err, IsNil)
  359. str, err = readBody(body)
  360. c.Assert(err, IsNil)
  361. c.Assert(str, Equals, objectValue)
  362. err = s.bucket.DeleteObject(objectKey)
  363. c.Assert(err, IsNil)
  364. // Put
  365. objectKey = objectName + "~!@#$%^&*()_-+=|\\[]{}<>,./?"
  366. err = s.bucket.PutObject(objectKey, strings.NewReader(objectValue))
  367. c.Assert(err, IsNil)
  368. // Check
  369. body, err = s.bucket.GetObject(objectKey)
  370. c.Assert(err, IsNil)
  371. str, err = readBody(body)
  372. c.Assert(err, IsNil)
  373. c.Assert(str, Equals, objectValue)
  374. err = s.bucket.DeleteObject(objectKey)
  375. c.Assert(err, IsNil)
  376. // Put
  377. objectKey = "go/中国 日本 +-#&=*"
  378. err = s.bucket.PutObject(objectKey, strings.NewReader(objectValue))
  379. c.Assert(err, IsNil)
  380. // Check
  381. body, err = s.bucket.GetObject(objectKey)
  382. c.Assert(err, IsNil)
  383. str, err = readBody(body)
  384. c.Assert(err, IsNil)
  385. c.Assert(str, Equals, objectValue)
  386. err = s.bucket.DeleteObject(objectKey)
  387. c.Assert(err, IsNil)
  388. }
  389. // TestPutObjectNegative
  390. func (s *OssBucketSuite) TestPutObjectNegative(c *C) {
  391. objectName := objectNamePrefix + "tpon"
  392. objectValue := "大江东去,浪淘尽,千古风流人物。 "
  393. // Put
  394. objectName = objectNamePrefix + "tpox"
  395. err := s.bucket.PutObject(objectName, strings.NewReader(objectValue),
  396. Meta("meta-my", "myprop"))
  397. c.Assert(err, IsNil)
  398. // Check meta
  399. body, err := s.bucket.GetObject(objectName)
  400. c.Assert(err, IsNil)
  401. str, err := readBody(body)
  402. c.Assert(err, IsNil)
  403. c.Assert(str, Equals, objectValue)
  404. meta, err := s.bucket.GetObjectDetailedMeta(objectName)
  405. c.Assert(err, IsNil)
  406. c.Assert(meta.Get("X-Oss-Meta-My"), Not(Equals), "myprop")
  407. c.Assert(meta.Get("X-Oss-Meta-My"), Equals, "")
  408. err = s.bucket.DeleteObject(objectName)
  409. c.Assert(err, IsNil)
  410. // invalid option
  411. err = s.bucket.PutObject(objectName, strings.NewReader(objectValue),
  412. IfModifiedSince(pastDate))
  413. c.Assert(err, NotNil)
  414. err = s.bucket.PutObjectFromFile(objectName, "bucket.go", IfModifiedSince(pastDate))
  415. c.Assert(err, NotNil)
  416. err = s.bucket.PutObjectFromFile(objectName, "/tmp/xxx")
  417. c.Assert(err, NotNil)
  418. }
  419. // TestPutObjectFromFile
  420. func (s *OssBucketSuite) TestPutObjectFromFile(c *C) {
  421. objectName := objectNamePrefix + "tpoff"
  422. localFile := "../sample/BingWallpaper-2015-11-07.jpg"
  423. newFile := "newpic11.jpg"
  424. // Put
  425. err := s.bucket.PutObjectFromFile(objectName, localFile)
  426. c.Assert(err, IsNil)
  427. // Check
  428. err = s.bucket.GetObjectToFile(objectName, newFile)
  429. c.Assert(err, IsNil)
  430. eq, err := compareFiles(localFile, newFile)
  431. c.Assert(err, IsNil)
  432. c.Assert(eq, Equals, true)
  433. acl, err := s.bucket.GetObjectACL(objectName)
  434. c.Assert(err, IsNil)
  435. testLogger.Println("aclRes:", acl)
  436. c.Assert(acl.ACL, Equals, "default")
  437. err = s.bucket.DeleteObject(objectName)
  438. c.Assert(err, IsNil)
  439. // Put with properties
  440. options := []Option{
  441. Expires(futureDate),
  442. ObjectACL(ACLPublicRead),
  443. Meta("myprop", "mypropval"),
  444. }
  445. os.Remove(newFile)
  446. err = s.bucket.PutObjectFromFile(objectName, localFile, options...)
  447. c.Assert(err, IsNil)
  448. // Check
  449. err = s.bucket.GetObjectToFile(objectName, newFile)
  450. c.Assert(err, IsNil)
  451. eq, err = compareFiles(localFile, newFile)
  452. c.Assert(err, IsNil)
  453. c.Assert(eq, Equals, true)
  454. acl, err = s.bucket.GetObjectACL(objectName)
  455. c.Assert(err, IsNil)
  456. testLogger.Println("GetObjectACL:", acl)
  457. c.Assert(acl.ACL, Equals, string(ACLPublicRead))
  458. meta, err := s.bucket.GetObjectDetailedMeta(objectName)
  459. c.Assert(err, IsNil)
  460. testLogger.Println("GetObjectDetailedMeta:", meta)
  461. c.Assert(meta.Get("X-Oss-Meta-Myprop"), Equals, "mypropval")
  462. err = s.bucket.DeleteObject(objectName)
  463. c.Assert(err, IsNil)
  464. os.Remove(newFile)
  465. }
  466. // TestPutObjectFromFile
  467. func (s *OssBucketSuite) TestPutObjectFromFileType(c *C) {
  468. objectName := objectNamePrefix + "tpoffwm"
  469. localFile := "../sample/BingWallpaper-2015-11-07.jpg"
  470. newFile := "newpic11.jpg"
  471. // Put
  472. err := s.bucket.PutObjectFromFile(objectName, localFile)
  473. c.Assert(err, IsNil)
  474. // Check
  475. err = s.bucket.GetObjectToFile(objectName, newFile)
  476. c.Assert(err, IsNil)
  477. eq, err := compareFiles(localFile, newFile)
  478. c.Assert(err, IsNil)
  479. c.Assert(eq, Equals, true)
  480. meta, err := s.bucket.GetObjectDetailedMeta(objectName)
  481. c.Assert(err, IsNil)
  482. c.Assert(meta.Get("Content-Type"), Equals, "image/jpeg")
  483. err = s.bucket.DeleteObject(objectName)
  484. c.Assert(err, IsNil)
  485. os.Remove(newFile)
  486. }
  487. // TestGetObject
  488. func (s *OssBucketSuite) TestGetObject(c *C) {
  489. objectName := objectNamePrefix + "tgo"
  490. objectValue := "长忆观潮,满郭人争江上望。来疑沧海尽成空,万面鼓声中。弄潮儿向涛头立,手把红旗旗不湿。别来几向梦中看,梦觉尚心寒。"
  491. // Put
  492. err := s.bucket.PutObject(objectName, strings.NewReader(objectValue))
  493. c.Assert(err, IsNil)
  494. // Check
  495. body, err := s.bucket.GetObject(objectName)
  496. c.Assert(err, IsNil)
  497. data, err := ioutil.ReadAll(body)
  498. body.Close()
  499. str := string(data)
  500. c.Assert(str, Equals, objectValue)
  501. testLogger.Println("GetObjec:", str)
  502. // Range
  503. var subObjectValue = string(([]byte(objectValue))[15:36])
  504. body, err = s.bucket.GetObject(objectName, Range(15, 35))
  505. c.Assert(err, IsNil)
  506. data, err = ioutil.ReadAll(body)
  507. body.Close()
  508. str = string(data)
  509. c.Assert(str, Equals, subObjectValue)
  510. testLogger.Println("GetObject:", str, ",", subObjectValue)
  511. // If-Modified-Since
  512. _, err = s.bucket.GetObject(objectName, IfModifiedSince(futureDate))
  513. c.Assert(err, NotNil)
  514. // If-Unmodified-Since
  515. body, err = s.bucket.GetObject(objectName, IfUnmodifiedSince(futureDate))
  516. c.Assert(err, IsNil)
  517. data, err = ioutil.ReadAll(body)
  518. body.Close()
  519. c.Assert(string(data), Equals, objectValue)
  520. meta, err := s.bucket.GetObjectDetailedMeta(objectName)
  521. c.Assert(err, IsNil)
  522. // If-Match
  523. body, err = s.bucket.GetObject(objectName, IfMatch(meta.Get("Etag")))
  524. c.Assert(err, IsNil)
  525. data, err = ioutil.ReadAll(body)
  526. body.Close()
  527. c.Assert(string(data), Equals, objectValue)
  528. // If-None-Match
  529. _, err = s.bucket.GetObject(objectName, IfNoneMatch(meta.Get("Etag")))
  530. c.Assert(err, NotNil)
  531. err = s.bucket.DeleteObject(objectName)
  532. c.Assert(err, IsNil)
  533. }
  534. // TestGetObjectNegative
  535. func (s *OssBucketSuite) TestGetObjectToWriterNegative(c *C) {
  536. objectName := objectNamePrefix + "tgotwn"
  537. objectValue := "长忆观潮,满郭人争江上望。"
  538. // object not exist
  539. _, err := s.bucket.GetObject("NotExist")
  540. c.Assert(err, NotNil)
  541. // constraint invalid
  542. err = s.bucket.PutObject(objectName, strings.NewReader(objectValue))
  543. c.Assert(err, IsNil)
  544. // out of range
  545. _, err = s.bucket.GetObject(objectName, Range(15, 1000))
  546. c.Assert(err, IsNil)
  547. // no exist
  548. err = s.bucket.GetObjectToFile(objectName, "/root/123abc9874")
  549. c.Assert(err, NotNil)
  550. // invalid option
  551. _, err = s.bucket.GetObject(objectName, ACL(ACLPublicRead))
  552. c.Assert(err, IsNil)
  553. err = s.bucket.GetObjectToFile(objectName, "newpic15.jpg", ACL(ACLPublicRead))
  554. c.Assert(err, IsNil)
  555. err = s.bucket.DeleteObject(objectName)
  556. c.Assert(err, IsNil)
  557. }
  558. // TestGetObjectToFile
  559. func (s *OssBucketSuite) TestGetObjectToFile(c *C) {
  560. objectName := objectNamePrefix + "tgotf"
  561. objectValue := "江南好,风景旧曾谙;日出江花红胜火,春来江水绿如蓝。能不忆江南?江南忆,最忆是杭州;山寺月中寻桂子,郡亭枕上看潮头。何日更重游!"
  562. newFile := "newpic15.jpg"
  563. // Put
  564. var val = []byte(objectValue)
  565. err := s.bucket.PutObject(objectName, strings.NewReader(objectValue))
  566. c.Assert(err, IsNil)
  567. // Check
  568. err = s.bucket.GetObjectToFile(objectName, newFile)
  569. c.Assert(err, IsNil)
  570. eq, err := compareFileData(newFile, val)
  571. c.Assert(err, IsNil)
  572. c.Assert(eq, Equals, true)
  573. os.Remove(newFile)
  574. // Range
  575. err = s.bucket.GetObjectToFile(objectName, newFile, Range(15, 35))
  576. c.Assert(err, IsNil)
  577. eq, err = compareFileData(newFile, val[15:36])
  578. c.Assert(err, IsNil)
  579. c.Assert(eq, Equals, true)
  580. os.Remove(newFile)
  581. err = s.bucket.GetObjectToFile(objectName, newFile, NormalizedRange("15-35"))
  582. c.Assert(err, IsNil)
  583. eq, err = compareFileData(newFile, val[15:36])
  584. c.Assert(err, IsNil)
  585. c.Assert(eq, Equals, true)
  586. os.Remove(newFile)
  587. err = s.bucket.GetObjectToFile(objectName, newFile, NormalizedRange("15-"))
  588. c.Assert(err, IsNil)
  589. eq, err = compareFileData(newFile, val[15:])
  590. c.Assert(err, IsNil)
  591. c.Assert(eq, Equals, true)
  592. os.Remove(newFile)
  593. err = s.bucket.GetObjectToFile(objectName, newFile, NormalizedRange("-10"))
  594. c.Assert(err, IsNil)
  595. eq, err = compareFileData(newFile, val[(len(val)-10):len(val)])
  596. c.Assert(err, IsNil)
  597. c.Assert(eq, Equals, true)
  598. os.Remove(newFile)
  599. // If-Modified-Since
  600. err = s.bucket.GetObjectToFile(objectName, newFile, IfModifiedSince(futureDate))
  601. c.Assert(err, NotNil)
  602. // If-Unmodified-Since
  603. err = s.bucket.GetObjectToFile(objectName, newFile, IfUnmodifiedSince(futureDate))
  604. c.Assert(err, IsNil)
  605. eq, err = compareFileData(newFile, val)
  606. c.Assert(err, IsNil)
  607. c.Assert(eq, Equals, true)
  608. os.Remove(newFile)
  609. meta, err := s.bucket.GetObjectDetailedMeta(objectName)
  610. c.Assert(err, IsNil)
  611. testLogger.Println("GetObjectDetailedMeta:", meta)
  612. // If-Match
  613. err = s.bucket.GetObjectToFile(objectName, newFile, IfMatch(meta.Get("Etag")))
  614. c.Assert(err, IsNil)
  615. eq, err = compareFileData(newFile, val)
  616. c.Assert(err, IsNil)
  617. c.Assert(eq, Equals, true)
  618. os.Remove(newFile)
  619. // If-None-Match
  620. err = s.bucket.GetObjectToFile(objectName, newFile, IfNoneMatch(meta.Get("Etag")))
  621. c.Assert(err, NotNil)
  622. err = s.bucket.DeleteObject(objectName)
  623. c.Assert(err, IsNil)
  624. }
  625. // TestListObjects
  626. func (s *OssBucketSuite) TestListObjects(c *C) {
  627. objectName := objectNamePrefix + "tlo"
  628. // list empty bucket
  629. lor, err := s.bucket.ListObjects()
  630. c.Assert(err, IsNil)
  631. left := len(lor.Objects)
  632. // Put three object
  633. err = s.bucket.PutObject(objectName+"1", strings.NewReader(""))
  634. c.Assert(err, IsNil)
  635. err = s.bucket.PutObject(objectName+"2", strings.NewReader(""))
  636. c.Assert(err, IsNil)
  637. err = s.bucket.PutObject(objectName+"3", strings.NewReader(""))
  638. c.Assert(err, IsNil)
  639. // list
  640. lor, err = s.bucket.ListObjects()
  641. c.Assert(err, IsNil)
  642. c.Assert(len(lor.Objects), Equals, left+3)
  643. // list with prefix
  644. lor, err = s.bucket.ListObjects(Prefix(objectName + "2"))
  645. c.Assert(err, IsNil)
  646. c.Assert(len(lor.Objects), Equals, 1)
  647. lor, err = s.bucket.ListObjects(Prefix(objectName + "22"))
  648. c.Assert(err, IsNil)
  649. c.Assert(len(lor.Objects), Equals, 0)
  650. // list with max keys
  651. lor, err = s.bucket.ListObjects(Prefix(objectName), MaxKeys(2))
  652. c.Assert(err, IsNil)
  653. c.Assert(len(lor.Objects), Equals, 2)
  654. // list with marker
  655. lor, err = s.bucket.ListObjects(Marker(objectName+"1"), MaxKeys(1))
  656. c.Assert(err, IsNil)
  657. c.Assert(len(lor.Objects), Equals, 1)
  658. err = s.bucket.DeleteObject(objectName + "1")
  659. c.Assert(err, IsNil)
  660. err = s.bucket.DeleteObject(objectName + "2")
  661. c.Assert(err, IsNil)
  662. err = s.bucket.DeleteObject(objectName + "3")
  663. c.Assert(err, IsNil)
  664. }
  665. // TestListObjects
  666. func (s *OssBucketSuite) TestListObjectsEncodingType(c *C) {
  667. objectName := objectNamePrefix + "床前明月光,疑是地上霜。举头望明月,低头思故乡。" + "tloet"
  668. for i := 0; i < 10; i++ {
  669. err := s.bucket.PutObject(objectName+strconv.Itoa(i), strings.NewReader(""))
  670. c.Assert(err, IsNil)
  671. }
  672. lor, err := s.bucket.ListObjects(Prefix(objectNamePrefix + "床前明月光,"))
  673. c.Assert(err, IsNil)
  674. c.Assert(len(lor.Objects), Equals, 10)
  675. lor, err = s.bucket.ListObjects(Prefix(objectNamePrefix + "床前明月光,"))
  676. c.Assert(err, IsNil)
  677. c.Assert(len(lor.Objects), Equals, 10)
  678. lor, err = s.bucket.ListObjects(Marker(objectNamePrefix + "床前明月光,疑是地上霜。举头望明月,低头思故乡。"))
  679. c.Assert(err, IsNil)
  680. c.Assert(len(lor.Objects), Equals, 10)
  681. lor, err = s.bucket.ListObjects(Prefix(objectNamePrefix + "床前明月光"))
  682. c.Assert(err, IsNil)
  683. for i, obj := range lor.Objects {
  684. c.Assert(obj.Key, Equals, objectNamePrefix+"床前明月光,疑是地上霜。举头望明月,低头思故乡。tloet"+strconv.Itoa(i))
  685. }
  686. for i := 0; i < 10; i++ {
  687. err = s.bucket.DeleteObject(objectName + strconv.Itoa(i))
  688. c.Assert(err, IsNil)
  689. }
  690. // 特殊字符
  691. objectName = "go go ` ~ ! @ # $ % ^ & * () - _ + =[] {} \\ | < > , . ? / 0"
  692. err = s.bucket.PutObject(objectName, strings.NewReader("明月几时有,把酒问青天"))
  693. c.Assert(err, IsNil)
  694. lor, err = s.bucket.ListObjects(Prefix(objectName))
  695. c.Assert(err, IsNil)
  696. c.Assert(len(lor.Objects), Equals, 1)
  697. err = s.bucket.DeleteObject(objectName)
  698. c.Assert(err, IsNil)
  699. objectName = "go/中国 日本 +-#&=*"
  700. err = s.bucket.PutObject(objectName, strings.NewReader("明月几时有,把酒问青天"))
  701. c.Assert(err, IsNil)
  702. lor, err = s.bucket.ListObjects(Prefix(objectName))
  703. c.Assert(err, IsNil)
  704. c.Assert(len(lor.Objects), Equals, 1)
  705. err = s.bucket.DeleteObject(objectName)
  706. c.Assert(err, IsNil)
  707. }
  708. // TestIsBucketExist
  709. func (s *OssBucketSuite) TestIsObjectExist(c *C) {
  710. objectName := objectNamePrefix + "tibe"
  711. // Put three object
  712. err := s.bucket.PutObject(objectName+"1", strings.NewReader(""))
  713. c.Assert(err, IsNil)
  714. err = s.bucket.PutObject(objectName+"11", strings.NewReader(""))
  715. c.Assert(err, IsNil)
  716. err = s.bucket.PutObject(objectName+"111", strings.NewReader(""))
  717. c.Assert(err, IsNil)
  718. // exist
  719. exist, err := s.bucket.IsObjectExist(objectName + "11")
  720. c.Assert(err, IsNil)
  721. c.Assert(exist, Equals, true)
  722. exist, err = s.bucket.IsObjectExist(objectName + "1")
  723. c.Assert(err, IsNil)
  724. c.Assert(exist, Equals, true)
  725. exist, err = s.bucket.IsObjectExist(objectName + "111")
  726. c.Assert(err, IsNil)
  727. c.Assert(exist, Equals, true)
  728. // not exist
  729. exist, err = s.bucket.IsObjectExist(objectName + "1111")
  730. c.Assert(err, IsNil)
  731. c.Assert(exist, Equals, false)
  732. exist, err = s.bucket.IsObjectExist(objectName)
  733. c.Assert(err, IsNil)
  734. c.Assert(exist, Equals, false)
  735. err = s.bucket.DeleteObject(objectName + "1")
  736. c.Assert(err, IsNil)
  737. err = s.bucket.DeleteObject(objectName + "11")
  738. c.Assert(err, IsNil)
  739. err = s.bucket.DeleteObject(objectName + "111")
  740. c.Assert(err, IsNil)
  741. }
  742. // TestDeleteObject
  743. func (s *OssBucketSuite) TestDeleteObject(c *C) {
  744. objectName := objectNamePrefix + "tdo"
  745. err := s.bucket.PutObject(objectName, strings.NewReader(""))
  746. c.Assert(err, IsNil)
  747. lor, err := s.bucket.ListObjects(Prefix(objectName))
  748. c.Assert(err, IsNil)
  749. c.Assert(len(lor.Objects), Equals, 1)
  750. // delete
  751. err = s.bucket.DeleteObject(objectName)
  752. c.Assert(err, IsNil)
  753. // duplicate delete
  754. err = s.bucket.DeleteObject(objectName)
  755. c.Assert(err, IsNil)
  756. lor, err = s.bucket.ListObjects(Prefix(objectName))
  757. c.Assert(err, IsNil)
  758. c.Assert(len(lor.Objects), Equals, 0)
  759. }
  760. // TestDeleteObjects
  761. func (s *OssBucketSuite) TestDeleteObjects(c *C) {
  762. objectName := objectNamePrefix + "tdos"
  763. // delete object
  764. err := s.bucket.PutObject(objectName, strings.NewReader(""))
  765. c.Assert(err, IsNil)
  766. res, err := s.bucket.DeleteObjects([]string{objectName})
  767. c.Assert(err, IsNil)
  768. c.Assert(len(res.DeletedObjects), Equals, 1)
  769. lor, err := s.bucket.ListObjects(Prefix(objectName))
  770. c.Assert(err, IsNil)
  771. c.Assert(len(lor.Objects), Equals, 0)
  772. // delete objects
  773. err = s.bucket.PutObject(objectName+"1", strings.NewReader(""))
  774. c.Assert(err, IsNil)
  775. err = s.bucket.PutObject(objectName+"2", strings.NewReader(""))
  776. c.Assert(err, IsNil)
  777. res, err = s.bucket.DeleteObjects([]string{objectName + "1", objectName + "2"})
  778. c.Assert(err, IsNil)
  779. c.Assert(len(res.DeletedObjects), Equals, 2)
  780. lor, err = s.bucket.ListObjects(Prefix(objectName))
  781. c.Assert(err, IsNil)
  782. c.Assert(len(lor.Objects), Equals, 0)
  783. // delete 0
  784. _, err = s.bucket.DeleteObjects([]string{})
  785. c.Assert(err, NotNil)
  786. // DeleteObjectsQuiet
  787. err = s.bucket.PutObject(objectName+"1", strings.NewReader(""))
  788. c.Assert(err, IsNil)
  789. err = s.bucket.PutObject(objectName+"2", strings.NewReader(""))
  790. c.Assert(err, IsNil)
  791. res, err = s.bucket.DeleteObjects([]string{objectName + "1", objectName + "2"},
  792. DeleteObjectsQuiet(false))
  793. c.Assert(err, IsNil)
  794. c.Assert(len(res.DeletedObjects), Equals, 2)
  795. lor, err = s.bucket.ListObjects(Prefix(objectName))
  796. c.Assert(err, IsNil)
  797. c.Assert(len(lor.Objects), Equals, 0)
  798. // DeleteObjectsQuiet
  799. err = s.bucket.PutObject(objectName+"1", strings.NewReader(""))
  800. c.Assert(err, IsNil)
  801. err = s.bucket.PutObject(objectName+"2", strings.NewReader(""))
  802. c.Assert(err, IsNil)
  803. res, err = s.bucket.DeleteObjects([]string{objectName + "1", objectName + "2"},
  804. DeleteObjectsQuiet(true))
  805. c.Assert(err, IsNil)
  806. c.Assert(len(res.DeletedObjects), Equals, 0)
  807. lor, err = s.bucket.ListObjects(Prefix(objectName))
  808. c.Assert(err, IsNil)
  809. c.Assert(len(lor.Objects), Equals, 0)
  810. // EncodingType
  811. err = s.bucket.PutObject("中国人", strings.NewReader(""))
  812. c.Assert(err, IsNil)
  813. res, err = s.bucket.DeleteObjects([]string{"中国人"})
  814. c.Assert(err, IsNil)
  815. c.Assert(len(res.DeletedObjects), Equals, 1)
  816. c.Assert(res.DeletedObjects[0], Equals, "中国人")
  817. // EncodingType
  818. err = s.bucket.PutObject("中国人", strings.NewReader(""))
  819. c.Assert(err, IsNil)
  820. res, err = s.bucket.DeleteObjects([]string{"中国人"}, DeleteObjectsQuiet(false))
  821. c.Assert(err, IsNil)
  822. c.Assert(len(res.DeletedObjects), Equals, 1)
  823. c.Assert(res.DeletedObjects[0], Equals, "中国人")
  824. // EncodingType
  825. err = s.bucket.PutObject("中国人", strings.NewReader(""))
  826. c.Assert(err, IsNil)
  827. res, err = s.bucket.DeleteObjects([]string{"中国人"}, DeleteObjectsQuiet(true))
  828. c.Assert(err, IsNil)
  829. c.Assert(len(res.DeletedObjects), Equals, 0)
  830. // 特殊字符
  831. key := "A ' < > \" & ~ ` ! @ # $ % ^ & * ( ) [] {} - _ + = / | \\ ? . , : ; A"
  832. err = s.bucket.PutObject(key, strings.NewReader("value"))
  833. c.Assert(err, IsNil)
  834. _, err = s.bucket.DeleteObjects([]string{key})
  835. c.Assert(err, IsNil)
  836. ress, err := s.bucket.ListObjects(Prefix(key))
  837. c.Assert(err, IsNil)
  838. c.Assert(len(ress.Objects), Equals, 0)
  839. // not exist
  840. _, err = s.bucket.DeleteObjects([]string{"NotExistObject"})
  841. c.Assert(err, IsNil)
  842. }
  843. // TestSetObjectMeta
  844. func (s *OssBucketSuite) TestSetObjectMeta(c *C) {
  845. objectName := objectNamePrefix + "tsom"
  846. err := s.bucket.PutObject(objectName, strings.NewReader(""))
  847. c.Assert(err, IsNil)
  848. err = s.bucket.SetObjectMeta(objectName,
  849. Expires(futureDate),
  850. Meta("myprop", "mypropval"))
  851. c.Assert(err, IsNil)
  852. meta, err := s.bucket.GetObjectDetailedMeta(objectName)
  853. c.Assert(err, IsNil)
  854. testLogger.Println("Meta:", meta)
  855. c.Assert(meta.Get("Expires"), Equals, futureDate.Format(http.TimeFormat))
  856. c.Assert(meta.Get("X-Oss-Meta-Myprop"), Equals, "mypropval")
  857. acl, err := s.bucket.GetObjectACL(objectName)
  858. c.Assert(err, IsNil)
  859. c.Assert(acl.ACL, Equals, "default")
  860. // invalid option
  861. err = s.bucket.SetObjectMeta(objectName, AcceptEncoding("url"))
  862. c.Assert(err, IsNil)
  863. // invalid option value
  864. err = s.bucket.SetObjectMeta(objectName, ServerSideEncryption("invalid"))
  865. c.Assert(err, NotNil)
  866. err = s.bucket.DeleteObject(objectName)
  867. c.Assert(err, IsNil)
  868. // no exist
  869. err = s.bucket.SetObjectMeta(objectName, Expires(futureDate))
  870. c.Assert(err, NotNil)
  871. }
  872. // TestGetObjectMeta
  873. func (s *OssBucketSuite) TestGetObjectMeta(c *C) {
  874. objectName := objectNamePrefix + "tgom"
  875. // Put
  876. err := s.bucket.PutObject(objectName, strings.NewReader(""))
  877. c.Assert(err, IsNil)
  878. meta, err := s.bucket.GetObjectMeta(objectName)
  879. c.Assert(err, IsNil)
  880. c.Assert(len(meta) > 0, Equals, true)
  881. err = s.bucket.DeleteObject(objectName)
  882. c.Assert(err, IsNil)
  883. _, err = s.bucket.GetObjectMeta("NotExistObject")
  884. c.Assert(err, NotNil)
  885. }
  886. // TestGetObjectDetailedMeta
  887. func (s *OssBucketSuite) TestGetObjectDetailedMeta(c *C) {
  888. objectName := objectNamePrefix + "tgodm"
  889. // Put
  890. err := s.bucket.PutObject(objectName, strings.NewReader(""),
  891. Expires(futureDate), Meta("myprop", "mypropval"))
  892. c.Assert(err, IsNil)
  893. // Check
  894. meta, err := s.bucket.GetObjectDetailedMeta(objectName)
  895. c.Assert(err, IsNil)
  896. testLogger.Println("GetObjectDetailedMeta:", meta)
  897. c.Assert(meta.Get("Expires"), Equals, futureDate.Format(http.TimeFormat))
  898. c.Assert(meta.Get("X-Oss-Meta-Myprop"), Equals, "mypropval")
  899. c.Assert(meta.Get("Content-Length"), Equals, "0")
  900. c.Assert(len(meta.Get("Date")) > 0, Equals, true)
  901. c.Assert(len(meta.Get("X-Oss-Request-Id")) > 0, Equals, true)
  902. c.Assert(len(meta.Get("Last-Modified")) > 0, Equals, true)
  903. // IfModifiedSince/IfModifiedSince
  904. _, err = s.bucket.GetObjectDetailedMeta(objectName, IfModifiedSince(futureDate))
  905. c.Assert(err, NotNil)
  906. meta, err = s.bucket.GetObjectDetailedMeta(objectName, IfUnmodifiedSince(futureDate))
  907. c.Assert(err, IsNil)
  908. c.Assert(meta.Get("Expires"), Equals, futureDate.Format(http.TimeFormat))
  909. c.Assert(meta.Get("X-Oss-Meta-Myprop"), Equals, "mypropval")
  910. // IfMatch/IfNoneMatch
  911. _, err = s.bucket.GetObjectDetailedMeta(objectName, IfNoneMatch(meta.Get("Etag")))
  912. c.Assert(err, NotNil)
  913. meta, err = s.bucket.GetObjectDetailedMeta(objectName, IfMatch(meta.Get("Etag")))
  914. c.Assert(err, IsNil)
  915. c.Assert(meta.Get("Expires"), Equals, futureDate.Format(http.TimeFormat))
  916. c.Assert(meta.Get("X-Oss-Meta-Myprop"), Equals, "mypropval")
  917. err = s.bucket.DeleteObject(objectName)
  918. c.Assert(err, IsNil)
  919. _, err = s.bucket.GetObjectDetailedMeta("NotExistObject")
  920. c.Assert(err, NotNil)
  921. }
  922. // TestSetAndGetObjectAcl
  923. func (s *OssBucketSuite) TestSetAndGetObjectAcl(c *C) {
  924. objectName := objectNamePrefix + "tsgba"
  925. err := s.bucket.PutObject(objectName, strings.NewReader(""))
  926. c.Assert(err, IsNil)
  927. // default
  928. acl, err := s.bucket.GetObjectACL(objectName)
  929. c.Assert(err, IsNil)
  930. c.Assert(acl.ACL, Equals, "default")
  931. // set ACL_PUBLIC_RW
  932. err = s.bucket.SetObjectACL(objectName, ACLPublicReadWrite)
  933. c.Assert(err, IsNil)
  934. acl, err = s.bucket.GetObjectACL(objectName)
  935. c.Assert(err, IsNil)
  936. c.Assert(acl.ACL, Equals, string(ACLPublicReadWrite))
  937. // set ACL_PRIVATE
  938. err = s.bucket.SetObjectACL(objectName, ACLPrivate)
  939. c.Assert(err, IsNil)
  940. acl, err = s.bucket.GetObjectACL(objectName)
  941. c.Assert(err, IsNil)
  942. c.Assert(acl.ACL, Equals, string(ACLPrivate))
  943. // set ACL_PUBLIC_R
  944. err = s.bucket.SetObjectACL(objectName, ACLPublicRead)
  945. c.Assert(err, IsNil)
  946. acl, err = s.bucket.GetObjectACL(objectName)
  947. c.Assert(err, IsNil)
  948. c.Assert(acl.ACL, Equals, string(ACLPublicRead))
  949. err = s.bucket.DeleteObject(objectName)
  950. c.Assert(err, IsNil)
  951. }
  952. // TestSetAndGetObjectAclNegative
  953. func (s *OssBucketSuite) TestSetAndGetObjectAclNegative(c *C) {
  954. objectName := objectNamePrefix + "tsgban"
  955. // object not exist
  956. err := s.bucket.SetObjectACL(objectName, ACLPublicRead)
  957. c.Assert(err, NotNil)
  958. }
  959. // TestCopyObject
  960. func (s *OssBucketSuite) TestCopyObject(c *C) {
  961. objectName := objectNamePrefix + "tco"
  962. objectValue := "男儿何不带吴钩,收取关山五十州。请君暂上凌烟阁,若个书生万户侯?"
  963. err := s.bucket.PutObject(objectName, strings.NewReader(objectValue),
  964. ACL(ACLPublicRead), Meta("my", "myprop"))
  965. c.Assert(err, IsNil)
  966. // copy
  967. var objectNameDest = objectName + "dest"
  968. _, err = s.bucket.CopyObject(objectName, objectNameDest)
  969. c.Assert(err, IsNil)
  970. // check
  971. lor, err := s.bucket.ListObjects(Prefix(objectName))
  972. c.Assert(err, IsNil)
  973. testLogger.Println("objects:", lor.Objects)
  974. c.Assert(len(lor.Objects), Equals, 2)
  975. body, err := s.bucket.GetObject(objectName)
  976. c.Assert(err, IsNil)
  977. str, err := readBody(body)
  978. c.Assert(err, IsNil)
  979. c.Assert(str, Equals, objectValue)
  980. err = s.bucket.DeleteObject(objectNameDest)
  981. c.Assert(err, IsNil)
  982. // copy with constraints x-oss-copy-source-if-modified-since
  983. _, err = s.bucket.CopyObject(objectName, objectNameDest, CopySourceIfModifiedSince(futureDate))
  984. c.Assert(err, NotNil)
  985. testLogger.Println("CopyObject:", err)
  986. // copy with constraints x-oss-copy-source-if-unmodified-since
  987. _, err = s.bucket.CopyObject(objectName, objectNameDest, CopySourceIfUnmodifiedSince(futureDate))
  988. c.Assert(err, IsNil)
  989. // check
  990. lor, err = s.bucket.ListObjects(Prefix(objectName))
  991. c.Assert(err, IsNil)
  992. testLogger.Println("objects:", lor.Objects)
  993. c.Assert(len(lor.Objects), Equals, 2)
  994. body, err = s.bucket.GetObject(objectName)
  995. c.Assert(err, IsNil)
  996. str, err = readBody(body)
  997. c.Assert(err, IsNil)
  998. c.Assert(str, Equals, objectValue)
  999. err = s.bucket.DeleteObject(objectNameDest)
  1000. c.Assert(err, IsNil)
  1001. // copy with constraints x-oss-copy-source-if-match
  1002. meta, err := s.bucket.GetObjectDetailedMeta(objectName)
  1003. c.Assert(err, IsNil)
  1004. testLogger.Println("GetObjectDetailedMeta:", meta)
  1005. _, err = s.bucket.CopyObject(objectName, objectNameDest, CopySourceIfMatch(meta.Get("Etag")))
  1006. c.Assert(err, IsNil)
  1007. // check
  1008. body, err = s.bucket.GetObject(objectName)
  1009. c.Assert(err, IsNil)
  1010. str, err = readBody(body)
  1011. c.Assert(err, IsNil)
  1012. c.Assert(str, Equals, objectValue)
  1013. err = s.bucket.DeleteObject(objectNameDest)
  1014. c.Assert(err, IsNil)
  1015. // copy with constraints x-oss-copy-source-if-none-match
  1016. _, err = s.bucket.CopyObject(objectName, objectNameDest, CopySourceIfNoneMatch(meta.Get("Etag")))
  1017. c.Assert(err, NotNil)
  1018. // copy with constraints x-oss-metadata-directive
  1019. _, err = s.bucket.CopyObject(objectName, objectNameDest, Meta("my", "mydestprop"),
  1020. MetadataDirective(MetaCopy))
  1021. c.Assert(err, IsNil)
  1022. // check
  1023. body, err = s.bucket.GetObject(objectName)
  1024. c.Assert(err, IsNil)
  1025. str, err = readBody(body)
  1026. c.Assert(err, IsNil)
  1027. c.Assert(str, Equals, objectValue)
  1028. destMeta, err := s.bucket.GetObjectDetailedMeta(objectNameDest)
  1029. c.Assert(err, IsNil)
  1030. c.Assert(meta.Get("X-Oss-Meta-My"), Equals, "myprop")
  1031. acl, err := s.bucket.GetObjectACL(objectNameDest)
  1032. c.Assert(err, IsNil)
  1033. c.Assert(acl.ACL, Equals, "default")
  1034. err = s.bucket.DeleteObject(objectNameDest)
  1035. c.Assert(err, IsNil)
  1036. // copy with constraints x-oss-metadata-directive and self defined dest object meta
  1037. options := []Option{
  1038. ObjectACL(ACLPublicReadWrite),
  1039. Meta("my", "mydestprop"),
  1040. MetadataDirective(MetaReplace),
  1041. }
  1042. _, err = s.bucket.CopyObject(objectName, objectNameDest, options...)
  1043. c.Assert(err, IsNil)
  1044. // check
  1045. body, err = s.bucket.GetObject(objectName)
  1046. c.Assert(err, IsNil)
  1047. str, err = readBody(body)
  1048. c.Assert(err, IsNil)
  1049. c.Assert(str, Equals, objectValue)
  1050. destMeta, err = s.bucket.GetObjectDetailedMeta(objectNameDest)
  1051. c.Assert(err, IsNil)
  1052. c.Assert(destMeta.Get("X-Oss-Meta-My"), Equals, "mydestprop")
  1053. acl, err = s.bucket.GetObjectACL(objectNameDest)
  1054. c.Assert(err, IsNil)
  1055. c.Assert(acl.ACL, Equals, string(ACLPublicReadWrite))
  1056. err = s.bucket.DeleteObject(objectNameDest)
  1057. c.Assert(err, IsNil)
  1058. err = s.bucket.DeleteObject(objectName)
  1059. c.Assert(err, IsNil)
  1060. }
  1061. // TestCopyObjectToOrFrom
  1062. func (s *OssBucketSuite) TestCopyObjectToOrFrom(c *C) {
  1063. objectName := objectNamePrefix + "tcotof"
  1064. objectValue := "男儿何不带吴钩,收取关山五十州。请君暂上凌烟阁,若个书生万户侯?"
  1065. destBucket := bucketName + "-dest"
  1066. objectNameDest := objectName + "dest"
  1067. s.client.CreateBucket(destBucket)
  1068. destBuck, err := s.client.Bucket(destBucket)
  1069. c.Assert(err, IsNil)
  1070. err = s.bucket.PutObject(objectName, strings.NewReader(objectValue))
  1071. c.Assert(err, IsNil)
  1072. // copy from
  1073. _, err = destBuck.CopyObjectFrom(bucketName, objectName, objectNameDest)
  1074. c.Assert(err, IsNil)
  1075. // check
  1076. body, err := destBuck.GetObject(objectNameDest)
  1077. c.Assert(err, IsNil)
  1078. str, err := readBody(body)
  1079. c.Assert(err, IsNil)
  1080. c.Assert(str, Equals, objectValue)
  1081. err = s.bucket.DeleteObject(objectName)
  1082. c.Assert(err, IsNil)
  1083. // copy to
  1084. _, err = destBuck.CopyObjectTo(bucketName, objectName, objectNameDest)
  1085. c.Assert(err, IsNil)
  1086. // check
  1087. body, err = s.bucket.GetObject(objectName)
  1088. c.Assert(err, IsNil)
  1089. str, err = readBody(body)
  1090. c.Assert(err, IsNil)
  1091. c.Assert(str, Equals, objectValue)
  1092. // clean
  1093. err = destBuck.DeleteObject(objectNameDest)
  1094. c.Assert(err, IsNil)
  1095. err = s.bucket.DeleteObject(objectName)
  1096. c.Assert(err, IsNil)
  1097. err = s.client.DeleteBucket(destBucket)
  1098. c.Assert(err, IsNil)
  1099. }
  1100. // TestCopyObjectToOrFromNegative
  1101. func (s *OssBucketSuite) TestCopyObjectToOrFromNegative(c *C) {
  1102. objectName := objectNamePrefix + "tcotofn"
  1103. destBucket := bucketName + "-destn"
  1104. objectNameDest := objectName + "destn"
  1105. // object no exist
  1106. _, err := s.bucket.CopyObjectTo(bucketName, objectName, objectNameDest)
  1107. c.Assert(err, NotNil)
  1108. // bucket no exist
  1109. _, err = s.bucket.CopyObjectFrom(destBucket, objectNameDest, objectName)
  1110. c.Assert(err, NotNil)
  1111. }
  1112. // TestAppendObject
  1113. func (s *OssBucketSuite) TestAppendObject(c *C) {
  1114. objectName := objectNamePrefix + "tao"
  1115. objectValue := "昨夜雨疏风骤,浓睡不消残酒。试问卷帘人,却道海棠依旧。知否?知否?应是绿肥红瘦。"
  1116. var val = []byte(objectValue)
  1117. var localFile = "testx.txt"
  1118. var nextPos int64
  1119. var midPos = 1 + rand.Intn(len(val)-1)
  1120. var err = createFileAndWrite(localFile+"1", val[0:midPos])
  1121. c.Assert(err, IsNil)
  1122. err = createFileAndWrite(localFile+"2", val[midPos:])
  1123. c.Assert(err, IsNil)
  1124. // string append
  1125. nextPos, err = s.bucket.AppendObject(objectName, strings.NewReader("昨夜雨疏风骤,浓睡不消残酒。试问卷帘人,"), nextPos)
  1126. c.Assert(err, IsNil)
  1127. nextPos, err = s.bucket.AppendObject(objectName, strings.NewReader("却道海棠依旧。知否?知否?应是绿肥红瘦。"), nextPos)
  1128. c.Assert(err, IsNil)
  1129. body, err := s.bucket.GetObject(objectName)
  1130. c.Assert(err, IsNil)
  1131. str, err := readBody(body)
  1132. c.Assert(err, IsNil)
  1133. c.Assert(str, Equals, objectValue)
  1134. err = s.bucket.DeleteObject(objectName)
  1135. c.Assert(err, IsNil)
  1136. // byte append
  1137. nextPos = 0
  1138. nextPos, err = s.bucket.AppendObject(objectName, bytes.NewReader(val[0:midPos]), nextPos)
  1139. c.Assert(err, IsNil)
  1140. nextPos, err = s.bucket.AppendObject(objectName, bytes.NewReader(val[midPos:]), nextPos)
  1141. c.Assert(err, IsNil)
  1142. body, err = s.bucket.GetObject(objectName)
  1143. c.Assert(err, IsNil)
  1144. str, err = readBody(body)
  1145. c.Assert(err, IsNil)
  1146. c.Assert(str, Equals, objectValue)
  1147. err = s.bucket.DeleteObject(objectName)
  1148. c.Assert(err, IsNil)
  1149. // file append
  1150. options := []Option{
  1151. ObjectACL(ACLPublicReadWrite),
  1152. Meta("my", "myprop"),
  1153. }
  1154. fd, err := os.Open(localFile + "1")
  1155. c.Assert(err, IsNil)
  1156. defer fd.Close()
  1157. nextPos = 0
  1158. nextPos, err = s.bucket.AppendObject(objectName, fd, nextPos, options...)
  1159. c.Assert(err, IsNil)
  1160. meta, err := s.bucket.GetObjectDetailedMeta(objectName)
  1161. c.Assert(err, IsNil)
  1162. testLogger.Println("GetObjectDetailedMeta:", meta, ",", nextPos)
  1163. c.Assert(meta.Get("X-Oss-Object-Type"), Equals, "Appendable")
  1164. c.Assert(meta.Get("X-Oss-Meta-My"), Equals, "myprop")
  1165. c.Assert(meta.Get("x-oss-Meta-Mine"), Equals, "")
  1166. c.Assert(meta.Get("X-Oss-Next-Append-Position"), Equals, strconv.FormatInt(nextPos, 10))
  1167. acl, err := s.bucket.GetObjectACL(objectName)
  1168. c.Assert(err, IsNil)
  1169. testLogger.Println("GetObjectACL:", acl)
  1170. c.Assert(acl.ACL, Equals, string(ACLPublicReadWrite))
  1171. // second append
  1172. options = []Option{
  1173. ObjectACL(ACLPublicRead),
  1174. Meta("my", "myproptwo"),
  1175. Meta("mine", "mypropmine"),
  1176. }
  1177. fd, err = os.Open(localFile + "2")
  1178. c.Assert(err, IsNil)
  1179. defer fd.Close()
  1180. nextPos, err = s.bucket.AppendObject(objectName, fd, nextPos, options...)
  1181. c.Assert(err, IsNil)
  1182. body, err = s.bucket.GetObject(objectName)
  1183. c.Assert(err, IsNil)
  1184. str, err = readBody(body)
  1185. c.Assert(err, IsNil)
  1186. c.Assert(str, Equals, objectValue)
  1187. meta, err = s.bucket.GetObjectDetailedMeta(objectName)
  1188. c.Assert(err, IsNil)
  1189. testLogger.Println("GetObjectDetailedMeta xxx:", meta)
  1190. c.Assert(meta.Get("X-Oss-Object-Type"), Equals, "Appendable")
  1191. c.Assert(meta.Get("X-Oss-Meta-My"), Equals, "myprop")
  1192. c.Assert(meta.Get("x-Oss-Meta-Mine"), Equals, "")
  1193. c.Assert(meta.Get("X-Oss-Next-Append-Position"), Equals, strconv.FormatInt(nextPos, 10))
  1194. acl, err = s.bucket.GetObjectACL(objectName)
  1195. c.Assert(err, IsNil)
  1196. c.Assert(acl.ACL, Equals, string(ACLPublicRead))
  1197. err = s.bucket.DeleteObject(objectName)
  1198. c.Assert(err, IsNil)
  1199. }
  1200. // TestAppendObjectNegative
  1201. func (s *OssBucketSuite) TestAppendObjectNegative(c *C) {
  1202. objectName := objectNamePrefix + "taon"
  1203. nextPos := int64(0)
  1204. nextPos, err := s.bucket.AppendObject(objectName, strings.NewReader("ObjectValue"), nextPos)
  1205. c.Assert(err, IsNil)
  1206. nextPos, err = s.bucket.AppendObject(objectName, strings.NewReader("ObjectValue"), 0)
  1207. c.Assert(err, NotNil)
  1208. err = s.bucket.DeleteObject(objectName)
  1209. c.Assert(err, IsNil)
  1210. }
  1211. // TestContentType
  1212. func (s *OssBucketSuite) TestAddContentType(c *C) {
  1213. opts := addContentType(nil, "abc.txt")
  1214. typ, err := findOption(opts, HTTPHeaderContentType, "")
  1215. c.Assert(err, IsNil)
  1216. c.Assert(typ, Equals, "text/plain; charset=utf-8")
  1217. opts = addContentType(nil)
  1218. typ, err = findOption(opts, HTTPHeaderContentType, "")
  1219. c.Assert(err, IsNil)
  1220. c.Assert(len(opts), Equals, 1)
  1221. c.Assert(typ, Equals, "application/octet-stream")
  1222. opts = addContentType(nil, "abc.txt", "abc.pdf")
  1223. typ, err = findOption(opts, HTTPHeaderContentType, "")
  1224. c.Assert(err, IsNil)
  1225. c.Assert(typ, Equals, "text/plain; charset=utf-8")
  1226. opts = addContentType(nil, "abc", "abc.txt", "abc.pdf")
  1227. typ, err = findOption(opts, HTTPHeaderContentType, "")
  1228. c.Assert(err, IsNil)
  1229. c.Assert(typ, Equals, "text/plain; charset=utf-8")
  1230. opts = addContentType(nil, "abc", "abc", "edf")
  1231. typ, err = findOption(opts, HTTPHeaderContentType, "")
  1232. c.Assert(err, IsNil)
  1233. c.Assert(typ, Equals, "application/octet-stream")
  1234. opts = addContentType([]Option{Meta("meta", "my")}, "abc", "abc.txt", "abc.pdf")
  1235. typ, err = findOption(opts, HTTPHeaderContentType, "")
  1236. c.Assert(err, IsNil)
  1237. c.Assert(len(opts), Equals, 2)
  1238. c.Assert(typ, Equals, "text/plain; charset=utf-8")
  1239. }
  1240. func (s *OssBucketSuite) TestGetConfig(c *C) {
  1241. client, err := New(endpoint, accessID, accessKey, UseCname(true),
  1242. Timeout(11, 12), SecurityToken("token"), EnableMD5(false))
  1243. c.Assert(err, IsNil)
  1244. bucket, err := client.Bucket(bucketName)
  1245. c.Assert(err, IsNil)
  1246. c.Assert(bucket.getConfig().HTTPTimeout.ConnectTimeout, Equals, time.Second*11)
  1247. c.Assert(bucket.getConfig().HTTPTimeout.ReadWriteTimeout, Equals, time.Second*12)
  1248. c.Assert(bucket.getConfig().HTTPTimeout.HeaderTimeout, Equals, time.Second*12)
  1249. c.Assert(bucket.getConfig().HTTPTimeout.LongTimeout, Equals, time.Second*12*10)
  1250. c.Assert(bucket.getConfig().SecurityToken, Equals, "token")
  1251. c.Assert(bucket.getConfig().IsCname, Equals, true)
  1252. c.Assert(bucket.getConfig().IsEnableMD5, Equals, false)
  1253. }
  1254. // TestSTSTonek
  1255. func (s *OssBucketSuite) _TestSTSTonek(c *C) {
  1256. objectName := objectNamePrefix + "tst"
  1257. objectValue := "红藕香残玉簟秋。轻解罗裳,独上兰舟。云中谁寄锦书来?雁字回时,月满西楼。"
  1258. stsServer := ""
  1259. stsEndpoint := ""
  1260. stsBucketName := ""
  1261. stsRes, err := getSTSToken(stsServer)
  1262. c.Assert(err, IsNil)
  1263. testLogger.Println("sts:", stsRes)
  1264. client, err := New(stsEndpoint, stsRes.AccessID, stsRes.AccessKey,
  1265. SecurityToken(stsRes.SecurityToken))
  1266. c.Assert(err, IsNil)
  1267. bucket, err := client.Bucket(stsBucketName)
  1268. c.Assert(err, IsNil)
  1269. // Put
  1270. err = bucket.PutObject(objectName, strings.NewReader(objectValue))
  1271. c.Assert(err, IsNil)
  1272. // Get
  1273. body, err := s.bucket.GetObject(objectName)
  1274. c.Assert(err, IsNil)
  1275. str, err := readBody(body)
  1276. c.Assert(err, IsNil)
  1277. c.Assert(str, Equals, objectValue)
  1278. // List
  1279. lor, err := bucket.ListObjects()
  1280. c.Assert(err, IsNil)
  1281. testLogger.Println("Objects:", lor.Objects)
  1282. // Delete
  1283. err = bucket.DeleteObject(objectName)
  1284. c.Assert(err, IsNil)
  1285. }
  1286. func (s *OssBucketSuite) TestSTSTonekNegative(c *C) {
  1287. objectName := objectNamePrefix + "tstg"
  1288. localFile := objectName + ".jpg"
  1289. client, err := New(endpoint, accessID, accessKey, SecurityToken("Invalid"))
  1290. c.Assert(err, IsNil)
  1291. _, err = client.ListBuckets()
  1292. c.Assert(err, NotNil)
  1293. bucket, err := client.Bucket(bucketName)
  1294. c.Assert(err, IsNil)
  1295. err = bucket.PutObject(objectName, strings.NewReader(""))
  1296. c.Assert(err, NotNil)
  1297. err = bucket.PutObjectFromFile(objectName, "")
  1298. c.Assert(err, NotNil)
  1299. _, err = bucket.GetObject(objectName)
  1300. c.Assert(err, NotNil)
  1301. err = bucket.GetObjectToFile(objectName, "")
  1302. c.Assert(err, NotNil)
  1303. _, err = bucket.ListObjects()
  1304. c.Assert(err, NotNil)
  1305. err = bucket.SetObjectACL(objectName, ACLPublicRead)
  1306. c.Assert(err, NotNil)
  1307. _, err = bucket.GetObjectACL(objectName)
  1308. c.Assert(err, NotNil)
  1309. err = bucket.UploadFile(objectName, localFile, MinPartSize)
  1310. c.Assert(err, NotNil)
  1311. err = bucket.DownloadFile(objectName, localFile, MinPartSize)
  1312. c.Assert(err, NotNil)
  1313. _, err = bucket.IsObjectExist(objectName)
  1314. c.Assert(err, NotNil)
  1315. _, err = bucket.ListMultipartUploads()
  1316. c.Assert(err, NotNil)
  1317. err = bucket.DeleteObject(objectName)
  1318. c.Assert(err, NotNil)
  1319. _, err = bucket.DeleteObjects([]string{objectName})
  1320. c.Assert(err, NotNil)
  1321. err = client.DeleteBucket(bucketName)
  1322. c.Assert(err, NotNil)
  1323. _, err = getSTSToken("")
  1324. c.Assert(err, NotNil)
  1325. _, err = getSTSToken("http://me.php")
  1326. c.Assert(err, NotNil)
  1327. }
  1328. func (s *OssBucketSuite) TestUploadBigFile(c *C) {
  1329. objectName := objectNamePrefix + "tubf"
  1330. bigFile := "D:\\tmp\\bigfile.zip"
  1331. newFile := "D:\\tmp\\newbigfile.zip"
  1332. exist, err := isFileExist(bigFile)
  1333. c.Assert(err, IsNil)
  1334. if !exist {
  1335. return
  1336. }
  1337. // Put
  1338. start := GetNowSec()
  1339. err = s.bucket.PutObjectFromFile(objectName, bigFile)
  1340. c.Assert(err, IsNil)
  1341. end := GetNowSec()
  1342. testLogger.Println("Put big file:", bigFile, "use sec:", end-start)
  1343. // Check
  1344. start = GetNowSec()
  1345. err = s.bucket.GetObjectToFile(objectName, newFile)
  1346. c.Assert(err, IsNil)
  1347. end = GetNowSec()
  1348. testLogger.Println("Get big file:", bigFile, "use sec:", end-start)
  1349. start = GetNowSec()
  1350. eq, err := compareFiles(bigFile, newFile)
  1351. c.Assert(err, IsNil)
  1352. c.Assert(eq, Equals, true)
  1353. end = GetNowSec()
  1354. testLogger.Println("Compare big file:", bigFile, "use sec:", end-start)
  1355. err = s.bucket.DeleteObject(objectName)
  1356. c.Assert(err, IsNil)
  1357. }
  1358. func (s *OssBucketSuite) TestSymlink(c *C) {
  1359. objectName := objectNamePrefix + "符号链接"
  1360. targetObjectName := objectNamePrefix + "符号链接目标文件"
  1361. err := s.bucket.DeleteObject(objectName)
  1362. c.Assert(err, IsNil)
  1363. err = s.bucket.DeleteObject(targetObjectName)
  1364. c.Assert(err, IsNil)
  1365. meta, err := s.bucket.GetSymlink(objectName)
  1366. c.Assert(err, NotNil)
  1367. // Put symlink
  1368. err = s.bucket.PutSymlink(objectName, targetObjectName)
  1369. c.Assert(err, IsNil)
  1370. err = s.bucket.PutObject(targetObjectName, strings.NewReader("target"))
  1371. c.Assert(err, IsNil)
  1372. err = s.bucket.PutSymlink(objectName, targetObjectName)
  1373. c.Assert(err, IsNil)
  1374. meta, err = s.bucket.GetSymlink(objectName)
  1375. c.Assert(err, IsNil)
  1376. c.Assert(meta.Get(HTTPHeaderOssSymlinkTarget), Equals, targetObjectName)
  1377. // List object
  1378. lor, err := s.bucket.ListObjects()
  1379. c.Assert(err, IsNil)
  1380. exist, v := s.getObject(lor.Objects, objectName)
  1381. c.Assert(exist, Equals, true)
  1382. c.Assert(v.Type, Equals, "Symlink")
  1383. body, err := s.bucket.GetObject(objectName)
  1384. c.Assert(err, IsNil)
  1385. str, err := readBody(body)
  1386. c.Assert(err, IsNil)
  1387. c.Assert(str, Equals, "target")
  1388. meta, err = s.bucket.GetSymlink(targetObjectName)
  1389. c.Assert(err, NotNil)
  1390. err = s.bucket.PutObject(objectName, strings.NewReader("src"))
  1391. c.Assert(err, IsNil)
  1392. body, err = s.bucket.GetObject(objectName)
  1393. c.Assert(err, IsNil)
  1394. str, err = readBody(body)
  1395. c.Assert(err, IsNil)
  1396. c.Assert(str, Equals, "src")
  1397. err = s.bucket.DeleteObject(objectName)
  1398. c.Assert(err, IsNil)
  1399. err = s.bucket.DeleteObject(targetObjectName)
  1400. c.Assert(err, IsNil)
  1401. // put symlink again
  1402. objectName = objectNamePrefix + "symlink"
  1403. targetObjectName = objectNamePrefix + "symlink-target"
  1404. err = s.bucket.PutSymlink(objectName, targetObjectName)
  1405. c.Assert(err, IsNil)
  1406. err = s.bucket.PutObject(targetObjectName, strings.NewReader("target1"))
  1407. c.Assert(err, IsNil)
  1408. meta, err = s.bucket.GetSymlink(objectName)
  1409. c.Assert(err, IsNil)
  1410. c.Assert(meta.Get(HTTPHeaderOssSymlinkTarget), Equals, targetObjectName)
  1411. body, err = s.bucket.GetObject(objectName)
  1412. c.Assert(err, IsNil)
  1413. str, err = readBody(body)
  1414. c.Assert(err, IsNil)
  1415. c.Assert(str, Equals, "target1")
  1416. err = s.bucket.DeleteObject(objectName)
  1417. c.Assert(err, IsNil)
  1418. err = s.bucket.DeleteObject(targetObjectName)
  1419. c.Assert(err, IsNil)
  1420. }
  1421. // TestRestoreObject
  1422. func (s *OssBucketSuite) TestRestoreObject(c *C) {
  1423. objectName := objectNamePrefix + "restore"
  1424. // List Object
  1425. lor, err := s.archiveBucket.ListObjects()
  1426. c.Assert(err, IsNil)
  1427. left := len(lor.Objects)
  1428. // Put three object
  1429. err = s.archiveBucket.PutObject(objectName, strings.NewReader(""))
  1430. c.Assert(err, IsNil)
  1431. // List
  1432. lor, err = s.archiveBucket.ListObjects()
  1433. c.Assert(err, IsNil)
  1434. c.Assert(len(lor.Objects), Equals, left+1)
  1435. for _, object := range lor.Objects {
  1436. c.Assert(object.StorageClass, Equals, string(StorageArchive))
  1437. c.Assert(object.Type, Equals, "Normal")
  1438. }
  1439. // Head Object
  1440. meta, err := s.archiveBucket.GetObjectDetailedMeta(objectName)
  1441. c.Assert(err, IsNil)
  1442. _, ok := meta["X-Oss-Restore"]
  1443. c.Assert(ok, Equals, false)
  1444. c.Assert(meta.Get("X-Oss-Storage-Class"), Equals, "Archive")
  1445. // Error Restore
  1446. err = s.archiveBucket.RestoreObject("notexistobject")
  1447. c.Assert(err, NotNil)
  1448. // Restore Object
  1449. err = s.archiveBucket.RestoreObject(objectName)
  1450. c.Assert(err, IsNil)
  1451. // Head Object
  1452. meta, err = s.archiveBucket.GetObjectDetailedMeta(objectName)
  1453. c.Assert(err, IsNil)
  1454. c.Assert(meta.Get("X-Oss-Restore"), Equals, "ongoing-request=\"true\"")
  1455. c.Assert(meta.Get("X-Oss-Storage-Class"), Equals, "Archive")
  1456. }
  1457. // private
  1458. func createFileAndWrite(fileName string, data []byte) error {
  1459. os.Remove(fileName)
  1460. fo, err := os.Create(fileName)
  1461. if err != nil {
  1462. return err
  1463. }
  1464. defer fo.Close()
  1465. bytes, err := fo.Write(data)
  1466. if err != nil {
  1467. return err
  1468. }
  1469. if bytes != len(data) {
  1470. return fmt.Errorf(fmt.Sprintf("write %d bytes not equal data length %d", bytes, len(data)))
  1471. }
  1472. return nil
  1473. }
  1474. // compare the content between fileL and fileR
  1475. func compareFiles(fileL string, fileR string) (bool, error) {
  1476. finL, err := os.Open(fileL)
  1477. if err != nil {
  1478. return false, err
  1479. }
  1480. defer finL.Close()
  1481. finR, err := os.Open(fileR)
  1482. if err != nil {
  1483. return false, err
  1484. }
  1485. defer finR.Close()
  1486. statL, err := finL.Stat()
  1487. if err != nil {
  1488. return false, err
  1489. }
  1490. statR, err := finR.Stat()
  1491. if err != nil {
  1492. return false, err
  1493. }
  1494. if statL.Size() != statR.Size() {
  1495. return false, nil
  1496. }
  1497. size := statL.Size()
  1498. if size > 102400 {
  1499. size = 102400
  1500. }
  1501. bufL := make([]byte, size)
  1502. bufR := make([]byte, size)
  1503. for {
  1504. n, _ := finL.Read(bufL)
  1505. if 0 == n {
  1506. break
  1507. }
  1508. n, _ = finR.Read(bufR)
  1509. if 0 == n {
  1510. break
  1511. }
  1512. if !bytes.Equal(bufL, bufR) {
  1513. return false, nil
  1514. }
  1515. }
  1516. return true, nil
  1517. }
  1518. // compare the content of file and data
  1519. func compareFileData(file string, data []byte) (bool, error) {
  1520. fin, err := os.Open(file)
  1521. if err != nil {
  1522. return false, err
  1523. }
  1524. defer fin.Close()
  1525. stat, err := fin.Stat()
  1526. if err != nil {
  1527. return false, err
  1528. }
  1529. if stat.Size() != (int64)(len(data)) {
  1530. return false, nil
  1531. }
  1532. buf := make([]byte, stat.Size())
  1533. n, err := fin.Read(buf)
  1534. if err != nil {
  1535. return false, err
  1536. }
  1537. if stat.Size() != (int64)(n) {
  1538. return false, errors.New("read error")
  1539. }
  1540. if !bytes.Equal(buf, data) {
  1541. return false, nil
  1542. }
  1543. return true, nil
  1544. }
  1545. func walkDir(dirPth, suffix string) ([]string, error) {
  1546. var files = []string{}
  1547. suffix = strings.ToUpper(suffix)
  1548. err := filepath.Walk(dirPth,
  1549. func(filename string, fi os.FileInfo, err error) error {
  1550. if err != nil {
  1551. return err
  1552. }
  1553. if fi.IsDir() {
  1554. return nil
  1555. }
  1556. if strings.HasSuffix(strings.ToUpper(fi.Name()), suffix) {
  1557. files = append(files, filename)
  1558. }
  1559. return nil
  1560. })
  1561. return files, err
  1562. }
  1563. func removeTempFiles(path string, prefix string) error {
  1564. files, err := walkDir(path, prefix)
  1565. if err != nil {
  1566. return nil
  1567. }
  1568. for _, file := range files {
  1569. os.Remove(file)
  1570. }
  1571. return nil
  1572. }
  1573. func isFileExist(filename string) (bool, error) {
  1574. _, err := os.Stat(filename)
  1575. if err != nil && os.IsNotExist(err) {
  1576. return false, nil
  1577. } else if err != nil {
  1578. return false, err
  1579. } else {
  1580. return true, nil
  1581. }
  1582. }
  1583. // STS Server的GET请求返回的数据
  1584. type getSTSResult struct {
  1585. Status int `json:"status"` // 返回状态码, 200表示获取成功,非200表示失败
  1586. AccessID string `json:"accessId"` //STS AccessId
  1587. AccessKey string `json:"accessKey"` // STS AccessKey
  1588. Expiration string `json:"expiration"` // STS Token
  1589. SecurityToken string `json:"securityToken"` // Token失效的时间, GMT时间
  1590. Bucket string `json:"bucket"` // 可以使用的bucket
  1591. Endpoint string `json:"bucket"` // 要访问的endpoint
  1592. }
  1593. // 从STS Server获取STS信息。返回值中当error为nil时,GetSTSResult有效。
  1594. func getSTSToken(STSServer string) (getSTSResult, error) {
  1595. result := getSTSResult{}
  1596. resp, err := http.Get(STSServer)
  1597. if err != nil {
  1598. return result, err
  1599. }
  1600. defer resp.Body.Close()
  1601. body, err := ioutil.ReadAll(resp.Body)
  1602. if err != nil {
  1603. return result, err
  1604. }
  1605. err = json.Unmarshal(body, &result)
  1606. if err != nil {
  1607. return result, err
  1608. }
  1609. if result.Status != 200 {
  1610. return result, errors.New("Server Return Status:" + strconv.Itoa(result.Status))
  1611. }
  1612. return result, nil
  1613. }
  1614. func readBody(body io.ReadCloser) (string, error) {
  1615. data, err := ioutil.ReadAll(body)
  1616. body.Close()
  1617. if err != nil {
  1618. return "", err
  1619. }
  1620. return string(data), nil
  1621. }
  1622. func (s *OssBucketSuite) getObject(objects []ObjectProperties, object string) (bool, ObjectProperties) {
  1623. for _, v := range objects {
  1624. if v.Key == object {
  1625. return true, v
  1626. }
  1627. }
  1628. return false, ObjectProperties{}
  1629. }