InMemoryDataSource should index by stringified IDs.

Author: jhgg

epoxy/contrib/relay/data_source/memory.py

@@ -1,5 +1,6 @@
 from collections import defaultdict
 from operator import attrgetter
+from six import text_type
 
 from ..connections.sorted_collection import SortedCollection
 from .base import BaseDataSource
@@ -11,11 +12,11 @@ def __init__(self):
         self.objects_by_type = defaultdict(lambda: SortedCollection(key=attrgetter('id')))
 
     def add(self, obj):
-        self.objects_by_type_and_id[obj.T][obj.id] = obj
+        self.objects_by_type_and_id[obj.T][text_type(obj.id)] = obj
         self.objects_by_type[obj.T].insert(obj)
 
     def fetch_node(self, object_type, id, resolve_info):
-        return self.objects_by_type_and_id[object_type].get(id)
+        return self.objects_by_type_and_id[object_type].get(text_type(id))
 
     def make_connection_resolver(self, relay, object_type_thunk):
         def resolver(obj, args, info):

tests/test_relay/test_relay_node.py

@@ -18,11 +18,11 @@ class Query(R.ObjectType):
     schema = R.Schema(R.Query)
 
     pets = {
-        '5': Pet(id='5', name='Garfield'),
-        '6': Pet(id='6', name='Odis')
+        5: Pet(id=5, name='Garfield'),
+        6: Pet(id=6, name='Odis')
     }
 
-    data = Query(pets=[pets['5'], pets['6']])
+    data = Query(pets=[pets[5], pets[6]])
     result = graphql(schema, '{ pets { id, name } }', data)
     assert result.data == {'pets': [{'id': 'UGV0OjU=', 'name': 'Garfield'}, {'id': 'UGV0OjY=', 'name': 'Odis'}]}
     assert not result.errors
@@ -48,11 +48,11 @@ def __init__(self, id, name):
             self.name = name
 
     pets = {
-        '5': MyPet(id='5', name='Garfield'),
-        '6': MyPet(id='6', name='Odis')
+        5: MyPet(id=5, name='Garfield'),
+        6: MyPet(id=6, name='Odis')
     }
 
-    data = Query(pets=[pets['5'], pets['6']])
+    data = Query(pets=[pets[5], pets[6]])
     result = graphql(schema, '{ pets { id, name } }', data)
     assert result.data == {'pets': [{'id': 'UGV0OjU=', 'name': 'Garfield'}, {'id': 'UGV0OjY=', 'name': 'Odis'}]}
     assert not result.errors
@@ -73,8 +73,8 @@ class Query(R.ObjectType):
 
     schema = R.Schema(R.Query)
 
-    data_source.add(Pet(id='5', name='Garfield'))
-    data_source.add(Pet(id='6', name='Odis'))
+    data_source.add(Pet(id=5, name='Garfield'))
+    data_source.add(Pet(id=6, name='Odis'))
 
     result = graphql(schema, '''
     {