Merge lp:~julian-edwards/gwacl/get-container-properties into lp:gwacl

[Julian Edwards]

Implement GetContainerProperties.

Implement GetContainerProperties - it's useful to use as a "tell me if this container exists or not" thing.

[Julian Edwards (julian-edwards)]

The response from this call is strangely all in response headers instead of XML in the body like most other calls, so I had to re-jig performRequest a bit so the headers are returned if you provide an non-nil http.Header in the request parameters.

It's annoying to have to copy the headers like it does, and could be avoided if we choose to turn the request parameter struct into a pointer so that arbitrary contained pointers could be returned.

[Julian Edwards (julian-edwards)]

On 31/07/13 14:03, Julian Edwards wrote:
> The response from this call is strangely all in response headers instead of XML in the body like most other calls, so I had to re-jig performRequest a bit so the headers are returned if you provide an non-nil http.Header in the request parameters.
>
> It's annoying to have to copy the headers like it does, and could be avoided if we choose to turn the request parameter struct into a pointer so that arbitrary contained pointers could be returned.
>

I caved and updated all the call sites to take an additional headers
returned.

[Jeroen T. Vermeulen (jtv)]

Something weird in the diff... I see the number of return values changing for performRequest, but only one of the "return" statements changes. Do you perchance have some more diff sitting on your machine?

[Julian Edwards (julian-edwards)]

On 31/07/13 16:06, Jeroen T. Vermeulen wrote:
> Something weird in the diff... I see the number of return values changing for performRequest, but only one of the "return" statements changes. Do you perchance have some more diff sitting on your machine?
>

No that's it, it all works!

[Jeroen T. Vermeulen (jtv)]

Ah, of course it just returned the result from a send() call. Basically the language supports tuples, it's just that it won't let us use them.

Anyway, good branch. The documentation comment for GetContainerProperties is in the imperative — python-style. In Go it should be in the third person, and start with the name of the function.

TestServerError could do with one small change: the c.Check(ok, Equals, true) should be an Assert, because the next check becomes meaningless if you continue — and you'll get a panic drowning out the root cause in the output.

review: Approve

[Julian Edwards (julian-edwards)]

On 31/07/13 16:18, Jeroen T. Vermeulen wrote:
> Review: Approve
>
> Ah, of course it just returned the result from a send() call. Basically the language supports tuples, it's just that it won't let us use them.
>
> Anyway, good branch. The documentation comment for GetContainerProperties is in the imperative — python-style. In Go it should be in the third person, and start with the name of the function.

It's the same for all the functions like that, I kept it consistent. If
we're going to change it, we should do them all at once.

> TestServerError could do with one small change: the c.Check(ok, Equals, true) should be an Assert, because the next check becomes meaningless if you continue — and you'll get a panic drowning out the root cause in the output.

Right, thanks!

=== modified file 'storage_base.go'
--- storage_base.go	2013-07-30 06:49:29 +0000
+++ storage_base.go	2013-07-31 06:34:25 +0000
@@ -259,12 +259,12 @@
 
 // performRequest issues an HTTP request to Azure.
 //
-// It returns the response body contents.
-func (context *StorageContext) performRequest(params requestParams) ([]byte, error) {
+// It returns the response body contents and the response headers.
+func (context *StorageContext) performRequest(params requestParams) ([]byte, http.Header, error) {
     params.Check()
     req, err := http.NewRequest(params.Method, params.URL, params.Body)
     if err != nil {
-        return nil, err
+        return nil, nil, err
     }
     // net/http has no way of adding headers en-masse, hence this abomination.
     for header, values := range params.ExtraHeaders {
@@ -285,37 +285,37 @@
 //
 // If the response's HTTP status code is not the same as "expectedStatus"
 // then an HTTPError will be returned as the error.
-func (context *StorageContext) send(req *http.Request, res Deserializer, expectedStatus HTTPStatus) ([]byte, error) {
+func (context *StorageContext) send(req *http.Request, res Deserializer, expectedStatus HTTPStatus) ([]byte, http.Header, error) {
     client := context.getClient()
     resp, err := client.Do(req)
     if err != nil {
-        return nil, err
+        return nil, nil, err
     }
 
     body, err := readAndClose(resp.Body)
     if err != nil {
-        return nil, fmt.Errorf("failed to read response data: %v", err)
+        return nil, nil, fmt.Errorf("failed to read response data: %v", err)
     }
 
     if resp.StatusCode != int(expectedStatus) {
         msg := newHTTPError(resp.StatusCode, body, "Azure request failed")
-        return body, msg
+        return body, resp.Header, msg
     }
 
     // If the caller didn't supply an object to deserialize the message into
     // then just return.
     if res == nil {
-        return body, nil
+        return body, resp.Header, nil
     }
 
     // TODO: Also deserialize response headers into the "res" object.
     err = res.Deserialize(body)
     if err != nil {
         msg := fmt.Errorf("Failed to deserialize data: %s", err)
-        return body, msg
+        return body, resp.Header, msg
     }
 
-    return body, nil
+    return body, resp.Header, nil
 }
 
 // getAccountURL returns the base URL for the context's storage account.
@@ -360,7 +360,7 @@
         uri = addURLQueryParams(uri, "marker", request.Marker)
     }
     containers := ContainerEnumerationResults{}
-    _, err := context.performRequest(requestParams{
+    _, _, err := context.performRequest(requestParams{
         Method:         "GET",
         URL:            uri,
         APIVersion:     "2012-02-12",
@@ -397,7 +397,7 @@
         uri = addURLQueryParams(uri, "prefix", request.Prefix)
     }
     blobs := BlobEnumerationResults{}
-    _, err := context.performRequest(requestParams{
+    _, _, err := context.performRequest(requestParams{
         Method:         "GET",
         URL:            uri,
         APIVersion:     "2012-02-12",
@@ -417,7 +417,7 @@
     uri := addURLQueryParams(
         context.getContainerURL(container),
         "restype", "container")
-    _, err := context.performRequest(requestParams{
+    _, _, err := context.performRequest(requestParams{
         Method:         "PUT",
         URL:            uri,
         APIVersion:     "2012-02-12",
@@ -436,7 +436,7 @@
     uri := addURLQueryParams(
         context.getContainerURL(container),
         "restype", "container")
-    _, err := context.performRequest(requestParams{
+    _, _, err := context.performRequest(requestParams{
         Method:         "DELETE",
         URL:            uri,
         APIVersion:     "2012-02-12",
@@ -449,6 +449,35 @@
     return nil
 }
 
+// Send a request to the storage service to retrieve a container's properties.
+// Also doubles as a handy way to see if a container exists.
+func (context *StorageContext) GetContainerProperties(container string) (*Properties, error) {
+    uri := addURLQueryParams(
+        context.getContainerURL(container),
+        "restype", "container")
+    params := requestParams{
+        Method:         "GET",
+        URL:            uri,
+        APIVersion:     "2012-02-12",
+        ExpectedStatus: http.StatusOK,
+    }
+    _, headers, err := context.performRequest(params)
+    if err != nil {
+        msg := fmt.Sprintf("failed to find container %s: ", container)
+        return nil, extendError(err, msg)
+    }
+
+    props := &Properties{
+        LastModified:  headers[http.CanonicalHeaderKey("Last-Modified")][0],
+        ETag:          headers[http.CanonicalHeaderKey("ETag")][0],
+        LeaseStatus:   headers[http.CanonicalHeaderKey("X-Ms-Lease-Status")][0],
+        LeaseState:    headers[http.CanonicalHeaderKey("X-Ms-Lease-State")][0],
+        LeaseDuration: headers[http.CanonicalHeaderKey("X-Ms-Lease-Duration")][0],
+    }
+
+    return props, nil
+}
+
 type PutBlobRequest struct {
     Container string // Container name in the storage account
     BlobType  string // Pass "page" or "block"
@@ -482,7 +511,7 @@
         extraHeaders.Add("x-ms-blob-content-length", size)
     }
 
-    _, err := context.performRequest(requestParams{
+    _, _, err := context.performRequest(requestParams{
         Method:         "PUT",
         URL:            context.GetFileURL(req.Container, req.Filename),
         APIVersion:     "2012-02-12",
@@ -523,7 +552,7 @@
     extraHeaders.Add("x-ms-range", rangeData)
     extraHeaders.Add("x-ms-page-write", "update")
 
-    _, err := context.performRequest(requestParams{
+    _, _, err := context.performRequest(requestParams{
         Method:         "PUT",
         URL:            uri,
         Body:           req.Data,
@@ -546,7 +575,7 @@
         "comp", "blocklist",
         "blocklisttype", "all")
     bl := GetBlockList{}
-    _, err := context.performRequest(requestParams{
+    _, _, err := context.performRequest(requestParams{
         Method:         "GET",
         URL:            uri,
         APIVersion:     "2012-02-12",
@@ -568,7 +597,7 @@
         context.GetFileURL(container, filename),
         "comp", "block",
         "blockid", base64ID)
-    _, err := context.performRequest(requestParams{
+    _, _, err := context.performRequest(requestParams{
         Method:         "PUT",
         URL:            uri,
         Body:           data,
@@ -593,7 +622,7 @@
     }
     dataReader := bytes.NewReader(data)
 
-    _, err = context.performRequest(requestParams{
+    _, _, err = context.performRequest(requestParams{
         Method:         "PUT",
         URL:            uri,
         Body:           dataReader,
@@ -610,7 +639,7 @@
 // Delete the specified blob from the given container.  Deleting a non-existant
 // blob will return without an error.
 func (context *StorageContext) DeleteBlob(container, filename string) error {
-    _, err := context.performRequest(requestParams{
+    _, _, err := context.performRequest(requestParams{
         Method:         "DELETE",
         URL:            context.GetFileURL(container, filename),
         APIVersion:     "2012-02-12",
@@ -630,7 +659,7 @@
 
 // Get the specified blob from the given container.
 func (context *StorageContext) GetBlob(container, filename string) (io.ReadCloser, error) {
-    body, err := context.performRequest(requestParams{
+    body, _, err := context.performRequest(requestParams{
         Method:         "GET",
         URL:            context.GetFileURL(container, filename),
         APIVersion:     "2012-02-12",
@@ -666,7 +695,7 @@
         panic("Access must be one of 'container', 'blob' or 'private'")
     }
 
-    _, err := context.performRequest(requestParams{
+    _, _, err := context.performRequest(requestParams{
         Method:         "PUT",
         URL:            uri,
         APIVersion:     "2009-09-19",
 
=== modified file 'storage_base_test.go'
--- storage_base_test.go	2013-07-29 00:43:12 +0000
+++ storage_base_test.go	2013-07-31 06:34:25 +0000
@@ -743,6 +743,71 @@
     c.Check(serverError.HTTPStatus.StatusCode(), Equals, http.StatusNotFound)
 }
 
+type TestGetContainerProperties struct{}
+
+var _ = Suite(&TestGetContainerProperties{})
+
+// The GetContainerProperties Storage API call returns without error when the
+// container has been created successfully.
+func (suite *TestGetContainerProperties) Test(c *C) {
+    header := make(http.Header)
+    header.Add("Last-Modified", "last-modified")
+    header.Add("ETag", "etag")
+    header.Add("X-Ms-Lease-Status", "status")
+    header.Add("X-Ms-Lease-State", "state")
+    header.Add("X-Ms-Lease-Duration", "duration")
+    response := &http.Response{
+        Status:     fmt.Sprintf("%d", http.StatusOK),
+        StatusCode: http.StatusOK,
+        Body:       makeResponseBody(""),
+        Header:     header,
+    }
+
+    transport := &TestTransport{Response: response}
+    context := makeStorageContext(transport)
+    containerName := MakeRandomString(10)
+    props, err := context.GetContainerProperties(containerName)
+    c.Assert(err, IsNil)
+    c.Check(transport.Request.URL.String(), Equals, fmt.Sprintf(
+        "http://%s.blob.core.windows.net/%s?restype=container",
+        context.Account, containerName))
+    c.Check(transport.Request.Method, Equals, "GET")
+    c.Check(transport.Request.Header.Get("Authorization"), Not(Equals), "")
+
+    c.Check(props.LastModified, Equals, "last-modified")
+    c.Check(props.ETag, Equals, "etag")
+    c.Check(props.LeaseStatus, Equals, "status")
+    c.Check(props.LeaseState, Equals, "state")
+    c.Check(props.LeaseDuration, Equals, "duration")
+}
+
+// Client-side errors from the HTTP client are propagated back to the caller.
+func (suite *TestGetContainerProperties) TestError(c *C) {
+    error := fmt.Errorf("canned-error")
+    context := makeStorageContext(&TestTransport{Error: error})
+    _, err := context.GetContainerProperties("container")
+    c.Assert(err, ErrorMatches, ".*canned-error.*")
+}
+
+// Server-side errors are propagated back to the caller.
+func (suite *TestGetContainerProperties) TestErrorResponse(c *C) {
+    response := makeHttpResponse(http.StatusNotFound, "not found")
+    context := makeStorageContext(&TestTransport{Response: response})
+    _, err := context.GetContainerProperties("container")
+    c.Assert(err, ErrorMatches, ".*Not Found.*")
+}
+
+// Azure HTTP errors (for instance 404 responses) are propagated back to
+// the caller as ServerError objects.
+func (suite *TestGetContainerProperties) TestServerError(c *C) {
+    response := makeHttpResponse(http.StatusNotFound, "not found")
+    context := makeStorageContext(&TestTransport{Response: response})
+    _, err := context.GetContainerProperties("container")
+    serverError, ok := err.(*ServerError)
+    c.Assert(ok, Equals, true)
+    c.Check(serverError.HTTPStatus.StatusCode(), Equals, http.StatusNotFound)
+}
+
 type TestPutPage struct{}
 
 var _ = Suite(&TestPutPage{})