def test_show_security_group_not_found(self, m_list):
tenant_id = fakes.tenants["baz"]["id"]
sec_group = openstack_helper.build_security_group_from_neutron(
fakes.security_groups[tenant_id])
m_list.return_value = sec_group
req = fakes.create_req_test(None, None)
self.assertRaises(exception.NotFound, self.controller.show, req, None)
def test_list_security_group_empty(self, m_list):
tenant_id = fakes.tenants["foo"]["id"]
sec_group = openstack_helper.build_security_group_from_neutron(
fakes.security_groups[tenant_id])
req = fakes.create_req_test(None, None)
m_list.return_value = sec_group
result = self.controller.index(req)
self.assertEqual(result.resources.__len__(), 0)
def test_list_security_group_empty(self, m_list):
tenant_id = fakes.tenants["foo"]["id"]
sec_group = openstack_helper.build_security_group_from_neutron(
fakes.security_groups[tenant_id]
)
req = fakes.create_req_test(None, None)
m_list.return_value = sec_group
result = self.controller.index(req)
self.assertEqual(result.resources.__len__(), 0)
def test_list_security_group(self, m_list):
tenant_id = fakes.tenants["baz"]["id"]
sec_group = fakes.security_groups[tenant_id]
expected = openstack_helper.build_security_group_from_neutron(
sec_group)
m_list.return_value = sec_group
ret = self.helper.list_security_groups(None)
self.assertEqual(2, ret.__len__())
self.assertEqual(expected, ret)
m_list.assert_called_with(None, 'security-groups',
response_resource="security_groups")
def test_show_security_group_not_found(self, m_list):
tenant_id = fakes.tenants["baz"]["id"]
sec_group = openstack_helper.build_security_group_from_neutron(
fakes.security_groups[tenant_id]
)
m_list.return_value = sec_group
req = fakes.create_req_test(None, None)
self.assertRaises(exception.NotFound,
self.controller.show,
req,
None)
def test_show_security_group(self, m_list):
tenant_id = fakes.tenants["baz"]["id"]
sec_group = fakes.security_groups[tenant_id][0]
expected = openstack_helper.build_security_group_from_neutron(
[sec_group])[0]
list_sec = sec_group
m_list.return_value = list_sec
ret = self.helper.get_security_group_details(None, None)
self.assertEqual(expected, ret)
m_list.assert_called_with(None, 'security-groups', None,
response_resource="security_group")
def test_list_security_group(self, m_list):
tenant_id = fakes.tenants["baz"]["id"]
sec_group = fakes.security_groups[tenant_id]
expected = openstack_helper.build_security_group_from_neutron(
sec_group)
m_list.return_value = sec_group
ret = self.helper.list_security_groups(None)
self.assertEqual(2, ret.__len__())
self.assertEqual(expected, ret)
m_list.assert_called_with(None,
'security-groups',
response_resource="security_groups")
def test_list_security_group(self, m_list):
tenant_id = fakes.tenants["baz"]["id"]
sec_group = openstack_helper.build_security_group_from_neutron(
fakes.security_groups[tenant_id])
req = fakes.create_req_test(None, None)
m_list.return_value = sec_group
result = self.controller.index(req)
expected = self.controller._get_security_group_resources(sec_group)
self.assertEqual(result.resources.__len__(), expected.__len__())
for r in result.resources:
self.assertIsInstance(r, occi_security_group.SecurityGroupResource)
m_list.assert_called_with(req)
def test_show_security_group(self, m_list):
tenant_id = fakes.tenants["baz"]["id"]
sec_group = openstack_helper.build_security_group_from_neutron(
[fakes.security_groups[tenant_id][0]])
req = fakes.create_req_test(None, None)
m_list.return_value = sec_group[0]
result = self.controller.show(req, None)
expected = self.controller._get_security_group_resources(sec_group)[0]
self.assertIsInstance(result,
occi_security_group.SecurityGroupResource)
self.assertEqual(result, expected)
m_list.assert_called_with(req, None)
def test_show_security_group(self, m_list):
tenant_id = fakes.tenants["baz"]["id"]
sec_group = fakes.security_groups[tenant_id][0]
expected = openstack_helper.build_security_group_from_neutron(
[sec_group])[0]
list_sec = sec_group
m_list.return_value = list_sec
ret = self.helper.get_security_group_details(None, None)
self.assertEqual(expected, ret)
m_list.assert_called_with(None,
'security-groups',
None,
response_resource="security_group")
def test_show_security_group(self, m_list):
tenant_id = fakes.tenants["baz"]["id"]
sec_group = openstack_helper.build_security_group_from_neutron(
[fakes.security_groups[tenant_id][0]]
)
req = fakes.create_req_test(None, None)
m_list.return_value = sec_group[0]
result = self.controller.show(req, None)
expected = self.controller._get_security_group_resources(sec_group)[0]
self.assertIsInstance(
result,
occi_security_group.SecurityGroupResource)
self.assertEqual(result, expected)
m_list.assert_called_with(req, None)
def test_list_security_group(self, m_list):
tenant_id = fakes.tenants["baz"]["id"]
sec_group = openstack_helper.build_security_group_from_neutron(
fakes.security_groups[tenant_id]
)
req = fakes.create_req_test(None, None)
m_list.return_value = sec_group
result = self.controller.index(req)
expected = self.controller._get_security_group_resources(sec_group)
self.assertEqual(result.resources.__len__(),
expected.__len__())
for r in result.resources:
self.assertIsInstance(r, occi_security_group.SecurityGroupResource)
m_list.assert_called_with(req)
def list_security_groups(self, req):
"""List security groups
:param req: the incoming request
"""
try:
secgroup = self.list_resources(req, 'security-groups',
response_resource="security_groups")
ooi_sec = os_helpers.build_security_group_from_neutron(
secgroup
)
return ooi_sec
except Exception:
raise exception.NotFound()
def get_security_group_details(self, req, sec_id):
"""Get info about a security group.
:param req: the incoming request
:param sec_id: security group id to show
"""
try:
secgroup = self.get_resource(req, 'security-groups', sec_id,
response_resource="security_group")
ooi_sec = os_helpers.build_security_group_from_neutron(
[secgroup]
)
return ooi_sec[0]
except Exception:
raise exception.NotFound()
def test_create_security_groups(self, m_tenant, m_create):
tenant_id = fakes.tenants["baz"]["id"]
sec_group = fakes.security_groups[tenant_id][0]
expected = openstack_helper.build_security_group_from_neutron(
[sec_group])[0]
m_tenant.return_value = uuid.uuid4().hex
group_info = {"name": sec_group["name"], "id": sec_group["id"],
"description": sec_group["description"]}
rules_out_1 = sec_group["security_group_rules"][0]
rules_out_2 = sec_group["security_group_rules"][1]
m_create.side_effect = [group_info, rules_out_1, rules_out_2]
ret = self.helper.create_security_group(None, expected["title"],
expected["summary"],
expected["rules"])
self.assertEqual(expected, ret)
self.assertEqual(3, m_create.call_count)
def create_security_group(self, req, name, description, rules):
"""Create security group
:param req: the incoming request
:param name: security group name
:param description: security group description
:param rules: security group rules
"""
try:
tenant_id = self.tenant_from_req(req)
param_group = {"tenant_id": tenant_id,
"description": description,
"name": name,
}
secgroup = self.create_resource(
req, 'security-groups', param_group,
response_resource="security_group")
sec_id = secgroup["id"]
secgroup["security_group_rules"] = []
for rule in rules:
port_min, port_max = os_helpers.security_group_rule_port(
rule["port"]
)
param_rule = {
"ethertype": rule.get("ipversion", "IPv4"),
"port_range_max": port_max,
"port_range_min": port_min,
"direction": os_helpers.security_group_rule_type(
rule["type"]),
"remote_ip_prefix": rule.get("range", "0.0.0.0/0"),
"protocol": rule["protocol"],
"security_group_id": sec_id,
}
secrule = self.create_resource(
req,
'security-group-rules', param_rule,
response_resource="security_group_rule")
secgroup["security_group_rules"].append(secrule)
ooi_sec = os_helpers.build_security_group_from_neutron(
[secgroup]
)
return ooi_sec[0]
except Exception as ex:
raise ex
def test_create_security_groups(self, m_tenant, m_create):
tenant_id = fakes.tenants["baz"]["id"]
sec_group = fakes.security_groups[tenant_id][0]
expected = openstack_helper.build_security_group_from_neutron(
[sec_group])[0]
m_tenant.return_value = uuid.uuid4().hex
group_info = {
"name": sec_group["name"],
"id": sec_group["id"],
"description": sec_group["description"]
}
rules_out_1 = sec_group["security_group_rules"][0]
rules_out_2 = sec_group["security_group_rules"][1]
m_create.side_effect = [group_info, rules_out_1, rules_out_2]
ret = self.helper.create_security_group(None, expected["title"],
expected["summary"],
expected["rules"])
self.assertEqual(expected, ret)
self.assertEqual(3, m_create.call_count)
def test_create_security_groups(self, m_create):
tenant_id = fakes.tenants["baz"]["id"]
sec_group = openstack_helper.build_security_group_from_neutron(
fakes.security_groups[tenant_id])[0]
params = {
"occi.core.title": sec_group["title"],
"occi.securitygroup.rules": sec_group["rules"],
"occi.core.summary": sec_group["summary"]
}
categories = {occi_security_group.SecurityGroupResource.kind}
req = fakes.create_req_json_occi(params, categories)
m_create.return_value = sec_group
ret = self.controller.create(req, params)
expected = self.controller._get_security_group_resources([sec_group])
self.assertIsInstance(ret.resources[0],
occi_security_group.SecurityGroupResource)
self.assertEqual(expected[0], ret.resources[0])
m_create.assert_called_with(req, sec_group["title"],
sec_group["summary"], sec_group["rules"])
def test_create_security_groups(self, m_create):
tenant_id = fakes.tenants["baz"]["id"]
sec_group = openstack_helper.build_security_group_from_neutron(
fakes.security_groups[tenant_id]
)[0]
params = {"occi.core.title": sec_group["title"],
"occi.securitygroup.rules": sec_group["rules"],
"occi.core.summary": sec_group["summary"]
}
categories = {occi_security_group.SecurityGroupResource.kind}
req = fakes.create_req_json_occi(params, categories)
m_create.return_value = sec_group
ret = self.controller.create(req, params)
expected = self.controller._get_security_group_resources(
[sec_group])
self.assertIsInstance(ret.resources[0],
occi_security_group.SecurityGroupResource)
self.assertEqual(expected[0], ret.resources[0])
m_create.assert_called_with(req, sec_group["title"],
sec_group["summary"],
sec_group["rules"])