from py.test import raises from sqlalchemy import Column, Table, types from sqlalchemy.ext.declarative import declarative_base from sqlalchemy_utils.types.choice import ChoiceType from sqlalchemy.dialects import postgresql import graphene from graphene.types.json import JSONString from ..converter import (convert_sqlalchemy_column, convert_sqlalchemy_relationship) from ..fields import SQLAlchemyConnectionField from ..types import SQLAlchemyObjectType, SQLAlchemyNode from ..registry import Registry from .models import Article, Pet, Reporter def assert_column_conversion(sqlalchemy_type, graphene_field, **kwargs): column = Column(sqlalchemy_type, doc='Custom Help Text', **kwargs) graphene_type = convert_sqlalchemy_column(column) assert isinstance(graphene_type, graphene_field) field = graphene_type.as_field() assert field.description == 'Custom Help Text' return field def test_should_unknown_sqlalchemy_field_raise_exception(): with raises(Exception) as excinfo: convert_sqlalchemy_column(None) assert 'Don\'t know how to convert the SQLAlchemy field' in str(excinfo.value) def test_should_date_convert_string(): assert_column_conversion(types.Date(), graphene.String) def test_should_datetime_convert_string(): assert_column_conversion(types.DateTime(), graphene.String) def test_should_time_convert_string(): assert_column_conversion(types.Time(), graphene.String) def test_should_string_convert_string(): assert_column_conversion(types.String(), graphene.String) def test_should_text_convert_string(): assert_column_conversion(types.Text(), graphene.String) def test_should_unicode_convert_string(): assert_column_conversion(types.Unicode(), graphene.String) def test_should_unicodetext_convert_string(): assert_column_conversion(types.UnicodeText(), graphene.String) def test_should_enum_convert_string(): assert_column_conversion(types.Enum(), graphene.String) def test_should_small_integer_convert_int(): assert_column_conversion(types.SmallInteger(), graphene.Int) def test_should_big_integer_convert_int(): assert_column_conversion(types.BigInteger(), graphene.Int) def test_should_integer_convert_int(): assert_column_conversion(types.Integer(), graphene.Int) def test_should_integer_convert_id(): assert_column_conversion(types.Integer(), graphene.ID, primary_key=True) def test_should_boolean_convert_boolean(): assert_column_conversion(types.Boolean(), graphene.Boolean) def test_should_float_convert_float(): assert_column_conversion(types.Float(), graphene.Float) def test_should_numeric_convert_float(): assert_column_conversion(types.Numeric(), graphene.Float) def test_should_choice_convert_enum(): TYPES = [ (u'es', u'Spanish'), (u'en', u'English') ] column = Column(ChoiceType(TYPES), doc='Language', name='language') Base = declarative_base() Table('translatedmodel', Base.metadata, column) graphene_type = convert_sqlalchemy_column(column) assert issubclass(graphene_type, graphene.Enum) assert graphene_type._meta.graphql_type.name == 'TRANSLATEDMODEL_LANGUAGE' assert graphene_type._meta.graphql_type.description == 'Language' assert graphene_type._meta.enum.__members__['es'].value == 'Spanish' assert graphene_type._meta.enum.__members__['en'].value == 'English' def test_should_manytomany_convert_connectionorlist(): registry = Registry() graphene_type = convert_sqlalchemy_relationship(Reporter.pets.property, registry) assert not graphene_type def test_should_manytomany_convert_connectionorlist_list(): class A(SQLAlchemyObjectType): class Meta: model = Pet graphene_type = convert_sqlalchemy_relationship(Reporter.pets.property, A._meta.registry) assert isinstance(graphene_type, graphene.List) assert graphene_type.of_type == A._meta.graphql_type def test_should_manytomany_convert_connectionorlist_connection(): class A(SQLAlchemyNode, SQLAlchemyObjectType): class Meta: model = Pet graphene_type = convert_sqlalchemy_relationship(Reporter.pets.property, A._meta.registry) assert isinstance(graphene_type, SQLAlchemyConnectionField) def test_should_manytoone_convert_connectionorlist(): registry = Registry() graphene_type = convert_sqlalchemy_relationship(Article.reporter.property, registry) assert not graphene_type def test_should_manytoone_convert_connectionorlist_list(): class A(SQLAlchemyObjectType): class Meta: model = Reporter graphene_type = convert_sqlalchemy_relationship(Article.reporter.property, A._meta.registry) assert isinstance(graphene_type, graphene.Field) assert graphene_type.type == A._meta.graphql_type def test_should_manytoone_convert_connectionorlist_connection(): class A(SQLAlchemyNode, SQLAlchemyObjectType): class Meta: model = Reporter graphene_type = convert_sqlalchemy_relationship(Article.reporter.property, A._meta.registry) assert isinstance(graphene_type, graphene.Field) assert graphene_type.type == A._meta.graphql_type def test_should_postgresql_uuid_convert(): assert_column_conversion(postgresql.UUID(), graphene.String) def test_should_postgresql_enum_convert(): assert_column_conversion(postgresql.ENUM(), graphene.String) def test_should_postgresql_array_convert(): assert_column_conversion(postgresql.ARRAY(types.Integer), graphene.List) def test_should_postgresql_json_convert(): assert_column_conversion(postgresql.JSON(), JSONString) def test_should_postgresql_jsonb_convert(): assert_column_conversion(postgresql.JSONB(), JSONString) def test_should_postgresql_hstore_convert(): assert_column_conversion(postgresql.HSTORE(), JSONString)