POSTGRES_USER, POSTGRES_HOST = 'postgres', 'localhost' def db_connection(dbname=None): conn = psycopg2.connect(user=POSTGRES_USER, host=POSTGRES_HOST, database=dbname) conn.autocommit = True return conn try: conn = db_connection() CAN_CONNECT_TO_DB = True SERVER_VERSION = conn.server_version json_types = register_json_typecasters(conn, lambda x: x) JSON_AVAILABLE = 'json' in json_types JSONB_AVAILABLE = 'jsonb' in json_types except: CAN_CONNECT_TO_DB = JSON_AVAILABLE = JSONB_AVAILABLE = False SERVER_VERSION = 0 dbtest = pytest.mark.skipif( not CAN_CONNECT_TO_DB, reason="Need a postgres instance at localhost accessible by user 'postgres'" ) requires_json = pytest.mark.skipif( not JSON_AVAILABLE, reason='Postgres server unavailable or json type not defined')
# TODO: should this be somehow be divined from environment? POSTGRES_USER, POSTGRES_HOST = 'postgres', 'localhost' def db_connection(dbname=None): conn = psycopg2.connect(user=POSTGRES_USER, host=POSTGRES_HOST, database=dbname) conn.autocommit = True return conn try: conn = db_connection() CAN_CONNECT_TO_DB = True SERVER_VERSION = conn.server_version json_types = register_json_typecasters(conn, lambda x: x) JSON_AVAILABLE = 'json' in json_types JSONB_AVAILABLE = 'jsonb' in json_types except: CAN_CONNECT_TO_DB = JSON_AVAILABLE = JSONB_AVAILABLE = False SERVER_VERSION = 0 dbtest = pytest.mark.skipif( not CAN_CONNECT_TO_DB, reason="Need a postgres instance at localhost accessible by user 'postgres'") requires_json = pytest.mark.skipif( not JSON_AVAILABLE, reason='Postgres server unavailable or json type not defined')