def test_main(self, mock_exit, mock_argv, mock_health_check):
mock_health_check.side_effect = [1, Exception('FAIL')]
haproxy_vrrp_check.main()
mock_exit.assert_called_once_with(1)
def test_main(self, mock_exit, mock_argv, mock_health_check):
mock_health_check.side_effect = [1, Exception('FAIL')]
haproxy_vrrp_check.main()
mock_exit.assert_called_once_with(1)