From d64e200f2f9d359591b385129e7ad8b320c0b7d7 Mon Sep 17 00:00:00 2001
From: JKorf <jankorf91@gmail.com>
Date: Thu, 2 May 2024 22:29:32 +0200
Subject: [PATCH] Test implementation fixes

---
 CryptoExchange.Net/Testing/Implementations/TestRequest.cs | 4 +++-
 CryptoExchange.Net/Testing/RestRequestValidator.cs        | 3 ++-
 2 files changed, 5 insertions(+), 2 deletions(-)

diff --git a/CryptoExchange.Net/Testing/Implementations/TestRequest.cs b/CryptoExchange.Net/Testing/Implementations/TestRequest.cs
index 91a7114..5f4d9b8 100644
--- a/CryptoExchange.Net/Testing/Implementations/TestRequest.cs
+++ b/CryptoExchange.Net/Testing/Implementations/TestRequest.cs
@@ -10,6 +10,7 @@ namespace CryptoExchange.Net.Testing.Implementations
 {
     internal class TestRequest : IRequest
     {
+        private readonly Dictionary<string, IEnumerable<string>> _headers = new Dictionary<string, IEnumerable<string>>();
         private readonly TestResponse _response;
 
         public string Accept { set { } }
@@ -31,9 +32,10 @@ namespace CryptoExchange.Net.Testing.Implementations
 
         public void AddHeader(string key, string value)
         {
+            _headers.Add(key, new[] { value });
         }
 
-        public Dictionary<string, IEnumerable<string>> GetHeaders() => new();
+        public Dictionary<string, IEnumerable<string>> GetHeaders() => _headers;
 
         public Task<IResponse> GetResponseAsync(CancellationToken cancellationToken) => Task.FromResult<IResponse>(_response);
 
diff --git a/CryptoExchange.Net/Testing/RestRequestValidator.cs b/CryptoExchange.Net/Testing/RestRequestValidator.cs
index 6865fb9..701bb54 100644
--- a/CryptoExchange.Net/Testing/RestRequestValidator.cs
+++ b/CryptoExchange.Net/Testing/RestRequestValidator.cs
@@ -170,8 +170,9 @@ namespace CryptoExchange.Net.Testing
             var expectedMethod = reader.ReadLine();
             var expectedPath = reader.ReadLine();
             var expectedAuth = bool.Parse(reader.ReadLine()!);
+            var response = reader.ReadToEnd();
 
-            TestHelpers.ConfigureRestClient(_client, "", System.Net.HttpStatusCode.OK);
+            TestHelpers.ConfigureRestClient(_client, response, System.Net.HttpStatusCode.OK);
             var result = await methodInvoke(_client).ConfigureAwait(false);
 
             // Check request/response properties