Beispiel #1
0
    def test_get_all_table_assignments(self):
        self.service_manager.allocate_scratch_tables(
            EnforcementController.APP_NAME, 1)
        self.service_manager.allocate_scratch_tables(
            EnforcementStatsController.APP_NAME, 2)

        result = self.service_manager.get_all_table_assignments()
        expected = OrderedDict([
            ('ingress', Tables(main_table=1, scratch_tables=[])),
            ('arpd', Tables(main_table=2, scratch_tables=[])),
            ('access_control', Tables(main_table=3, scratch_tables=[])),
            ('enforcement', Tables(main_table=4, scratch_tables=[21])),
            ('enforcement_stats', Tables(main_table=4, scratch_tables=[22,
                                                                       23])),
            ('dpi', Tables(main_table=5, scratch_tables=[])),
            ('meter', Tables(main_table=6, scratch_tables=[])),
            ('meter_stats', Tables(main_table=6, scratch_tables=[])),
            ('subscriber', Tables(main_table=6, scratch_tables=[])),
            ('egress', Tables(main_table=20, scratch_tables=[])),
        ])

        self.assertEqual(len(result), len(expected))
        for result_key, expected_key in zip(result, expected):
            self.assertEqual(result_key, expected_key)
            self.assertEqual(result[result_key].main_table,
                             expected[expected_key].main_table)
            self.assertEqual(result[result_key].scratch_tables,
                             expected[expected_key].scratch_tables)
    def test_get_all_table_assignments(self):
        self.service_manager.allocate_scratch_tables(
            EnforcementController.APP_NAME, 1)
        self.service_manager.allocate_scratch_tables(
            EnforcementStatsController.APP_NAME, 2)

        result = self.service_manager.get_all_table_assignments()
        print(result)
        expected = OrderedDict([
            ('mme',
             Tables(main_table=0,
                    scratch_tables=[],
                    type=ControllerType.SPECIAL)),
            ('ingress',
             Tables(main_table=1,
                    scratch_tables=[],
                    type=ControllerType.SPECIAL)),
            ('arpd',
             Tables(main_table=2,
                    scratch_tables=[],
                    type=ControllerType.PHYSICAL)),
            ('access_control',
             Tables(main_table=3,
                    scratch_tables=[],
                    type=ControllerType.PHYSICAL)),
            ('middle', Tables(main_table=10, scratch_tables=[], type=None)),
            ('enforcement',
             Tables(main_table=11,
                    scratch_tables=[21],
                    type=ControllerType.LOGICAL)),
            ('enforcement_stats',
             Tables(main_table=12,
                    scratch_tables=[22, 23],
                    type=ControllerType.LOGICAL)),
            ('dpi',
             Tables(main_table=13,
                    scratch_tables=[],
                    type=ControllerType.LOGICAL)),
            ('ipfix',
             Tables(main_table=14,
                    scratch_tables=[],
                    type=ControllerType.LOGICAL)),
            ('egress',
             Tables(main_table=20,
                    scratch_tables=[],
                    type=ControllerType.SPECIAL)),
        ])

        self.assertEqual(len(result), len(expected))
        for result_key, expected_key in zip(result, expected):
            self.assertEqual(result_key, expected_key)
            self.assertEqual(result[result_key].main_table,
                             expected[expected_key].main_table)
            self.assertEqual(result[result_key].scratch_tables,
                             expected[expected_key].scratch_tables)
Beispiel #3
0
def _display_flows(client, apps=None):
    pipelined_config = load_service_config('pipelined')
    bridge_name = pipelined_config['bridge_name']
    response = client.GetAllTableAssignments(Void())
    table_assignments = {
        table_assignment.app_name: Tables(
            main_table=table_assignment.main_table,
            type=None,
            scratch_tables=table_assignment.scratch_tables,
        )
        for table_assignment in response.table_assignments
    }
    try:
        flows = BridgeTools.get_annotated_flows_for_bridge(
            bridge_name,
            table_assignments,
            apps,
        )
    except subprocess.CalledProcessError as e:
        if e.returncode == errno.EPERM:
            print("Need to run as root to dump flows")
        return

    for flow in flows:
        print(flow)