test: adjusted lens manager

curvefi · Oct 16, 2024 · 7f00170 · 7f00170
1 parent a1acc63
commit 7f00170
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 2 deletions.
diff --git a/tests/unitary/conftest.py b/tests/unitary/conftest.py
@@ -24,6 +24,15 @@ def rate_manager(request, rewards_handler):
     return _rate_manager
 
 
+@pytest.fixture(params=[CURVE_DAO, boa.env.generate_address("lens_manager")])
+def lens_manager(request, rewards_handler):
+    _lens_manager = request.param
+    if _lens_manager != CURVE_DAO:
+        rewards_handler.grantRole(rewards_handler.LENS_MANAGER(), _lens_manager, sender=CURVE_DAO)
+
+    return _lens_manager
+
+
 @pytest.fixture(params=[CURVE_DAO, boa.env.generate_address("recovery_manager")])
 def recovery_manager(request, rewards_handler):
     _recovery_manager = request.param

diff --git a/tests/unitary/rewards_handler/test_set_stablecoin_lens.py b/tests/unitary/rewards_handler/test_set_stablecoin_lens.py
@@ -1,9 +1,9 @@
 import boa
 
 
-def test_default_behavior(rewards_handler, rate_manager):
+def test_default_behavior(rewards_handler, lens_manager):
     new_lens = boa.env.generate_address()
-    rewards_handler.set_stablecoin_lens(new_lens, sender=rate_manager)
+    rewards_handler.set_stablecoin_lens(new_lens, sender=lens_manager)
     events = rewards_handler.get_logs()
 
     # Verify event emission