catch unauthed error and imporve logging

cmd/utils.go

@@ -31,7 +31,12 @@ func prepareDefaultAirflowImageTag(airflowVersion string, httpClient *airflowver
 			defaultImageTag = fmt.Sprintf("%s-buster-onbuild", airflowVersion)
 		}
 	} else if airflowVersion != "" {
-		return "", err
+		switch t := err; t {
+		default:
+			return "", err
+		case houston.PermissionsErrorVerbose:
+			return "", errors.New("The --airflow-version flag is not supported if you're not authenticated to Astronomer. Please authenticate and try again.")
+		}
 	}
 
 	if len(defaultImageTag) == 0 {

cmd/utils_test.go

@@ -143,3 +143,77 @@ func Test_prepareDefaultAirflowImageTagHoustonBadRequest(t *testing.T) {
 	assert.Equal(t, err.Error(), fmt.Sprintf("API error (400): %s", mockErrorResponse))
 	assert.Equal(t, "", defaultTag)
 }
+
+
+func Test_prepareDefaultAirflowImageTagHoustonUnauthedRequest(t *testing.T) {
+	testUtil.InitTestConfig()
+	mockErrorResponse := `{
+	  "errors": [
+		{
+		  "message": "You do not have the appropriate permissions for that",
+		  "locations": [
+			{
+			  "line": 2,
+			  "column": 3
+			}
+		  ],
+		  "path": [
+			"deploymentConfig"
+		  ],
+		  "extensions": {
+			"code": "FORBIDDEN",
+			"exception": {
+			  "message": "You do not have the appropriate permissions for that"
+			}
+		  }
+		}
+	  ],
+	  "data": {
+		"deploymentConfig": null
+	  }
+	}`
+	// prepare fake response from updates.astronomer.io
+	okResponse := `{
+	  "version": "1.0",
+	  "available_releases": [
+		{
+		  "version": "1.10.5",
+		  "level": "new_feature",
+		  "url": "https://github.com/astronomer/airflow/releases/tag/1.10.5-11",
+		  "release_date": "2020-10-05T20:03:00+00:00",
+		  "tags": [
+			"1.10.5-alpine3.10-onbuild",
+			"1.10.5-buster-onbuild",
+			"1.10.5-alpine3.10",
+			"1.10.5-buster"
+		  ],
+		  "channel": "stable"
+		}
+	  ]
+	}`
+
+	client := testUtil.NewTestClient(func(req *http.Request) *http.Response {
+		return &http.Response{
+			StatusCode: 200,
+			Body:       ioutil.NopCloser(bytes.NewBufferString(okResponse)),
+			Header:     make(http.Header),
+		}
+	})
+	httpClient := airflowversions.NewClient(client)
+
+	// prepare fake response from houston
+	houstonClient := testUtil.NewTestClient(func(req *http.Request) *http.Response {
+		return &http.Response{
+			StatusCode: 200,
+			Body:       ioutil.NopCloser(bytes.NewBufferString(mockErrorResponse)),
+			Header:     make(http.Header),
+		}
+	})
+	api := houston.NewHoustonClient(houstonClient)
+
+	output := new(bytes.Buffer)
+
+	defaultTag, err := prepareDefaultAirflowImageTag("2.0.2", httpClient, api, output)
+	assert.Equal(t, "The --airflow-version flag is not supported if you're not authenticated to Astronomer. Please authenticate and try again.", err.Error())
+	assert.Equal(t, "", defaultTag)
+}

houston/houston.go

@@ -11,6 +11,7 @@ import (
 )
 
 var PermissionsError = errors.New("You do not have the appropriate permissions for that")
+var PermissionsErrorVerbose = errors.New("You do not have the appropriate permissions for that: Your token has expired. Please log in again.")
 
 // Client containers the logger and HTTPClient used to communicate with the HoustonAPI
 type Client struct {
@@ -85,12 +86,11 @@ func (c *Client) Do(doOpts httputil.DoOptions) (*Response, error) {
 	if err != nil {
 		return nil, errors.Wrap(err, "Failed to JSON decode Houston response")
 	}
-
 	// Houston Specific Errors
 	if decode.Errors != nil {
 		err = errors.New(decode.Errors[0].Message)
 		if err.Error() == PermissionsError.Error() {
-			return nil, errors.Wrap(errors.New("Your token has expired. Please log in again."), err.Error())
+			return nil, PermissionsErrorVerbose
 		}
 		return nil, err
 	}