Yank invoice (#4)

deislabs · Apr 22, 2021 · 480f005 · 480f005
1 parent 5453d12
commit 480f005
Show file tree

Hide file tree

Showing 2 changed files with 19 additions and 0 deletions.
diff --git a/Bindle.Tests/Integration.cs b/Bindle.Tests/Integration.cs
@@ -100,5 +100,17 @@ public async Task CanCreateInvoices()
             Assert.Equal(invoice.Parcels[1].Label.Feature["test"]["a"], fetched.Parcels[1].Label.Feature["test"]["a"]);
             Assert.Equal(invoice.Groups.Count, fetched.Groups.Count);
         }
+
+        [Fact]
+        public async Task CanYankInvoice()
+        {
+            var client = new BindleClient(DEMO_SERVER_URL);
+            await client.YankInvoice("your/fancy/bindle/0.3.0");  // TODO: use one that doesn't conflict with CanFetchInvoice (because Xunit parallelisation)
+            await Assert.ThrowsAsync<System.Net.WebException>(async () => {
+                await client.GetInvoice("your/fancy/bindle/0.3.0");
+            });
+            var invoice = await client.GetInvoice("your/fancy/bindle/0.3.0", IncludeYanked);
+            Assert.Equal("your/fancy/bindle", invoice.Bindle.Name);
+        }
     }
 }
diff --git a/Bindle/BindleClient.cs b/Bindle/BindleClient.cs
@@ -67,6 +67,13 @@ public async Task<CreateInvoiceResult> CreateInvoice(Invoice invoice)
             return Parser.ParseCreateInvoiceResult(toml);
         }
 
+        public async Task YankInvoice(string invoiceId)
+        {
+            var httpClient = new HttpClient();
+            var uri = new Uri(_baseUri, INVOICE_PATH + "/" + invoiceId);
+            await httpClient.DeleteAsync(uri);
+        }
+
         private static string SlashSafe(string uri)
         {
             if (uri.EndsWith('/'))