Home / Class/ TestRedactStrategy Class — langchain Architecture

TestRedactStrategy Class — langchain Architecture

Architecture documentation for the TestRedactStrategy class in test_pii.py from the langchain codebase.

Entity Profile

Dependency Diagram

graph TD
  f0334f9f_5b74_843e_1a42_a995ed9c8b18["TestRedactStrategy"]
  6dcffc03_fae9_01f8_ed9b_d30d8a73dd13["test_pii.py"]
  f0334f9f_5b74_843e_1a42_a995ed9c8b18 -->|defined in| 6dcffc03_fae9_01f8_ed9b_d30d8a73dd13
  437a7f4a_3ed3_ce3f_fbf8_ab71d470e594["test_redact_email()"]
  f0334f9f_5b74_843e_1a42_a995ed9c8b18 -->|method| 437a7f4a_3ed3_ce3f_fbf8_ab71d470e594
  ba145a13_365b_f950_d747_2937e78c7fe2["test_redact_multiple_pii()"]
  f0334f9f_5b74_843e_1a42_a995ed9c8b18 -->|method| ba145a13_365b_f950_d747_2937e78c7fe2

Relationship Graph

Source Code

libs/langchain_v1/tests/unit_tests/agents/middleware/implementations/test_pii.py lines 231–254

class TestRedactStrategy:
    """Test redact strategy."""

    def test_redact_email(self) -> None:
        middleware = PIIMiddleware("email", strategy="redact")
        state = AgentState[Any](messages=[HumanMessage("Email me at test@example.com")])

        result = middleware.before_model(state, Runtime())

        assert result is not None
        assert "[REDACTED_EMAIL]" in result["messages"][0].content
        assert "test@example.com" not in result["messages"][0].content

    def test_redact_multiple_pii(self) -> None:
        middleware = PIIMiddleware("email", strategy="redact")
        state = AgentState[Any](messages=[HumanMessage("Contact alice@test.com or bob@test.com")])

        result = middleware.before_model(state, Runtime())

        assert result is not None
        content = result["messages"][0].content
        assert content.count("[REDACTED_EMAIL]") == 2
        assert "alice@test.com" not in content
        assert "bob@test.com" not in content

Frequently Asked Questions

What is the TestRedactStrategy class?
TestRedactStrategy is a class in the langchain codebase, defined in libs/langchain_v1/tests/unit_tests/agents/middleware/implementations/test_pii.py.
Where is TestRedactStrategy defined?
TestRedactStrategy is defined in libs/langchain_v1/tests/unit_tests/agents/middleware/implementations/test_pii.py at line 231.

Analyze Your Own Codebase

Get architecture documentation, dependency graphs, and domain analysis for your codebase in minutes.

Try Supermodel Free