iframe-proxy | Sunbelt Computer Software

120 lines (86 loc) · 4.11 KB
from pathlib import Path
from typing import Optional
import pytest
import responses
from cycode.cli.user_settings.credentials_manager import CredentialsManager
from cycode.cyclient.client_creator import create_scan_client
from cycode.cyclient.cycode_oidc_based_client import CycodeOidcBasedClient
from cycode.cyclient.cycode_token_based_client import CycodeTokenBasedClient
from cycode.cyclient.scan_client import ScanClient
# not real JWT with userId and tenantId fields
_EXPECTED_API_TOKEN = 'eyJhbGciOiJIUzI1NiIsInR5cCI6IkpXVCJ9.eyJzdWIiOiIxMjM0NTY3ODkwIiwibmFtZSI6IkpvaG4gRG9lIiwiaWF0IjoxNTE2MjM5MDIyLCJ1c2VySWQiOiJibGFibGEiLCJ0ZW5hbnRJZCI6ImJsYWJsYSJ9.8RfoWBfciuj8nwc7UB8uOUJchVuaYpYlgf1G2QHiWTk'  # noqa: E501
_CLIENT_ID = 'b1234568-0eaa-1234-beb8-6f0c12345678'
_CLIENT_SECRET = 'a12345a-42b2-1234-3bdd-c0130123456'
_ID_TOKEN = 'eyJhbGciOiJSUzI1NiIsImtpZCI6IjEyMzQ1NiJ9.eyJzdWIiOiI4NzY1NDMyMSIsImF1ZCI6ImN5Y29kZSIsImV4cCI6MTUxNjIzOTAyMiwiaXNfb2lkYyI6MX0.Rrby2hPzsoMM3'  # noqa: E501
CLI_ENV_VARS = {'CYCODE_CLIENT_ID': _CLIENT_ID, 'CYCODE_CLIENT_SECRET': _CLIENT_SECRET}
TEST_FILES_PATH = Path(__file__).parent.joinpath('test_files').absolute()
MOCKED_RESPONSES_PATH = Path(__file__).parent.joinpath('cyclient/mocked_responses/data').absolute()
ZIP_CONTENT_PATH = TEST_FILES_PATH.joinpath('zip_content').absolute()
@pytest.fixture(scope='session')
def test_files_path() -> Path:
    return TEST_FILES_PATH
@pytest.fixture(scope='session')
def scan_client() -> ScanClient:
    return create_scan_client(_CLIENT_ID, _CLIENT_SECRET, hide_response_log=False)
def create_token_based_client(
    client_id: Optional[str] = None, client_secret: Optional[str] = None
) -> CycodeTokenBasedClient:
    CredentialsManager.FILE_NAME = 'unit-tests-credentials.yaml'
    if client_id is None:
        client_id = _CLIENT_ID
    if client_secret is None:
        client_secret = _CLIENT_SECRET
    return CycodeTokenBasedClient(client_id, client_secret)
def create_oidc_based_client(client_id: Optional[str] = None, id_token: Optional[str] = None) -> CycodeOidcBasedClient:
    CredentialsManager.FILE_NAME = 'unit-tests-credentials.yaml'
    if client_id is None:
        client_id = _CLIENT_ID
    if id_token is None:
        id_token = _ID_TOKEN
    return CycodeOidcBasedClient(client_id, id_token)
@pytest.fixture(scope='session')
def token_based_client() -> CycodeTokenBasedClient:
    return create_token_based_client()
@pytest.fixture(scope='session')
def api_token_url(token_based_client: CycodeTokenBasedClient) -> str:
    return f'{token_based_client.api_url}/api/v1/auth/api-token'
@pytest.fixture(scope='session')
def api_token_response(api_token_url: str) -> responses.Response:
    return responses.Response(
        method=responses.POST,
        url=api_token_url,
        json={
            'token': _EXPECTED_API_TOKEN,
            'refresh_token': '12345678-0c68-1234-91ba-a13123456789',
            'expires_in': 86400,
        status=200,
@pytest.fixture(scope='session')
@responses.activate
def api_token(token_based_client: CycodeTokenBasedClient, api_token_response: responses.Response) -> str:
    responses.add(api_token_response)
    return token_based_client.get_access_token()
@pytest.fixture(scope='session')
def oidc_based_client() -> CycodeOidcBasedClient:
    return create_oidc_based_client()
@pytest.fixture(scope='session')
def oidc_api_token_url(oidc_based_client: CycodeOidcBasedClient) -> str:
    return f'{oidc_based_client.api_url}/api/v1/auth/oidc/api-token'
@pytest.fixture(scope='session')
@responses.activate
def oidc_api_token(oidc_based_client: CycodeOidcBasedClient, oidc_api_token_response: responses.Response) -> str:
    responses.add(oidc_api_token_response)
    return oidc_based_client.get_access_token()
@pytest.fixture(scope='session')
def oidc_api_token_response(oidc_api_token_url: str) -> responses.Response:
    return responses.Response(
        method=responses.POST,
        url=oidc_api_token_url,
        json={
            'token': _EXPECTED_API_TOKEN,
            'refresh_token': '12345678-0c68-1234-91ba-a13123456789',
            'expires_in': 86400,
        status=200,
Sunbelt Computer Software

PL/B Language Development and Support

Provide feedback

Saved searches

Use saved searches to filter your results more quickly

Uh oh!

Sunbelt Computer Software

PL/B Language Development and Support

Uh oh!

FilesExpand file tree

conftest.py

Latest commit

History

conftest.py

File metadata and controls
