reformat private key

This commit is contained in:
Kenneth Kehl
2023-12-15 12:16:03 -08:00
parent cec2d7ea02
commit f54a79bdf9
2 changed files with 6 additions and 6 deletions

View File

@@ -26,7 +26,7 @@ from app.utils import hide_from_search_engines
from app.utils.login import is_safe_redirect_url
def _reformulate_keystring(orig):
def _reformat_keystring(orig):
new_keystring = orig.replace("-----BEGIN PRIVATE KEY-----", "")
new_keystring = new_keystring.replace("-----END PRIVATE KEY-----", "")
new_keystring = new_keystring.strip()
@@ -42,7 +42,7 @@ def _get_access_token(code, state):
access_token_url = os.getenv("LOGIN_DOT_GOV_ACCESS_TOKEN_URL")
keystring = os.getenv("LOGIN_PEM")
if " " in keystring:
keystring = _reformulate_keystring(keystring)
keystring = _reformat_keystring(keystring)
payload = {
"iss": client_id,

View File

@@ -3,7 +3,7 @@ import uuid
import pytest
from flask import url_for
from app.main.views.sign_in import _reformulate_keystring
from app.main.views.sign_in import _reformat_keystring
from app.models.user import User
from tests.conftest import SERVICE_ONE_ID, normalize_spaces
@@ -40,14 +40,14 @@ def test_render_sign_in_template_with_next_link_for_password_reset(client_reques
)
def test_reformulate_keystring():
def test_reformat_keystring():
orig = "-----BEGIN PRIVATE KEY----- blahblahblah -----END PRIVATE KEY-----"
expected = """-----BEGIN PRIVATE KEY-----
blahblahblah
-----END PRIVATE KEY-----
"""
reformulated = _reformulate_keystring(orig)
assert reformulated == expected
reformatted = _reformat_keystring(orig)
assert reformatted == expected
def test_sign_in_explains_session_timeout(client_request):