fix tests

tsuru · May 18, 2021 · af5b180 · af5b180
1 parent 00ec864
commit af5b180
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 6 deletions.
diff --git a/pkg/rpaas/client/autoscale.go b/pkg/rpaas/client/autoscale.go
@@ -123,12 +123,7 @@ func (c *client) UpdateAutoscale(ctx context.Context, args UpdateAutoscaleArgs)
 		return err
 	}
 
-	expectedStatus := http.StatusCreated
-	if shouldCreate {
-		expectedStatus = http.StatusOK
-	}
-
-	if resp.StatusCode != expectedStatus {
+	if resp.StatusCode > http.StatusCreated {
 		return newErrUnexpectedStatusCodeFromResponse(resp)
 	}
 

diff --git a/pkg/rpaas/client/internal_client_test.go b/pkg/rpaas/client/internal_client_test.go
@@ -117,6 +117,7 @@ func TestNewClientThroughTsuruWithOptions(t *testing.T) {
 				assert.EqualError(t, err, tt.expectedError)
 			} else {
 				require.NoError(t, err)
+				rpaasClient.(*client).client.Transport = nil // void compare
 				assert.Equal(t, tt.expected, rpaasClient.(*client))
 			}
 			if tt.teardown != nil {