13964: Detect and report rate limit and quota exceeded errors

[arvados.git] / lib / dispatchcloud / azure_test.go

diff --git a/lib/dispatchcloud/azure_test.go b/lib/dispatchcloud/azure_test.go
index bcba51bfdf809880f6ac3576bfea37c3db295f2a..0c8c8d534d8d1ab6c0f22b38f4f7da7496830186 100644 (file)
--- a/lib/dispatchcloud/azure_test.go
+++ b/lib/dispatchcloud/azure_test.go
@@ -14,6 +14,8 @@ import (
        "git.curoverse.com/arvados.git/sdk/go/config"
        "github.com/Azure/azure-sdk-for-go/services/compute/mgmt/2018-06-01/compute"
        "github.com/Azure/azure-sdk-for-go/services/network/mgmt/2018-06-01/network"
+       "github.com/Azure/go-autorest/autorest"
+       "github.com/Azure/go-autorest/autorest/azure"
        "github.com/Azure/go-autorest/autorest/to"
        check "gopkg.in/check.v1"
 )
@@ -156,3 +158,49 @@ func (*AzureProviderSuite) TestDestroyInstances(c *check.C) {
                c.Check(i.Destroy(context.Background()), check.IsNil)
        }
 }
+
+func (*AzureProviderSuite) TestDeleteFake(c *check.C) {
+       ap, _, _, err := GetProvider()
+       if err != nil {
+               c.Fatal("Error making provider", err)
+       }
+
+       _, err = ap.(*AzureProvider).netClient.Delete(context.Background(), "fakefakefake", "fakefakefake")
+
+       rq := err.(autorest.DetailedError).Original.(*azure.RequestError)
+
+       log.Printf("%v %q %q", rq.Response.StatusCode, rq.ServiceError.Code, rq.ServiceError.Message)
+}
+
+func (*AzureProviderSuite) TestWrapError(c *check.C) {
+       retryError := autorest.DetailedError{
+               Original: &azure.RequestError{
+                       DetailedError: autorest.DetailedError{
+                               Response: &http.Response{
+                                       StatusCode: 429,
+                                       Header:     map[string][]string{"Retry-After": []string{"123"}},
+                               },
+                       },
+                       ServiceError: &azure.ServiceError{},
+               },
+       }
+       wrapped := WrapAzureError(retryError)
+       _, ok := wrapped.(RateLimitError)
+       c.Check(ok, check.Equals, true)
+
+       quotaError := autorest.DetailedError{
+               Original: &azure.RequestError{
+                       DetailedError: autorest.DetailedError{
+                               Response: &http.Response{
+                                       StatusCode: 503,
+                               },
+                       },
+                       ServiceError: &azure.ServiceError{
+                               Message: "No more quota",
+                       },
+               },
+       }
+       wrapped = WrapAzureError(quotaError)
+       _, ok = wrapped.(QuotaError)
+       c.Check(ok, check.Equals, true)
+}