def set_allowed_hosts(self):
""" First operation will fail and the second operation will succeed"""
allowedhosts = "[\"*.couchbase.com\",\"10.112.0.0/16\",\"172.23.0.0/16\"]"
host = "[\"*.couchbase.com\"]"
for node in self.cluster.nodes_in_cluster:
shell = RemoteMachineShellConnection(node)
output = shell.set_allowedhosts("localhost", self.cluster.master.rest_username,
self.cluster.master.rest_password, host)
self.log.info("expected failure from set_allowedhosts {0}".format(output))
if "errors" not in output[0]:
self.fail("Invalid address should fail, address {0}".format(host))
output = shell.set_allowedhosts("localhost", self.cluster.master.rest_username,
self.cluster.master.rest_password, allowedhosts)
if len(output) > 2:
self.fail("Allowed hosts is not changed and error is {0}".format(output))
output = shell.get_allowedhosts(self.cluster.master.rest_username,
self.cluster.master.rest_password)
self.assertEqual(output, allowedhosts)
shell.disconnect()
def test_allowed_hosts_rest_apis(self):
self.__create_user("user1", "password", "data_writer[*]:admin")
self.__create_user("user2", "password", "data_writer[*]", False)
for node in self.cluster.nodes_in_cluster:
shell = RemoteMachineShellConnection(node)
for user in self.users_list:
if user["Writer"]:
output = shell.set_allowedhosts("localhost", user["name"],
user["password"],
self.allowedhosts)
if len(output[0]) > 2:
self.fail(
"Allowed hosts is not changed and error is {0}".
format(output))
output = shell.get_allowedhosts(user["name"],
user["password"])
self.assertEqual(output, self.allowedhosts)
shell.disconnect()
self.add_node_and_rebalance()
