Skip to content

Commit bb95b7c

Browse files
committed
fix: registerAndGetOwnerId is private method
1 parent 517507f commit bb95b7c

File tree

2 files changed

+12
-10
lines changed

2 files changed

+12
-10
lines changed

src/DIRAC/WorkloadManagementSystem/DB/SandboxMetadataDB.py

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -12,7 +12,7 @@ def __init__(self):
1212
DB.__init__(self, "SandboxMetadataDB", "WorkloadManagement/SandboxMetadataDB")
1313
result = self.__initializeDB()
1414
if not result["OK"]:
15-
raise RuntimeError("Can't create tables: %s" % result["Message"])
15+
raise RuntimeError(f"Can't create tables: {result['Message']}")
1616
self.__assignedSBGraceDays = 0
1717
self.__unassignedSBGraceDays = 15
1818

@@ -73,7 +73,7 @@ def __initializeDB(self):
7373

7474
return self._createTables(tablesToCreate)
7575

76-
def registerAndGetOwnerId(self, owner, ownerDN, ownerGroup):
76+
def __registerAndGetOwnerId(self, owner, ownerDN, ownerGroup):
7777
"""
7878
Get the owner ID and register it if it's not there
7979
"""
@@ -112,7 +112,7 @@ def registerAndGetSandbox(self, owner, ownerDN, ownerGroup, sbSE, sbPFN, size=0)
112112
Register a new sandbox in the metadata catalog
113113
Returns ( sbid, newSandbox )
114114
"""
115-
result = self.registerAndGetOwnerId(owner, ownerDN, ownerGroup)
115+
result = self.__registerAndGetOwnerId(owner, ownerDN, ownerGroup)
116116
if not result["OK"]:
117117
return result
118118
ownerId = result["Value"]
@@ -144,7 +144,7 @@ def registerAndGetSandbox(self, owner, ownerDN, ownerGroup, sbSE, sbPFN, size=0)
144144
return S_OK((result["lastRowId"], True))
145145
result = self._query("SELECT LAST_INSERT_ID()")
146146
if not result["OK"]:
147-
return S_ERROR("Can't determine sand box id after insertion")
147+
return S_ERROR("Can't determine sandbox id after insertion")
148148
return S_OK((result["Value"][0][0], True))
149149

150150
def accessedSandboxById(self, sbId):
@@ -273,7 +273,7 @@ def unassignEntities(self, entitiesDict, requesterName, requesterGroup):
273273
escapedSetup = self._escapeString(entitySetup)["Value"]
274274
result = self.__filterEntitiesByRequester(entitiesIds, escapedSetup, requesterName, requesterGroup)
275275
if not result["OK"]:
276-
gLogger.error("Cannot filter entities: %s" % result["Message"])
276+
gLogger.error(f"Cannot filter entities: {result['Message']}")
277277
continue
278278
ids = result["Value"]
279279
if not ids:
@@ -283,7 +283,7 @@ def unassignEntities(self, entitiesDict, requesterName, requesterGroup):
283283
sqlCmd = "DELETE FROM `sb_EntityMapping` WHERE %s" % " AND ".join(sqlCond)
284284
result = self._update(sqlCmd)
285285
if not result["OK"]:
286-
gLogger.error("Cannot unassign entities: %s" % result["Message"])
286+
gLogger.error(f"Cannot unassign entities: {result['Message']}")
287287
else:
288288
updated += 1
289289
return S_OK(updated)

tests/Integration/WorkloadManagementSystem/Test_SandboxStoreClient.py

Lines changed: 6 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -29,9 +29,6 @@
2929
python -m pytest -c ../pytest.ini -vv tests/Integration/WorkloadManagementSystem/Test_SandboxStoreClient.py
3030
"""
3131

32-
from __future__ import print_function
33-
from __future__ import absolute_import
34-
from __future__ import division
3532
from DIRAC.Core.Base.Script import parseCommandLine
3633

3734
parseCommandLine()
@@ -47,7 +44,7 @@
4744
gLogger.setLevel("DEBUG")
4845

4946

50-
def test_SSCChain(self):
47+
def test_SSCChain():
5148
"""full test of functionalities"""
5249
ssc = SandboxStoreClient()
5350
smDB = SandboxMetadataDB()
@@ -71,3 +68,8 @@ def test_SSCChain(self):
7168
# # cleaning
7269
# res = smDB.deleteSandboxes(SBIdList)
7370
# assert res['OK'] is True
71+
72+
73+
def test_SandboxMetadataDB():
74+
smDB = SandboxMetadataDB()
75+
smDB.registerAndGetSandbox("owner_1", "owner_1_DN", "owner_1_group", "sbSE", "/sb/pfn/1.tar.bz2")

0 commit comments

Comments
 (0)