def _load_chain_data(self, virtual_topology_data):
""" Load a chain data with a single rule """
# Load chain and rule virtual topology data from file.
self._chain_data = virtual_topology_data['chains'][0].get('chain')
self._chain = Chain(self._api, self._context, self._chain_data)
self._chain._get_tenant_id = MagicMock(return_value = 'tenant_0')
self._chain.build()
def _load_chain_data(self, virtual_topology_data):
""" Load a chain data with a single rule """
# Load chain and rule virtual topology data from file.
self._chain_data = virtual_topology_data["chains"][0].get("chain")
self._chain = Chain(self._api, self._context, self._chain_data)
self._chain._get_tenant_id = MagicMock(return_value="tenant_0")
self._chain.build()
def add_chain(self, chain_data):
chain = Chain(self._api, self, chain_data)
chain.build()
self._chains[chain_data['name']] = chain
class ChainTest(unittest.TestCase):
def setUp(self):
self._empty_chain = yaml.load("""
chains:
- chain:
id: 1
name: in_filter_001
""")
self._simple_chain = yaml.load("""
chains:
- chain:
id: 1 # can be different to MN chain id
name: in_filter_001
rules:
- rule:
id: 101
position: 1
dl_type: 0x86DD
type: drop
""")
self._multiple_rules_chain = yaml.load("""
chains:
- chain:
id: 1 # can be different to MN chain id
name: in_filter_001
rules:
- rule:
id: 101
position: 1
dl_type: 0x86DD
type: drop
- rule:
id: 102
position: 2
type: accept
""")
self._api = MagicMock()
self._context = MagicMock()
self._mock_chain = MagicMock()
self._mock_rule = MagicMock()
# Mock Chain MidoNet Client resource.
self._api.add_chain.return_value = self._mock_chain
self._mock_chain.tenant_id.return_value = self._mock_chain
self._mock_chain.name.return_value = self._mock_chain
self._mock_chain.create.return_value = self._mock_chain
self._mock_chain.add_rule.return_value = self._mock_rule
self._mock_chain.get_id.return_value = 'chain_0'
def _load_chain_data(self, virtual_topology_data):
""" Load a chain data with a single rule """
# Load chain and rule virtual topology data from file.
self._chain_data = virtual_topology_data['chains'][0].get('chain')
self._chain = Chain(self._api, self._context, self._chain_data)
self._chain._get_tenant_id = MagicMock(return_value = 'tenant_0')
self._chain.build()
def test_load_empty_chain(self):
""" Tests if empty rule chain data can be correctly loaded. """
self._load_chain_data(self._empty_chain)
self._mock_chain.tenant_id.assert_called_with('tenant_0')
self._mock_chain.name.assert_called_with('in_filter_001')
self._mock_chain.create.assert_called_with()
def test_load_rule_chain(self):
""" Tests if rule chain data can be correctly loaded from the yaml
file and corresponding resource creation / update operations are
performed.
"""
self._load_chain_data(self._simple_chain)
rule = self._chain.get_rule(101)
self.assertNotEqual(None, rule)
self.assertEqual(101, rule.get_id())
self.assertEqual(1, rule.get_chain_id())
self.assertEqual('drop', rule.get_type())
self._mock_chain.tenant_id.assert_called_with('tenant_0')
self._mock_chain.name.assert_called_with('in_filter_001')
self._mock_chain.create.assert_called_with()
self.assertEqual([call.chain_id('chain_0'),
call.position(1),
call.dl_type(0x86DD),
call.type('drop'),
call.create()], self._mock_rule.mock_calls)
def test_load_multiple_rules_chain(self):
""" Tests if chain data with multiple rules can be correctly loaded from
the yaml file.
"""
self._load_chain_data(self._multiple_rules_chain)
rule = self._chain.get_rule(101)
self.assertNotEqual(None, rule)
self.assertEqual(101, rule.get_id())
self.assertEqual(1, rule.get_chain_id())
self.assertEqual('drop', rule.get_type())
rule = self._chain.get_rule(102)
self.assertNotEqual(None, rule)
self.assertEqual(102, rule.get_id())
self.assertEqual(1, rule.get_chain_id())
self.assertEqual('accept', rule.get_type())
self._mock_chain.tenant_id.assert_called_with('tenant_0')
self._mock_chain.name.assert_called_with('in_filter_001')
self._mock_chain.create.assert_called_with()
self.assertEqual([call.chain_id('chain_0'),
call.position(1),
call.dl_type(0x86DD),
call.type('drop'),
call.create(),
call.chain_id('chain_0'),
call.position(2),
call.type('accept'),
call.create()], self._mock_rule.mock_calls)
class ChainTest(unittest.TestCase):
def setUp(self):
self._empty_chain = yaml.load(
"""
chains:
- chain:
id: 1
name: in_filter_001
"""
)
self._simple_chain = yaml.load(
"""
chains:
- chain:
id: 1 # can be different to MN chain id
name: in_filter_001
rules:
- rule:
id: 101
position: 1
dl_type: 0x86DD
type: drop
"""
)
self._multiple_rules_chain = yaml.load(
"""
chains:
- chain:
id: 1 # can be different to MN chain id
name: in_filter_001
rules:
- rule:
id: 101
position: 1
dl_type: 0x86DD
type: drop
- rule:
id: 102
position: 2
type: accept
"""
)
self._api = MagicMock()
self._context = MagicMock()
self._mock_chain = MagicMock()
self._mock_rule = MagicMock()
# Mock Chain MidoNet Client resource.
self._api.add_chain.return_value = self._mock_chain
self._mock_chain.tenant_id.return_value = self._mock_chain
self._mock_chain.name.return_value = self._mock_chain
self._mock_chain.create.return_value = self._mock_chain
self._mock_chain.add_rule.return_value = self._mock_rule
self._mock_chain.get_id.return_value = "chain_0"
def _load_chain_data(self, virtual_topology_data):
""" Load a chain data with a single rule """
# Load chain and rule virtual topology data from file.
self._chain_data = virtual_topology_data["chains"][0].get("chain")
self._chain = Chain(self._api, self._context, self._chain_data)
self._chain._get_tenant_id = MagicMock(return_value="tenant_0")
self._chain.build()
def test_load_empty_chain(self):
""" Tests if empty rule chain data can be correctly loaded. """
self._load_chain_data(self._empty_chain)
self._mock_chain.tenant_id.assert_called_with("tenant_0")
self._mock_chain.name.assert_called_with("in_filter_001")
self._mock_chain.create.assert_called_with()
def test_load_rule_chain(self):
""" Tests if rule chain data can be correctly loaded from the yaml
file and corresponding resource creation / update operations are
performed.
"""
self._load_chain_data(self._simple_chain)
rule = self._chain.get_rule(101)
self.assertNotEqual(None, rule)
self.assertEqual(101, rule.get_id())
self.assertEqual(1, rule.get_chain_id())
self.assertEqual("drop", rule.get_type())
self._mock_chain.tenant_id.assert_called_with("tenant_0")
self._mock_chain.name.assert_called_with("in_filter_001")
self._mock_chain.create.assert_called_with()
self.assertEqual(
[call.chain_id("chain_0"), call.position(1), call.dl_type(0x86DD), call.type("drop"), call.create()],
self._mock_rule.mock_calls,
)
def test_load_multiple_rules_chain(self):
""" Tests if chain data with multiple rules can be correctly loaded from
the yaml file.
"""
self._load_chain_data(self._multiple_rules_chain)
rule = self._chain.get_rule(101)
self.assertNotEqual(None, rule)
self.assertEqual(101, rule.get_id())
self.assertEqual(1, rule.get_chain_id())
self.assertEqual("drop", rule.get_type())
rule = self._chain.get_rule(102)
self.assertNotEqual(None, rule)
self.assertEqual(102, rule.get_id())
self.assertEqual(1, rule.get_chain_id())
self.assertEqual("accept", rule.get_type())
self._mock_chain.tenant_id.assert_called_with("tenant_0")
self._mock_chain.name.assert_called_with("in_filter_001")
self._mock_chain.create.assert_called_with()
self.assertEqual(
[
call.chain_id("chain_0"),
call.position(1),
call.dl_type(0x86DD),
call.type("drop"),
call.create(),
call.chain_id("chain_0"),
call.position(2),
call.type("accept"),
call.create(),
],
self._mock_rule.mock_calls,
)
def add_chain(self, chain_data):
chain = Chain(self._api, self, chain_data)
chain.build()
self._chains[chain_data['name']] = chain
