<reponame>amenezes/config-client
import json

import pytest
import requests

from config import http
from config.auth import OAuth2
from config.cf import CF
from config.cfenv import CFenv
from config.spring import ConfigClient

CUSTOM_VCAP_SERVICES = json.dumps(
    {
        "p-config-server": [
            {
                "credentials": {
                    "uri": "http://example_uri",
                    "access_token_uri": "http://example_access_token_uri",
                    "client_id": "example_client_id",
                    "client_secret": "example_client_secret",
                }
            }
        ]
    }
)


CUSTOM_VCAP_APPLICATION = json.dumps(
    {
        "application_name": "myapp",
        "space_name": "test",
        "organization_name": "test",
        "uris": [],
    }
)


CONFIG = {
    "health": {"config": {"enabled": False}},
    "spring": {
        "cloud": {
            "consul": {
                "discovery": {
                    "health-check-interval": "10s",
                    "health-check-path": "/manage/health",
                    "instance-id": "myapp:${random.value}",
                    "prefer-ip-address": True,
                    "register-health-check": True,
                },
                "host": "discovery",
                "port": 8500,
            }
        }
    },
}


class ResponseMock:
    def __init__(self, *args, **kwargs):
        self.ok = kwargs.get("ok") or False
        self.status_code = kwargs.get("status_code") or 404
        self.headers = {"Content-Type": "application/json"}
        self.text = kwargs.get("text", "")

    def json(self):
        return {"access_token": "<KEY>"}


def response_mock_success(*args, **kwargs):
    return ResponseMock(ok=True, status_code=202)


def response_mock_error(*args, **kwargs):
    return ResponseMock()


def connection_error(*args, **kwargs):
    raise ConnectionError


def value_error(*args, **kwargs):
    raise ValueError


def base_exception_error(*args, **kwargs):
    raise Exception


def system_error(*args, **kwargs):
    raise SystemExit


def http_error(*args, **kwargs):
    raise requests.exceptions.HTTPError


def missing_schema_error(*args, **kwargs):
    raise requests.exceptions.MissingSchema


@pytest.fixture
def client(monkeypatch, scope="module"):
    return ConfigClient(app_name="test_app")


@pytest.fixture
def client_with_auth(monkeypatch, mocker, oauth2):
    monkeypatch.setattr(http, "post", response_mock_success)
    mocker.patch.object(http, "get")
    http.get.return_value = ResponseMock()
    return ConfigClient(app_name="test_app", oauth2=oauth2)


@pytest.fixture
def oauth2():
    return OAuth2(
        access_token_uri="https://p-spring-cloud-services.uaa.sys.example.com/oauth/token",
        client_id="p-config-server-example-client-id",
        client_secret="EXAMPLE_SECRET",
    )


@pytest.fixture
def cfenv():
    return CFenv()


@pytest.fixture
def custom_cfenv():
    return CFenv(
        vcap_services=json.loads(CUSTOM_VCAP_SERVICES),
        vcap_application=json.loads(CUSTOM_VCAP_APPLICATION),
    )


@pytest.fixture
def cf():
    return CF()
