Add headers support to .NET server (#302)

Also fix the embarassing mistake of never actually sending the proper headers from the TDK client to make the backend server failover simulation work

Author: borrrden

client/src/cbltest/api/replicator.py

@@ -141,6 +141,7 @@ async def start(self) -> None:
             payload.pinnedServerCert = self.pinned_server_cert
             payload.reset = self.reset
             payload.collections = self.collections
+            payload.headers = self.headers
             req = self.__request_factory.create_request(
                 TestServerRequestType.START_REPLICATOR, payload
             )

client/src/cbltest/v1/requests.py

@@ -459,6 +459,9 @@ def __init__(self, database: str, endpoint: str):
         self.pinnedServerCert: str | None = None
         """The PEM representation of the TLS certificate that the remote is using"""
 
+        self.headers: dict[str, str] | None = None
+        """The headers to include in the replication requests"""
+
     def to_json(self) -> Any:
         """Serializes the :class:`PostStartReplicatorRequestBody` to a JSON string"""
         raw = {
@@ -469,6 +472,7 @@ def to_json(self) -> Any:
             "enableDocumentListener": self.enableDocumentListener,
             "enableAutoPurge": self.enableAutoPurge,
             "pinnedServerCert": self.pinnedServerCert,
+            "headers": self.headers,
         }
 
         if self.collections is not None:

servers/dotnet/testserver.logic/Handlers/StartReplicatorHandler.cs

@@ -261,12 +261,14 @@ internal readonly record struct StartReplicatorConfig
         public bool enableAutoPurge { get; init; }
 
         public string? pinnedServerCert { get; init; }
+        
+        public IReadOnlyDictionary<string, string?>? headers { get; init; }
 
         [JsonConstructor]
         public StartReplicatorConfig(string database, string endpoint,
             string replicatorType, bool continuous, IReadOnlyList<StartReplicatorCollection> collections,
             StartReplicatorAuthenticator? authenticator = null, bool enableDocumentListener = false,
-            bool enableAutoPurge = true, string? pinnedServerCert = null)
+            bool enableAutoPurge = true, string? pinnedServerCert = null, IReadOnlyDictionary<string, string?>? headers = null)
         {
             this.database = database;
             this.endpoint = endpoint;
@@ -277,6 +279,7 @@ public StartReplicatorConfig(string database, string endpoint,
             this.enableDocumentListener = enableDocumentListener;
             this.enableAutoPurge = enableAutoPurge;
             this.pinnedServerCert = pinnedServerCert;
+            this.headers = headers;
 
             if (replicatorType.ToLowerInvariant() == "pull") {
                 ReplicatorType = ReplicatorType.Pull;
@@ -360,7 +363,8 @@ public static Task StartReplicatorHandler(int version, Session session, JsonDocu
             EnableAutoPurge = deserializedBody.config.enableAutoPurge,
             PinnedServerCertificate = deserializedBody.config.pinnedServerCert != null 
                 ? new(Encoding.ASCII.GetBytes(deserializedBody.config.pinnedServerCert)) 
-                : null
+                : null,
+            Headers = deserializedBody.config.headers?.ToImmutableDictionary() ?? ImmutableDictionary<string, string?>.Empty,
         };
 
         var (repl, id) = session.ObjectManager.RegisterObject(() => new Replicator(replConfig));

servers/dotnet/testserver.logic/testserver.logic.csproj

@@ -5,7 +5,7 @@
         <ImplicitUsings>enable</ImplicitUsings>
         <Nullable>enable</Nullable>
         <LangVersion>latest</LangVersion>
-        <AssemblyVersion>1.1.0</AssemblyVersion>
+        <AssemblyVersion>1.2.2</AssemblyVersion>
     </PropertyGroup>
 
     <ItemGroup>