improve test coverage

bihealth · Jan 4, 2024 · 50a3843 · 50a3843
1 parent 33243f3
commit 50a3843
Showing 1 changed file with 10 additions and 1 deletion.
diff --git a/tests/test_irods_common.py b/tests/test_irods_common.py
@@ -38,6 +38,16 @@ def test_init_irods(mocksession, fs):
     iRODSCommon()._init_irods()
     mocksession.assert_called()
 
+@patch("cubi_tk.irods_common.iRODSCommon._init_irods")
+def test_get_irods_sessions(mockinit):
+    with iRODSCommon()._get_irods_sessions(count = 4) as sessions:
+        [s for s in sessions]
+    assert mockinit.call_count == 4
+
+    mockinit.reset_mock()
+    with iRODSCommon()._get_irods_sessions(count = -1) as sessions:
+        [s for s in sessions]
+    assert mockinit.call_count == 1
 
 @patch("getpass.getpass")
 @patch("cubi_tk.irods_common.iRODSSession")
@@ -65,7 +75,6 @@ def test_save_irods_token(mocksession, mockencode, fs):
     assert icommon.irods_env_path.parent.joinpath(".irodsA").exists()
     mockencode.assert_called_with("secure")
 
-
 # Test iRODSTransfer #########
 @pytest.fixture
 def jobs():