Map pg's uuid to graphene's uuid

Author: ticosax

Diff for: graphene_sqlalchemy/converter.py

@@ -4,7 +4,7 @@
 from sqlalchemy.orm import interfaces
 
 from graphene import (ID, Boolean, Dynamic, Enum, Field, Float, Int, List,
-                      String)
+                      String, UUID)
 from graphene.types.json import JSONString
 
 from .enums import enum_for_sa_enum
@@ -191,3 +191,8 @@ def convert_json_to_string(type, column, registry=None):
 @convert_sqlalchemy_type.register(JSONType)
 def convert_json_type_to_string(type, column, registry=None):
     return JSONString
+
+
+@convert_sqlalchemy_type.register(postgresql.UUID)
+def convert_column_to_uuid(type, column, registry=None):
+    return UUID

Diff for: graphene_sqlalchemy/tests/test_converter.py

@@ -252,7 +252,7 @@ class Meta:
 
 
 def test_should_postgresql_uuid_convert():
-    assert get_field(postgresql.UUID()).type == graphene.String
+    assert get_field(postgresql.UUID()).type == graphene.UUID
 
 
 def test_should_postgresql_enum_convert():