Added DNS firewall Reverse DNS

This commit is contained in:
2025-11-06 20:51:51 +01:00
parent df0f60ef29
commit dbcda1685a
7 changed files with 303 additions and 0 deletions

View File

@@ -0,0 +1,79 @@
using System.Linq;
using System.Threading;
using System.Threading.Tasks;
using AMWD.Net.Api.Cloudflare;
using AMWD.Net.Api.Cloudflare.Dns;
using Moq;
namespace Cloudflare.Dns.Tests.DnsFirewallExtensions.ReverseDns
{
[TestClass]
public class ShowDNSFirewallClusterReverseDNSTest
{
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, IQueryParameterFilter QueryFilter)> _callbacks;
[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>
{
["192.0.2.1"] = "ptr.example.com"
}
}
};
}
[TestMethod]
public async Task ShouldShowDnsFirewallClusterReverseDns()
{
// Arrange
var client = GetClient();
// Act
var response = await client.ShowDNSFirewallClusterReverseDNS(AccountId, ClusterId, TestContext.CancellationToken);
// Assert
Assert.IsNotNull(response);
Assert.IsTrue(response.Success);
Assert.IsNotNull(response.Result);
Assert.IsTrue(response.Result.ReverseDNS?.ContainsKey("192.0.2.1") ?? false);
Assert.AreEqual("ptr.example.com", response.Result.ReverseDNS!["192.0.2.1"]);
Assert.HasCount(1, _callbacks);
var (requestPath, queryFilter) = _callbacks.First();
Assert.AreEqual($"/accounts/{AccountId}/dns_firewall/{ClusterId}/reverse_dns", requestPath);
Assert.IsNull(queryFilter);
_clientMock.Verify(m => m.GetAsync<ReverseDnsResponse>($"/accounts/{AccountId}/dns_firewall/{ClusterId}/reverse_dns", null, TestContext.CancellationToken), Times.Once);
_clientMock.VerifyNoOtherCalls();
}
private ICloudflareClient GetClient()
{
_clientMock = new Mock<ICloudflareClient>();
_clientMock
.Setup(m => m.GetAsync<ReverseDnsResponse>(It.IsAny<string>(), It.IsAny<IQueryParameterFilter>(), It.IsAny<CancellationToken>()))
.Callback<string, IQueryParameterFilter, CancellationToken>((requestPath, queryFilter, _) => _callbacks.Add((requestPath, queryFilter)))
.ReturnsAsync(() => _response);
return _clientMock.Object;
}
}
}

View File

@@ -0,0 +1,118 @@
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;
}
}
}