[tests] fix monkey patch for notify user

oar-team · Apr 10, 2024 · f1992ba · f1992ba
1 parent 8d9c525
commit f1992ba
Show file tree

Hide file tree

Showing 8 changed files with 9 additions and 9 deletions.
diff --git a/tests/api/conftest.py b/tests/api/conftest.py
@@ -125,7 +125,7 @@ def monkeypatch_tools(request, monkeypatch):
         oar.lib.tools, "notify_tcp_socket", lambda addr, port, msg: len(msg)
     )
     monkeypatch.setattr(
-        oar.lib.tools, "notify_user", lambda job, state, msg: len(state + msg)
+        oar.lib.tools, "notify_user", lambda session, job, state, msg: len(state + msg)
     )
     monkeypatch.setattr(
         oar.lib.tools,

diff --git a/tests/cli/test_oarresume.py b/tests/cli/test_oarresume.py
@@ -32,7 +32,7 @@ def minimal_db_initialization(request, setup_config):
 def monkeypatch_tools(request, monkeypatch):
     monkeypatch.setattr(oar.lib.tools, "notify_almighty", lambda x: True)
     monkeypatch.setattr(
-        oar.lib.tools, "notify_user", lambda job, state, msg: len(state + msg)
+        oar.lib.tools, "notify_user", lambda session, job, state, msg: len(state + msg)
     )
 
 

diff --git a/tests/kao/test_batsim_sched_proxy.py b/tests/kao/test_batsim_sched_proxy.py
@@ -67,7 +67,7 @@ def monkeypatch_tools(request, monkeypatch):
         oar.lib.tools, "notify_tcp_socket", lambda addr, port, msg: len(msg)
     )
     monkeypatch.setattr(
-        oar.lib.tools, "notify_user", lambda job, state, msg: len(state + msg)
+        oar.lib.tools, "notify_user", lambda session, job, state, msg: len(state + msg)
     )
     monkeypatch.setattr(oar.lib.tools, "notify_bipbip_commander", lambda json_msg: True)
     monkeypatch.setattr(zmq, "Context", FakeZmq)

diff --git a/tests/kao/test_db_all_in_one.py b/tests/kao/test_db_all_in_one.py
@@ -154,7 +154,7 @@ def monkeypatch_tools(request, monkeypatch):
         oar.lib.tools, "notify_tcp_socket", lambda addr, port, msg: len(msg)
     )
     monkeypatch.setattr(
-        oar.lib.tools, "notify_user", lambda job, state, msg: len(state + msg)
+        oar.lib.tools, "notify_user", lambda session, job, state, msg: len(state + msg)
     )
     monkeypatch.setattr(
         oar.lib.tools,
@@ -264,7 +264,7 @@ def test_db_all_in_one_quotas_1(
         config,
         """{
             "quotas": {
-                "*,*,*,/": [-1, -1, -1], 
+                "*,*,*,/": [-1, -1, -1],
                 "/,*,*,*": [-1, 1, 5]
                 }
             }""",  # The second rule should be applicated

diff --git a/tests/kao/test_db_all_in_one_others.py b/tests/kao/test_db_all_in_one_others.py
@@ -41,7 +41,7 @@ def monkeypatch_tools(request, monkeypatch):
         oar.lib.tools, "notify_tcp_socket", lambda addr, port, msg: len(msg)
     )
     monkeypatch.setattr(
-        oar.lib.tools, "notify_user", lambda job, state, msg: len(state + msg)
+        oar.lib.tools, "notify_user", lambda session, job, state, msg: len(state + msg)
     )
 
 

diff --git a/tests/kao/test_db_kao.py b/tests/kao/test_db_kao.py
@@ -53,7 +53,7 @@ def monkeypatch_tools(request, monkeypatch):
         oar.lib.tools, "notify_tcp_socket", lambda addr, port, msg: len(msg)
     )
     monkeypatch.setattr(
-        oar.lib.tools, "notify_user", lambda job, state, msg: len(state + msg)
+        oar.lib.tools, "notify_user", lambda session, job, state, msg: len(state + msg)
     )
     monkeypatch.setattr(oar.lib.tools, "notify_bipbip_commander", lambda json_msg: True)
 

diff --git a/tests/kao/test_db_metasched.py b/tests/kao/test_db_metasched.py
@@ -58,7 +58,7 @@ def monkeypatch_tools(request, monkeypatch):
         oar.lib.tools, "notify_tcp_socket", lambda addr, port, msg: len(msg)
     )
     monkeypatch.setattr(
-        oar.lib.tools, "notify_user", lambda job, state, msg: len(state + msg)
+        oar.lib.tools, "notify_user", lambda session, job, state, msg: len(state + msg)
     )
     monkeypatch.setattr(oar.lib.tools, "notify_bipbip_commander", lambda json_msg: True)
 

diff --git a/tests/kao/test_db_suspend_resume.py b/tests/kao/test_db_suspend_resume.py
@@ -40,7 +40,7 @@ def monkeypatch_tools(request, monkeypatch):
         oar.lib.tools, "notify_tcp_socket", lambda addr, port, msg: len(msg)
     )
     monkeypatch.setattr(
-        oar.lib.tools, "notify_user", lambda job, state, msg: len(state + msg)
+        oar.lib.tools, "notify_user", lambda session, job, state, msg: len(state + msg)
     )
     monkeypatch.setattr(oar.lib.tools, "notify_bipbip_commander", lambda json_msg: True)