def test_edit_rules_add_rule(self):
SECGROUP_ID = '1'
RULE_ID = '1'
FROM_PORT = '-1'
TO_PORT = '-1'
IP_PROTOCOL = 'icmp'
CIDR = '0.0.0.0/0'
new_rule = self.mox.CreateMock(api.SecurityGroup)
new_rule.from_port = FROM_PORT
new_rule.to_port = TO_PORT
new_rule.ip_protocol = IP_PROTOCOL
new_rule.cidr = CIDR
new_rule.security_group_id = SECGROUP_ID
new_rule.id = RULE_ID
formData = {
'method': 'AddRule',
'tenant_id': self.TEST_TENANT,
'security_group_id': SECGROUP_ID,
'from_port': FROM_PORT,
'to_port': TO_PORT,
'ip_protocol': IP_PROTOCOL,
'cidr': CIDR
}
self.mox.StubOutWithMock(api, 'security_group_rule_create')
api.security_group_rule_create(IsA(http.HttpRequest),
SECGROUP_ID, IP_PROTOCOL, FROM_PORT, TO_PORT, CIDR)\
.AndReturn(new_rule)
self.mox.StubOutWithMock(messages, 'info')
messages.info(IsA(http.HttpRequest), IsA(str))
self.mox.ReplayAll()
res = self.client.post(
reverse('dash_security_groups_edit_rules',
args=[self.TEST_TENANT, SECGROUP_ID]), formData)
self.assertRedirectsNoFollow(
res,
reverse('dash_security_groups_edit_rules',
args=[self.TEST_TENANT, SECGROUP_ID]))
self.mox.VerifyAll()
def test_edit_rules_add_rule(self):
SECGROUP_ID = "1"
RULE_ID = "1"
FROM_PORT = "-1"
TO_PORT = "-1"
IP_PROTOCOL = "icmp"
CIDR = "0.0.0.0/0"
new_rule = self.mox.CreateMock(api.SecurityGroup)
new_rule.from_port = FROM_PORT
new_rule.to_port = TO_PORT
new_rule.ip_protocol = IP_PROTOCOL
new_rule.cidr = CIDR
new_rule.security_group_id = SECGROUP_ID
new_rule.id = RULE_ID
formData = {
"method": "AddRule",
"tenant_id": self.TEST_TENANT,
"security_group_id": SECGROUP_ID,
"from_port": FROM_PORT,
"to_port": TO_PORT,
"ip_protocol": IP_PROTOCOL,
"cidr": CIDR,
}
self.mox.StubOutWithMock(api, "security_group_rule_create")
api.security_group_rule_create(
IsA(http.HttpRequest), SECGROUP_ID, IP_PROTOCOL, FROM_PORT, TO_PORT, CIDR
).AndReturn(new_rule)
self.mox.StubOutWithMock(messages, "info")
messages.info(IsA(http.HttpRequest), IsA(str))
self.mox.ReplayAll()
res = self.client.post(
reverse("dash_security_groups_edit_rules", args=[self.TEST_TENANT, SECGROUP_ID]), formData
)
self.assertRedirectsNoFollow(
res, reverse("dash_security_groups_edit_rules", args=[self.TEST_TENANT, SECGROUP_ID])
)
self.mox.VerifyAll()
def test_edit_rules_add_rule(self):
SECGROUP_ID = '1'
RULE_ID = '1'
FROM_PORT = '-1'
TO_PORT = '-1'
IP_PROTOCOL = 'icmp'
CIDR = '0.0.0.0/0'
new_rule = self.mox.CreateMock(api.SecurityGroup)
new_rule.from_port = FROM_PORT
new_rule.to_port = TO_PORT
new_rule.ip_protocol = IP_PROTOCOL
new_rule.cidr = CIDR
new_rule.security_group_id = SECGROUP_ID
new_rule.id = RULE_ID
formData = {'method': 'AddRule',
'tenant_id': self.TEST_TENANT,
'security_group_id': SECGROUP_ID,
'from_port': FROM_PORT,
'to_port': TO_PORT,
'ip_protocol': IP_PROTOCOL,
'cidr': CIDR}
self.mox.StubOutWithMock(api, 'security_group_rule_create')
api.security_group_rule_create(IsA(http.HttpRequest),
SECGROUP_ID, IP_PROTOCOL, FROM_PORT, TO_PORT, CIDR)\
.AndReturn(new_rule)
self.mox.StubOutWithMock(messages, 'info')
messages.info(IsA(http.HttpRequest), IsA(basestring))
self.mox.ReplayAll()
res = self.client.post(reverse('dash_security_groups_edit_rules',
args=[self.TEST_TENANT, SECGROUP_ID]),
formData)
self.assertRedirectsNoFollow(res,
reverse('dash_security_groups_edit_rules',
args=[self.TEST_TENANT, SECGROUP_ID]))
self.mox.VerifyAll()
def test_edit_rules_add_rule_exception(self):
exception = novaclient_exceptions.ClientException(
'ClientException', message='ClientException')
SECGROUP_ID = '1'
RULE_ID = '1'
FROM_PORT = '-1'
TO_PORT = '-1'
IP_PROTOCOL = 'icmp'
CIDR = '0.0.0.0/0'
formData = {
'method': 'AddRule',
'tenant_id': self.TEST_TENANT,
'security_group_id': SECGROUP_ID,
'from_port': FROM_PORT,
'to_port': TO_PORT,
'ip_protocol': IP_PROTOCOL,
'cidr': CIDR
}
self.mox.StubOutWithMock(api, 'security_group_rule_create')
api.security_group_rule_create(IsA(http.HttpRequest), SECGROUP_ID,
IP_PROTOCOL, FROM_PORT, TO_PORT,
CIDR).AndRaise(exception)
self.mox.StubOutWithMock(messages, 'error')
messages.error(IsA(http.HttpRequest), IsA(str))
self.mox.ReplayAll()
res = self.client.post(
reverse('dash_security_groups_edit_rules',
args=[self.TEST_TENANT, SECGROUP_ID]), formData)
self.assertRedirectsNoFollow(
res,
reverse('dash_security_groups_edit_rules',
args=[self.TEST_TENANT, SECGROUP_ID]))
self.mox.VerifyAll()
def test_edit_rules_add_rule_exception(self):
exception = novaclient_exceptions.ClientException("ClientException", message="ClientException")
SECGROUP_ID = "1"
RULE_ID = "1"
FROM_PORT = "-1"
TO_PORT = "-1"
IP_PROTOCOL = "icmp"
CIDR = "0.0.0.0/0"
formData = {
"method": "AddRule",
"tenant_id": self.TEST_TENANT,
"security_group_id": SECGROUP_ID,
"from_port": FROM_PORT,
"to_port": TO_PORT,
"ip_protocol": IP_PROTOCOL,
"cidr": CIDR,
}
self.mox.StubOutWithMock(api, "security_group_rule_create")
api.security_group_rule_create(
IsA(http.HttpRequest), SECGROUP_ID, IP_PROTOCOL, FROM_PORT, TO_PORT, CIDR
).AndRaise(exception)
self.mox.StubOutWithMock(messages, "error")
messages.error(IsA(http.HttpRequest), IsA(str))
self.mox.ReplayAll()
res = self.client.post(
reverse("dash_security_groups_edit_rules", args=[self.TEST_TENANT, SECGROUP_ID]), formData
)
self.assertRedirectsNoFollow(
res, reverse("dash_security_groups_edit_rules", args=[self.TEST_TENANT, SECGROUP_ID])
)
self.mox.VerifyAll()
def test_edit_rules_add_rule_exception(self):
exception = novaclient_exceptions.ClientException('ClientException',
message='ClientException')
SECGROUP_ID = '1'
RULE_ID = '1'
FROM_PORT = '-1'
TO_PORT = '-1'
IP_PROTOCOL = 'icmp'
CIDR = '0.0.0.0/0'
formData = {'method': 'AddRule',
'tenant_id': self.TEST_TENANT,
'security_group_id': SECGROUP_ID,
'from_port': FROM_PORT,
'to_port': TO_PORT,
'ip_protocol': IP_PROTOCOL,
'cidr': CIDR}
self.mox.StubOutWithMock(api, 'security_group_rule_create')
api.security_group_rule_create(IsA(http.HttpRequest),
SECGROUP_ID, IP_PROTOCOL, FROM_PORT,
TO_PORT, CIDR).AndRaise(exception)
self.mox.StubOutWithMock(messages, 'error')
messages.error(IsA(http.HttpRequest), IsA(basestring))
self.mox.ReplayAll()
res = self.client.post(reverse('dash_security_groups_edit_rules',
args=[self.TEST_TENANT, SECGROUP_ID]),
formData)
self.assertRedirectsNoFollow(res,
reverse('dash_security_groups_edit_rules',
args=[self.TEST_TENANT, SECGROUP_ID]))
self.mox.VerifyAll()
def handle(self, request, data):
tenant_id = data['tenant_id']
try:
LOG.info('Add security_group_rule: "%s"' % data)
rule = api.security_group_rule_create(
request, data['security_group_id'], data['ip_protocol'],
data['from_port'], data['to_port'], data['cidr'])
messages.info(request, 'Successfully added rule: %s' \
% rule.id)
except novaclient_exceptions.ClientException, e:
LOG.error("ClientException in AddRule", exc_info=True)
messages.error(request,
'Error adding rule security group: %s' % e.message)
def handle(self, request, data):
tenant_id = data["tenant_id"]
try:
LOG.info('Add security_group_rule: "%s"' % data)
rule = api.security_group_rule_create(
request,
data["security_group_id"],
data["ip_protocol"],
data["from_port"],
data["to_port"],
data["cidr"],
)
messages.info(request, "Successfully added rule: %s" % rule.id)
except novaclient_exceptions.ClientException, e:
LOG.exception("ClientException in AddRule")
messages.error(request, "Error adding rule security group: %s" % e.message)
def handle(self, request, data):
tenant_id = data['tenant_id']
try:
LOG.info('Add security_group_rule: "%s"' % data)
rule = api.security_group_rule_create(request,
data['security_group_id'],
data['ip_protocol'],
data['from_port'],
data['to_port'],
data['cidr'])
messages.info(request, _('Successfully added rule: %s') \
% rule.id)
except novaclient_exceptions.ClientException, e:
LOG.exception("ClientException in AddRule")
messages.error(request, _('Error adding rule security group: %s')
% e.message)
