Add integration test

Author: all-seeing-code

pydgraph/client_stub.py

@@ -19,7 +19,10 @@
 from pydgraph.meta import VERSION
 from pydgraph.proto import api_pb2_grpc as api_grpc
 
-from urllib.parse import urlparse
+try:
+    from urllib.parse import urlparse
+except ImportError:
+     from urlparse import urlparse
 
 __author__ = 'Garvit Pahal <[email protected]>'
 __maintainer__ = 'Martin Martinez Rivera <[email protected]>'

pydgraph/txn.py

@@ -192,12 +192,12 @@ def create_mutation(self, mutation=None, set_obj=None, del_obj=None,
         return mutation
 
     def create_request(self, query=None, variables=None, mutations=None, commit_now=None, resp_format="JSON"):
-        switch(resp_format) {
-        case "RDF":
-            resp_format = api.Request.RespFormat.RDF
-        default:
+        if resp_format=="JSON":
             resp_format = api.Request.RespFormat.JSON
-        };
+        elif resp_format =="RDF":
+            resp_format = api.Request.RespFormat.RDF
+        else: raise errors.TransactionError('Response format should be either RDF or JSON')
+
         """Creates a request object"""
         request = api.Request(start_ts = self._ctx.start_ts, commit_now=commit_now,
                               read_only=self._read_only, best_effort=self._best_effort, resp_format = resp_format)

tests/test_queries.py

@@ -69,6 +69,14 @@ def test_mutation_and_query(self):
             }
         }"""
 
+        queryRDF = """query q($a: string) {
+            q(func: anyofterms(name, "Alice"))
+            {
+                uid
+                name
+            }
+        }"""
+
         response = self.client.txn().query(query, variables={'$a': 'Alice'})
         self.assertEqual([{'name': 'Alice', 'follows': [{'name': 'Greg'}]}],
                          json.loads(response.json).get('me'))
@@ -79,6 +87,12 @@ def test_mutation_and_query(self):
         self.assertTrue(is_number(response.latency.encoding_ns),
                         'Encoding latency is not available')
 
+        """ Run query with JSON and RDF resp_format and verify the result """
+        response = self.client.txn().query(queryRDF, variables={'$a': 'Alice'})
+        uid = json.loads(response.json).get('q')[0]['uid']
+        expected_rdf = "<{}> <name> \"Alice\" .\n".format(uid)
+        response = self.client.txn().query(queryRDF, variables={'$a': 'Alice'}, resp_format="RDF")
+        self.assertEqual(expected_rdf,response.rdf)
 
 def is_number(number):
     """Returns true if object is a number. Compatible with Python 2 and 3."""