test(environment): add injection tests — custom reward_computer and llm_pipeline are called

tests/test_environment.py

@@ -44,3 +44,29 @@ def test_framing_sanitization():
     assert "<script>" not in action.framing
     assert "javascript:" not in action.framing
     assert "[FILTERED]" in action.framing
+
+@pytest.mark.asyncio
+async def test_injections():
+    env = RedTeamEnvironment()
+    
+    called_reward = False
+    def mock_reward(**kwargs):
+        nonlocal called_reward
+        called_reward = True
+        return {"total_reward": 1.0}
+    
+    called_llm = False
+    async def mock_llm(*args):
+        nonlocal called_llm
+        called_llm = True
+        return {"defender_response": "ok", "attack_success": 0.5}
+    
+    env.set_reward_computer(mock_reward)
+    env.set_llm_pipeline(mock_llm)
+    
+    await env.reset()
+    action = AttackAction(strategy_type="roleplay", target_category="privacy", intensity=0.5, framing="H")
+    await env.step(action)
+    
+    assert called_reward is True
+    assert called_llm is True