TestBgpMultihop

Complexity

Total Complexity

1

Size/Duplication

Total Lines	38
Duplicated Lines	0 %

Importance

Changes

0

1 Method

Rating	Name	Duplication	Size	Complexity
B	test_action()	0	33	1

"""Generated test for checking pynos based actions
"""
import xml.etree.ElementTree as ET
from st2tests.base import BaseActionTestCase
from bgp_multihop import bgp_multihop

__all__ = [
    'TestBgpMultihop'
]


class MockCallback(object):  # pylint:disable=too-few-public-methods
    """Class to hold mock callback and result
    """
    returned_data = None

    def callback(self, call, **kwargs):  # pylint:disable=unused-argument
        """Mock callback method
        """
        xml_result = ET.tostring(call)
        self.returned_data = xml_result


class TestBgpMultihop(BaseActionTestCase):
    """Test holder class
    """
    action_cls = bgp_multihop

    def test_action(self):
        """Generated test to check action
        """
        action = self.get_action_instance()
        mock_callback = MockCallback()
        kwargs = {
            'count': '3',
            'username': '',
            'rbridge_id': '224',
            'get': False,
            'ip': '',
            'vrf': 'test',
            'neighbor': '10.0.2.1',
            'password': '',
            'port': '22',
            '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:m'
            'gmt:brocade-bgp"><vrf-name>test</vrf-name><router-bgp-cmds-holder'
            '><router-bgp-attributes><neighbor-ips><neighbor-addr><router-bgp-'
            'neighbor-address>10.0.2.1</router-bgp-neighbor-address><ebgp-mult'
            'ihop><ebgp-multihop-count>3</ebgp-multihop-count></ebgp-multihop>'
            '</neighbor-addr></neighbor-ips></router-bgp-attributes></router-b'
            'gp-cmds-holder></bgp></router></rbridge-id></config>'
        )

        self.assertTrue(expected_xml, mock_callback.returned_data)