langchain Architecture Docs
Home / Class/ TestMaskStrategy Class — langchain Architecture

TestMaskStrategy Class — langchain Architecture

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

Class python

Entity Profile

Dependency Diagram

graph TD
  6da4bcad_d3fd_cbf4_a386_d46859fa4911["TestMaskStrategy"]
  6dcffc03_fae9_01f8_ed9b_d30d8a73dd13["test_pii.py"]
  6da4bcad_d3fd_cbf4_a386_d46859fa4911 -->|defined in| 6dcffc03_fae9_01f8_ed9b_d30d8a73dd13
  05d9f48d_c4ca_16de_28f2_1435499e31e8["test_mask_email()"]
  6da4bcad_d3fd_cbf4_a386_d46859fa4911 -->|method| 05d9f48d_c4ca_16de_28f2_1435499e31e8
  4503dffe_2185_933d_6706_fd3d9ae63af0["test_mask_credit_card()"]
  6da4bcad_d3fd_cbf4_a386_d46859fa4911 -->|method| 4503dffe_2185_933d_6706_fd3d9ae63af0
  c7c241e4_441b_6349_6ef5_b6afc8e88599["test_mask_ip()"]
  6da4bcad_d3fd_cbf4_a386_d46859fa4911 -->|method| c7c241e4_441b_6349_6ef5_b6afc8e88599

Relationship Graph

Source Code

libs/langchain_v1/tests/unit_tests/agents/middleware/implementations/test_pii.py lines 257–292

class TestMaskStrategy:
    """Test mask strategy."""

    def test_mask_email(self) -> None:
        middleware = PIIMiddleware("email", strategy="mask")
        state = AgentState[Any](messages=[HumanMessage("Email: user@example.com")])

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

        assert result is not None
        content = result["messages"][0].content
        assert "user@****.com" in content
        assert "user@example.com" not in content

    def test_mask_credit_card(self) -> None:
        middleware = PIIMiddleware("credit_card", strategy="mask")
        # Valid test card
        state = AgentState[Any](messages=[HumanMessage("Card: 4532015112830366")])

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

        assert result is not None
        content = result["messages"][0].content
        assert "0366" in content  # Last 4 digits visible
        assert "4532015112830366" not in content

    def test_mask_ip(self) -> None:
        middleware = PIIMiddleware("ip", strategy="mask")
        state = AgentState[Any](messages=[HumanMessage("IP: 192.168.1.100")])

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

        assert result is not None
        content = result["messages"][0].content
        assert "*.*.*.100" in content
        assert "192.168.1.100" not in content

Defined In

Source

Frequently Asked Questions

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

Analyze Your Own Codebase

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

Try Supermodel Free