diff --git a/SATS/integration_test.py b/SATS/integration_test.py index 0e63bd2..ae73462 100644 --- a/SATS/integration_test.py +++ b/SATS/integration_test.py @@ -24,13 +24,7 @@ def post_to_page(self, page, **kwargs) -> requests.models.Response: def uaa_pick_idp(self) -> str: r = self.s.get(self.uaa_url + "/login") - params = { - "returnIDParam": "idp", - "entityID": self.uaa_url[8:], - "idp": self.idp_name, - "isPassive": "true", - } - r = self.s.get(f"{self.uaa_url}/saml/discovery", params=params) + r = self.s.get(f"{self.uaa_url}/saml2/authenticate/{self.idp_name}") soup = BeautifulSoup(r.text, features="html.parser") saml_request = soup.find(attrs={"name": "SAMLRequest"}).attrs["value"] relay_state = soup.find(attrs={"name": "RelayState"}).attrs["value"] @@ -107,7 +101,7 @@ def log_in(self, username, password, totp_seed=None) -> typing.Tuple[str, bool]: handles registering totp if user does not have one currently returns a tuple of str, bool representing the user's totp_seed, and whether a new totp was registered """ - + r = self.uaa_pick_idp() soup = BeautifulSoup(r.text, features="html.parser") form = soup.find("form")