def test_server_warning(self): headers = [ ("warning", '''299 pilosa/2.0 "Deprecated PQL version: PQL v2 will remove support for SetBit() in Pilosa 2.1. Please update your client to support Set() (See https://docs.pilosa.com/pql#versioning)." "Sat, 25 Aug 2019 23:34:45 GMT"''') ] server = MockServer(200, headers=headers) with server: client = Client(server.uri) client.query(self.field.row(1))
def test_create_client(self): # create default client c = Client() self.assertEquals(URI(), c.cluster.hosts[0][0]) # create with cluster c = Client(Cluster(URI.address(":15000"))) self.assertEquals(URI.address(":15000"), c.cluster.hosts[0][0]) # create with URI c = Client(URI.address(":20000")) self.assertEquals(URI.address(":20000"), c.cluster.hosts[0][0]) # create with invalid type self.assertRaises(PilosaError, Client, 15000)
def test_failover_coordinator_fail(self): content = """ {"state":"NORMAL","nodes":[{"id":"827c7196-8875-4467-bee2-3604a4346f2b","uri":{"scheme":"%(SCHEME)s","host":"nonexistent","port":%(PORT)s},"isCoordinator":true}],"localID":"827c7196-8875-4467-bee2-3604a4346f2b"} """ server = MockServer(200, content=content, interpolate=True) with server: client = Client(server.uri) self.assertRaises(PilosaError, client.query, self.key_index.set_column_attrs("foo", {"foo": "bar"}))
def get_client(cls): return Client(SERVER_ADDRESS)
def test_failover_fail(self): uris = [URI.address("nonexistent%s" % i) for i in range(20)] client = Client(Cluster(*uris)) self.assertRaises(PilosaError, client.query, self.frame.bitmap(5))
def test_failed_connection(self): client = Client("http://non-existent-sub.pilosa.com:22222") self.assertRaises(PilosaError, client.query, self.frame.setbit(15, 10))
def get_client(cls): server_address = cls.get_server_address() return Client(server_address, tls_skip_verify=True)
def test_create_index_fail(self): server = MockServer(404) with server: client = Client(server.uri) self.assertRaises(PilosaServerError, client.create_index, self.index)
def test_fetch_coordinator_node_failure(self): server = MockServer(content=b'{"nodes":[]}') with server: client = Client(server.uri) self.assertRaises(PilosaError, client._fetch_coordinator_node)
def get_client(cls): import os server_address = os.environ.get("PILOSA_BIND", "") if not server_address: server_address = "http://:10101" return Client(server_address, tls_skip_verify=True)
def get_client(cls): # setting tls_client_certificate_path just for coverage, # it has no effect on non-https addresses return Client(SERVER_ADDRESS)