feat(modelarmor): Added samples for get and update floor settings

modelarmor/api/ModelArmor.Samples.Tests/AssemblyInfo.cs

@@ -0,0 +1,20 @@
+/*
+ * Copyright 2025 Google LLC
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     https://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+*/
+
+using Xunit;
+
+[assembly: CollectionBehavior(DisableTestParallelization = true)]
+[assembly: TestCollectionOrderer("ModelArmor.Samples.Tests.ModelArmorCollectionOrder", "ModelArmor.Samples.Tests")]

modelarmor/api/ModelArmor.Samples.Tests/FloorSettingFixture.cs

@@ -0,0 +1,175 @@
+/*
+ * Copyright 2025 Google LLC
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     https://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+*/
+
+using Google.Cloud.ModelArmor.V1;
+using System;
+using Xunit;
+
+[CollectionDefinition("FloorSettingCollection")]
+public class FloorSettingFixture : IDisposable, ICollectionFixture<FloorSettingFixture>
+{
+    private const string EnvProjectId = "GOOGLE_PROJECT_ID";
+
+    public ModelArmorClient Client { get; }
+    public string ProjectId { get; }
+    public string FolderId { get; }
+    public string OrganizationId { get; }
+
+    public FloorSettingFixture()
+    {
+        ProjectId = GetRequiredEnvVar(EnvProjectId);
+        FolderId = GetRequiredEnvVar("MA_FOLDER_ID");
+        OrganizationId = GetRequiredEnvVar("MA_ORG_ID");
+
+        // Create the Model Armor client.
+        Client = ModelArmorClient.Create();
+
+        ResetFloorSettings();
+    }
+
+    public string GetRequiredEnvVar(string name)
+    {
+        var value = Environment.GetEnvironmentVariable(name);
+        if (string.IsNullOrEmpty(value))
+        {
+            throw new Exception($"Missing {name} environment variable");
+        }
+        return value;
+    }
+
+    public void Dispose()
+    {
+        ResetFloorSettings();
+    }
+
+    // Reset floor settings to default values for project, folder, and organization
+    private void ResetFloorSettings()
+    {
+        // Reset project floor settings if project ID is available
+        if (!string.IsNullOrEmpty(ProjectId))
+        {
+            Console.WriteLine($"Resetting project floor settings for project ID: {ProjectId}");
+            ResetProjectFloorSettings(ProjectId);
+        }
+
+        // Reset folder floor settings if folder ID is available
+        string folderId = FolderId;
+        if (!string.IsNullOrEmpty(folderId))
+        {
+            Console.WriteLine($"Resetting folder floor settings for folder ID: {folderId}");
+            ResetFolderFloorSettings(folderId);
+        }
+
+        // Reset organization floor settings if organization ID is available
+        string organizationId = OrganizationId;
+        if (!string.IsNullOrEmpty(organizationId))
+        {
+            Console.WriteLine(
+                $"Resetting organization floor settings for organization ID: {organizationId}"
+            );
+            ResetOrganizationFloorSettings(organizationId);
+        }
+    }
+
+    // Reset project floor settings to default
+    public void ResetProjectFloorSettings(string projectId)
+    {
+        try
+        {
+            // Create default floor setting with empty RAI filters and enforcement disabled
+            FloorSetting defaultFloorSetting = new FloorSetting
+            {
+                Name = $"projects/{projectId}/locations/global/floorSetting",
+                FilterConfig = new FilterConfig
+                {
+                    RaiSettings = new RaiFilterSettings { RaiFilters = { } },
+                },
+                EnableFloorSettingEnforcement = false,
+            };
+
+            // Update the floor setting to reset it
+            Client.UpdateFloorSetting(
+                new UpdateFloorSettingRequest { FloorSetting = defaultFloorSetting }
+            );
+            Console.WriteLine($"Project floor settings reset for project ID: {projectId}");
+        }
+        catch (Exception ex)
+        {
+            // Log but don't throw to avoid breaking test cleanup
+            Console.WriteLine($"Error resetting project floor settings: {ex.Message}");
+        }
+    }
+
+    // Reset folder floor settings to default
+    public void ResetFolderFloorSettings(string folderId)
+    {
+        try
+        {
+            // Create default floor setting with empty RAI filters and enforcement disabled
+            FloorSetting defaultFloorSetting = new FloorSetting
+            {
+                Name = $"folders/{folderId}/locations/global/floorSetting",
+                FilterConfig = new FilterConfig
+                {
+                    RaiSettings = new RaiFilterSettings { RaiFilters = { } },
+                },
+                EnableFloorSettingEnforcement = false,
+            };
+
+            // Update the floor setting to reset it
+            Client.UpdateFloorSetting(
+                new UpdateFloorSettingRequest { FloorSetting = defaultFloorSetting }
+            );
+            Console.WriteLine($"Folder floor settings reset for folder ID: {folderId}");
+        }
+        catch (Exception ex)
+        {
+            // Log but don't throw to avoid breaking test cleanup
+            Console.WriteLine($"Error resetting folder floor settings: {ex.Message}");
+        }
+    }
+
+    // Reset organization floor settings to default
+    public void ResetOrganizationFloorSettings(string organizationId)
+    {
+        try
+        {
+            // Create default floor setting with empty RAI filters and enforcement disabled
+            FloorSetting defaultFloorSetting = new FloorSetting
+            {
+                Name = $"organizations/{organizationId}/locations/global/floorSetting",
+                FilterConfig = new FilterConfig
+                {
+                    RaiSettings = new RaiFilterSettings { RaiFilters = { } },
+                },
+                EnableFloorSettingEnforcement = false,
+            };
+
+            // Update the floor setting to reset it
+            Client.UpdateFloorSetting(
+                new UpdateFloorSettingRequest { FloorSetting = defaultFloorSetting }
+            );
+            Console.WriteLine(
+                $"Organization floor settings reset for organization ID: {organizationId}"
+            );
+        }
+        catch (Exception ex)
+        {
+            // Log but don't throw to avoid breaking test cleanup
+            Console.WriteLine($"Error resetting organization floor settings: {ex.Message}");
+        }
+    }
+}

modelarmor/api/ModelArmor.Samples.Tests/GetFolderFloorSettingsTest.cs

@@ -0,0 +1,51 @@
+/*
+ * Copyright 2025 Google LLC
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     https://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+using Google.Cloud.ModelArmor.V1;
+using Xunit;
+using Xunit.Abstractions;
+
+namespace ModelArmor.Samples.Tests
+{
+    [Collection("FloorSettingCollection")]
+    public class GetFolderFloorSettingsTests : IClassFixture<FloorSettingFixture>
+    {
+        private readonly FloorSettingFixture _fixture;
+        private readonly GetFolderFloorSettings _sample;
+        private readonly ITestOutputHelper _output;
+
+        public GetFolderFloorSettingsTests(FloorSettingFixture fixture, ITestOutputHelper output)
+        {
+            _fixture = fixture;
+            _sample = new GetFolderFloorSettings();
+            _output = output;
+        }
+
+        [Fact]
+        public void GetFolderFloorSettingTest()
+        {
+            // Get folder ID from environment variable
+            string folderId = _fixture.GetRequiredEnvVar("MA_FOLDER_ID");
+
+            // Run the sample
+            FloorSetting response = _sample.GetFolderFloorSetting(folderId);
+
+            // Verify response
+            Assert.NotNull(response);
+            Assert.Equal($"folders/{folderId}/locations/global/floorSetting", response.Name);
+        }
+    }
+}

modelarmor/api/ModelArmor.Samples.Tests/GetOrganizationFloorSettingsTest.cs

@@ -0,0 +1,57 @@
+/*
+ * Copyright 2025 Google LLC
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     https://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+using Google.Cloud.ModelArmor.V1;
+using Xunit;
+using Xunit.Abstractions;
+
+namespace ModelArmor.Samples.Tests
+{
+    [Collection("FloorSettingCollection")]
+    public class GetOrganizationFloorSettingsTests : IClassFixture<FloorSettingFixture>
+    {
+        private readonly FloorSettingFixture _fixture;
+        private readonly GetOrganizationFloorSettings _sample;
+        private readonly ITestOutputHelper _output;
+
+        public GetOrganizationFloorSettingsTests(
+            FloorSettingFixture fixture,
+            ITestOutputHelper output
+        )
+        {
+            _fixture = fixture;
+            _sample = new GetOrganizationFloorSettings();
+            _output = output;
+        }
+
+        [Fact]
+        public void GetOrganizationFloorSettingTest()
+        {
+            // Get organization ID from environment variable
+            string organizationId = _fixture.GetRequiredEnvVar("MA_ORG_ID");
+
+            // Run the sample
+            FloorSetting response = _sample.GetOrganizationFloorSetting(organizationId);
+
+            // Verify response
+            Assert.NotNull(response);
+            Assert.Equal(
+                $"organizations/{organizationId}/locations/global/floorSetting",
+                response.Name
+            );
+        }
+    }
+}

modelarmor/api/ModelArmor.Samples.Tests/GetProjectFloorSettingsTest.cs

@@ -0,0 +1,51 @@
+/*
+ * Copyright 2025 Google LLC
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     https://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+using Google.Cloud.ModelArmor.V1;
+using Xunit;
+using Xunit.Abstractions;
+
+namespace ModelArmor.Samples.Tests
+{
+    [Collection("FloorSettingCollection")]
+    public class GetProjectFloorSettingsTests : IClassFixture<FloorSettingFixture>
+    {
+        private readonly FloorSettingFixture _fixture;
+        private readonly GetProjectFloorSettings _sample;
+        private readonly ITestOutputHelper _output;
+
+        public GetProjectFloorSettingsTests(FloorSettingFixture fixture, ITestOutputHelper output)
+        {
+            _fixture = fixture;
+            _sample = new GetProjectFloorSettings();
+            _output = output;
+        }
+
+        [Fact]
+        public void GetProjectFloorSettingTest()
+        {
+            // Get project ID from environment variable
+            string projectId = _fixture.GetRequiredEnvVar("GOOGLE_PROJECT_ID");
+
+            // Run the sample
+            FloorSetting response = _sample.GetProjectFloorSetting(projectId);
+
+            // Verify response
+            Assert.NotNull(response);
+            Assert.Equal($"projects/{projectId}/locations/global/floorSetting", response.Name);
+        }
+    }
+}

modelarmor/api/ModelArmor.Samples.Tests/ModelArmor.Samples.Tests.csproj

@@ -1,4 +1,4 @@
-<Project Sdk="Microsoft.NET.Sdk">
+<Project Sdk="Microsoft.NET.Sdk">
   <PropertyGroup>
     <TargetFramework>net8.0</TargetFramework>
     <IsPackable>false</IsPackable>

modelarmor/api/ModelArmor.Samples.Tests/ModelArmorCollectionOrder.cs

@@ -0,0 +1,34 @@
+/*
+ * Copyright 2025 Google LLC
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     https://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+*/
+
+using System.Collections.Generic;
+using System.Linq;
+using Xunit;
+using Xunit.Abstractions;
+using Xunit.Sdk;
+
+namespace ModelArmor.Samples.Tests
+{
+    public class ModelArmorCollectionOrder : ITestCollectionOrderer
+    {
+        public IEnumerable<ITestCollection> OrderTestCollections(IEnumerable<ITestCollection> testCollections)
+        {
+            // Place FloorSettingCollection first, then the rest
+            return testCollections.OrderBy(tc =>
+                tc.DisplayName.Contains("FloorSettingCollection") ? 0 : 1);
+        }
+    }
+}