first commit

packages/@n8n/nodes-langchain/nodes/Guardrails/test/checks/pii.test.ts

@@ -0,0 +1,28 @@
+import type { PIIConfig } from '../../actions/checks/pii';
+import { PIIEntity, createPiiCheckFn } from '../../actions/checks/pii';
+
+describe('pii guardrail', () => {
+	it('masks detected PII and triggers tripwire', async () => {
+		const config: PIIConfig = {
+			entities: [PIIEntity.EMAIL_ADDRESS, PIIEntity.US_SSN],
+		};
+		const text = 'Contact john@example.com SSN: 111-22-3333';
+
+		const result = await createPiiCheckFn(config)(text);
+
+		expect(result.tripwireTriggered).toBe(true);
+		expect(result.info?.maskEntities?.EMAIL_ADDRESS).toEqual(['john@example.com']);
+		expect(result.info?.maskEntities?.US_SSN).toEqual(['111-22-3333']);
+	});
+
+	it('returns no findings on empty input', async () => {
+		const config: PIIConfig = {
+			entities: [PIIEntity.EMAIL_ADDRESS],
+		};
+
+		const result = await createPiiCheckFn(config)('');
+		expect(result.tripwireTriggered).toBe(false);
+		expect(result.info?.maskEntities).toEqual({});
+		expect(result.info?.analyzerResults).toEqual([]);
+	});
+});