Python patch Exemples

Exemple #1

Fichier : test_junos_security_policies_global.py Projet : ansible-collections/junipernetworks.junos

    def setUp(self):
        super(TestJunosSecurity_policies_globalModule, self).setUp()

        self.mock_lock_configuration = patch(
            "ansible_collections.junipernetworks.junos.plugins.module_utils.network.junos.junos.lock_configuration"
        )
        self.lock_configuration = self.mock_lock_configuration.start()

        self.mock_unlock_configuration = patch(
            "ansible_collections.junipernetworks.junos.plugins.module_utils.network.junos.junos.unlock_configuration"
        )
        self.unlock_configuration = self.mock_unlock_configuration.start()

        self.mock_load_config = patch(
            "ansible_collections.junipernetworks.junos.plugins.module_utils.network.junos.config.security_policies_global.security_policies_global.load_config"
        )
        self.load_config = self.mock_load_config.start()

        self.mock_commit_configuration = patch(
            "ansible_collections.junipernetworks.junos.plugins.module_utils.network.junos.config.security_policies_global."
            "security_policies_global.commit_configuration")
        self.mock_commit_configuration = self.mock_commit_configuration.start()

        self.mock_execute_show_command = patch(
            "ansible_collections.junipernetworks.junos.plugins.module_utils.network.junos.facts.security_policies_global.security_policies_global."
            "Security_policies_globalFacts._get_device_data")
        self.execute_show_command = self.mock_execute_show_command.start()
        self.maxDiff = None

Exemple #2

Fichier : test_junos_routing_options.py Projet : ansible-collections/junipernetworks.junos

    def setUp(self):
        super(TestJunosRouting_optionsModule, self).setUp()

        self.mock_lock_configuration = patch(
            "ansible_collections.junipernetworks.junos.plugins.module_utils.network.junos.junos.lock_configuration"
        )
        self.lock_configuration = self.mock_lock_configuration.start()

        self.mock_unlock_configuration = patch(
            "ansible_collections.junipernetworks.junos.plugins.module_utils.network.junos.junos.unlock_configuration"
        )
        self.unlock_configuration = self.mock_unlock_configuration.start()

        self.mock_load_config = patch(
            "ansible_collections.junipernetworks.junos.plugins.module_utils.network.junos.config.routing_options.routing_options.load_config"
        )
        self.load_config = self.mock_load_config.start()

        self.mock_commit_configuration = patch(
            "ansible_collections.junipernetworks.junos.plugins.module_utils.network.junos.config.routing_options.routing_options.commit_configuration"
        )
        self.mock_commit_configuration = self.mock_commit_configuration.start()

        self.mock_execute_show_command = patch(
            "ansible_collections.junipernetworks.junos.plugins.module_utils.network.junos.facts.routing_options.routing_options."
            "Routing_optionsFacts.get_device_data"
        )
        self.execute_show_command = self.mock_execute_show_command.start()

Exemple #3

    def setUp(self):
        super(TestJunosScpModule, self).setUp()
        self.mock_get_device = patch(
            "ansible_collections.junipernetworks.junos.plugins.modules.junos_scp.get_device"
        )
        self.get_device = self.mock_get_device.start()

        self.mock_scp = patch(
            "ansible_collections.junipernetworks.junos.plugins.modules.junos_scp.SCP"
        )
        self.scp = self.mock_scp.start()

        self.scp_mock = MagicMock()
        self.scp().__enter__.return_value = self.scp_mock

Exemple #4

 def setUp(self):
     self.mock_module = patch.multiple(basic.AnsibleModule, exit_json=exit_json, fail_json=fail_json)
     self.mock_module.start()
     self.mock_sleep = patch('time.sleep')
     self.mock_sleep.start()
     set_module_args({})
     self.addCleanup(self.mock_module.stop)
     self.addCleanup(self.mock_sleep.stop)

Exemple #5

Fichier : test_junos_rpc.py Projet : routenull0/junipernetworks.junos

    def setUp(self):
        super(TestJunosCommandModule, self).setUp()
        self.mock_conn = patch("ansible.module_utils.connection.Connection")
        self.conn = self.mock_conn.start()

        self.mock_netconf = patch(
            "ansible_collections.junipernetworks.junos.plugins.module_utils.network.junos.junos.NetconfConnection"
        )
        self.netconf_conn = self.mock_netconf.start()

        self.mock_netconf_rpc = patch(
            "ansible_collections.ansible.netcommon.plugins.module_utils.network.common.netconf.NetconfConnection"
        )
        self.netconf_rpc = self.mock_netconf_rpc.start()

        self.mock_exec_rpc = patch(
            "ansible_collections.junipernetworks.junos.plugins.modules.junos_rpc.exec_rpc"
        )
        self.exec_rpc = self.mock_exec_rpc.start()

Exemple #6

Fichier : test_junos_ping.py Projet : mshaaban-redhat/firewall

    def setUp(self):
        super(TestJunosPingModule, self).setUp()

        self.mock_get_connection = patch(
            "ansible_collections.junipernetworks.junos.plugins.modules.junos_ping.get_connection"
        )
        self.get_connection = self.mock_get_connection.start()

        self.conn = self.get_connection()
        self.conn.get = MagicMock()

Exemple #7

Fichier : test_junos_command.py Projet : routenull0/junipernetworks.junos

    def setUp(self):
        super(TestJunosCommandModule, self).setUp()

        self.mock_conn = patch(
            "ansible_collections.junipernetworks.junos.plugins.module_utils.network.junos.junos.Connection"
        )
        self.conn = self.mock_conn.start()

        self.mock_netconf = patch(
            "ansible_collections.junipernetworks.junos.plugins.module_utils.network.junos.junos.NetconfConnection"
        )
        self.netconf_conn = self.mock_netconf.start()

        self.mock_exec_rpc = patch(
            "ansible_collections.junipernetworks.junos.plugins.modules.junos_command.exec_rpc"
        )
        self.exec_rpc = self.mock_exec_rpc.start()

        self.mock_netconf_rpc = patch(
            "ansible_collections.ansible.netcommon.plugins.module_utils.network.common.netconf.NetconfConnection"
        )
        self.netconf_rpc = self.mock_netconf_rpc.start()

        self.mock_get_connection = patch(
            "ansible_collections.junipernetworks.junos.plugins.modules.junos_command.get_connection"
        )
        self.get_connection = self.mock_get_connection.start()

        self.mock_get_capabilities = patch(
            "ansible_collections.junipernetworks.junos.plugins.modules.junos_command.get_capabilities"
        )
        self.get_capabilities = self.mock_get_capabilities.start()
        self.get_capabilities.return_value = {"network_api": "netconf"}

Exemple #8

Fichier : test_junos_interfaces.py Projet : routenull0/junipernetworks.junos

    def setUp(self):
        super(TestJunosInterfacesModule, self).setUp()
        self.mock_lock_configuration = patch(
            "ansible_collections.junipernetworks.junos.plugins.module_utils.network.junos.junos.lock_configuration"
        )
        self.lock_configuration = self.mock_lock_configuration.start()
        self.mock_unlock_configuration = patch(
            "ansible_collections.junipernetworks.junos.plugins.module_utils.network.junos.junos.unlock_configuration"
        )
        self.unlock_configuration = self.mock_unlock_configuration.start()
        self.mock_load_config = patch(
            "ansible_collections.junipernetworks.junos.plugins.module_utils.network.junos.config.interfaces.interfaces.load_config"
        )
        self.load_config = self.mock_load_config.start()

        self.mock_validate_config = patch(
            "ansible_collections.ansible.netcommon.plugins.module_utils.network.common.utils.validate_config"
        )
        self.validate_config = self.mock_validate_config.start()

        self.mock_commit_configuration = patch(
            "ansible_collections.junipernetworks.junos.plugins.module_utils.network.junos.config.interfaces.interfaces.commit_configuration"
        )
        self.mock_commit_configuration = self.mock_commit_configuration.start()

        self.mock_get_config = patch(
            "ansible_collections.junipernetworks.junos.plugins.module_utils.network.junos.facts.interfaces.interfaces."
            "InterfacesFacts.get_config")
        self.get_config = self.mock_get_config.start()

Exemple #9

Fichier : test_junos_netconf.py Projet : coll-test/junipernetworks.junos

    def setUp(self):
        super(TestJunosCommandModule, self).setUp()

        self.mock_lock_configuration = patch(
            'ansible_collections.junipernetworks.junos.plugins.module_utils.network.junos.junos.lock_configuration'
        )
        self.lock_configuration = self.mock_lock_configuration.start()

        self.mock_unlock_configuration = patch(
            'ansible_collections.junipernetworks.junos.plugins.module_utils.network.junos.junos.unlock_configuration'
        )
        self.unlock_configuration = self.mock_unlock_configuration.start()

        self.mock_conn = patch('ansible.module_utils.connection.Connection')
        self.conn = self.mock_conn.start()

        self.mock_netconf = patch(
            'ansible_collections.junipernetworks.junos.plugins.module_utils.network.junos.junos.NetconfConnection'
        )
        self.netconf_conn = self.mock_netconf.start()

        self.mock_netconf_rpc = patch(
            'ansible_collections.ansible.netcommon.plugins.module_utils.network.common.netconf.NetconfConnection'
        )
        self.netconf_rpc = self.mock_netconf_rpc.start()

        self.mock_get_capabilities = patch(
            'ansible_collections.junipernetworks.junos.plugins.module_utils.network.junos.junos.get_capabilities'
        )
        self.get_capabilities = self.mock_get_capabilities.start()
        self.get_capabilities.return_value = {'network_api': 'netconf'}

Exemple #10

Fichier : test_junos_bgp_global.py Projet : raubvogel/junipernetworks.junos

    def setUp(self):
        super(TestJunosBgp_globalModule, self).setUp()
        self.mock_lock_configuration = patch(
            "ansible_collections.junipernetworks.junos.plugins.module_utils.network.junos.junos.lock_configuration"
        )
        self.lock_configuration = self.mock_lock_configuration.start()
        self.mock_unlock_configuration = patch(
            "ansible_collections.junipernetworks.junos.plugins.module_utils.network.junos.junos.unlock_configuration"
        )
        self.unlock_configuration = self.mock_unlock_configuration.start()
        self.mock_load_config = patch(
            "ansible_collections.junipernetworks.junos.plugins.module_utils.network.junos.config.bgp_global.bgp_global.load_config"
        )
        self.load_config = self.mock_load_config.start()

        self.mock_validate_config = patch(
            "ansible_collections.ansible.netcommon.plugins.module_utils.network.common.utils.validate_config"
        )
        self.validate_config = self.mock_validate_config.start()

        self.mock_commit_configuration = patch(
            "ansible_collections.junipernetworks.junos.plugins.module_utils.network.junos.config.bgp_global.bgp_global.commit_configuration"
        )
        self.mock_commit_configuration = self.mock_commit_configuration.start()

        self.mock_get_connection = patch(
            "ansible_collections.junipernetworks.junos.plugins.module_utils.network.junos.facts.bgp_global.bgp_global."
            "Bgp_globalFacts.get_connection"
        )
        self.get_connection = self.mock_get_connection.start()

        self.conn = self.get_connection()
        self.conn.get = MagicMock()

        self.mock_get_xml_dict = patch(
            "ansible_collections.junipernetworks.junos.plugins.module_utils.network.junos.facts."
            "bgp_global.bgp_global.Bgp_globalFacts._get_xml_dict"
        )
        self._get_xml_dict = self.mock_get_xml_dict.start()

Exemple #11

Fichier : test_junos_facts.py Projet : mshaaban-redhat/firewall

    def setUp(self):
        super(TestJunosCommandModule, self).setUp()

        self.mock_get_config = patch(
            "ansible_collections.junipernetworks.junos.plugins.module_utils.network.junos.facts.legacy.base.get_configuration"
        )
        self.get_config = self.mock_get_config.start()

        self.mock_netconf = patch(
            "ansible_collections.junipernetworks.junos.plugins.module_utils.network.junos.junos.NetconfConnection"
        )
        self.netconf_conn = self.mock_netconf.start()

        self.mock_exec_rpc = patch(
            "ansible_collections.junipernetworks.junos.plugins.module_utils.network.junos.facts.legacy.base.exec_rpc"
        )
        self.exec_rpc = self.mock_exec_rpc.start()

        self.mock_netconf_rpc = patch(
            "ansible_collections.ansible.netcommon.plugins.module_utils.network.common.netconf.NetconfConnection"
        )
        self.netconf_rpc = self.mock_netconf_rpc.start()

        self.mock_get_resource_connection = patch(
            "ansible_collections.ansible.netcommon.plugins.module_utils.network.common.facts.facts.get_resource_connection"
        )
        self.get_resource_connection = (
            self.mock_get_resource_connection.start()
        )

        self.mock_get_capabilities = patch(
            "ansible_collections.junipernetworks.junos.plugins.module_utils.network.junos.facts.legacy.base.get_capabilities"
        )
        self.get_capabilities = self.mock_get_capabilities.start()

        self.get_capabilities.return_value = {
            "device_info": {
                "network_os": "junos",
                "network_os_hostname": "vsrx01",
                "network_os_model": "vsrx",
                "network_os_version": "17.3R1.10",
            },
            "network_api": "netconf",
        }

Exemple #12

Fichier : test_junos_facts.py Projet : coll-test/junipernetworks.junos

    def setUp(self):
        super(TestJunosCommandModule, self).setUp()

        self.mock_get_config = patch(
            'ansible_collections.junipernetworks.junos.plugins.module_utils.network.junos.facts.legacy.base.get_configuration'
        )
        self.get_config = self.mock_get_config.start()

        self.mock_netconf = patch(
            'ansible_collections.junipernetworks.junos.plugins.module_utils.network.junos.junos.NetconfConnection'
        )
        self.netconf_conn = self.mock_netconf.start()

        self.mock_exec_rpc = patch(
            'ansible_collections.junipernetworks.junos.plugins.module_utils.network.junos.facts.legacy.base.exec_rpc'
        )
        self.exec_rpc = self.mock_exec_rpc.start()

        self.mock_netconf_rpc = patch(
            'ansible_collections.ansible.netcommon.plugins.module_utils.network.common.netconf.NetconfConnection'
        )
        self.netconf_rpc = self.mock_netconf_rpc.start()

        self.mock_get_resource_connection = patch(
            'ansible_collections.ansible.netcommon.plugins.module_utils.network.common.facts.facts.get_resource_connection'
        )
        self.get_resource_connection = self.mock_get_resource_connection.start(
        )

        self.mock_get_capabilities = patch(
            'ansible_collections.junipernetworks.junos.plugins.module_utils.network.junos.facts.legacy.base.get_capabilities'
        )
        self.get_capabilities = self.mock_get_capabilities.start()

        self.get_capabilities.return_value = {
            'device_info': {
                'network_os': 'junos',
                'network_os_hostname': 'vsrx01',
                'network_os_model': 'vsrx',
                'network_os_version': '17.3R1.10'
            },
            'network_api': 'netconf'
        }

Exemple #13

    def setUp(self):
        super(TestJunosConfigModule, self).setUp()

        self.mock_get_config = patch(
            "ansible_collections.junipernetworks.junos.plugins.modules.junos_config.get_configuration"
        )
        self.get_config = self.mock_get_config.start()

        self.mock_load_config = patch(
            "ansible_collections.junipernetworks.junos.plugins.modules.junos_config.load_config"
        )
        self.load_config = self.mock_load_config.start()

        self.mock_load_configuration = patch(
            "ansible_collections.junipernetworks.junos.plugins.modules.junos_config.load_configuration"
        )
        self.load_configuration = self.mock_load_configuration.start()

        self.mock_lock_configuration = patch(
            "ansible_collections.junipernetworks.junos.plugins.module_utils.network.junos.junos.lock_configuration"
        )
        self.lock_configuration = self.mock_lock_configuration.start()

        self.mock_unlock_configuration = patch(
            "ansible_collections.junipernetworks.junos.plugins.module_utils.network.junos.junos.unlock_configuration"
        )
        self.unlock_configuration = self.mock_unlock_configuration.start()

        self.mock_commit_configuration = patch(
            "ansible_collections.junipernetworks.junos.plugins.modules.junos_config.commit_configuration"
        )
        self.commit_configuration = self.mock_commit_configuration.start()

        self.mock_get_diff = patch(
            "ansible_collections.junipernetworks.junos.plugins.modules.junos_config.get_diff"
        )
        self.get_diff = self.mock_get_diff.start()

        self.mock_conn = patch("ansible.module_utils.connection.Connection")
        self.conn = self.mock_conn.start()

        self.mock_netconf = patch(
            "ansible_collections.junipernetworks.junos.plugins.module_utils.network.junos.junos.NetconfConnection"
        )
        self.netconf_conn = self.mock_netconf.start()

        self.mock_exec_rpc = patch(
            "ansible_collections.junipernetworks.junos.plugins.modules.junos_config.exec_rpc"
        )
        self.exec_rpc = self.mock_exec_rpc.start()

        self.mock_netconf_rpc = patch(
            "ansible_collections.ansible.netcommon.plugins.module_utils.network.common.netconf.NetconfConnection"
        )
        self.netconf_rpc = self.mock_netconf_rpc.start()

Exemple #14

Fichier : test_junos_package.py Projet : coll-test/junipernetworks.junos

 def setUp(self):
     super(TestJunosPackageModule, self).setUp()
     self.mock_get_device = patch(
         'ansible_collections.junipernetworks.junos.plugins.modules.junos_package.get_device'
     )
     self.get_device = self.mock_get_device.start()