def test_create_thread():
session = MagicMock()
refresh_seconds = 1
retry_seconds = 2
heartbeat_thread = _HeartbeatThread(session, refresh_seconds, retry_seconds)
assert_equals(heartbeat_thread.livy_session, session)
assert_equals(heartbeat_thread.refresh_seconds, refresh_seconds)
assert_equals(heartbeat_thread.retry_seconds, retry_seconds)
def test_create_thread():
session = MagicMock()
refresh_seconds = 1
retry_seconds = 2
heartbeat_thread = _HeartbeatThread(session, refresh_seconds,
retry_seconds)
assert_equals(heartbeat_thread.livy_session, session)
assert_equals(heartbeat_thread.refresh_seconds, refresh_seconds)
assert_equals(heartbeat_thread.retry_seconds, retry_seconds)
def test_run_stops():
session = MagicMock()
refresh_seconds = 0.01
retry_seconds = 2
heartbeat_thread = _HeartbeatThread(session, refresh_seconds, retry_seconds)
heartbeat_thread.start()
sleep(0.1)
heartbeat_thread.stop()
assert session.refresh_status_and_info.called
assert heartbeat_thread.livy_session is None
def test_run_once():
session = MagicMock()
refresh_seconds = 0.1
retry_seconds = 2
heartbeat_thread = _HeartbeatThread(session, refresh_seconds, retry_seconds, 1)
heartbeat_thread.start()
sleep(0.15)
heartbeat_thread.stop()
session.refresh_status_and_info.assert_called_once_with()
assert heartbeat_thread.livy_session is None
def test_run_stops():
session = MagicMock()
refresh_seconds = 0.01
retry_seconds = 2
heartbeat_thread = _HeartbeatThread(session, refresh_seconds,
retry_seconds)
heartbeat_thread.start()
sleep(0.1)
heartbeat_thread.stop()
assert session.refresh_status_and_info.called
assert heartbeat_thread.livy_session is None
def test_run_once():
session = MagicMock()
refresh_seconds = 0.1
retry_seconds = 2
heartbeat_thread = _HeartbeatThread(session, refresh_seconds,
retry_seconds, 1)
heartbeat_thread.start()
sleep(0.15)
heartbeat_thread.stop()
session.refresh_status_and_info.assert_called_once_with()
assert heartbeat_thread.livy_session is None
def test_run_retries():
msg = "oh noes!"
session = MagicMock()
session.refresh_status_and_info = MagicMock(side_effect=ValueError(msg))
refresh_seconds = 0.1
retry_seconds = 0.1
heartbeat_thread = _HeartbeatThread(session, refresh_seconds, retry_seconds, 1)
heartbeat_thread.start()
sleep(0.15)
heartbeat_thread.stop()
session.refresh_status_and_info.assert_called_once_with()
session.logger.error.assert_called_once_with(msg)
assert heartbeat_thread.livy_session is None
def test_run_retries_stops():
msg = "oh noes!"
session = MagicMock()
session.refresh_status_and_info = MagicMock(side_effect=ValueError(msg))
refresh_seconds = 0.01
retry_seconds = 0.01
heartbeat_thread = _HeartbeatThread(session, refresh_seconds,
retry_seconds)
heartbeat_thread.start()
sleep(0.1)
heartbeat_thread.stop()
assert session.refresh_status_and_info.called
assert session.logger.error.called
assert heartbeat_thread.livy_session is None
def test_run_retries():
msg = "oh noes!"
session = MagicMock()
session.refresh_status = MagicMock(side_effect=ValueError(msg))
refresh_seconds = 0.1
retry_seconds = 0.1
heartbeat_thread = _HeartbeatThread(session, refresh_seconds,
retry_seconds, 1)
heartbeat_thread.start()
sleep(0.15)
heartbeat_thread.stop()
session.refresh_status.assert_called_once_with()
session.logger.error.assert_called_once_with(msg)
assert heartbeat_thread.livy_session is None
