fixed read test that was breaking

shipyardapp · Nov 18, 2024 · 950bf26 · 950bf26
1 parent e10de13
commit 950bf26
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/shipyard_blueprints/magnite/tests/test_read.py b/shipyard_blueprints/magnite/tests/test_read.py
@@ -3,6 +3,7 @@
 
 from dotenv import load_dotenv, find_dotenv
 from shipyard_magnite import MagniteClient
+from shipyard_magnite.errs import ReadError
 from pytest import MonkeyPatch
 from copy import deepcopy
 from requests.exceptions import HTTPError
@@ -29,12 +30,12 @@ def test_read_invalid_campaign(client):
     campaign_id = "bad_campaign_id"
     client.connect()
 
-    with pytest.raises(HTTPError):
+    with pytest.raises(ReadError):
         client.read("campaigns", campaign_id)
 
 
 def test_read_invalid_endpoint(client):
     campaign_id = os.getenv("CAMPAIGN_ID")
     client.connect()
-    with pytest.raises(HTTPError):
+    with pytest.raises(ReadError):
         client.read("bad-endpoint", campaign_id)