35 lines
1.1 KiB
Python
35 lines
1.1 KiB
Python
import unittest
|
|
|
|
from nova.virt.ec2.openstack_rule_transformer import OpenstackRuleTransformer
|
|
|
|
|
|
class TestOpenstackRuleTransformer(unittest.TestCase):
|
|
|
|
def setUp(self):
|
|
self.openstack_rule_transformer = OpenstackRuleTransformer()
|
|
|
|
def test_should_copy_to_port_as_string(self):
|
|
openstack_rule = {
|
|
'ip_protocol': 'abc',
|
|
'from_port': 123,
|
|
'to_port': 456,
|
|
'group': {},
|
|
'parent_group_id': 55,
|
|
'ip_range': {'cidr': '9.8.7.6/55'},
|
|
'id': 18
|
|
}
|
|
rule = self.openstack_rule_transformer.to_rule(openstack_rule)
|
|
self.assertEqual(rule.to_port, str(openstack_rule['to_port']))
|
|
|
|
def test_should_copy_from_port_as_string(self):
|
|
openstack_rule = {
|
|
'ip_protocol': 'abc',
|
|
'from_port': 123,
|
|
'to_port': 456,
|
|
'group': {},
|
|
'parent_group_id': 55,
|
|
'ip_range': {'cidr': '9.8.7.6/55'},
|
|
'id': 18
|
|
}
|
|
rule = self.openstack_rule_transformer.to_rule(openstack_rule)
|
|
self.assertEqual(rule.from_port, str(openstack_rule['from_port'])) |