def testUnknownProperty_raisesException(self):
with self.assertRaises(Exception) as context:
prop = SomeWeirdUnknownProperty()
prop._code_name = "my_prop"
convert_ndb_property(prop)
self.assertTrue("Don't know how to convert" in context.exception.message, msg=context.exception.message)
def testUnknownProperty_raisesException(self):
with self.assertRaises(Exception) as context:
prop = SomeWeirdUnknownProperty()
prop._code_name = "my_prop"
convert_ndb_property(prop)
self.assertTrue("Don't know how to convert" in context.exception.message, msg=context.exception.message)
def testNoneResult_raisesException(self, patch_convert):
from graphene_gae.ndb.converter import convert_ndb_property
patch_convert.get.return_value = lambda _, __: None
with self.assertRaises(Exception) as context:
prop = ndb.StringProperty()
prop._code_name = "my_prop"
convert_ndb_property(prop)
self.assertTrue("Failed to convert NDB field my_prop" in context.exception.message, msg=context.exception.message)
def testNoneResult_raisesException(self, patch_convert):
from graphene_gae.ndb.converter import convert_ndb_property
patch_convert.get.return_value = lambda _: None
with self.assertRaises(Exception) as context:
prop = ndb.StringProperty()
prop._code_name = "my_prop"
convert_ndb_property(prop)
expected_message = 'Failed to convert NDB propeerty to a GraphQL field my_prop (StringProperty())'
self.assertTrue(expected_message in context.exception.message, msg=context.exception.message)
def testKeyProperty_withSuffix(self):
my_registry = Registry()
class User(ndb.Model):
name = ndb.StringProperty()
class UserType(NdbObjectType):
class Meta:
model = User
registry = my_registry
prop = ndb.KeyProperty(kind='User')
prop._code_name = 'user_key'
conversion = convert_ndb_property(prop, my_registry)
self.assertLength(conversion, 2)
self.assertEqual(conversion[0].name, 'user_id')
self.assertIsInstance(conversion[0].field, DynamicNdbKeyStringField)
_type = conversion[0].field.get_type()
self.assertIsInstance(_type, NdbKeyStringField)
self.assertEqual(_type._type, String)
self.assertEqual(conversion[1].name, 'user')
self.assertIsInstance(conversion[1].field, DynamicNdbKeyReferenceField)
_type = conversion[1].field.get_type()
self.assertIsInstance(_type, NdbKeyReferenceField)
self.assertEqual(_type._type, UserType)
def testStringProperty_required_shouldConvertToList(self):
ndb_prop = ndb.StringProperty(required=True)
result = convert_ndb_property(ndb_prop)
graphene_type = result.field._type
self.assertIsInstance(graphene_type, graphene.NonNull)
self.assertEqual(graphene_type.of_type, graphene.String)
def testKeyProperty_withSuffix(self):
my_registry = Registry()
class User(ndb.Model):
name = ndb.StringProperty()
class UserType(NdbObjectType):
class Meta:
model = User
registry = my_registry
prop = ndb.KeyProperty(kind='User')
prop._code_name = 'user_key'
conversion = convert_ndb_property(prop, my_registry)
self.assertLength(conversion, 2)
self.assertEqual(conversion[0].name, 'user_id')
self.assertIsInstance(conversion[0].field, DynamicNdbKeyStringField)
_type = conversion[0].field.get_type()
self.assertIsInstance(_type, NdbKeyStringField)
self.assertEqual(_type._type, String)
self.assertEqual(conversion[1].name, 'user')
self.assertIsInstance(conversion[1].field, DynamicNdbKeyReferenceField)
_type = conversion[1].field.get_type()
self.assertIsInstance(_type, NdbKeyReferenceField)
self.assertEqual(_type._type, UserType)
def testStringProperty_required_shouldConvertToList(self):
ndb_prop = ndb.StringProperty(required=True)
result = convert_ndb_property(ndb_prop)
graphene_type = result.field._type
self.assertIsInstance(graphene_type, graphene.NonNull)
self.assertEqual(graphene_type.of_type, graphene.String)
def testStringProperty_repeated_shouldConvertToList(self):
ndb_prop = ndb.StringProperty(repeated=True)
conversion_result = convert_ndb_property(ndb_prop)
graphene_type = conversion_result.field
self.assertIsInstance(graphene_type, graphene.List)
self.assertIsInstance(graphene_type.of_type, graphene.String)
def fields_for_ndb_model(options):
ndb_model = options.model
only_fields = options.only_fields
already_created_fields = {
name
for name, _ in options.local_fields.iteritems()
}
ndb_fields = OrderedDict()
for prop_name, prop in ndb_model._properties.iteritems():
name = prop._code_name
is_not_in_only = only_fields and name not in only_fields
is_excluded = name in options.exclude_fields or name in already_created_fields
if is_not_in_only or is_excluded:
continue
results = convert_ndb_property(prop)
if not results:
continue
if not isinstance(results, list):
results = [results]
for r in results:
ndb_fields[r.name] = r.field
ndb_fields['ndb_id'] = Field(
String, resolver=lambda entity, *_: str(entity.key.id()))
return ndb_fields
def testKeyProperty_withSuffix_repeated(self):
class User(ndb.Model):
name = ndb.StringProperty()
class UserType(NdbObjectType):
class Meta:
model = User
prop = ndb.KeyProperty(kind='User', repeated=True)
prop._code_name = 'user_keys'
conversion = convert_ndb_property(prop)
self.assertLength(conversion, 2)
self.assertEqual(conversion[0].name, 'user_ids')
self.assertIsInstance(conversion[0].field, DynamicNdbKeyStringField)
_type = conversion[0].field.get_type()
self.assertIsInstance(_type, NdbKeyStringField)
self.assertIsInstance(_type._type, List)
self.assertEqual(_type._type.of_type, String)
self.assertEqual(conversion[1].name, 'users')
self.assertIsInstance(conversion[1].field, DynamicNdbKeyReferenceField)
_type = conversion[1].field.get_type()
self.assertIsInstance(_type, NdbKeyReferenceField)
self.assertIsInstance(_type._type, List)
self.assertEqual(_type._type.of_type, UserType)
def testKeyProperty_repeatedPlural_withSuffixRemoval_removesSuffixAndPluralName(self):
prop = ndb.KeyProperty()
prop._code_name = "user_keys"
conversion = convert_ndb_property(prop)
self.assertEqual(conversion.name, "users")
prop = ndb.KeyProperty()
prop._code_name = "tag_name_keys"
conversion = convert_ndb_property(prop)
self.assertEqual(conversion.name, "tag_names")
prop = ndb.KeyProperty()
prop._code_name = "person_keys"
conversion = convert_ndb_property(prop)
self.assertEqual(conversion.name, "people")
prop = ndb.KeyProperty()
prop._code_name = "universal_category_keys"
conversion = convert_ndb_property(prop)
self.assertEqual(conversion.name, "universal_categories")
def __assert_conversion(self, ndb_property_type, expected_graphene_type,
*args, **kwargs):
ndb_property = ndb_property_type(*args, **kwargs)
result = convert_ndb_property(ndb_property)
graphene_field = result.field
self.assertEqual(graphene_field._type, expected_graphene_type)
def __assert_conversion(self, ndb_property_type, expected_graphene_type, *args, **kwargs):
ndb_property = ndb_property_type(*args, **kwargs)
result = convert_ndb_property(ndb_property)
graphene_field = result.field
self.assertEqual(graphene_field._type, expected_graphene_type)
def testKeyProperty_withSuffixRemoval_removesSuffix(self):
prop = ndb.KeyProperty()
prop._code_name = "user_key"
conversion = convert_ndb_property(prop)
self.assertEqual(conversion.name, "user")
def __assert_conversion(self, ndb_property_type, expected_graphene_type, *args, **kwargs):
ndb_property = ndb_property_type(*args, **kwargs)
conversion_result = convert_ndb_property(ndb_property)
graphene_field_type = conversion_result.field
self.assertIsInstance(graphene_field_type, expected_graphene_type)