def test_nifi_ambari_good(self, mock, mock2, mock3):  #Also HDP select bad
     try:
         service_installer.install_nifi()
         self.fail('Cannot continue installation without hdp-select')
     except EnvironmentError as e:
         assert str(
             e
         ) == 'hdp-select could not be installed. Please install it manually and then re-run the setup.'
 def test_nifi_ambari_bad(self, mock):
     try:
         service_installer.install_nifi()
         self.fail('Cannot continue installation without Ambari')
     except EnvironmentError as e:
         assert str(
             e
         ) == 'You must install the demo on the same node as the Ambari server. Install Ambari here or move to another node with Ambari installed before continuing'
 def test_nifi_no_ambari_contact_no_continue(self, mock, mock2, mock3, mock4, mock5):
     assert service_installer.install_nifi() == False
 def test_nifi_check_is_good(self, mock, mock2, mock3, mock4, mock5):
     assert service_installer.install_nifi() == True
 def test_nifi_ambari_good(self, mock, mock2, mock3): #Also HDP select bad
   try:
     service_installer.install_nifi()
     self.fail('Cannot continue installation without hdp-select')
   except EnvironmentError as e:
     assert str(e) == 'hdp-select could not be installed. Please install it manually and then re-run the setup.'
 def test_nifi_ambari_bad(self, mock):
   try:
     service_installer.install_nifi()
     self.fail('Cannot continue installation without Ambari')
   except EnvironmentError as e:
     assert str(e) == 'You must install the demo on the same node as the Ambari server. Install Ambari here or move to another node with Ambari installed before continuing'
 def test_nifi_no_ambari_contact_no_continue(self, mock, mock2, mock3,
                                             mock4, mock5):
     assert service_installer.install_nifi() == False
 def test_nifi_check_is_good(self, mock, mock2, mock3, mock4, mock5):
     assert service_installer.install_nifi() == True