Testbgp_max_paths

Complexity

Total Complexity

1

Size/Duplication

Total Lines	25
Duplicated Lines	0 %

Importance

Changes	1
Bugs	0	Features	0

1 Method

Rating	Name	Duplication	Size	Complexity
A	test_action()	0	22	1

import xml.etree.ElementTree as ET
from st2tests.base import BaseActionTestCase

from bgp_max_paths import bgp_max_paths

__all__ = [
    'Testbgp_max_paths'
]


class MockCallback(object):
    returned_data = None

    def callback(self, call, **kwargs):
        xml_result = ET.tostring(call)
        self.returned_data = xml_result


class Testbgp_max_paths(BaseActionTestCase):
    action_cls = bgp_max_paths

    def test_action(self):
        action = self.get_action_instance()
        mock_callback = MockCallback()
        kwargs = {
            'username': '',
            'paths': '10',
            'afi': 'ipv4',
            'get': 'False',
            'ip': '',
            'vrf': 'test',
            'password': '',
            'port': '22',
            'rbridge_id': '224',
            'test': True,
            'callback': mock_callback.callback
        }

        action.run(**kwargs)

        expected_xml = """<config><rbridge-id xmlns="urn:brocade.com:mgmt:brocade-rbridge"><rbridge-id>224</rbridge-id><router><bgp xmlns="urn:brocade.com:mgmt:brocade-bgp"><vrf-name>test</vrf-name><router-bgp-cmds-holder><address-family><ipv4><ipv4-unicast><af-common-cmds-holder><maximum-paths><load-sharing-value>10</load-sharing-value></maximum-paths></af-common-cmds-holder></ipv4-unicast></ipv4></address-family></router-bgp-cmds-holder></bgp></router></rbridge-id></config>"""

        self.assertTrue(expected_xml, mock_callback.returned_data)