Update test client setup for the new swarm/aeolus structure

ESA-VirES · Aug 1, 2018 · f6cece7 · f6cece7
1 parent e2879b1
commit f6cece7
Showing 1 changed file with 10 additions and 1 deletion.
diff --git a/test/test_viresclient.py b/test/test_viresclient.py
@@ -3,7 +3,8 @@
 import pandas as pd
 
 from viresclient._data_handling import ReturnedData
-from viresclient import ClientRequest
+from viresclient._client import ClientRequest
+from viresclient import SwarmRequest, AeolusRequest
 import viresclient
 
 SUPPORTED_FILETYPES = ('csv', 'cdf')
@@ -75,3 +76,11 @@ def test_ClientRequest():
     assert isinstance(request._wps_service,
                       viresclient._wps.wps_vires.ViresWPS10Service
                       )
+    request = SwarmRequest('', '', '')
+    assert isinstance(request._wps_service,
+                      viresclient._wps.wps_vires.ViresWPS10Service
+                      )
+    request = AeolusRequest('', '', '')
+    assert isinstance(request._wps_service,
+                      viresclient._wps.wps_vires.ViresWPS10Service
+                      )