Fix test

Author: zliang-akamai

test/integration/models/object_storage/test_obj_quotas.py

@@ -45,11 +45,10 @@ def test_get_obj_storage_quota_usage(test_linode_client):
         quota_id = quotas[0].quota_id
         quota = test_linode_client.load(ObjectStorageQuota, quota_id)
 
-        with pytest.raises(ApiError) as exc:
+        # quota without usage should return an API error on usage retrieval
+        with pytest.raises(ApiError):
             quota.usage()
 
-        assert exc.value.status == 404
-        assert "Usage not supported" in str(exc.value)
         return
 
     quota_id = quota_with_usage.quota_id
@@ -65,12 +64,7 @@ def test_get_obj_storage_quota_usage(test_linode_client):
 
 
 def test_list_and_get_obj_storage_global_quotas(test_linode_client):
-    try:
-        quotas = test_linode_client.object_storage.global_quotas()
-    except ApiError as err:
-        if err.status == 404:
-            pytest.skip("Object Storage is not enabled on this account.")
-        raise
+    quotas = test_linode_client.object_storage.global_quotas()
 
     if len(quotas) < 1:
         pytest.skip("No available global quota for testing. Skipping now...")
@@ -91,12 +85,7 @@ def test_list_and_get_obj_storage_global_quotas(test_linode_client):
 
 
 def test_get_obj_storage_global_quota_usage(test_linode_client):
-    try:
-        quotas = test_linode_client.object_storage.global_quotas()
-    except ApiError as err:
-        if err.status == 404:
-            pytest.skip("Object Storage is not enabled on this account.")
-        raise
+    quotas = test_linode_client.object_storage.global_quotas()
 
     if len(quotas) < 1:
         pytest.skip("No available global quota for testing. Skipping now...")
@@ -109,11 +98,10 @@ def test_get_obj_storage_global_quota_usage(test_linode_client):
         quota_id = quotas[0].quota_id
         quota = test_linode_client.load(ObjectStorageGlobalQuota, quota_id)
 
-        with pytest.raises(ApiError) as exc:
+        # quota without usage should return an API error on usage retrieval
+        with pytest.raises(ApiError):
             quota.usage()
 
-        assert exc.value.status == 404
-        assert "Usage not supported" in str(exc.value)
         return
 
     quota_id = quota_with_usage.quota_id