def test_get_logs_returns_false_with_value_error(): err = "err" mock_spark_session = MagicMock() type(mock_spark_session).logs = PropertyMock(side_effect=ValueError(err)) client = LivyClient(mock_spark_session) res, logs_r = client.get_logs() assert not res assert logs_r == err
def test_get_logs_returns_false_with_value_error(): err = "err" mock_spark_session = MagicMock() mock_spark_session.get_logs = MagicMock(side_effect=ValueError(err)) client = LivyClient(mock_spark_session) res, logs_r = client.get_logs() assert not res assert logs_r == err
def test_get_logs_returns_session_logs(): logs = "hi" mock_spark_session = MagicMock() mock_spark_session.logs = logs client = LivyClient(mock_spark_session) res, logs_r = client.get_logs() assert res assert logs_r == logs
def test_get_logs_returns_session_logs(): logs = "hi" mock_spark_session = MagicMock() mock_spark_session.get_logs = MagicMock(return_value=logs) client = LivyClient(mock_spark_session) res, logs_r = client.get_logs() assert res assert logs_r == logs