Skip to content

Commit 9d1e3c6

Browse files
committed
Added extra test for column properties
1 parent 5e5d0df commit 9d1e3c6

File tree

1 file changed

+19
-2
lines changed

1 file changed

+19
-2
lines changed

graphene_sqlalchemy/tests/test_converter.py

Lines changed: 19 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,8 +1,8 @@
11
from py.test import raises
2-
from sqlalchemy import Column, Table, case, types
2+
from sqlalchemy import Column, Table, case, types, select, func
33
from sqlalchemy.dialects import postgresql
44
from sqlalchemy.ext.declarative import declarative_base
5-
from sqlalchemy.orm import composite
5+
from sqlalchemy.orm import composite, column_property
66
from sqlalchemy.sql.elements import Label
77
from sqlalchemy_utils import ChoiceType, JSONType, ScalarListType
88

@@ -136,6 +136,23 @@ def test_should_choice_convert_enum():
136136
assert graphene_type._meta.enum.__members__['en'].value == 'English'
137137

138138

139+
def test_should_columproperty_convert():
140+
141+
Base = declarative_base()
142+
143+
class Test(Base):
144+
__tablename__ = 'test'
145+
id = Column(types.Integer, primary_key=True)
146+
column = column_property(
147+
select([func.sum(func.cast(id, types.Integer))]).where(
148+
id==1
149+
)
150+
)
151+
152+
graphene_type = convert_sqlalchemy_column(Test.column)
153+
assert graphene_type.kwargs['required'] == False
154+
155+
139156
def test_should_scalar_list_convert_list():
140157
assert_column_conversion(ScalarListType(), graphene.List)
141158

0 commit comments

Comments
 (0)