#NEWLINE# @ mock . patch ( 'routersploit.utils.print_success' ) #NEWLINE# def test_command_check_target_vulnerable( self , mock_print_success ) : #NEWLINE# #INDENT# with mock . patch . object ( mock_check . interpreter . current_module , 'check' ) as mock_check : #NEWLINE# #INDENT# mock_check . return_value = True #NEWLINE# self . interpreter . command_check ( ) #NEWLINE# mock_check . assert_called_once_with ( ) #NEWLINE# mock_print_success . assert_called_once_with ( 'Target is vulnerable' )