mirror of
https://github.com/microsoft/onefuzz.git
synced 2025-06-17 04:18:07 +00:00
Implement the containers
function for C# (#2078)
Fairly straightforward but required implementation of a few more functions on the `IContainers`/`Containers` class.
This commit is contained in:
175
src/ApiService/IntegrationTests/ContainersTests.cs
Normal file
175
src/ApiService/IntegrationTests/ContainersTests.cs
Normal file
@ -0,0 +1,175 @@
|
||||
|
||||
using System;
|
||||
using System.Collections.Generic;
|
||||
using System.IO;
|
||||
using System.Linq;
|
||||
using System.Net;
|
||||
using Azure.Storage.Blobs;
|
||||
using IntegrationTests.Fakes;
|
||||
using Microsoft.OneFuzz.Service;
|
||||
using Xunit;
|
||||
using Xunit.Abstractions;
|
||||
|
||||
using Async = System.Threading.Tasks;
|
||||
|
||||
namespace IntegrationTests;
|
||||
|
||||
[Trait("Category", "Live")]
|
||||
public class AzureStorageContainersTest : ContainersTestBase {
|
||||
public AzureStorageContainersTest(ITestOutputHelper output)
|
||||
: base(output, Integration.AzureStorage.FromEnvironment()) { }
|
||||
}
|
||||
|
||||
public class AzuriteContainersTest : ContainersTestBase {
|
||||
public AzuriteContainersTest(ITestOutputHelper output)
|
||||
: base(output, new Integration.AzuriteStorage()) { }
|
||||
}
|
||||
|
||||
public abstract class ContainersTestBase : FunctionTestBase {
|
||||
public ContainersTestBase(ITestOutputHelper output, IStorage storage)
|
||||
: base(output, storage) { }
|
||||
|
||||
[Theory]
|
||||
[InlineData("GET")]
|
||||
[InlineData("POST")]
|
||||
[InlineData("DELETE")]
|
||||
public async Async.Task WithoutAuthorization_IsRejected(string method) {
|
||||
var auth = new TestEndpointAuthorization(RequestType.NoAuthorization, Logger, Context);
|
||||
var func = new ContainersFunction(Logger, auth, Context);
|
||||
|
||||
var result = await func.Run(TestHttpRequestData.Empty(method));
|
||||
Assert.Equal(HttpStatusCode.Unauthorized, result.StatusCode);
|
||||
|
||||
var err = BodyAs<Error>(result);
|
||||
Assert.Equal(ErrorCode.UNAUTHORIZED, err.Code);
|
||||
}
|
||||
|
||||
|
||||
[Fact]
|
||||
public async Async.Task CanDelete() {
|
||||
var containerName = "test";
|
||||
var client = GetContainerClient(containerName);
|
||||
await client.CreateIfNotExistsAsync();
|
||||
|
||||
var msg = TestHttpRequestData.FromJson("DELETE", new ContainerDelete(new Container(containerName)));
|
||||
|
||||
var auth = new TestEndpointAuthorization(RequestType.User, Logger, Context);
|
||||
var func = new ContainersFunction(Logger, auth, Context);
|
||||
var result = await func.Run(msg);
|
||||
Assert.Equal(HttpStatusCode.OK, result.StatusCode);
|
||||
|
||||
// container should be gone
|
||||
Assert.False(await client.ExistsAsync());
|
||||
}
|
||||
|
||||
|
||||
[Fact]
|
||||
public async Async.Task CanPost_New() {
|
||||
var meta = new Dictionary<string, string> { { "some", "value" } };
|
||||
var containerName = "test";
|
||||
var msg = TestHttpRequestData.FromJson("POST", new ContainerCreate(new Container(containerName), meta));
|
||||
|
||||
var auth = new TestEndpointAuthorization(RequestType.User, Logger, Context);
|
||||
var func = new ContainersFunction(Logger, auth, Context);
|
||||
var result = await func.Run(msg);
|
||||
Assert.Equal(HttpStatusCode.OK, result.StatusCode);
|
||||
|
||||
// container should be created with metadata:
|
||||
var client = GetContainerClient(containerName);
|
||||
Assert.True(await client.ExistsAsync());
|
||||
var props = await client.GetPropertiesAsync();
|
||||
Assert.Equal(meta, props.Value.Metadata);
|
||||
|
||||
var response = BodyAs<ContainerInfo>(result);
|
||||
await AssertCanCRUD(response.SasUrl);
|
||||
}
|
||||
|
||||
[Fact]
|
||||
public async Async.Task CanPost_Existing() {
|
||||
var containerName = "test";
|
||||
var client = GetContainerClient(containerName);
|
||||
await client.CreateIfNotExistsAsync();
|
||||
|
||||
var metadata = new Dictionary<string, string> { { "some", "value" } };
|
||||
var msg = TestHttpRequestData.FromJson("POST", new ContainerCreate(new Container(containerName), metadata));
|
||||
|
||||
var auth = new TestEndpointAuthorization(RequestType.User, Logger, Context);
|
||||
var func = new ContainersFunction(Logger, auth, Context);
|
||||
var result = await func.Run(msg);
|
||||
Assert.Equal(HttpStatusCode.OK, result.StatusCode);
|
||||
|
||||
// metadata should _not_ be updated:
|
||||
var props = await client.GetPropertiesAsync();
|
||||
Assert.Empty(props.Value.Metadata);
|
||||
|
||||
var response = BodyAs<ContainerInfo>(result);
|
||||
await AssertCanCRUD(response.SasUrl);
|
||||
}
|
||||
|
||||
|
||||
[Fact]
|
||||
public async Async.Task Get_Existing() {
|
||||
var containerName = "test";
|
||||
{
|
||||
var client = GetContainerClient(containerName);
|
||||
await client.CreateIfNotExistsAsync();
|
||||
}
|
||||
|
||||
var msg = TestHttpRequestData.FromJson("GET", new ContainerGet(new Container(containerName)));
|
||||
|
||||
var auth = new TestEndpointAuthorization(RequestType.User, Logger, Context);
|
||||
var func = new ContainersFunction(Logger, auth, Context);
|
||||
var result = await func.Run(msg);
|
||||
Assert.Equal(HttpStatusCode.OK, result.StatusCode);
|
||||
|
||||
// we should get back a SAS URI that works (create, delete, list, read):
|
||||
var info = BodyAs<ContainerInfo>(result);
|
||||
await AssertCanCRUD(info.SasUrl);
|
||||
}
|
||||
|
||||
[Fact]
|
||||
public async Async.Task Get_Missing_Fails() {
|
||||
var msg = TestHttpRequestData.FromJson("GET", new ContainerGet(new Container("container")));
|
||||
|
||||
var auth = new TestEndpointAuthorization(RequestType.User, Logger, Context);
|
||||
var func = new ContainersFunction(Logger, auth, Context);
|
||||
var result = await func.Run(msg);
|
||||
Assert.Equal(HttpStatusCode.BadRequest, result.StatusCode);
|
||||
}
|
||||
|
||||
[Fact]
|
||||
public async Async.Task List_Existing() {
|
||||
var meta1 = new Dictionary<string, string> { { "key1", "value1" } };
|
||||
var meta2 = new Dictionary<string, string> { { "key2", "value2" } };
|
||||
await GetContainerClient("one").CreateIfNotExistsAsync(metadata: meta1);
|
||||
await GetContainerClient("two").CreateIfNotExistsAsync(metadata: meta2);
|
||||
|
||||
var msg = TestHttpRequestData.Empty("GET"); // this means list all
|
||||
|
||||
var auth = new TestEndpointAuthorization(RequestType.User, Logger, Context);
|
||||
var func = new ContainersFunction(Logger, auth, Context);
|
||||
var result = await func.Run(msg);
|
||||
Assert.Equal(HttpStatusCode.OK, result.StatusCode);
|
||||
|
||||
var list = BodyAs<ContainerInfoBase[]>(result);
|
||||
// other tests can run in parallel, so filter to just our containers:
|
||||
var cs = list.Where(ci => ci.Name.ContainerName.StartsWith(Context.ServiceConfiguration.OneFuzzStoragePrefix)).ToList();
|
||||
Assert.Equal(2, cs.Count);
|
||||
|
||||
// ensure correct metadata was returned.
|
||||
// these will be in order as "one"<"two"
|
||||
Assert.Equal(meta1, cs[0].Metadata);
|
||||
Assert.Equal(meta2, cs[1].Metadata);
|
||||
}
|
||||
|
||||
private static async Async.Task AssertCanCRUD(Uri sasUrl) {
|
||||
var client = new BlobContainerClient(sasUrl);
|
||||
await client.UploadBlobAsync("blob", new BinaryData("content")); // create
|
||||
var b = Assert.Single(await client.GetBlobsAsync().ToListAsync()); // list
|
||||
using (var s = await client.GetBlobClient(b.Name).OpenReadAsync())
|
||||
using (var sr = new StreamReader(s)) {
|
||||
Assert.Equal("content", await sr.ReadToEndAsync()); // read
|
||||
}
|
||||
await client.DeleteBlobAsync("blob"); // delete
|
||||
}
|
||||
}
|
Reference in New Issue
Block a user