bucket_test.go 64 KB

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