mirror of
https://github.com/ckaczor/sqltoolsservice.git
synced 2026-01-14 09:35:43 -05:00
7acc668c04
Implementation of the resource provider APIs in order to support Create Firewall Rule. Provides definition for a ResourceProvider and Authentication service. The ResourceProvider supports firewall rules for now, and since authentication is routed through that method it will call into the auth service to set up the current account to be used. Additional notes: - Fixed deserialization by adding an Accept header. This shouldn't be necessary, but for some reason the firewall rule defaults to XML without this - Use generic server list and parse the ID to get the resource group, avoiding a large number of extra calls for each RG - Errors now include error message from the API
72 lines
2.8 KiB
C#
72 lines
2.8 KiB
C#
//
|
|
// Copyright (c) Microsoft. All rights reserved.
|
|
// Licensed under the MIT license. See LICENSE file in the project root for full license information.
|
|
|
|
using System;
|
|
using Microsoft.SqlTools.ResourceProvider.Core.Firewall;
|
|
using Xunit;
|
|
|
|
namespace Microsoft.SqlTools.ServiceLayer.UnitTests.ResourceProvider
|
|
{
|
|
/// <summary>
|
|
/// Tests to verify FirewallErrorParser
|
|
/// </summary>
|
|
public class FirewallErrorParserTest
|
|
{
|
|
private const int SqlAzureFirewallBlockedErrorNumber = 40615;
|
|
private const int SqlAzureLoginFailedErrorNumber = 18456;
|
|
private string _errorMessage = "error Message with 1.2.3.4 as IP address";
|
|
private FirewallErrorParser _firewallErrorParser = new FirewallErrorParser();
|
|
|
|
|
|
[Fact]
|
|
public void ParseExceptionShouldThrowExceptionGivenNullErrorMessage()
|
|
{
|
|
string errorMessage = null;
|
|
int errorCode = SqlAzureFirewallBlockedErrorNumber;
|
|
|
|
Assert.Throws<ArgumentNullException>("errorMessage", () =>
|
|
{
|
|
FirewallParserResponse response = _firewallErrorParser.ParseErrorMessage(errorMessage, errorCode);
|
|
Assert.False(response.FirewallRuleErrorDetected);
|
|
});
|
|
}
|
|
|
|
[Fact]
|
|
public void ParseExceptionShouldReturnFireWallRuleNotDetectedGivenDifferentError()
|
|
{
|
|
int errorCode = 123;
|
|
|
|
FirewallParserResponse response = _firewallErrorParser.ParseErrorMessage(_errorMessage, errorCode);
|
|
Assert.False(response.FirewallRuleErrorDetected);
|
|
}
|
|
|
|
[Fact]
|
|
public void ParseExceptionShouldReturnFireWallRuleNotDetectedGivenLoginFailedError()
|
|
{
|
|
int errorCode = SqlAzureLoginFailedErrorNumber;
|
|
|
|
FirewallParserResponse response = _firewallErrorParser.ParseErrorMessage(_errorMessage, errorCode);
|
|
Assert.False(response.FirewallRuleErrorDetected);
|
|
}
|
|
|
|
[Fact]
|
|
public void ParseExceptionShouldReturnFireWallRuleNotDetectedGivenInvalidErrorMessage()
|
|
{
|
|
int errorCode = SqlAzureFirewallBlockedErrorNumber;
|
|
string errorMessage = "error Message with no IP address";
|
|
FirewallParserResponse response = _firewallErrorParser.ParseErrorMessage(errorMessage, errorCode);
|
|
Assert.False(response.FirewallRuleErrorDetected);
|
|
}
|
|
|
|
[Fact]
|
|
public void ParseExceptionShouldReturnFireWallRuleDetectedGivenValidErrorMessage()
|
|
{
|
|
int errorCode = SqlAzureFirewallBlockedErrorNumber;
|
|
FirewallParserResponse response = _firewallErrorParser.ParseErrorMessage(_errorMessage, errorCode);
|
|
Assert.True(response.FirewallRuleErrorDetected);
|
|
Assert.Equal(response.BlockedIpAddress.ToString(), "1.2.3.4");
|
|
}
|
|
}
|
|
}
|