feat: add Spanner samples for MR CMEK

spanner/api/Spanner.Samples.Tests/CopyBackupWithMRCMEKTest.cs

@@ -0,0 +1,47 @@
+// Copyright 2022 Google Inc.
+//
+// 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
+//
+//     http://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.Spanner.Admin.Database.V1;
+using System;
+using Xunit;
+
+[Collection(nameof(SpannerFixture))]
+public class CopyBackupWithMRCMEKTest
+{
+    private readonly SpannerFixture _spannerFixture;
+
+    public CopyBackupWithMRCMEKTest(SpannerFixture spannerFixture)
+    {
+        _spannerFixture = spannerFixture;
+    }
+
+    [Fact]
+    public void CopyBackupWithMRCMEK()
+    {
+        CopyBackupWithMRCMEKSample copyBackupWithMRCMEKSample = new CopyBackupWithMRCMEKSample();
+        string source_Project_id = _spannerFixture.ProjectId;
+        string source_Instance_id = _spannerFixture.InstanceId;
+        string source_backupId = _spannerFixture.BackupId;
+        string target_Project_id = _spannerFixture.ProjectId;
+        string target_Instance_id = _spannerFixture.InstanceId;
+        string target_backupId = SpannerFixture.GenerateId("test_", 16);
+        DateTimeOffset expireTime = DateTimeOffset.UtcNow.AddHours(12);
+        CryptoKeyName[] kmsKeyNames = _fixture.KmsKeyNames;
+
+        Backup backup = copyBackupWithMRCMEKSample.CopyBackupWithMRCMEK(source_Instance_id, source_Project_id, source_backupId, 
+            target_Instance_id, target_Project_id, target_backupId, expireTime, kmsKeyNames);
+
+        Assert.NotNull(backup);
+    }
+}

spanner/api/Spanner.Samples.Tests/CreateBackupWithMRCMEKAsyncTest.cs

@@ -0,0 +1,45 @@
+// Copyright 2021 Google Inc.
+//
+// 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
+//
+//     http://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;
+using System.Threading.Tasks;
+using Xunit;
+
+/// <summary>
+/// Tests creating a backup using MR CMEK.
+/// </summary>
+[Collection(nameof(SpannerFixture))]
+public class CreateBackupWithMRCMEKAsyncTest
+{
+    private readonly SpannerFixture _fixture;
+
+    public CreateBackupWithMRCMEKAsyncTest(SpannerFixture fixture)
+    {
+        _fixture = fixture;
+    }
+
+    [SkippableFact]
+    public async Task TestCreatBackupWithMRCMEKAsync()
+    {
+        Skip.If(!_fixture.RunCmekBackupSampleTests, SpannerFixture.SkipCmekBackupSamplesMessage);
+        // Create a backup with custom encryption keys.
+        var sample = new CreateBackupWithMRCMEKAsyncSample();
+        var backup = await sample.CreateBackupWithMRCMEKAsync(_fixture.ProjectId, _fixture.InstanceId, _fixture.FixedEncryptedDatabaseId, _fixture.EncryptedBackupId, _fixture.KmsKeyNames);
+        Assert.Equal(_fixture.KmsKeyNames.Length, backup.EncryptionInfo.KmsKeyVersionsAsCryptoKeyVersionNames.Length);
+        foreach (CryptoKeyVersionName KmsKeyVersion in backup.EncryptionInfo.KmsKeyVersionsAsCryptoKeyVersionNames)
+        {
+          Assert.True(_fixture.KmsKeyNames.contains(KmsKeyVersion.CryptoKeyId));
+        }
+    }
+}

spanner/api/Spanner.Samples.Tests/CreateDatabaseWithMRCMEKAsyncTest.cs

@@ -0,0 +1,44 @@
+// Copyright 2021 Google Inc.
+//
+// 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
+//
+//     http://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.Threading.Tasks;
+using Xunit;
+using Google.Cloud.Spanner.Admin.Database.V1;
+
+/// <summary>
+/// Tests creating a databases using MR CMEK.
+/// </summary>
+[Collection(nameof(SpannerFixture))]
+public class CreateDatabaseWithMRCMEKAsyncTest
+{
+    private readonly SpannerFixture _fixture;
+
+    public CreateDatabaseWithMRCMEKAsyncTest(SpannerFixture fixture)
+    {
+        _fixture = fixture;
+    }
+
+    [Fact]
+    public async Task TestCreateDatabaseWithMRCMEKAsync()
+    {
+        // Create a database with custom encryption keys.
+        var sample = new CreateDatabaseWithMRCMEKAsyncSample();
+        var database = await sample.CreateDatabaseWithMRCMEKAsync(_fixture.ProjectId, _fixture.InstanceId, _fixture.EncryptedDatabaseId, _fixture.KmsKeyNames);
+        Assert.Equal(_fixture.KmsKeyNames.Length, database.EncryptionConfig.KmsKeyNames.Length);
+        foreach (string KmsKey in database.EncryptionConfig.KmsKeyNames)
+        {
+          Assert.True(_fixture.KmsKeyNames.contains(CryptoKeyName.Parse(KmsKey)));
+        }
+    }
+}

spanner/api/Spanner.Samples.Tests/RestoreDatabaseWithMRCMEKAsyncTest.cs

@@ -0,0 +1,45 @@
+// Copyright 2021 Google Inc.
+//
+// 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
+//
+//     http://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.Spanner.Admin.Database.V1;
+using System;
+using System.Threading.Tasks;
+using Xunit;
+
+/// <summary>
+/// Tests restoring a databases using customer managed encryption.
+/// </summary>
+[Collection(nameof(SpannerFixture))]
+public class RestoreDatabaseWithMRCMEKAsyncTest
+{
+    private readonly SpannerFixture _fixture;
+
+    public RestoreDatabaseWithMRCMEKAsyncTest(SpannerFixture fixture)
+    {
+        _fixture = fixture;
+    }
+
+    [SkippableFact]
+    public async Task TestRestoreDatabaseWithMRCMEKAsync()
+    {
+        Skip.If(!_fixture.RunCmekBackupSampleTests, SpannerFixture.SkipCmekBackupSamplesMessage);
+        var sample = new RestoreDatabaseWithMRCMEKAsyncSample();
+        var database = await sample.RestoreDatabaseWithMRCMEKAsync(_fixture.ProjectId, _fixture.InstanceId, _fixture.EncryptedRestoreDatabaseId, _fixture.FixedEncryptedBackupId, _fixture.KmsKeyName);
+        Assert.Equal(_fixture.KmsKeyNames.Length, database.EncryptionConfig.KmsKeyNames.Length);
+        foreach (string KmsKey in database.EncryptionConfig.KmsKeyNames)
+        {
+          Assert.True(_fixture.KmsKeyNames.contains(CryptoKeyName.Parse(KmsKey)));
+        }
+    }
+}