upd (converter): add sqlalchemy.types.Variant handling

Author: TLDMain

graphene_sqlalchemy/converter.py

@@ -252,3 +252,8 @@ def convert_json_to_string(type, column, registry=None):
 @convert_sqlalchemy_type.register(types.JSON)
 def convert_json_type_to_string(type, column, registry=None):
     return JSONString
+
+
+@convert_sqlalchemy_type.register(types.Variant)
+def convert_variant_to_impl_type(type, column, registry=None):
+    return convert_sqlalchemy_type(type.impl, column, registry=registry)

graphene_sqlalchemy/tests/test_converter.py

@@ -287,6 +287,14 @@ class Meta:
     assert graphene_type.type == A
 
 
+def test_should_variant_int_convert_int():
+    assert get_field(types.Variant(types.Integer(), {})).type == graphene.Int
+
+
+def test_should_variant_string_convert_string():
+    assert get_field(types.Variant(types.String(), {})).type == graphene.String
+
+
 def test_should_postgresql_uuid_convert():
     assert get_field(postgresql.UUID()).type == graphene.String