def test_get_named_config(test_type, environment, expected): """Assert that the named configurations can be loaded. Or that a KeyError is returned for missing config types. """ if test_type == 'valid': assert isinstance(config.get_named_config(environment), expected) else: with pytest.raises(KeyError): config.get_named_config(environment)
import psycopg2 import pycountry import requests from dotenv import load_dotenv, find_dotenv from flask import Flask from legal_api import db from legal_api.config import get_named_config from legal_api.models import Address, Business, Director, Filing from sqlalchemy_continuum import versioning_manager from sqlalchemy import text load_dotenv(find_dotenv()) FLASK_APP = Flask(__name__) FLASK_APP.config.from_object(get_named_config('production')) db.init_app(FLASK_APP) COLIN_API = os.getenv('COLIN_API', None) if not COLIN_API: print('ERROR, no COLIN_API defined.') def create_business(db, business_json): business = Business() business.identifier = business_json['business']['identifier'] business.founding_date = business_json['business']['foundingDate'] business.last_ledger_timestamp = business_json['business'][ 'lastLedgerTimestamp'] business.legal_name = business_json['business']['legalName'] business.founding_date = business_json['business']['foundingDate']