From 21ad5cd9942044939c8203dd076ea080b6d55a61 Mon Sep 17 00:00:00 2001
From: Bryan Newbold <bnewbold@archive.org>
Date: Thu, 5 May 2022 11:21:29 -0700
Subject: ingest spn2: fix tests

---
 python/sandcrawler/__init__.py | 1 +
 python/sandcrawler/ia.py       | 2 +-
 2 files changed, 2 insertions(+), 1 deletion(-)

(limited to 'python/sandcrawler')

diff --git a/python/sandcrawler/__init__.py b/python/sandcrawler/__init__.py
index 6718c57..469c2a2 100644
--- a/python/sandcrawler/__init__.py
+++ b/python/sandcrawler/__init__.py
@@ -7,6 +7,7 @@ from .ia import (
     CdxRow,
     PetaboxError,
     ResourceResult,
+    SavePageNowBackoffError,
     SavePageNowClient,
     SavePageNowError,
     WarcResource,
diff --git a/python/sandcrawler/ia.py b/python/sandcrawler/ia.py
index 9c727ce..7365383 100644
--- a/python/sandcrawler/ia.py
+++ b/python/sandcrawler/ia.py
@@ -1012,7 +1012,7 @@ class SavePageNowClient:
                     break
 
         # check if SPNv2 user has capacity available
-        resp = self.v2_session.get("https://web.archive.org/save/status/user")
+        resp = self.v2_session.get(f"{self.v2endpoint}/status/user")
         if resp.status_code == 429:
             raise SavePageNowBackoffError(
                 f"SPNv2 availability API status_code: {resp.status_code}"
-- 
cgit v1.2.3