70 lines
2.2 KiB
Python
70 lines
2.2 KiB
Python
import pytest
|
|
import requests
|
|
from flask import url_for
|
|
from urllib.parse import urljoin
|
|
from .browsers import BROWSERSTACK_CONFIG
|
|
from atst.domain.users import Users
|
|
import atst.domain.exceptions as exceptions
|
|
from tests.test_auth import _login
|
|
|
|
import cryptography.x509 as x509
|
|
from cryptography.hazmat.backends import default_backend
|
|
|
|
|
|
USER_CERT = "ssl/client-certs/atat.mil.crt"
|
|
|
|
|
|
@pytest.mark.parametrize("browser_type", BROWSERSTACK_CONFIG.keys())
|
|
@pytest.mark.usefixtures('live_server')
|
|
def test_can_get_title(browser_type, app, drivers):
|
|
driver = drivers[browser_type]
|
|
driver.get(url_for("atst.root", _external=True))
|
|
assert "JEDI" in driver.title
|
|
|
|
|
|
def _get_common_name(cert_path):
|
|
with open(USER_CERT, "rb") as cert_file:
|
|
cert = x509.load_pem_x509_certificate(cert_file.read(), default_backend())
|
|
common_names = cert.subject.get_attributes_for_oid(x509.NameOID.COMMON_NAME)
|
|
return common_names[0].value
|
|
|
|
|
|
@pytest.fixture(scope="module")
|
|
def valid_user_from_cert():
|
|
cn = _get_common_name(USER_CERT)
|
|
cn_parts = cn.split(".")
|
|
user_info = {
|
|
"last_name": cn_parts[0],
|
|
"first_name": cn_parts[1],
|
|
"dod_id": cn_parts[-1],
|
|
"atat_role_name": "developer",
|
|
}
|
|
return Users.get_or_create_by_dod_id(**user_info)
|
|
|
|
|
|
def _valid_login(client, driver):
|
|
with open(USER_CERT) as cert:
|
|
response = _login(client, cert=cert.read())
|
|
cookie = [cookie for cookie in client.cookie_jar][0]
|
|
driver.add_cookie(
|
|
{
|
|
"domain": None,
|
|
"name": cookie.name,
|
|
"value": cookie.value,
|
|
"path": cookie.path,
|
|
"secure": cookie.secure,
|
|
"expiry": cookie.expires,
|
|
}
|
|
)
|
|
|
|
|
|
@pytest.mark.usefixtures('live_server')
|
|
def test_login(drivers, client, app, valid_user_from_cert):
|
|
driver = drivers["win10_chrome62"]
|
|
driver.get(url_for("atst.root", _external=True))
|
|
cookie = _valid_login(client, driver)
|
|
requests_page = url_for("requests.requests_form_new", screen=1, _external=True)
|
|
driver.get(requests_page)
|
|
user = valid_user_from_cert
|
|
assert user.last_name in driver.page_source
|