119 lines
4.2 KiB
C#
119 lines
4.2 KiB
C#
using System.Linq;
|
|
using System.Threading;
|
|
using System.Threading.Tasks;
|
|
using AMWD.Net.Api.Cloudflare;
|
|
using AMWD.Net.Api.Cloudflare.Dns;
|
|
using AMWD.Net.Api.Cloudflare.Dns.Internals;
|
|
using Moq;
|
|
|
|
namespace Cloudflare.Dns.Tests.DnsFirewallExtensions.ReverseDns
|
|
{
|
|
[TestClass]
|
|
public class UpdateDNSFirewallClusterReverseDNSTest
|
|
{
|
|
public TestContext TestContext { get; set; }
|
|
|
|
private const string AccountId = "023e105f4ecef8ad9ca31a8372d0c353";
|
|
private const string ClusterId = "023e105f4ecef8ad9ca31a8372d0c355";
|
|
|
|
private Mock<ICloudflareClient> _clientMock;
|
|
private CloudflareResponse<ReverseDnsResponse> _response;
|
|
private List<(string RequestPath, InternalUpdateDNSFirewallClusterReverseDNSRequest Request)> _callbacks;
|
|
private UpdateDNSFirewallClusterReverseDNSRequest _request;
|
|
|
|
[TestInitialize]
|
|
public void Initialize()
|
|
{
|
|
_callbacks = [];
|
|
|
|
_response = new CloudflareResponse<ReverseDnsResponse>
|
|
{
|
|
Success = true,
|
|
Messages = [new ResponseInfo(1000, "Message 1")],
|
|
Errors = [new ResponseInfo(1000, "Error 1")],
|
|
Result = new ReverseDnsResponse
|
|
{
|
|
ReverseDNS = new Dictionary<string, string>
|
|
{
|
|
["10.0.0.1"] = "ptr1.example.com"
|
|
}
|
|
}
|
|
};
|
|
|
|
_request = new UpdateDNSFirewallClusterReverseDNSRequest(AccountId, ClusterId)
|
|
{
|
|
ReverseDNS = new Dictionary<string, string>
|
|
{
|
|
["10.0.0.1"] = "ptr1.example.com"
|
|
}
|
|
};
|
|
}
|
|
|
|
[TestMethod]
|
|
public async Task ShouldUpdateDnsFirewallClusterReverseDns()
|
|
{
|
|
// Arrange
|
|
var client = GetClient();
|
|
|
|
// Act
|
|
var response = await client.UpdateDNSFirewallClusterReverseDNS(_request, TestContext.CancellationToken);
|
|
|
|
// Assert
|
|
Assert.IsNotNull(response);
|
|
Assert.IsTrue(response.Success);
|
|
Assert.IsNotNull(response.Result);
|
|
Assert.IsTrue(response.Result.ReverseDNS?.ContainsKey("10.0.0.1") ?? false);
|
|
Assert.AreEqual("ptr1.example.com", response.Result.ReverseDNS!["10.0.0.1"]);
|
|
|
|
Assert.HasCount(1, _callbacks);
|
|
|
|
var (requestPath, request) = _callbacks.First();
|
|
Assert.AreEqual($"/accounts/{AccountId}/dns_firewall/{ClusterId}/reverse_dns", requestPath);
|
|
|
|
Assert.IsNotNull(request);
|
|
Assert.IsNotNull(request.Ptr);
|
|
Assert.IsTrue(request.Ptr.ContainsKey("10.0.0.1"));
|
|
Assert.AreEqual("ptr1.example.com", request.Ptr["10.0.0.1"]);
|
|
|
|
_clientMock.Verify(m => m.PatchAsync<ReverseDnsResponse, InternalUpdateDNSFirewallClusterReverseDNSRequest>($"/accounts/{AccountId}/dns_firewall/{ClusterId}/reverse_dns", It.IsAny<InternalUpdateDNSFirewallClusterReverseDNSRequest>(), TestContext.CancellationToken), Times.Once);
|
|
_clientMock.VerifyNoOtherCalls();
|
|
}
|
|
|
|
[TestMethod]
|
|
public async Task ShouldUpdateDnsFirewallClusterReverseDnsMinimalSet()
|
|
{
|
|
// Arrange
|
|
_request = new UpdateDNSFirewallClusterReverseDNSRequest(AccountId, ClusterId);
|
|
var client = GetClient();
|
|
|
|
// Act
|
|
var response = await client.UpdateDNSFirewallClusterReverseDNS(_request, TestContext.CancellationToken);
|
|
|
|
// Assert
|
|
Assert.IsNotNull(response);
|
|
Assert.IsTrue(response.Success);
|
|
Assert.AreEqual(_response.Result, response.Result);
|
|
|
|
Assert.HasCount(1, _callbacks);
|
|
|
|
var (requestPath, request) = _callbacks.First();
|
|
Assert.AreEqual($"/accounts/{AccountId}/dns_firewall/{ClusterId}/reverse_dns", requestPath);
|
|
Assert.IsNull(request.Ptr);
|
|
|
|
_clientMock.Verify(m => m.PatchAsync<ReverseDnsResponse, InternalUpdateDNSFirewallClusterReverseDNSRequest>($"/accounts/{AccountId}/dns_firewall/{ClusterId}/reverse_dns", It.IsAny<InternalUpdateDNSFirewallClusterReverseDNSRequest>(), TestContext.CancellationToken), Times.Once);
|
|
_clientMock.VerifyNoOtherCalls();
|
|
}
|
|
|
|
private ICloudflareClient GetClient()
|
|
{
|
|
_clientMock = new Mock<ICloudflareClient>();
|
|
_clientMock
|
|
.Setup(m => m.PatchAsync<ReverseDnsResponse, InternalUpdateDNSFirewallClusterReverseDNSRequest>(It.IsAny<string>(), It.IsAny<InternalUpdateDNSFirewallClusterReverseDNSRequest>(), It.IsAny<CancellationToken>()))
|
|
.Callback<string, InternalUpdateDNSFirewallClusterReverseDNSRequest, CancellationToken>((requestPath, request, _) => _callbacks.Add((requestPath, request)))
|
|
.ReturnsAsync(() => _response);
|
|
|
|
return _clientMock.Object;
|
|
}
|
|
}
|
|
}
|