def test_01_contructor(self):
# Verify that an error is risen when no addresses are
# configured (neither with host/port, nor with addrs).
with self.assertRaises(ConfigurationError):
tarantool.MeshConnection()
# Verify that a bad address given at initialization leads
# to an error.
bad_addrs = [
{"port": 1234}, # no host
{"host": "localhost"}, # no port
{"host": "localhost", "port": "1234"}, # port is str
]
for bad_addr in bad_addrs:
with self.assertRaises(ConfigurationError):
con = tarantool.MeshConnection(bad_addr.get('host'),
bad_addr.get('port'))
with self.assertRaises(ConfigurationError):
con = tarantool.MeshConnection(addrs=[bad_addr])
# Verify that identical addresses are squashed.
addrs = [{"host": "localhost", "port": 1234}]
con = tarantool.MeshConnection("localhost", 1234, addrs=addrs,
connect_now=False)
self.assertEqual(len(con.strategy.addrs), 1)
def test_00_basic(self):
def assert_srv_id(con, srv_id):
with warnings.catch_warnings():
warnings.simplefilter('ignore', category=NetworkWarning)
resp = con.call('srv_id')
self.assertEqual(resp.data and resp.data[0], srv_id)
con = tarantool.MeshConnection(addrs=[
{'host': self.host_1, 'port': self.port_1},
{'host': self.host_2, 'port': self.port_2},
], user='******', password='******')
# Response from instance#1.
assert_srv_id(con, 1)
# Stop instance#1 -- response from instance#2.
self.srv.stop()
assert_srv_id(con, 2)
# Start instance#1, stop instance#2 -- response from
# instance#1 again.
self.srv.start()
self.srv.admin('function srv_id() return 1 end')
self.srv2.stop()
assert_srv_id(con, 1)
# Stop instance #2 -- NetworkError (because we have no
# alive servers anymore).
self.srv.stop()
with self.assertRaises(NetworkError):
with warnings.catch_warnings():
warnings.simplefilter('ignore', category=NetworkWarning)
con.ping()
def test_06_reconnection(self):
# Create a mesh connection; pass only the first server
# address.
con = tarantool.MeshConnection(
self.host_1, self.port_1, user='******', password='******',
cluster_discovery_function=self.get_all_nodes_func_name)
con.last_nodes_refresh = 0
resp = con.call('srv_id')
self.assertEqual(resp.data and resp.data[0], 1)
# Verify that the last discovery was successful and the
# strategy has 2 addresses.
self.assertEqual(len(con.strategy.addrs), 2)
self.srv.stop()
# Verify that we switched to the second server.
with warnings.catch_warnings():
# Suppress reconnection warnings.
warnings.simplefilter("ignore")
resp = con.call('srv_id')
self.assertEqual(resp.data and resp.data[0], 2)
con.close()
def test_05_discovery_delay(self):
# Create a mesh connection, pass only the first server address.
con = tarantool.MeshConnection(
self.host_1, self.port_1, user='******', password='******',
cluster_discovery_function=self.get_all_nodes_func_name,
cluster_discovery_delay=1)
# Verify that the strategy has two addresses come from
# the function right after connecting.
self.assertEqual(len(con.strategy.addrs), 2)
# Drop addresses list to the initial state.
con.strategy.update([con.strategy.addrs[0], ])
# Verify that the discovery will not be performed until
# 'cluster_discovery_delay' seconds will be passed.
con.ping()
self.assertEqual(len(con.strategy.addrs), 1)
sleep(1.1)
# Refresh after cluster_discovery_delay.
con.ping()
self.assertEqual(len(con.strategy.addrs), 2)
con.close()
def test_02_discovery_bad_address(self):
retvals = [
"",
"1",
"'localhost:1234'",
"{}",
"error('raise an error')",
"{'localhost:foo'}",
"{'localhost:0'}",
"{'localhost:65536'}",
"{'localhost:1234:5678'}",
"{':1234'}",
"{'localhost:'}",
]
for retval in retvals:
func_name = 'bad_cluster_discovery'
self.define_custom_cluster_function(func_name, retval)
con = tarantool.MeshConnection(self.host_1, self.port_1,
user='******', password='******')
con.cluster_discovery_function = func_name
# Verify that a cluster discovery (that is triggered
# by ping) give one or two warnings.
with warnings.catch_warnings(record=True) as ws:
con.ping()
self.assertTrue(len(ws) in (1, 2))
for w in ws:
self.assertIs(w.category, ClusterDiscoveryWarning)
# Verify that incorrect or empty result was discarded.
self.assertEqual(len(con.strategy.addrs), 1)
self.assertEqual(con.strategy.addrs[0]['host'], self.host_1)
self.assertEqual(con.strategy.addrs[0]['port'], self.port_1)
con.close()
def test_04_discovery_add_address(self):
# Create a mesh connection; pass only the first server
# address.
con = tarantool.MeshConnection(
self.host_1,
self.port_1,
user='******',
password='******',
cluster_discovery_function=self.get_all_nodes_func_name,
connect_now=False)
# Verify that the strategy has one address that comes from
# the constructor arguments.
self.assertEqual(len(con.strategy.addrs), 1)
con.connect()
# Verify that we work with the first server.
resp = con.call('srv_id')
self.assertEqual(resp.data and resp.data[0], 1)
# Verify that the refresh was successful and the strategy
# has 2 addresses.
self.assertEqual(len(con.strategy.addrs), 2)
con.close()
def test_07_discovery_exclude_address(self):
# Define function to get back only second server.
func_name = 'get_second_node'
self.define_cluster_function(func_name, [self.srv2])
# Create a mesh connection, pass only the first server address.
con = tarantool.MeshConnection(
self.host_1, self.port_1, user='******', password='******',
cluster_discovery_function=func_name)
# Verify that discovery was successful and the strategy
# has 1 address.
self.assertEqual(len(con.strategy.addrs), 1)
# Verify that the current server is second one.
resp = con.call('srv_id')
self.assertEqual(resp.data and resp.data[0], 2)
con.close()
def test_03_discovery_bad_good_addresses(self):
func_name = 'bad_and_good_addresses'
retval = "{'localhost:', '%s:%d'}" % (self.host_2, self.port_2)
self.define_custom_cluster_function(func_name, retval)
con = tarantool.MeshConnection(self.host_1, self.port_1,
user='******', password='******')
con.cluster_discovery_function = func_name
# Verify that a cluster discovery (that is triggered
# by ping) give one warning.
with warnings.catch_warnings(record=True) as ws:
con.ping()
self.assertEqual(len(ws), 1)
self.assertIs(ws[0].category, ClusterDiscoveryWarning)
# Verify that only second address was accepted.
self.assertEqual(len(con.strategy.addrs), 1)
self.assertEqual(con.strategy.addrs[0]['host'], self.host_2)
self.assertEqual(con.strategy.addrs[0]['port'], self.port_2)
con.close()
