fix(modelarmor): Fix tests and upgrade libraries

harshnasitcrest · harshnasitcrest · commit 1bf5a0352f38 · 2025-05-30T12:53:36.000+05:30
diff --git a/model_armor/snippets/requirements.txt b/model_armor/snippets/requirements.txt
@@ -1,2 +1,2 @@
-google-cloud-modelarmor==0.2.2
-google-cloud-dlp==3.27.0
+google-cloud-modelarmor==0.2.5
+google-cloud-dlp==3.30.0
diff --git a/model_armor/snippets/snippets_test.py b/model_armor/snippets/snippets_test.py
@@ -515,7 +515,7 @@ def test_create_model_armor_template_with_basic_sdp(
 
     assert (
         filter_enforcement.name
-        == modelarmor_v1.SdpBasicConfig.filter_enforcement.ENABLED.name
+        == modelarmor_v1.SdpBasicConfig.SdpBasicConfigEnforcement.ENABLED.name
     )
 
 
@@ -672,7 +672,7 @@ def test_update_model_armor_template_with_mask_configuration(
     )
     assert (
         filter_enforcement.name
-        != modelarmor_v1.SdpBasicConfig.filter_enforcement.ENABLED.name
+        != modelarmor_v1.SdpBasicConfig.SdpBasicConfigEnforcement.ENABLED.name
     )
 
 

Original file line number	Diff line number	Diff line change
`@@ -515,7 +515,7 @@ def test_create_model_armor_template_with_basic_sdp(`
`515`	`515`
`516`	`516`	`assert (`
`517`	`517`	`filter_enforcement.name`
`518`		`- == modelarmor_v1.SdpBasicConfig.filter_enforcement.ENABLED.name`
	`518`	`+ == modelarmor_v1.SdpBasicConfig.SdpBasicConfigEnforcement.ENABLED.name`
`519`	`519`	`)`
`520`	`520`
`521`	`521`
`@@ -672,7 +672,7 @@ def test_update_model_armor_template_with_mask_configuration(`
`672`	`672`	`)`
`673`	`673`	`assert (`
`674`	`674`	`filter_enforcement.name`
`675`		`- != modelarmor_v1.SdpBasicConfig.filter_enforcement.ENABLED.name`
	`675`	`+ != modelarmor_v1.SdpBasicConfig.SdpBasicConfigEnforcement.ENABLED.name`
`676`	`676`	`)`
`677`	`677`
`678`	`678`