<s> from __future__ import absolute_import <EOL> from salt . states import win_certutil as certutil <EOL> from salttesting import TestCase <EOL> from salttesting . helpers import ensure_in_syspath <EOL> from salttesting . mock import ( <EOL> MagicMock , <EOL> patch <EOL> ) <EOL> ensure_in_syspath ( '<STR_LIT>' ) <EOL> certutil . __salt__ = { } <EOL> class CertUtilTestCase ( TestCase ) : <EOL> def test_add_serial ( self ) : <EOL> '''<STR_LIT>''' <EOL> expected = { <EOL> '<STR_LIT>' : { '<STR_LIT>' : '<STR_LIT>' } , <EOL> '<STR_LIT>' : '<STR_LIT>' , <EOL> '<STR_LIT:name>' : '<STR_LIT>' , <EOL> '<STR_LIT:result>' : True <EOL> } <EOL> cache_mock = MagicMock ( return_value = '<STR_LIT>' ) <EOL> get_cert_serial_mock = MagicMock ( return_value = '<STR_LIT>' ) <EOL> get_store_serials_mock = MagicMock ( return_value = [ '<STR_LIT>' ] ) <EOL> add_mock = MagicMock ( return_value = '<STR_LIT>' ) <EOL> with patch . dict ( certutil . __salt__ , { '<STR_LIT>' : cache_mock , <EOL> '<STR_LIT>' : get_cert_serial_mock , <EOL> '<STR_LIT>' : get_store_serials_mock , <EOL> '<STR_LIT>' : add_mock } ) : <EOL> out = certutil . add_store ( '<STR_LIT>' , '<STR_LIT>' ) <EOL> cache_mock . assert_called_once_with ( '<STR_LIT>' , '<STR_LIT>' ) <EOL> get_cert_serial_mock . assert_called_once_with ( '<STR_LIT>' ) <EOL> get_store_serials_mock . assert_called_once_with ( '<STR_LIT>' ) <EOL> add_mock . assert_called_once_with ( '<STR_LIT>' , '<STR_LIT>' ) <EOL> self . assertEqual ( expected , out ) <EOL> def test_add_serial_missing ( self ) : <EOL> '''<STR_LIT>''' <EOL> expected = { <EOL> '<STR_LIT>' : { } , <EOL> '<STR_LIT>' : '<STR_LIT>' , <EOL> '<STR_LIT:name>' : '<STR_LIT>' , <EOL> '<STR_LIT:result>' : False <EOL> } <EOL> cache_mock = MagicMock ( return_value = False ) <EOL> get_cert_serial_mock = MagicMock ( return_value = '<STR_LIT>' ) <EOL> get_store_serials_mock = MagicMock ( return_value = [ '<STR_LIT>' ] ) <EOL> add_mock = MagicMock ( return_value = '<STR_LIT>' ) <EOL> with patch . dict ( certutil . __salt__ , { '<STR_LIT>' : cache_mock , <EOL> '<STR_LIT>' : get_cert_serial_mock , <EOL> '<STR_LIT>' : get_store_serials_mock , <EOL> '<STR_LIT>' : add_mock } ) : <EOL> out = certutil . add_store ( '<STR_LIT>' , '<STR_LIT>' ) <EOL> cache_mock . assert_called_once_with ( '<STR_LIT>' , '<STR_LIT>' ) <EOL> assert not get_cert_serial_mock . called <EOL> assert not get_store_serials_mock . called <EOL> assert not add_mock . called <EOL> self . assertEqual ( expected , out ) <EOL> def test_add_serial_exists ( self ) : <EOL> '''<STR_LIT>''' <EOL> expected = { <EOL> '<STR_LIT>' : { } , <EOL> '<STR_LIT>' : '<STR_LIT>' , <EOL> '<STR_LIT:name>' : '<STR_LIT>' , <EOL> '<STR_LIT:result>' : True <EOL> } <EOL> cache_mock = MagicMock ( return_value = '<STR_LIT>' ) <EOL> get_cert_serial_mock = MagicMock ( return_value = '<STR_LIT>' ) <EOL> get_store_serials_mock = MagicMock ( return_value = [ '<STR_LIT>' , '<STR_LIT>' ] ) <EOL> add_mock = MagicMock ( return_value = '<STR_LIT>' ) <EOL> with patch . dict ( certutil . __salt__ , { '<STR_LIT>' : cache_mock , <EOL> '<STR_LIT>' : get_cert_serial_mock , <EOL> '<STR_LIT>' : get_store_serials_mock , <EOL> '<STR_LIT>' : add_mock } ) : <EOL> out = certutil . add_store ( '<STR_LIT>' , '<STR_LIT>' ) <EOL> cache_mock . assert_called_once_with ( '<STR_LIT>' , '<STR_LIT>' ) <EOL> get_cert_serial_mock . assert_called_once_with ( '<STR_LIT>' ) <EOL> get_store_serials_mock . assert_called_once_with ( '<STR_LIT>' ) <EOL> assert not add_mock . called <EOL> self . assertEqual ( expected , out ) <EOL> def test_add_serial_fail ( self ) : <EOL> '''<STR_LIT>''' <EOL> expected = { <EOL> '<STR_LIT>' : { } , <EOL> '<STR_LIT>' : '<STR_LIT>' , <EOL> '<STR_LIT:name>' : '<STR_LIT>' , <EOL> '<STR_LIT:result>' : False <EOL> } <EOL> cache_mock = MagicMock ( return_value = '<STR_LIT>' ) <EOL> get_cert_serial_mock = MagicMock ( return_value = '<STR_LIT>' ) <EOL> get_store_serials_mock = MagicMock ( return_value = [ '<STR_LIT>' ] ) <EOL> add_mock = MagicMock ( return_value = '<STR_LIT>' ) <EOL> with patch . dict ( certutil . __salt__ , { '<STR_LIT>' : cache_mock , <EOL> '<STR_LIT>' : get_cert_serial_mock , <EOL> '<STR_LIT>' : get_store_serials_mock , <EOL> '<STR_LIT>' : add_mock } ) : <EOL> out = certutil . add_store ( '<STR_LIT>' , '<STR_LIT>' ) <EOL> cache_mock . assert_called_once_with ( '<STR_LIT>' , '<STR_LIT>' ) <EOL> get_cert_serial_mock . assert_called_once_with ( '<STR_LIT>' ) <EOL> get_store_serials_mock . assert_called_once_with ( '<STR_LIT>' ) <EOL> add_mock . assert_called_once_with ( '<STR_LIT>' , '<STR_LIT>' ) <EOL> self . assertEqual ( expected , out ) <EOL> def test_del_serial ( self ) : <EOL> '''<STR_LIT>''' <EOL> expected = { <EOL> '<STR_LIT>' : { '<STR_LIT>' : '<STR_LIT>' } , <EOL> '<STR_LIT>' : '<STR_LIT>' , <EOL> '<STR_LIT:name>' : '<STR_LIT>' , <EOL> '<STR_LIT:result>' : True <EOL> } <EOL> cache_mock = MagicMock ( return_value = '<STR_LIT>' ) <EOL> get_cert_serial_mock = MagicMock ( return_value = '<STR_LIT>' ) <EOL> get_store_serials_mock = MagicMock ( return_value = [ '<STR_LIT>' , '<STR_LIT>' ] ) <EOL> del_mock = MagicMock ( return_value = '<STR_LIT>' ) <EOL> with patch . dict ( certutil . __salt__ , { '<STR_LIT>' : cache_mock , <EOL> '<STR_LIT>' : get_cert_serial_mock , <EOL> '<STR_LIT>' : get_store_serials_mock , <EOL> '<STR_LIT>' : del_mock } ) : <EOL> out = certutil . del_store ( '<STR_LIT>' , '<STR_LIT>' ) <EOL> cache_mock . assert_called_once_with ( '<STR_LIT>' , '<STR_LIT>' ) <EOL> get_cert_serial_mock . assert_called_once_with ( '<STR_LIT>' ) <EOL> get_store_serials_mock . assert_called_once_with ( '<STR_LIT>' ) <EOL> del_mock . assert_called_once_with ( '<STR_LIT>' , '<STR_LIT>' ) <EOL> self . assertEqual ( expected , out ) <EOL> def test_del_serial_missing ( self ) : <EOL> '''<STR_LIT>''' <EOL> expected = { <EOL> '<STR_LIT>' : { } , <EOL> '<STR_LIT>' : '<STR_LIT>' , <EOL> '<STR_LIT:name>' : '<STR_LIT>' , <EOL> '<STR_LIT:result>' : False <EOL> } <EOL> cache_mock = MagicMock ( return_value = False ) <EOL> get_cert_serial_mock = MagicMock ( return_value = '<STR_LIT>' ) <EOL> get_store_serials_mock = MagicMock ( return_value = [ '<STR_LIT>' ] ) <EOL> del_mock = MagicMock ( return_value = '<STR_LIT>' ) <EOL> with patch . dict ( certutil . __salt__ , { '<STR_LIT>' : cache_mock , <EOL> '<STR_LIT>' : get_cert_serial_mock , <EOL> '<STR_LIT>' : get_store_serials_mock , <EOL> '<STR_LIT>' : del_mock } ) : <EOL> out = certutil . del_store ( '<STR_LIT>' , '<STR_LIT>' ) <EOL> cache_mock . assert_called_once_with ( '<STR_LIT>' , '<STR_LIT>' ) <EOL> assert not get_cert_serial_mock . called <EOL> assert not get_store_serials_mock . called <EOL> assert not del_mock . called <EOL> self . assertEqual ( expected , out ) <EOL> def test_del_serial_doesnt_exists ( self ) : <EOL> '''<STR_LIT>''' <EOL> expected = { <EOL> '<STR_LIT>' : { } , <EOL> '<STR_LIT>' : '<STR_LIT>' , <EOL> '<STR_LIT:name>' : '<STR_LIT>' , <EOL> '<STR_LIT:result>' : True <EOL> } <EOL> cache_mock = MagicMock ( return_value = '<STR_LIT>' ) <EOL> get_cert_serial_mock = MagicMock ( return_value = '<STR_LIT>' ) <EOL> get_store_serials_mock = MagicMock ( return_value = [ '<STR_LIT>' ] ) <EOL> del_mock = MagicMock ( return_value = '<STR_LIT>' ) <EOL> with patch . dict ( certutil . __salt__ , { '<STR_LIT>' : cache_mock , <EOL> '<STR_LIT>' : get_cert_serial_mock , <EOL> '<STR_LIT>' : get_store_serials_mock , <EOL> '<STR_LIT>' : del_mock } ) : <EOL> out = certutil . del_store ( '<STR_LIT>' , '<STR_LIT>' ) <EOL> cache_mock . assert_called_once_with ( '<STR_LIT>' , '<STR_LIT>' ) <EOL> get_cert_serial_mock . assert_called_once_with ( '<STR_LIT>' ) <EOL> get_store_serials_mock . assert_called_once_with ( '<STR_LIT>' ) <EOL> assert not del_mock . called <EOL> self . assertEqual ( expected , out ) <EOL> def test_del_serial_fail ( self ) : <EOL> '''<STR_LIT>''' <EOL> expected = { <EOL> '<STR_LIT>' : { } , <EOL> '<STR_LIT>' : '<STR_LIT>' , <EOL> '<STR_LIT:name>' : '<STR_LIT>' , <EOL> '<STR_LIT:result>' : False <EOL> } <EOL> cache_mock = MagicMock ( return_value = '<STR_LIT>' ) <EOL> get_cert_serial_mock = MagicMock ( return_value = '<STR_LIT>' ) <EOL> get_store_serials_mock = MagicMock ( return_value = [ '<STR_LIT>' , '<STR_LIT>' ] ) <EOL> del_mock = MagicMock ( return_value = '<STR_LIT>' ) <EOL> with patch . dict ( certutil . __salt__ , { '<STR_LIT>' : cache_mock , <EOL> '<STR_LIT>' : get_cert_serial_mock , <EOL> '<STR_LIT>' : get_store_serials_mock , <EOL> '<STR_LIT>' : del_mock } ) : <EOL> out = certutil . del_store ( '<STR_LIT>' , '<STR_LIT>' ) <EOL> cache_mock . assert_called_once_with ( '<STR_LIT>' , '<STR_LIT>' ) <EOL> get_cert_serial_mock . assert_called_once_with ( '<STR_LIT>' ) <EOL> get_store_serials_mock . assert_called_once_with ( '<STR_LIT>' ) <EOL> del_mock . assert_called_once_with ( '<STR_LIT>' , '<STR_LIT>' ) <EOL> self . assertEqual ( expected , out ) <EOL> if __name__ == '<STR_LIT:__main__>' : <EOL> from integration import run_tests <EOL> run_tests ( CertUtilTestCase , needs_daemon = False ) </s>
