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)
