#NEWLINE# def test_new_zone( self ) : #NEWLINE# #INDENT# '\n        Test for Add a new zone\n        ' #NEWLINE# with patch . object ( firewalld , '__mgmt' , return_value = 'success' ) : #NEWLINE# #INDENT# mock = MagicMock ( return_value = 'A' ) #NEWLINE# with patch . object ( firewalld , '__firewall_cmd' , mock ) : #NEWLINE# #INDENT# self . assertEqual ( firewalld . new_zone ( 'zone' ) , 'A' ) #NEWLINE# #UNINDENT# #UNINDENT# with patch . object ( firewalld , '__mgmt' , return_value = 'A' ) : #NEWLINE# #INDENT# self . assertEqual ( firewalld . new_zone ( 'zone' ) , 'A' ) #NEWLINE# #UNINDENT# with patch . object ( firewalld , '__mgmt' , return_value = 'A' ) : #NEWLINE# #INDENT# self . assertEqual ( firewalld . new_zone ( 'zone' , False ) , 'A' )