tests: get_config() and tests for SparcClient

src/sparc/client/client.py

@@ -33,30 +33,36 @@ class SparcClient:
     -----------
     module_names : list
         Stores the list of modules that are automatically loaded from the <projectbase>/services directory.
+    config : ConfigParser
+        Config used for sparc.client
+
 
     Methods:
     --------
     add_module(path, config, connect):
         Adds and optionally connects to a module in a given path with configuration variables defined in config.
-    connect()
+    connect():
         Connects all the modules by calling their connect() functions.
+    get_config():
+        Returns config used by sparc.client
+
     """
 
     def __init__(self, config_file: str = "config.ini", connect: bool = True) -> None:
 
         # Try to find config file, if not available, provide default
-        config = ConfigParser()
-        config['global'] = {'default_profile' : 'default'}
-        config['default'] = {'pennsieve_profile_name' : 'pennsieve'}
+        self.config = ConfigParser()
+        self.config['global'] = {'default_profile' : 'default'}
+        self.config['default'] = {'pennsieve_profile_name' : 'pennsieve'}
 
         try:
-            config.read(config_file)
+            self.config.read(config_file)
         except Exception as e:
             logging.warning("Configuration file not provided or incorrect, using default settings.")
             pass
 
-        logging.debug(config.sections())
-        current_config = config["global"]["default_profile"]
+        logging.debug(self.config.sections())
+        current_config = self.config["global"]["default_profile"]
 
         logging.debug("Using the following config:" + current_config)
         self.module_names = []
@@ -67,7 +73,7 @@ def __init__(self, config_file: str = "config.ini", connect: bool = True) -> Non
         for _, module_name, _ in iter_modules([package_dir]):
             # import the module and iterate through its attributes
             self.add_module(
-                f"{__package__}.services.{module_name}", config[current_config], connect
+                f"{__package__}.services.{module_name}", self.config[current_config], connect
             )
 
     def add_module(
@@ -121,3 +127,7 @@ def connect(self) -> bool:
             if hasattr(module, "connect"):
                 getattr(self, module_name).connect()
         return True
+
+    def get_config(self) -> ConfigParser:
+        """Returns config for sparc.client"""
+        return self.config

tests/test_client.py

@@ -4,31 +4,40 @@
 
 from sparc.client import SparcClient
 
-
+# Test creating a default instance
 def test_class(config_file):
     c = SparcClient(connect=False, config_file=config_file)
     assert len(c.module_names) > 0
 
-
-# Config file tests
-def test_config_non_existing(config_file=None):
-    with pytest.raises(RuntimeError):
-        a = SparcClient(config_file, connect=False)
-
-
 # Test config file with incorrect section pointer
 def test_config_no_section(test_resources_dir):
     config_file = os.path.join(test_resources_dir, "dummy_config.ini")
     with pytest.raises(KeyError):
         a = SparcClient(config_file, connect=False)
 
-
+# Config non existing config
+def test_config_non_existing(config_file=None):
+    client = SparcClient(config_file, connect=False)
+    c = client.get_config()
+    assert c['global']['default_profile'] == 'default'
+    assert c['default']['pennsieve_profile_name'] == 'pennsieve'
+
+# Test proper config provided
+def test_get_config(test_resources_dir):
+    config_file = os.path.join(test_resources_dir, "config.ini")
+    client = SparcClient(config_file, connect=False)
+    c = client.get_config()
+    print(c.sections())
+    assert c['global']['default_profile'] == 'ci'
+    assert c['ci']['pennsieve_profile_name'] == 'ci'
+
+# Test module addition
 def test_failed_add_module(config_file):
     client = SparcClient(connect=False, config_file=config_file)
     with pytest.raises(ModuleNotFoundError):
         client.add_module(paths="sparc.client.xyz", connect=False)
 
-
+# Test using a config with the module
 def test_add_module_connect(config_file):
     sc = SparcClient(config_file=config_file, connect=False)
 
@@ -46,7 +55,7 @@ def test_add_module_connect(config_file):
     assert d.init_config_arg == expected_module_config
     assert d.connect_method_called is True
 
-
+# Test adding a Pennsieve module
 def test_add_pennsieve(config_file):
     sc = SparcClient(config_file=config_file, connect=False)
     assert "pennsieve" in sc.module_names
@@ -55,8 +64,8 @@ def test_add_pennsieve(config_file):
 
     assert isinstance(sc.pennsieve, PennsieveService)
 
-
-def test_connect(config_file, monkeypatch):
+# Test connection to the module
+def test_module_connect(config_file, monkeypatch):
     sc = SparcClient(config_file=config_file, connect=False)
     mock_connect_results = []