Add interface for .Net generated client. (#284)

armadaproject · Dec 19, 2019 · 83dae67 · 83dae67
1 parent e642796
commit 83dae67
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 2 deletions.
diff --git a/client/DotNet/Armada.Client.Test/Tests.cs b/client/DotNet/Armada.Client.Test/Tests.cs
@@ -15,7 +15,7 @@ public async Task TestSimpleJobSubmitFlow()
         {
             var jobSet = $"set-{Guid.NewGuid()}";
 
-            var client = new ArmadaClient("http://localhost:8080", new HttpClient());
+            IArmadaClient client = new ArmadaClient("http://localhost:8080", new HttpClient());
             await client.CreateQueueAsync("test", new ApiQueue {PriorityFactor = 200});
 
             var pod = new V1PodSpec

diff --git a/client/DotNet/Armada.Client/Client.cs b/client/DotNet/Armada.Client/Client.cs
@@ -12,6 +12,14 @@ public interface IEvent
         string Queue { get;  }
         System.DateTimeOffset? Created { get; }
     }
+
+    public interface IArmadaClient
+    {
+        Task<ApiCancellationResult> CancelJobsAsync(ApiJobCancelRequest body);
+        Task<ApiJobSubmitResponse> SubmitJobsAsync(ApiJobSubmitRequest body);
+        Task<object> CreateQueueAsync(string name, ApiQueue body);
+        Task<IEnumerable<StreamResponse<ApiEventStreamMessage>>> GetJobEventsStream(string jobSetId, string fromMessage = null, bool watch = false);
+    }
 
     public partial class ApiEventMessage
     {
@@ -41,7 +49,7 @@ public class StreamResponse<T>
         public string Error { get; set; }
     }
 
-    public partial class ArmadaClient
+    public partial class ArmadaClient : IArmadaClient
     {
         public async Task<IEnumerable<StreamResponse<ApiEventStreamMessage>>> GetJobEventsStream(string jobSetId,
             string fromMessage = null, bool watch = false)