vendor/github.com/aliyun/aliyun-oss-go-sdk/oss/conn_test.go

   1 package oss
   2
   3 import (
   4         "net/http"
   5         "os"
   6         "strings"
   7         "time"
   8
   9         . "gopkg.in/check.v1"
  10 )
  11
  12 type OssConnSuite struct{}
  13
  14 var _ = Suite(&OssConnSuite{})
  15
  16 func (s *OssConnSuite) TestURLMarker(c *C) {
  17         um := urlMaker{}
  18         um.Init("docs.github.com", true, false)
  19         c.Assert(um.Type, Equals, urlTypeCname)
  20         c.Assert(um.Scheme, Equals, "http")
  21         c.Assert(um.NetLoc, Equals, "docs.github.com")
  22
  23         c.Assert(um.getURL("bucket", "object", "params").String(), Equals, "http://docs.github.com/object?params")
  24         c.Assert(um.getURL("bucket", "object", "").String(), Equals, "http://docs.github.com/object")
  25         c.Assert(um.getURL("", "object", "").String(), Equals, "http://docs.github.com/object")
  26
  27         var conn Conn
  28         conn.config = getDefaultOssConfig()
  29         conn.config.AuthVersion = AuthV1
  30         c.Assert(conn.getResource("bucket", "object", "subres"), Equals, "/bucket/object?subres")
  31         c.Assert(conn.getResource("bucket", "object", ""), Equals, "/bucket/object")
  32         c.Assert(conn.getResource("", "object", ""), Equals, "/")
  33
  34         um.Init("https://docs.github.com", true, false)
  35         c.Assert(um.Type, Equals, urlTypeCname)
  36         c.Assert(um.Scheme, Equals, "https")
  37         c.Assert(um.NetLoc, Equals, "docs.github.com")
  38
  39         um.Init("http://docs.github.com", true, false)
  40         c.Assert(um.Type, Equals, urlTypeCname)
  41         c.Assert(um.Scheme, Equals, "http")
  42         c.Assert(um.NetLoc, Equals, "docs.github.com")
  43
  44         um.Init("docs.github.com:8080", false, true)
  45         c.Assert(um.Type, Equals, urlTypeAliyun)
  46         c.Assert(um.Scheme, Equals, "http")
  47         c.Assert(um.NetLoc, Equals, "docs.github.com:8080")
  48
  49         c.Assert(um.getURL("bucket", "object", "params").String(), Equals, "http://bucket.docs.github.com:8080/object?params")
  50         c.Assert(um.getURL("bucket", "object", "").String(), Equals, "http://bucket.docs.github.com:8080/object")
  51         c.Assert(um.getURL("", "object", "").String(), Equals, "http://docs.github.com:8080/")
  52         c.Assert(conn.getResource("bucket", "object", "subres"), Equals, "/bucket/object?subres")
  53         c.Assert(conn.getResource("bucket", "object", ""), Equals, "/bucket/object")
  54         c.Assert(conn.getResource("", "object", ""), Equals, "/")
  55
  56         um.Init("https://docs.github.com:8080", false, true)
  57         c.Assert(um.Type, Equals, urlTypeAliyun)
  58         c.Assert(um.Scheme, Equals, "https")
  59         c.Assert(um.NetLoc, Equals, "docs.github.com:8080")
  60
  61         um.Init("127.0.0.1", false, true)
  62         c.Assert(um.Type, Equals, urlTypeIP)
  63         c.Assert(um.Scheme, Equals, "http")
  64         c.Assert(um.NetLoc, Equals, "127.0.0.1")
  65
  66         um.Init("http://127.0.0.1", false, false)
  67         c.Assert(um.Type, Equals, urlTypeIP)
  68         c.Assert(um.Scheme, Equals, "http")
  69         c.Assert(um.NetLoc, Equals, "127.0.0.1")
  70         c.Assert(um.getURL("bucket", "object", "params").String(), Equals, "http://127.0.0.1/bucket/object?params")
  71         c.Assert(um.getURL("", "object", "params").String(), Equals, "http://127.0.0.1/?params")
  72
  73         um.Init("https://127.0.0.1:8080", false, false)
  74         c.Assert(um.Type, Equals, urlTypeIP)
  75         c.Assert(um.Scheme, Equals, "https")
  76         c.Assert(um.NetLoc, Equals, "127.0.0.1:8080")
  77
  78         um.Init("http://[2401:b180::dc]", false, false)
  79         c.Assert(um.Type, Equals, urlTypeIP)
  80         c.Assert(um.Scheme, Equals, "http")
  81         c.Assert(um.NetLoc, Equals, "[2401:b180::dc]")
  82
  83         um.Init("https://[2401:b180::dc]:8080", false, false)
  84         c.Assert(um.Type, Equals, urlTypeIP)
  85         c.Assert(um.Scheme, Equals, "https")
  86         c.Assert(um.NetLoc, Equals, "[2401:b180::dc]:8080")
  87 }
  88
  89 func (s *OssConnSuite) TestAuth(c *C) {
  90         endpoint := "https://github.com/"
  91         cfg := getDefaultOssConfig()
  92         cfg.AuthVersion = AuthV1
  93         um := urlMaker{}
  94         um.Init(endpoint, false, false)
  95         conn := Conn{cfg, &um, nil}
  96         uri := um.getURL("bucket", "object", "")
  97         req := &http.Request{
  98                 Method:     "PUT",
  99                 URL:        uri,
 100                 Proto:      "HTTP/1.1",
 101                 ProtoMajor: 1,
 102                 ProtoMinor: 1,
 103                 Header:     make(http.Header),
 104                 Host:       uri.Host,
 105         }
 106
 107         req.Header.Set("Content-Type", "text/html")
 108         req.Header.Set("Date", "Thu, 17 Nov 2005 18:49:58 GMT")
 109         req.Header.Set("Host", endpoint)
 110         req.Header.Set("X-OSS-Meta-Your", "your")
 111         req.Header.Set("X-OSS-Meta-Author", "foo@bar.com")
 112         req.Header.Set("X-OSS-Magic", "abracadabra")
 113         req.Header.Set("Content-Md5", "ODBGOERFMDMzQTczRUY3NUE3NzA5QzdFNUYzMDQxNEM=")
 114
 115         conn.signHeader(req, conn.getResource("bucket", "object", ""))
 116         testLogger.Println("AUTHORIZATION:", req.Header.Get(HTTPHeaderAuthorization))
 117 }
 118
 119 func (s *OssConnSuite) TestConnToolFunc(c *C) {
 120         err := CheckRespCode(202, []int{})
 121         c.Assert(err, NotNil)
 122
 123         err = CheckRespCode(202, []int{404})
 124         c.Assert(err, NotNil)
 125
 126         err = CheckRespCode(202, []int{202, 404})
 127         c.Assert(err, IsNil)
 128
 129         srvErr, err := serviceErrFromXML([]byte(""), 312, "")
 130         c.Assert(err, NotNil)
 131         c.Assert(srvErr.StatusCode, Equals, 0)
 132
 133         srvErr, err = serviceErrFromXML([]byte("ABC"), 312, "")
 134         c.Assert(err, NotNil)
 135         c.Assert(srvErr.StatusCode, Equals, 0)
 136
 137         srvErr, err = serviceErrFromXML([]byte("<Error></Error>"), 312, "")
 138         c.Assert(err, IsNil)
 139         c.Assert(srvErr.StatusCode, Equals, 312)
 140
 141         unexpect := UnexpectedStatusCodeError{[]int{200}, 202}
 142         c.Assert(len(unexpect.Error()) > 0, Equals, true)
 143         c.Assert(unexpect.Got(), Equals, 202)
 144
 145         fd, err := os.Open("../sample/BingWallpaper-2015-11-07.jpg")
 146         c.Assert(err, IsNil)
 147         fd.Close()
 148         var out ProcessObjectResult
 149         err = jsonUnmarshal(fd, &out)
 150         c.Assert(err, NotNil)
 151 }
 152
 153 func (s *OssConnSuite) TestSignRtmpURL(c *C) {
 154         cfg := getDefaultOssConfig()
 155
 156         um := urlMaker{}
 157         um.Init(endpoint, false, false)
 158         conn := Conn{cfg, &um, nil}
 159
 160         //Anonymous
 161         channelName := "test-sign-rtmp-url"
 162         playlistName := "playlist.m3u8"
 163         expiration := time.Now().Unix() + 3600
 164         signedRtmpURL := conn.signRtmpURL(bucketName, channelName, playlistName, expiration)
 165         playURL := getPublishURL(bucketName, channelName)
 166         hasPrefix := strings.HasPrefix(signedRtmpURL, playURL)
 167         c.Assert(hasPrefix, Equals, true)
 168
 169         //empty playlist name
 170         playlistName = ""
 171         signedRtmpURL = conn.signRtmpURL(bucketName, channelName, playlistName, expiration)
 172         playURL = getPublishURL(bucketName, channelName)
 173         hasPrefix = strings.HasPrefix(signedRtmpURL, playURL)
 174         c.Assert(hasPrefix, Equals, true)
 175 }
 176
 177 func (s *OssConnSuite) TestGetRtmpSignedStr(c *C) {
 178         cfg := getDefaultOssConfig()
 179         um := urlMaker{}
 180         um.Init(endpoint, false, false)
 181         conn := Conn{cfg, &um, nil}
 182
 183         akIf := conn.config.GetCredentials()
 184
 185         //Anonymous
 186         channelName := "test-get-rtmp-signed-str"
 187         playlistName := "playlist.m3u8"
 188         expiration := time.Now().Unix() + 3600
 189         params := map[string]interface{}{}
 190         signedStr := conn.getRtmpSignedStr(bucketName, channelName, playlistName, expiration, akIf.GetAccessKeySecret(), params)
 191         c.Assert(signedStr, Equals, "")
 192 }