<s> '''<STR_LIT>''' <EOL> from __future__ import absolute_import <EOL> from salttesting import TestCase , skipIf <EOL> from salttesting . mock import MagicMock , patch , NO_MOCK , NO_MOCK_REASON <EOL> from salttesting . helpers import ensure_in_syspath <EOL> ensure_in_syspath ( '<STR_LIT>' ) <EOL> from salt . modules import dig <EOL> _SPF_VALUES = { <EOL> '<STR_LIT>' : { <EOL> '<STR_LIT>' : <NUM_LIT> , <EOL> '<STR_LIT>' : <NUM_LIT:0> , <EOL> '<STR_LIT>' : '<STR_LIT>' , <EOL> '<STR_LIT>' : '<STR_LIT>' <EOL> } , <EOL> '<STR_LIT>' : { <EOL> '<STR_LIT>' : <NUM_LIT> , <EOL> '<STR_LIT>' : <NUM_LIT:0> , <EOL> '<STR_LIT>' : '<STR_LIT>' , <EOL> '<STR_LIT>' : '<STR_LIT>' <EOL> } , <EOL> '<STR_LIT>' : { <EOL> '<STR_LIT>' : <NUM_LIT> , <EOL> '<STR_LIT>' : <NUM_LIT:0> , <EOL> '<STR_LIT>' : '<STR_LIT>' , <EOL> '<STR_LIT>' : '<STR_LIT>' <EOL> } , <EOL> '<STR_LIT>' : { <EOL> '<STR_LIT>' : <NUM_LIT> , <EOL> '<STR_LIT>' : <NUM_LIT:0> , <EOL> '<STR_LIT>' : '<STR_LIT>' , <EOL> '<STR_LIT>' : '<STR_LIT>' <EOL> } , <EOL> } <EOL> def _spf_side_effect ( key , python_shell = False ) : <EOL> return _SPF_VALUES . get ( '<STR_LIT:U+0020>' . join ( key ) , { '<STR_LIT>' : <NUM_LIT> , <EOL> '<STR_LIT>' : <NUM_LIT:0> , <EOL> '<STR_LIT>' : '<STR_LIT>' , <EOL> '<STR_LIT>' : '<STR_LIT>' } ) <EOL> @ skipIf ( NO_MOCK , NO_MOCK_REASON ) <EOL> @ skipIf ( dig . __virtual__ ( ) is False , '<STR_LIT>' ) <EOL> class DigTestCase ( TestCase ) : <EOL> def test_check_ip ( self ) : <EOL> self . assertTrue ( dig . check_ip ( '<STR_LIT:127.0.0.1>' ) , msg = '<STR_LIT>' ) <EOL> def test_check_ip_ipv6 ( self ) : <EOL> self . assertTrue ( <EOL> dig . check_ip ( '<STR_LIT>' ) , <EOL> msg = '<STR_LIT>' <EOL> ) <EOL> def test_check_ip_ipv6_valid ( self ) : <EOL> self . assertTrue ( dig . check_ip ( '<STR_LIT>' ) ) <EOL> def test_check_ip_neg ( self ) : <EOL> self . assertFalse ( <EOL> dig . check_ip ( '<STR_LIT>' ) , <EOL> msg = "<STR_LIT>" <EOL> ) <EOL> def test_check_ip_empty ( self ) : <EOL> self . assertFalse ( <EOL> dig . check_ip ( '<STR_LIT>' ) , <EOL> msg = "<STR_LIT>" <EOL> ) <EOL> def test_a ( self ) : <EOL> dig . __salt__ = { } <EOL> dig_mock = MagicMock ( <EOL> return_value = { <EOL> '<STR_LIT>' : <NUM_LIT> , <EOL> '<STR_LIT>' : <NUM_LIT:0> , <EOL> '<STR_LIT>' : '<STR_LIT>' , <EOL> '<STR_LIT>' : '<STR_LIT>' <EOL> '<STR_LIT>' <EOL> '<STR_LIT>' <EOL> '<STR_LIT>' <EOL> '<STR_LIT>' <EOL> '<STR_LIT>' <EOL> } <EOL> ) <EOL> with patch . dict ( dig . __salt__ , { '<STR_LIT>' : dig_mock } ) : <EOL> self . assertEqual ( <EOL> dig . A ( '<STR_LIT>' ) , <EOL> [ '<STR_LIT>' , <EOL> '<STR_LIT>' , <EOL> '<STR_LIT>' , <EOL> '<STR_LIT>' , <EOL> '<STR_LIT>' , <EOL> '<STR_LIT>' ] <EOL> ) <EOL> def test_aaaa ( self ) : <EOL> dig . __salt__ = { } <EOL> dig_mock = MagicMock ( <EOL> return_value = { <EOL> '<STR_LIT>' : <NUM_LIT> , <EOL> '<STR_LIT>' : <NUM_LIT:0> , <EOL> '<STR_LIT>' : '<STR_LIT>' , <EOL> '<STR_LIT>' : '<STR_LIT>' <EOL> } <EOL> ) <EOL> with patch . dict ( dig . __salt__ , { '<STR_LIT>' : dig_mock } ) : <EOL> self . assertEqual ( <EOL> dig . AAAA ( '<STR_LIT>' ) , <EOL> [ '<STR_LIT>' ] <EOL> ) <EOL> @ patch ( '<STR_LIT>' , MagicMock ( return_value = [ '<STR_LIT>' ] ) ) <EOL> def test_ns ( self ) : <EOL> dig . __salt__ = { } <EOL> dig_mock = MagicMock ( <EOL> return_value = { <EOL> '<STR_LIT>' : <NUM_LIT> , <EOL> '<STR_LIT>' : <NUM_LIT:0> , <EOL> '<STR_LIT>' : '<STR_LIT>' , <EOL> '<STR_LIT>' : '<STR_LIT>' <EOL> } <EOL> ) <EOL> with patch . dict ( dig . __salt__ , { '<STR_LIT>' : dig_mock } ) : <EOL> self . assertEqual ( dig . NS ( '<STR_LIT>' ) , [ '<STR_LIT>' ] ) <EOL> def test_spf ( self ) : <EOL> dig . __salt__ = { } <EOL> dig_mock = MagicMock ( side_effect = _spf_side_effect ) <EOL> with patch . dict ( dig . __salt__ , { '<STR_LIT>' : dig_mock } ) : <EOL> self . assertEqual ( <EOL> dig . SPF ( '<STR_LIT>' ) , <EOL> [ '<STR_LIT>' , '<STR_LIT>' ] <EOL> ) <EOL> def test_spf_redir ( self ) : <EOL> '''<STR_LIT>''' <EOL> dig . __salt__ = { } <EOL> dig_mock = MagicMock ( side_effect = _spf_side_effect ) <EOL> with patch . dict ( dig . __salt__ , { '<STR_LIT>' : dig_mock } ) : <EOL> self . assertEqual ( <EOL> dig . SPF ( '<STR_LIT>' ) , <EOL> [ '<STR_LIT>' , '<STR_LIT>' ] <EOL> ) <EOL> def test_spf_include ( self ) : <EOL> '''<STR_LIT>''' <EOL> dig . __salt__ = { } <EOL> dig_mock = MagicMock ( side_effect = _spf_side_effect ) <EOL> with patch . dict ( dig . __salt__ , { '<STR_LIT>' : dig_mock } ) : <EOL> self . assertEqual ( <EOL> dig . SPF ( '<STR_LIT>' ) , <EOL> [ '<STR_LIT>' , '<STR_LIT>' ] <EOL> ) <EOL> def test_mx ( self ) : <EOL> dig . __salt__ = { } <EOL> dig_mock = MagicMock ( <EOL> return_value = { <EOL> '<STR_LIT>' : <NUM_LIT> , <EOL> '<STR_LIT>' : <NUM_LIT:0> , <EOL> '<STR_LIT>' : '<STR_LIT>' , <EOL> '<STR_LIT>' : '<STR_LIT>' <EOL> '<STR_LIT>' <EOL> '<STR_LIT>' <EOL> '<STR_LIT>' <EOL> '<STR_LIT>' <EOL> } <EOL> ) <EOL> with patch . dict ( dig . __salt__ , { '<STR_LIT>' : dig_mock } ) : <EOL> self . assertEqual ( <EOL> dig . MX ( '<STR_LIT>' ) , <EOL> [ [ '<STR_LIT>' , '<STR_LIT>' ] , <EOL> [ '<STR_LIT>' , '<STR_LIT>' ] , <EOL> [ '<STR_LIT>' , '<STR_LIT>' ] , <EOL> [ '<STR_LIT>' , '<STR_LIT>' ] , <EOL> [ '<STR_LIT>' , '<STR_LIT>' ] ] <EOL> ) <EOL> if __name__ == '<STR_LIT:__main__>' : <EOL> from integration import run_tests <EOL> run_tests ( DigTestCase , needs_daemon = False ) </s>
