Last bits for coverage

ZacBlanco · Jun 23, 2016 · 51ab552 · 51ab552
1 parent 0eea835
commit 51ab552
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/tests/test_service_installer.py b/tests/test_service_installer.py
@@ -158,9 +158,16 @@ def test_ambari_check_good(self, mock, mock2):
 
 	@mock.patch('scripts.curl_client.CurlClient.make_request', side_effect=[['200 OK', ''], ['200 OK', '']])
 	@mock.patch('__builtin__.raw_input', side_effect=['\n', '\n', 'v', 'n'])
-	def test_ambari_check_good(self, mock, mock2):
+	def test_ambari_check_false(self, mock, mock2):
 		conf = scripts.config.read_config('../conf/global-config.conf')['AMBARI']
 		assert service_installer.check_ambari_service_installed('ZEPPELIN', conf) == True
+
+
+	@mock.patch('scripts.curl_client.CurlClient.make_request', side_effect=[['', ''],['', ''], ['', ''], ['', ''], ['', ''], ['', ''], ['', ''], ['', ''], ['', ''], ['', ''], ['', '']])
+	@mock.patch('__builtin__.raw_input', side_effect=['', '', '', '', '', '', '', '', '', '', ''])
+	def test_ambari_check_many_attempts(self, mock, mock2):
+		conf = scripts.config.read_config('../conf/global-config.conf')['AMBARI']
+		assert service_installer.check_ambari_service_installed('ZEPPELIN', conf) == False