Uname:Linux mail.sarafai.ru 6.1.0-43-amd64 #1 SMP PREEMPT_DYNAMIC Debian 6.1.162-1 (2026-02-08) x86_64

403WebShell
403Webshell
Server IP : 82.148.16.210  /  Your IP : 216.73.216.19
Web Server : nginx/1.29.5
System : Linux mail.sarafai.ru 6.1.0-43-amd64 #1 SMP PREEMPT_DYNAMIC Debian 6.1.162-1 (2026-02-08) x86_64
User : root ( 0)
PHP Version : 7.4.33
Disable Function : pcntl_alarm,pcntl_fork,pcntl_waitpid,pcntl_wait,pcntl_wifexited,pcntl_wifstopped,pcntl_wifsignaled,pcntl_wifcontinued,pcntl_wexitstatus,pcntl_wtermsig,pcntl_wstopsig,pcntl_signal,pcntl_signal_get_handler,pcntl_signal_dispatch,pcntl_get_last_error,pcntl_strerror,pcntl_sigprocmask,pcntl_sigwaitinfo,pcntl_sigtimedwait,pcntl_exec,pcntl_getpriority,pcntl_setpriority,pcntl_async_signals,pcntl_unshare,
MySQL : OFF  |  cURL : ON  |  WGET : ON  |  Perl : ON  |  Python : ON  |  Sudo : ON  |  Pkexec : OFF
Directory :  /lib/cloudinit/config/tests/

Upload File :
current_dir [ Writeable ] document_root [ Writeable ]

 

Command :


[ Back ]     

Current File : /lib/cloudinit/config/tests/test_disable_ec2_metadata.py
# This file is part of cloud-init. See LICENSE file for license information.

"""Tests cc_disable_ec2_metadata handler"""

import cloudinit.config.cc_disable_ec2_metadata as ec2_meta

from cloudinit.tests.helpers import CiTestCase, mock

import logging

LOG = logging.getLogger(__name__)

DISABLE_CFG = {'disable_ec2_metadata': 'true'}


class TestEC2MetadataRoute(CiTestCase):

    @mock.patch('cloudinit.config.cc_disable_ec2_metadata.subp.which')
    @mock.patch('cloudinit.config.cc_disable_ec2_metadata.subp.subp')
    def test_disable_ifconfig(self, m_subp, m_which):
        """Set the route if ifconfig command is available"""
        m_which.side_effect = lambda x: x if x == 'ifconfig' else None
        ec2_meta.handle('foo', DISABLE_CFG, None, LOG, None)
        m_subp.assert_called_with(
            ['route', 'add', '-host', '169.254.169.254', 'reject'],
            capture=False)

    @mock.patch('cloudinit.config.cc_disable_ec2_metadata.subp.which')
    @mock.patch('cloudinit.config.cc_disable_ec2_metadata.subp.subp')
    def test_disable_ip(self, m_subp, m_which):
        """Set the route if ip command is available"""
        m_which.side_effect = lambda x: x if x == 'ip' else None
        ec2_meta.handle('foo', DISABLE_CFG, None, LOG, None)
        m_subp.assert_called_with(
            ['ip', 'route', 'add', 'prohibit', '169.254.169.254'],
            capture=False)

    @mock.patch('cloudinit.config.cc_disable_ec2_metadata.subp.which')
    @mock.patch('cloudinit.config.cc_disable_ec2_metadata.subp.subp')
    def test_disable_no_tool(self, m_subp, m_which):
        """Log error when neither route nor ip commands are available"""
        m_which.return_value = None  # Find neither ifconfig nor ip
        ec2_meta.handle('foo', DISABLE_CFG, None, LOG, None)
        self.assertEqual(
            [mock.call('ip'), mock.call('ifconfig')], m_which.call_args_list)
        m_subp.assert_not_called()

# vi: ts=4 expandtab

Youez - 2016 - github.com/yon3zu
LinuXploit