def setUp(self): """Sets up the client and verifies the version is compatible.""" self.client = create_client(self.TEST_SERVER_ADDR) version = self.client.any_client().check_version(pydgraph.Check()) # version.tag string format is v<MAJOR>.<MINOR>.<PATCH> # version_tup = [MAJOR, MINOR, PATCH] version_tup = version.tag[1:].split('.') version_supported = int(version_tup[0]) > 0 self.assertTrue( version_supported, 'Dgraph server version must be >= v1.0.0, got %s' % version.tag)
def test_close(self): client_stub = pydgraph.DgraphClientStub(addr=self.TEST_SERVER_ADDR) self.check_version(client_stub) client_stub.close() with self.assertRaises(Exception): client_stub.check_version(pydgraph.Check())
def test_timeout(self): with self.assertRaises(Exception): pydgraph.DgraphClientStub(self.TEST_SERVER_ADDR).check_version( pydgraph.Check(), timeout=-1)
def check_version(self, stub): self.validate_version_object(stub.check_version(pydgraph.Check()))