Fix authorization

Spice/src/flight/SpiceFlightClient.cs

@@ -1,3 +1,4 @@
+using System.Diagnostics;
 using System.Net.Http.Headers;
 using System.Text;
 using Apache.Arrow;
@@ -27,7 +28,7 @@ internal SpiceFlightClient(string address, string? appId, string? apiKey)
                         Authorization = AuthenticationHeaderValue.Parse("Basic " +
                                                                         System.Convert.ToBase64String(Encoding
                                                                             .UTF8
-                                                                            .GetBytes(appId + ":" + apiKey)))
+                                                                            .GetBytes($"{appId}:{apiKey}")))
                     }
                 }
             };
@@ -38,14 +39,25 @@ internal SpiceFlightClient(string address, string? appId, string? apiKey)
         }
 
         _flightClient = new FlightClient(GrpcChannel.ForAddress(address, options));
-        _flightClient.Handshake();
+        var stream = _flightClient.Handshake();
+
+        stream.ResponseHeadersAsync.Wait();
+        var md = stream.ResponseHeadersAsync.Result;
+        var token = md.Get("authorization");
+        if (token == null || options.HttpClient == null)
+        {
+            throw new Exception("Failed to authenticate token");
+        }
+
+        options.HttpClient.DefaultRequestHeaders.Authorization = AuthenticationHeaderValue.Parse(token.Value);
     }
 
 
+
     internal async IAsyncEnumerator<RecordBatch> Query(string sql)
     {
         var descriptor = FlightDescriptor.CreateCommandDescriptor(sql);
-
+        
         var flightInfo = await _flightClient.GetInfo(descriptor);
         var endpoint = flightInfo.Endpoints.FirstOrDefault();
         if (endpoint == null)

SpiceTest/FlightQueryTest.cs

@@ -15,7 +15,8 @@ public async Task Test1()
 
         while (await result.MoveNextAsync())
         {
-            Console.WriteLine(result.Current);
+            var batch = result.Current;
+            Assert.That(batch.Length, Is.EqualTo(10));
         } 
     }
 }