Fixed tests

Author: syrusakbary

graphene_django/tests/test_converter.py

@@ -124,10 +124,10 @@ class Meta:
     graphene_type = convert_django_field_with_choices(field)
     assert isinstance(graphene_type, graphene.Enum)
     assert graphene_type._meta.name == 'TranslatedModelLanguage'
-    assert graphene_type._meta.enum.__members__['SPANISH'].value == 'es'
-    assert graphene_type._meta.enum.__members__['SPANISH'].description == 'Spanish'
-    assert graphene_type._meta.enum.__members__['ENGLISH'].value == 'en'
-    assert graphene_type._meta.enum.__members__['ENGLISH'].description == 'English'
+    assert graphene_type._meta.enum.__members__['ES'].value == 'es'
+    assert graphene_type._meta.enum.__members__['ES'].description == 'Spanish'
+    assert graphene_type._meta.enum.__members__['EN'].value == 'en'
+    assert graphene_type._meta.enum.__members__['EN'].description == 'English'
 
 
 def test_field_with_grouped_choices():

graphene_django/tests/test_types.py

@@ -39,7 +39,7 @@ def test_django_interface():
 @patch('graphene_django.tests.models.Article.objects.get', return_value=Article(id=1))
 def test_django_get_node(get):
     article = Article.get_node(1, None, None)
-    get.assert_called_with(id=1)
+    get.assert_called_with(pk=1)
     assert article.id == 1
 
 
@@ -81,13 +81,13 @@ def test_schema_representation():
 }
 
 enum ArticleImportance {
-  VERY_IMPORTANT
-  NOT_AS_IMPORTANT
+  A_1
+  A_2
 }
 
 enum ArticleLang {
-  SPANISH
-  ENGLISH
+  ES
+  EN
 }
 
 scalar DateTime
@@ -114,8 +114,8 @@ def test_schema_representation():
 }
 
 enum ReporterA_choice {
-  THIS
-  THAT
+  A_1
+  A_2
 }
 
 type RootQuery {