def test_get_proper_message(self):
limits = LimitsMixin()
limits.messages = [
{"type": "min", "value": "1", "message": "Message for min_1"},
{"type": "min", "value": "2", "message": "Message for min_2"},
{"type": "max", "value": "5", "message": "Message for max_5"},
{"type": "max", "value": "8", "message": "Message for max_8"},
]
self.assertEqual(limits._get_message("min"), "Message for min_2")
self.assertEqual(limits._get_message("max"), "Message for max_5")
def test_get_proper_message(self):
limits = LimitsMixin()
limits.messages = [
{'type': 'min', 'value': '1', 'message': 'Message for min_1'},
{'type': 'min', 'value': '2', 'message': 'Message for min_2'},
{'type': 'max', 'value': '5', 'message': 'Message for max_5'},
{'type': 'max', 'value': '8', 'message': 'Message for max_8'}
]
self.assertEqual(
limits._get_message('min'), 'Message for min_2')
self.assertEqual(
limits._get_message('max'), 'Message for max_5')
def test_get_proper_message(self):
limits = LimitsMixin()
limits.messages = [
{'type': 'min', 'value': '1', 'message': 'Message for min_1'},
{'type': 'min', 'value': '2', 'message': 'Message for min_2'},
{'type': 'max', 'value': '5', 'message': 'Message for max_5'},
{'type': 'max', 'value': '8', 'message': 'Message for max_8'}
]
self.assertEqual(
limits._get_message('min'), 'Message for min_2')
self.assertEqual(
limits._get_message('max'), 'Message for max_5')
def test_check_node_limits(self):
roles = self.data.get('roles_meta')
attributes = self.data.get('attributes')
for role, data in six.iteritems(roles):
result = LimitsMixin().check_node_limits(
models={'settings': attributes},
nodes=self.env.nodes,
role=role,
limits=data.get('limits'))
if role == 'cinder':
self.assertTrue(result.get('valid'))
if role == 'controller':
self.assertFalse(result.get('valid'))
self.assertEqual(
result.get('messages'),
'Default message')
if role == 'mongo':
self.assertFalse(result.get('valid'))
self.assertEqual(
result.get('messages'),
'At least 4 MongoDB node can be added if spam')
def test_check_override(self):
roles = self.data.get("roles_meta")
attributes = self.data.get("attributes")
limits = LimitsMixin()
# Set nodes count to 4
limits.count = 4
limits.limit_reached = True
limits.models = {"settings": attributes}
limits.nodes = self.env.nodes
# Set "cinder" role to working on
limits.role = "cinder"
limits.limit_types = ["max"]
limits.checked_limit_types = {}
limits.limit_values = {"max": None}
override_data = roles["cinder"]["limits"]["overrides"][0]
result = limits._check_override(override_data)
self.assertEqual(result[0]["message"], "At most one role_1 node can be added")
def test_check_override(self):
roles = self.data.get('roles_meta')
attributes = self.data.get('attributes')
limits = LimitsMixin()
# Set nodes count to 4
limits.count = 4
limits.limit_reached = True
limits.models = {'settings': attributes}
limits.nodes = self.env.nodes
# Set "cinder" role to working on
limits.role = 'cinder'
limits.limit_types = ['max']
limits.checked_limit_types = {}
limits.limit_values = {'max': None}
override_data = roles['cinder']['limits']['overrides'][0]
result = limits._check_override(override_data)
self.assertEqual(
result[0]['message'], 'At most one role_1 node can be added')
def test_check_node_limits(self):
roles = self.data.get('roles_meta')
attributes = self.data.get('attributes')
for role, data in six.iteritems(roles):
result = LimitsMixin().check_node_limits(
models={'settings': attributes},
nodes=self.env.nodes,
role=role,
limits=data.get('limits'))
if role == 'cinder':
self.assertTrue(result.get('valid'))
if role == 'controller':
self.assertFalse(result.get('valid'))
self.assertEqual(
result.get('messages'),
'Default message')
if role == 'mongo':
self.assertFalse(result.get('valid'))
self.assertEqual(
result.get('messages'),
'At least 4 MongoDB node can be added if spam')
def test_check_node_limits(self):
roles = self.data.get("roles_meta")
attributes = self.data.get("attributes")
for role, data in six.iteritems(roles):
result = LimitsMixin().check_node_limits(
models={"settings": attributes}, nodes=self.env.nodes, role=role, limits=data.get("limits")
)
if role == "cinder":
self.assertTrue(result.get("valid"))
if role == "controller":
self.assertFalse(result.get("valid"))
self.assertEqual(result.get("messages"), "Default message")
if role == "mongo":
self.assertFalse(result.get("valid"))
self.assertEqual(result.get("messages"), "At least 4 MongoDB node can be added if spam")
def test_check_override(self):
roles = self.data.get('roles_meta')
attributes = self.data.get('attributes')
limits = LimitsMixin()
# Set nodes count to 4
limits.count = 4
limits.limit_reached = True
limits.models = {'settings': attributes}
limits.nodes = self.env.nodes
# Set "cinder" role to working on
limits.role = 'cinder'
limits.limit_types = ['max']
limits.checked_limit_types = {}
limits.limit_values = {'max': None}
override_data = roles['cinder']['limits']['overrides'][0]
result = limits._check_override(override_data)
self.assertEqual(
result[0]['message'], 'At most one role_1 node can be added')