diff --git a/modelarmor/modelarmor_test.go b/modelarmor/modelarmor_test.go index 3f35878f0a..1432e255bd 100644 --- a/modelarmor/modelarmor_test.go +++ b/modelarmor/modelarmor_test.go @@ -37,8 +37,6 @@ import ( grpcstatus "google.golang.org/grpc/status" ) -// TODO: Floorsettings test cases will be added later - // testLocation retrieves the GOLANG_SAMPLES_LOCATION environment variable // used to determine the region for running the test. // Skip the test if the environment variable is not set. @@ -1297,8 +1295,6 @@ func TestUpdateTemplateWithMaskConfiguration(t *testing.T) { // TestGetProjectFloorSettings tests the retrieval of floor settings at the project level. // It verifies the output contains the expected confirmation string. func TestGetProjectFloorSettings(t *testing.T) { - - t.Skip("TODO(b/424365799): Update this once the mentioned issue is resolved") tc := testutil.SystemTest(t) var buf bytes.Buffer @@ -1314,7 +1310,6 @@ func TestGetProjectFloorSettings(t *testing.T) { // TestGetOrganizationFloorSettings tests the retrieval of floor settings at the organization level. // It checks that the output includes the expected string indicating success. func TestGetOrganizationFloorSettings(t *testing.T) { - t.Skip("TODO(b/424365799): Update this once the mentioned issue is resolved") organizationID := testOrganizationID(t) var buf bytes.Buffer if err := getOrganizationFloorSettings(&buf, organizationID); err != nil { @@ -1329,7 +1324,6 @@ func TestGetOrganizationFloorSettings(t *testing.T) { // TestGetFolderFloorSettings tests the retrieval of floor settings at the folder level. // It ensures the result contains the expected confirmation message. func TestGetFolderFloorSettings(t *testing.T) { - t.Skip("TODO(b/424365799): Update this once the mentioned issue is resolved") folderID := testFolderID(t) var buf bytes.Buffer if err := getFolderFloorSettings(&buf, folderID); err != nil { @@ -1344,7 +1338,6 @@ func TestGetFolderFloorSettings(t *testing.T) { // TestUpdateFolderFloorSettings tests updating floor settings for a specific folder. // It verifies that the output buffer contains a confirmation message indicating a successful update. func TestUpdateFolderFloorSettings(t *testing.T) { - t.Skip("TODO(b/424365799): Update this once the mentioned issue is resolved") folderID := testFolderID(t) locationID := testLocation(t) var buf bytes.Buffer @@ -1364,7 +1357,6 @@ func TestUpdateFolderFloorSettings(t *testing.T) { // TestUpdateOrganizationFloorSettings tests updating floor settings for a specific organization. // It ensures the output buffer includes a success message confirming the update. func TestUpdateOrganizationFloorSettings(t *testing.T) { - t.Skip("TODO(b/424365799): Update this once the mentioned issue is resolved") organizationID := testOrganizationID(t) locationID := testLocation(t) var buf bytes.Buffer @@ -1385,7 +1377,6 @@ func TestUpdateOrganizationFloorSettings(t *testing.T) { // TestUpdateProjectFloorSettings tests updating floor settings for a specific project. // It checks that the resulting output includes the expected confirmation message. func TestUpdateProjectFloorSettings(t *testing.T) { - t.Skip("TODO(b/424365799): Update this once the mentioned issue is resolved") tc := testutil.SystemTest(t) locationID := testLocation(t) var buf bytes.Buffer diff --git a/modelarmor/update_folder_floor_settings.go b/modelarmor/update_folder_floor_settings.go index 6718d98d48..447497c609 100644 --- a/modelarmor/update_folder_floor_settings.go +++ b/modelarmor/update_folder_floor_settings.go @@ -59,7 +59,7 @@ func updateFolderFloorSettings(w io.Writer, folderID, locationID string) error { RaiFilters: []*modelarmorpb.RaiFilterSettings_RaiFilter{ { FilterType: modelarmorpb.RaiFilterType_HATE_SPEECH, - ConfidenceLevel: modelarmorpb.DetectionConfidenceLevel_HIGH, + ConfidenceLevel: modelarmorpb.DetectionConfidenceLevel_LOW_AND_ABOVE, }, }, }, diff --git a/modelarmor/update_organization_floor_settings.go b/modelarmor/update_organization_floor_settings.go index 1979b96a10..a8b63c2e56 100644 --- a/modelarmor/update_organization_floor_settings.go +++ b/modelarmor/update_organization_floor_settings.go @@ -62,7 +62,7 @@ func updateOrganizationFloorSettings(w io.Writer, organizationID, locationID str RaiFilters: []*modelarmorpb.RaiFilterSettings_RaiFilter{ { FilterType: modelarmorpb.RaiFilterType_HATE_SPEECH, - ConfidenceLevel: modelarmorpb.DetectionConfidenceLevel_HIGH, + ConfidenceLevel: modelarmorpb.DetectionConfidenceLevel_LOW_AND_ABOVE, }, }, }, diff --git a/modelarmor/update_project_floor_settings.go b/modelarmor/update_project_floor_settings.go index 3fe8a8f3e1..da9e6b3585 100644 --- a/modelarmor/update_project_floor_settings.go +++ b/modelarmor/update_project_floor_settings.go @@ -62,7 +62,7 @@ func updateProjectFloorSettings(w io.Writer, projectID, locationID string) error RaiFilters: []*modelarmorpb.RaiFilterSettings_RaiFilter{ { FilterType: modelarmorpb.RaiFilterType_HATE_SPEECH, - ConfidenceLevel: modelarmorpb.DetectionConfidenceLevel_HIGH, + ConfidenceLevel: modelarmorpb.DetectionConfidenceLevel_LOW_AND_ABOVE, }, }, },