EUCLIDPCR-1863 New tests

Author: Jorge Fernandez Hernandez

astroquery/esa/euclid/tests/test_euclidtap.py

@@ -317,12 +317,35 @@ def test_launch_sync_job(tmp_path_factory):
     assert 3 == len(results), len(results)
 
 
-def test_launch_async_job():
-    conn_handler = DummyConnHandler()
+@patch.object(TapPlus, 'launch_job')
+def test_launch_sync_job_exception(mock_launch_job, caplog):
+    query = "select alpha, delta, source_id, table1_oid from caca"
 
+    conn_handler = DummyConnHandler()
     tap_plus = TapPlus(url="http://test:1111/tap", connhandler=conn_handler)
-    query = "select alpha, delta, source_id, table1_oid from caca"
 
+    launch_response = DummyResponse(200)
+    launch_response.set_data(method="POST", body=JOB_DATA)
+    conn_handler.set_default_response(launch_response)
+
+    tap = EuclidClass(tap_plus_conn_handler=conn_handler, datalink_handler=tap_plus, show_server_messages=False)
+
+    mock_launch_job.side_effect = HTTPError("launch_job HTTPError")
+
+    tap.launch_job(query, output_format="votable")
+
+    mssg = "Query failed: select alpha, delta, source_id, table1_oid from caca: HTTP error: launch_job HTTPError"
+    assert caplog.records[0].msg == mssg
+
+    mock_launch_job.side_effect = Exception("launch_job Exception")
+
+    tap.launch_job(query, output_format="votable")
+
+    mssg = "Query failed: select alpha, delta, source_id, table1_oid from caca, launch_job Exception"
+    assert caplog.records[1].msg == mssg
+
+
+def test_launch_async_job():
     conn_handler = DummyConnHandler()
     tap_plus = TapPlus(url="http://test:1111/tap", connhandler=conn_handler)
     jobid = '12345'
@@ -346,6 +369,8 @@ def test_launch_async_job():
     req = "async/" + jobid + "/results/result"
     conn_handler.set_response(req, responseResultsJob)
 
+    query = "select alpha, delta, source_id, table1_oid from caca"
+
     tap = EuclidClass(tap_plus_conn_handler=conn_handler, datalink_handler=tap_plus, show_server_messages=False)
 
     job = tap.launch_job_async(query, output_format="votable")
@@ -391,6 +416,29 @@ def test_launch_async_job():
     assert 3 == len(results), len(results)
 
 
+@patch.object(TapPlus, 'launch_job_async')
+def test_launch_async_job_exception(mock_launch_job_async, caplog):
+    conn_handler = DummyConnHandler()
+    tap_plus = TapPlus(url="http://test:1111/tap", connhandler=conn_handler)
+
+    tap = EuclidClass(tap_plus_conn_handler=conn_handler, datalink_handler=tap_plus, show_server_messages=False)
+    query = "select alpha, delta, source_id, table1_oid from caca"
+
+    mock_launch_job_async.side_effect = HTTPError("launch_job_async HTTPError")
+
+    tap.launch_job_async(query, output_format="votable")
+
+    mssg = "Query failed: select alpha, delta, source_id, table1_oid from caca: HTTP error: launch_job_async HTTPError"
+    assert caplog.records[0].msg == mssg
+
+    mock_launch_job_async.side_effect = Exception("launch_job_async Exception")
+
+    tap.launch_job_async(query, output_format="votable")
+
+    mssg = "Query failed: select alpha, delta, source_id, table1_oid from caca, launch_job_async Exception"
+    assert caplog.records[1].msg == mssg
+
+
 def test_list_async_jobs():
     conn_handler = DummyConnHandler()
     tap_plus = TapPlus(url="http://test:1111/tap", connhandler=conn_handler)
@@ -682,6 +730,38 @@ def test_get_obs_products():
 
     assert result is not None
 
+    result = tap.get_observation_products(id='13', product_type='mosaic', filter='VIS', output_file=None)
+
+    assert result is not None
+
+
+def test_get_obs_products_exceptions():
+    conn_handler = DummyConnHandler()
+    tap_plus = TapPlus(url="http://test:1111/tap", data_context='data', client_id='ASTROQUERY',
+                       connhandler=conn_handler)
+    # Launch response: we use default response because the query contains decimals
+    responseLaunchJob = DummyResponse(200)
+    responseLaunchJob.set_data(method='POST', context=None, body='', headers=None)
+
+    conn_handler.set_default_response(responseLaunchJob)
+
+    tap = EuclidClass(tap_plus_conn_handler=conn_handler, datalink_handler=tap_plus, show_server_messages=False)
+
+    with pytest.raises(ValueError) as exc_info:
+        tap.get_observation_products(id=None, product_type='observation', filter='VIS', output_file=None)
+
+    assert str(exc_info.value).startswith("Missing required argument: 'observation_id'")
+
+    with pytest.raises(ValueError) as exc_info:
+        tap.get_observation_products(id=12, product_type=None, filter='VIS', output_file=None)
+
+    assert str(exc_info.value).startswith("Missing required argument: 'product_type'")
+
+    with pytest.raises(ValueError) as exc_info:
+        tap.get_observation_products(id=12, product_type='XXXXXXXX', filter='VIS', output_file=None)
+
+    assert str(exc_info.value).startswith("Invalid product type XXXXXXXX. Valid values: ['observation', 'mosaic']")
+
 
 def test_get_cutout():
     conn_handler = DummyConnHandler()