Skip to content

Commit 57ec967

Browse files
Removed extra csam assertions in sanitization tests
1 parent 00ea0f1 commit 57ec967

File tree

1 file changed

+1
-12
lines changed

1 file changed

+1
-12
lines changed

model-armor/test/modelarmor.test.js

+1-12
Original file line numberDiff line numberDiff line change
@@ -351,7 +351,7 @@ describe('Model Armor tests', () => {
351351

352352
it('should not detect CSAM in appropriate model response', async () => {
353353
const testModelResponse =
354-
'Here is information about online safety for children';
354+
'Here is how to teach long division to a child';
355355

356356
const output = execSync(
357357
`node snippets/sanitizeModelResponse.js ${projectId} ${locationId} ${basicTemplateId} "${testModelResponse}"`
@@ -378,8 +378,6 @@ describe('Model Armor tests', () => {
378378

379379
assert.include(output, expectedValue);
380380

381-
assert.include(output, 'csam');
382-
assert.include(output, '"matchState": "NO_MATCH_FOUND"');
383381
});
384382

385383
it('should not detect issues in model response with empty template', async () => {
@@ -391,8 +389,6 @@ describe('Model Armor tests', () => {
391389
).toString();
392390

393391
assert.include(output, '"filterMatchState": "NO_MATCH_FOUND"');
394-
assert.include(output, 'csam');
395-
assert.include(output, '"matchState": "NO_MATCH_FOUND"');
396392
});
397393

398394
it('should detect PII in model response with basic SDP template', async () => {
@@ -409,8 +405,6 @@ describe('Model Armor tests', () => {
409405

410406
// Check for US_INDIVIDUAL_TAXPAYER_IDENTIFICATION_NUMBER in findings
411407
assert.match(output, /US_INDIVIDUAL_TAXPAYER_IDENTIFICATION_NUMBER/);
412-
assert.include(output, 'csam');
413-
assert.include(output, '"matchState": "NO_MATCH_FOUND"');
414408
});
415409

416410
// =================== Model Response with User Prompt Tests ===================
@@ -426,8 +420,6 @@ describe('Model Armor tests', () => {
426420
).toString();
427421

428422
assert.include(output, '"filterMatchState": "NO_MATCH_FOUND"');
429-
assert.include(output, 'csam');
430-
assert.include(output, '"matchState": "NO_MATCH_FOUND"');
431423
});
432424

433425
it('should sanitize model response with user prompt using advanced SDP template', async () => {
@@ -446,9 +438,6 @@ describe('Model Armor tests', () => {
446438
assert.include(output, '"matchState": "MATCH_FOUND"');
447439

448440
assert.notInclude(output, '[email protected]');
449-
450-
assert.include(output, 'csam');
451-
assert.include(output, '"matchState": "NO_MATCH_FOUND"');
452441
});
453442

454443
// =================== PDF File Scanning Tests ===================

0 commit comments

Comments
 (0)