--- /dev/null
+package oss
+
+import (
+ "strings"
+
+ . "gopkg.in/check.v1"
+)
+
+type OssUtilsSuite struct{}
+
+var _ = Suite(&OssUtilsSuite{})
+
+func (s *OssUtilsSuite) TestUtilsTime(c *C) {
+ c.Assert(GetNowSec() > 1448597674, Equals, true)
+ c.Assert(GetNowNanoSec() > 1448597674000000000, Equals, true)
+ c.Assert(len(GetNowGMT()), Equals, len("Fri, 27 Nov 2015 04:14:34 GMT"))
+}
+
+func (s *OssUtilsSuite) TestUtilsSplitFile(c *C) {
+ localFile := "../sample/BingWallpaper-2015-11-07.jpg"
+
+ // Num
+ parts, err := SplitFileByPartNum(localFile, 4)
+ c.Assert(err, IsNil)
+ c.Assert(len(parts), Equals, 4)
+ testLogger.Println("parts 4:", parts)
+ for i, part := range parts {
+ c.Assert(part.Number, Equals, i+1)
+ c.Assert(part.Offset, Equals, int64(i*120512))
+ c.Assert(part.Size, Equals, int64(120512))
+ }
+
+ parts, err = SplitFileByPartNum(localFile, 5)
+ c.Assert(err, IsNil)
+ c.Assert(len(parts), Equals, 5)
+ testLogger.Println("parts 5:", parts)
+ for i, part := range parts {
+ c.Assert(part.Number, Equals, i+1)
+ c.Assert(part.Offset, Equals, int64(i*96409))
+ }
+
+ _, err = SplitFileByPartNum(localFile, 10001)
+ c.Assert(err, NotNil)
+
+ _, err = SplitFileByPartNum(localFile, 0)
+ c.Assert(err, NotNil)
+
+ _, err = SplitFileByPartNum(localFile, -1)
+ c.Assert(err, NotNil)
+
+ _, err = SplitFileByPartNum("notexist", 1024)
+ c.Assert(err, NotNil)
+
+ // Size
+ parts, err = SplitFileByPartSize(localFile, 120512)
+ c.Assert(err, IsNil)
+ c.Assert(len(parts), Equals, 4)
+ testLogger.Println("parts 4:", parts)
+ for i, part := range parts {
+ c.Assert(part.Number, Equals, i+1)
+ c.Assert(part.Offset, Equals, int64(i*120512))
+ c.Assert(part.Size, Equals, int64(120512))
+ }
+
+ parts, err = SplitFileByPartSize(localFile, 96409)
+ c.Assert(err, IsNil)
+ c.Assert(len(parts), Equals, 6)
+ testLogger.Println("parts 6:", parts)
+ for i, part := range parts {
+ c.Assert(part.Number, Equals, i+1)
+ c.Assert(part.Offset, Equals, int64(i*96409))
+ }
+
+ _, err = SplitFileByPartSize(localFile, 0)
+ c.Assert(err, NotNil)
+
+ _, err = SplitFileByPartSize(localFile, -1)
+ c.Assert(err, NotNil)
+
+ _, err = SplitFileByPartSize(localFile, 10)
+ c.Assert(err, NotNil)
+
+ _, err = SplitFileByPartSize("noexist", 120512)
+ c.Assert(err, NotNil)
+}
+
+func (s *OssUtilsSuite) TestUtilsFileExt(c *C) {
+ c.Assert(strings.Contains(TypeByExtension("test.txt"), "text/plain"), Equals, true)
+ c.Assert(TypeByExtension("test.jpg"), Equals, "image/jpeg")
+ c.Assert(TypeByExtension("test.pdf"), Equals, "application/pdf")
+ c.Assert(TypeByExtension("test"), Equals, "")
+ c.Assert(strings.Contains(TypeByExtension("/root/dir/test.txt"), "text/plain"), Equals, true)
+ c.Assert(strings.Contains(TypeByExtension("root/dir/test.txt"), "text/plain"), Equals, true)
+ c.Assert(strings.Contains(TypeByExtension("root\\dir\\test.txt"), "text/plain"), Equals, true)
+ c.Assert(strings.Contains(TypeByExtension("D:\\work\\dir\\test.txt"), "text/plain"), Equals, true)
+}
+
+func (s *OssUtilsSuite) TestGetPartEnd(c *C) {
+ end := GetPartEnd(3, 10, 3)
+ c.Assert(end, Equals, int64(5))
+
+ end = GetPartEnd(9, 10, 3)
+ c.Assert(end, Equals, int64(9))
+
+ end = GetPartEnd(7, 10, 3)
+ c.Assert(end, Equals, int64(9))
+}
+
+func (s *OssUtilsSuite) TestParseRange(c *C) {
+ // InvalidRange bytes==M-N
+ _, err := ParseRange("bytes==M-N")
+ c.Assert(err, NotNil)
+ c.Assert(err.Error(), Equals, "InvalidRange bytes==M-N")
+
+ // InvalidRange ranges=M-N
+ _, err = ParseRange("ranges=M-N")
+ c.Assert(err, NotNil)
+ c.Assert(err.Error(), Equals, "InvalidRange ranges=M-N")
+
+ // InvalidRange ranges=M-N
+ _, err = ParseRange("bytes=M-N")
+ c.Assert(err, NotNil)
+ c.Assert(err.Error(), Equals, "InvalidRange bytes=M-N")
+
+ // InvalidRange ranges=M-
+ _, err = ParseRange("bytes=M-")
+ c.Assert(err, NotNil)
+ c.Assert(err.Error(), Equals, "InvalidRange bytes=M-")
+
+ // InvalidRange ranges=-N
+ _, err = ParseRange("bytes=-N")
+ c.Assert(err, NotNil)
+ c.Assert(err.Error(), Equals, "InvalidRange bytes=-N")
+
+ // InvalidRange ranges=-0
+ _, err = ParseRange("bytes=-0")
+ c.Assert(err, NotNil)
+ c.Assert(err.Error(), Equals, "InvalidRange bytes=-0")
+
+ // InvalidRange bytes=1-2-3
+ _, err = ParseRange("bytes=1-2-3")
+ c.Assert(err, NotNil)
+ c.Assert(err.Error(), Equals, "InvalidRange bytes=1-2-3")
+
+ // InvalidRange bytes=1-N
+ _, err = ParseRange("bytes=1-N")
+ c.Assert(err, NotNil)
+ c.Assert(err.Error(), Equals, "InvalidRange bytes=1-N")
+
+ // Ranges=M-N
+ ur, err := ParseRange("bytes=1024-4096")
+ c.Assert(err, IsNil)
+ c.Assert(ur.Start, Equals, (int64)(1024))
+ c.Assert(ur.End, Equals, (int64)(4096))
+ c.Assert(ur.HasStart, Equals, true)
+ c.Assert(ur.HasEnd, Equals, true)
+
+ // Ranges=M-N,X-Y
+ ur, err = ParseRange("bytes=1024-4096,2048-4096")
+ c.Assert(err, IsNil)
+ c.Assert(ur.Start, Equals, (int64)(1024))
+ c.Assert(ur.End, Equals, (int64)(4096))
+ c.Assert(ur.HasStart, Equals, true)
+ c.Assert(ur.HasEnd, Equals, true)
+
+ // Ranges=M-
+ ur, err = ParseRange("bytes=1024-")
+ c.Assert(err, IsNil)
+ c.Assert(ur.Start, Equals, (int64)(1024))
+ c.Assert(ur.End, Equals, (int64)(0))
+ c.Assert(ur.HasStart, Equals, true)
+ c.Assert(ur.HasEnd, Equals, false)
+
+ // Ranges=-N
+ ur, err = ParseRange("bytes=-4096")
+ c.Assert(err, IsNil)
+ c.Assert(ur.Start, Equals, (int64)(0))
+ c.Assert(ur.End, Equals, (int64)(4096))
+ c.Assert(ur.HasStart, Equals, false)
+ c.Assert(ur.HasEnd, Equals, true)
+}
+
+func (s *OssUtilsSuite) TestAdjustRange(c *C) {
+ // Nil
+ start, end := AdjustRange(nil, 8192)
+ c.Assert(start, Equals, (int64)(0))
+ c.Assert(end, Equals, (int64)(8192))
+
+ // 1024-4096
+ ur := &UnpackedRange{true, true, 1024, 4095}
+ start, end = AdjustRange(ur, 8192)
+ c.Assert(start, Equals, (int64)(1024))
+ c.Assert(end, Equals, (int64)(4096))
+
+ // 1024-
+ ur = &UnpackedRange{true, false, 1024, 4096}
+ start, end = AdjustRange(ur, 8192)
+ c.Assert(start, Equals, (int64)(1024))
+ c.Assert(end, Equals, (int64)(8192))
+
+ // -4096
+ ur = &UnpackedRange{false, true, 1024, 4096}
+ start, end = AdjustRange(ur, 8192)
+ c.Assert(start, Equals, (int64)(4096))
+ c.Assert(end, Equals, (int64)(8192))
+
+ // Invalid range 4096-1024
+ ur = &UnpackedRange{true, true, 4096, 1024}
+ start, end = AdjustRange(ur, 8192)
+ c.Assert(start, Equals, (int64)(0))
+ c.Assert(end, Equals, (int64)(8192))
+
+ // Invalid range -1-
+ ur = &UnpackedRange{true, false, -1, 0}
+ start, end = AdjustRange(ur, 8192)
+ c.Assert(start, Equals, (int64)(0))
+ c.Assert(end, Equals, (int64)(8192))
+
+ // Invalid range -9999
+ ur = &UnpackedRange{false, true, 0, 9999}
+ start, end = AdjustRange(ur, 8192)
+ c.Assert(start, Equals, (int64)(0))
+ c.Assert(end, Equals, (int64)(8192))
+}
+
+func (s *OssUtilsSuite) TestUtilCheckBucketName(c *C) {
+ err := CheckBucketName("a")
+ c.Assert(err, NotNil)
+
+ err = CheckBucketName("a11111111111111111111111111111nbbbbbbbbbbbbbbbbbbbbbbbbbbbqqqqqqqqqqqqqqqqqqqq")
+ c.Assert(err, NotNil)
+
+ err = CheckBucketName("-abcd")
+ c.Assert(err, NotNil)
+
+ err = CheckBucketName("abcd-")
+ c.Assert(err, NotNil)
+
+ err = CheckBucketName("abcD")
+ c.Assert(err, NotNil)
+
+ err = CheckBucketName("abc 1")
+ c.Assert(err, NotNil)
+
+ err = CheckBucketName("abc&1")
+ c.Assert(err, NotNil)
+
+ err = CheckBucketName("abc-1")
+ c.Assert(err, IsNil)
+
+ err = CheckBucketName("1bc-1")
+ c.Assert(err, IsNil)
+
+ err = CheckBucketName("111-1")
+ c.Assert(err, IsNil)
+
+ err = CheckBucketName("abc123-def1")
+ c.Assert(err, IsNil)
+}