Added datetime tests

Author: syrusakbary

graphene/types/tests/test_datetime.py

@@ -0,0 +1,41 @@
+import datetime
+import pytz
+
+from ..datetime import DateTime
+from ..objecttype import ObjectType
+from ..schema import Schema
+
+
+class Query(ObjectType):
+    datetime = DateTime(_in=DateTime(name='in'))
+
+    def resolve_datetime(self, args, context, info):
+        _in = args.get('in')
+        return _in
+
+schema = Schema(query=Query)
+
+
+def test_datetime_query():
+    now = datetime.datetime.now().replace(tzinfo=pytz.utc)
+    isoformat = now.isoformat()
+
+    result = schema.execute('''{ datetime(in: "%s") }'''%isoformat)
+    assert not result.errors
+    assert result.data == {
+        'datetime': isoformat
+    }
+
+
+def test_datetime_query_variable():
+    now = datetime.datetime.now().replace(tzinfo=pytz.utc)
+    isoformat = now.isoformat()
+
+    result = schema.execute(
+        '''query Test($date: DateTime){ datetime(in: $date) }''',
+        variable_values={'date': isoformat}
+    )
+    assert not result.errors
+    assert result.data == {
+        'datetime': isoformat
+    }

setup.py

@@ -78,6 +78,8 @@ def run_tests(self):
         'pytest>=2.7.2',
         'pytest-benchmark',
         'mock',
+        'pytz',
+        'iso8601',
     ],
     extras_require={
         'django': [