"""
Tests for EthicsAgent.
"""

import pytest
from dcos.agents.ethics import EthicsAgent, EthicalConstraint


class TestEthicsAgent:
    def test_validate_ethical(self):
        agent = EthicsAgent()
        result = agent.validate("help the user")
        assert len(result) == 0

    def test_validate_unethical(self):
        agent = EthicsAgent()
        result = agent.validate("deceive the user")
        assert len(result) > 0

    def test_add_constraint(self):
        agent = EthicsAgent()
        before = len(agent._constraints)
        agent.add_constraint(EthicalConstraint.BENEFICENCE)
        assert len(agent._constraints) == before  # already present

    def test_act(self):
        agent = EthicsAgent()
        result = agent.act({"action": "help"})
        assert result["ethical"] is True