dragonflow/dragonflow/tests/unit/test_classifier_app.py
Omer Anson 395eeabf24 Make Subnet a first-order model
Change-Id: I369f7d99626f07b8e22a13bf374dae06697468bc
Closes-Bug: #1549125
Related-Bug: #1708178
2017-12-18 09:14:50 +00:00

90 lines
3.7 KiB
Python

# Copyright (c) 2017 OpenStack Foundation.
# All Rights Reserved.
#
# Licensed under the Apache License, Version 2.0 (the "License"); you may
# not use this file except in compliance with the License. You may obtain
# a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
# License for the specific language governing permissions and limitations
# under the License.
import copy
import mock
import testscenarios
from dragonflow.common import constants
from dragonflow.controller.common import constants as const
from dragonflow.db.models import l2
from dragonflow.db.models import ovs
from dragonflow.tests.unit import test_app_base
make_fake_local_port = test_app_base.make_fake_local_port
SCENARIO_ORDER_DELETE_LPORT_FIRST = 'delete_lport_first'
SCENARIO_ORDER_DELETE_OVS_PORT_FIRST = 'delete_ovs_port_first'
class TestClassifierAppForVlan(testscenarios.WithScenarios,
test_app_base.DFAppTestBase):
apps_list = ["classifier"]
scenarios = [(SCENARIO_ORDER_DELETE_LPORT_FIRST,
{'order': SCENARIO_ORDER_DELETE_LPORT_FIRST}),
(SCENARIO_ORDER_DELETE_OVS_PORT_FIRST,
{'order': SCENARIO_ORDER_DELETE_OVS_PORT_FIRST})]
def setUp(self):
super(TestClassifierAppForVlan, self).setUp()
fake_vlan_switch1 = l2.LogicalSwitch(
network_type='vlan',
id='fake_vlan_switch1', mtu=1500,
is_external=False, segmentation_id=41,
topic='fake_tenant1', unique_key=2,
name='private')
self.controller.update(fake_vlan_switch1)
subnet = copy.deepcopy(test_app_base.fake_lswitch_default_subnets[0])
subnet.id = 'fake_vlan_subnet1'
subnet.lswitch = 'fake_vlan_switch1'
self.controller.update(subnet)
self.app = self.open_flow_app.dispatcher.apps['classifier']
def test_classifier_for_vlan_port(self):
fake_local_vlan_port = make_fake_local_port(
lswitch='fake_vlan_switch1')
self.controller.update(fake_local_vlan_port)
self.app.mod_flow.assert_not_called()
ovs_port = ovs.OvsPort(id='fake_ovs_port',
lport=fake_local_vlan_port.id,
ofport=1, admin_state='up',
type=constants.OVS_COMPUTE_INTERFACE)
self.controller.update(ovs_port)
port_key = fake_local_vlan_port.unique_key
match = self.app.parser.OFPMatch(reg7=port_key)
self.app.mod_flow.assert_called_with(
inst=mock.ANY,
table_id=const.INGRESS_DISPATCH_TABLE,
priority=const.PRIORITY_MEDIUM,
match=match)
self.app.mod_flow.reset_mock()
ofport = ovs_port.ofport
match = self.app.parser.OFPMatch(in_port=ofport)
if self.order == SCENARIO_ORDER_DELETE_LPORT_FIRST:
self.controller.delete(fake_local_vlan_port)
self.controller.delete(ovs_port)
elif self.order == SCENARIO_ORDER_DELETE_OVS_PORT_FIRST:
self.controller.delete(ovs_port)
self.controller.delete(fake_local_vlan_port)
else:
self.fail("Bad order")
self.app.mod_flow.assert_called_with(
table_id=const.INGRESS_CLASSIFICATION_DISPATCH_TABLE,
command=self.datapath.ofproto.OFPFC_DELETE,
priority=const.PRIORITY_MEDIUM,
match=match)
self.app.mod_flow.reset_mock()