Add support for Update CA Configuration and Test(#399) (#411)

G-Research · Dec 30, 2024 · bced1f4 · bced1f4
1 parent 6b0e582
commit bced1f4
Show file tree

Hide file tree

Showing 3 changed files with 37 additions and 0 deletions.
diff --git a/Consul.Test/ConnectTest.cs b/Consul.Test/ConnectTest.cs
@@ -54,5 +54,23 @@ public async Task Connect_GetCAConfigurationTest()
             Assert.NotEqual((ulong)0, result.CreateIndex);
             Assert.NotEqual((ulong)0, result.ModifyIndex);
         }
+
+        [Fact]
+        public async Task Connect_CASetConfig()
+        {
+            var req = await _client.Connect.CAGetConfig();
+            var config = req.Response;
+
+            config.Config["test_state"] = new Dictionary<string, string> { { "foo", "bar" } };
+            config.Config["PrivateKey"] = "";
+
+            await _client.Connect.CASetConfig(config);
+            req = await _client.Connect.CAGetConfig();
+            var updatedConfig = req.Response;
+
+            Assert.Equal("consul", updatedConfig.Provider);
+            Assert.Equal("bar", updatedConfig.State["foo"]);
+            Assert.Equal("", updatedConfig.Config["PrivateKey"]);
+        }
     }
 }
diff --git a/Consul/Connect.cs b/Consul/Connect.cs
@@ -90,6 +90,23 @@ public Task<QueryResult<CAConfig>> CAGetConfig(QueryOptions q, CancellationToken
         {
             return _client.Get<CAConfig>("/v1/connect/ca/configuration", q).Execute(ct);
         }
+
+        /// <summary>
+        /// CASetConfig sets the current CA configuration.
+        /// </summary>
+        public Task<WriteResult> CASetConfig(CAConfig config, CancellationToken ct = default)
+        {
+            return CASetConfig(config, WriteOptions.Default, ct);
+        }
+
+        /// <summary>
+        /// CASetConfig sets the current CA configuration.
+        /// </summary>
+        public Task<WriteResult> CASetConfig(CAConfig config, WriteOptions q, CancellationToken ct = default)
+        {
+            return _client.Put("/v1/connect/ca/configuration", config, q).Execute(ct);
+
+        }
     }
 
     public partial class ConsulClient : IConsulClient

diff --git a/Consul/Interfaces/IConnectEndpoint.cs b/Consul/Interfaces/IConnectEndpoint.cs
@@ -30,5 +30,7 @@ public interface IConnectEndpoint
         Task<QueryResult<CARoots>> CARoots(CancellationToken ct = default);
         Task<QueryResult<CAConfig>> CAGetConfig(QueryOptions q, CancellationToken ct = default);
         Task<QueryResult<CAConfig>> CAGetConfig(CancellationToken ct = default);
+        Task<WriteResult> CASetConfig(CAConfig config, WriteOptions q, CancellationToken ct = default);
+        Task<WriteResult> CASetConfig(CAConfig config, CancellationToken ct = default);
     }
 }