class GremlinRestClientTestCase(unittest.TestCase): def setUp(self): self.client = GremlinRestClient() def test_request(self): resp = self.client.execute("1 + 1") self.assertEqual(resp.data[0], 2) def test_bindings(self): resp = self.client.execute("x + x", bindings={"x": 1}) self.assertEqual(resp.data[0], 2) def test_error(self): try: self.client.execute("x + x.fasdfewq", bindings={"x": 1}) error = False except GremlinServerError: error = True self.assertTrue(error)
def setUp(self): self.client = GremlinRestClient()