Skip to content

Commit 281473a

Browse files
committed
feat: add batchSystem type in jobParameters
1 parent 53f617c commit 281473a

File tree

4 files changed

+5
-3
lines changed

4 files changed

+5
-3
lines changed

src/DIRAC/Resources/Computing/BatchSystems/TimeLeft/test/Test_PBSResourceUsage.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -76,7 +76,6 @@ def test_getResourcUsage(mocker):
7676
side_effect=[S_OK(RRCKI_OUT)],
7777
)
7878
mocker.patch("os.path.isfile", return_value=True)
79-
mocker.patch.dict(os.environ, {"PBS_O_QUEUE": "lhcb", "PBS_O_QPATH": "/some/path"})
8079

8180
pbsRU = PBSResourceUsage("55755440.seer.t1.grid.kiae.ru", {"Queue": "lhcb", "BinaryPath": "/some/path"})
8281
res = pbsRU.getResourceUsage()

src/DIRAC/Resources/Computing/BatchSystems/TimeLeft/test/Test_SGEResourceUsage.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -46,7 +46,7 @@ def test_getResourceUsage(mocker):
4646
return_value=S_OK(RESULT_FROM_SGE),
4747
)
4848

49-
sgeResourceUsage = SGEResourceUsage()
49+
sgeResourceUsage = SGEResourceUsage("1234", {"Queue": "Test"})
5050
res = sgeResourceUsage.getResourceUsage()
5151

5252
assert res["OK"]

src/DIRAC/Resources/Computing/BatchSystems/TimeLeft/test/Test_TimeLeft.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -24,7 +24,7 @@ def test_batchSystemNotDefined(mocker):
2424
tl.cpuPower = 10
2525
res = tl.getTimeLeft()
2626
assert not res["OK"]
27-
assert res["Message"] == "Current batch system is not supported"
27+
assert "Current batch system is not supported" in res["Message"]
2828

2929

3030
def test_getScaledCPU(mocker):

src/DIRAC/WorkloadManagementSystem/Agent/JobAgent.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -286,6 +286,9 @@ def execute(self):
286286
if queue:
287287
self.jobReport.setJobParameter(par_name="CEQueue", par_value=queue, sendFlag=False)
288288

289+
if batchSystem := gConfig.getValue("/LocalSite/BatchSystem/Type", ""):
290+
self.jobReport.setJobParameter(par_name="BatchSystem", par_value=batchSystem, sendFlag=False)
291+
289292
self.log.debug(f"Before self._submitJob() ({self.ceName}CE)")
290293
result = self._submitJob(
291294
jobID=jobID,

0 commit comments

Comments
 (0)