nhyatt/bitwarden
1
0
Fork 0
mirror of https://github.com/bitwarden/server.git synced 2025-04-06 13:38:13 -05:00
Code

Parameterize AmazonSQSClient to allow testing

Browse Source
This commit is contained in:
Mike Hanson 2020-06-18 10:13:22 -07:00
parent 61b15c55d0
commit 398867a839
1 changed files with 14 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
src/Core/Services/Implementations/AmazonSqsBlockIpService.cs
View File

@ -7,7 +7,7 @@ namespace Bit.Core.Services
{ {
public class AmazonSqsBlockIpService : IBlockIpService, IDisposable public class AmazonSqsBlockIpService : IBlockIpService, IDisposable
{ {
private readonly AmazonSQSClient _client; private readonly IAmazonSQS _client;
private string _blockIpQueueUrl; private string _blockIpQueueUrl;
private string _unblockIpQueueUrl; private string _unblockIpQueueUrl;
private bool _didInit = false; private bool _didInit = false;
@ -15,6 +15,17 @@ namespace Bit.Core.Services
public AmazonSqsBlockIpService( public AmazonSqsBlockIpService(
GlobalSettings globalSettings) GlobalSettings globalSettings)
: this(globalSettings, new AmazonSQSClient(
globalSettings.Amazon.AccessKeyId,
globalSettings.Amazon.AccessKeySecret,
RegionEndpoint.GetBySystemName(globalSettings.Amazon.Region))
)
{
}
public AmazonSqsBlockIpService(
GlobalSettings globalSettings,
IAmazonSQS amazonSqs)
{ {
if (string.IsNullOrWhiteSpace(globalSettings.Amazon?.AccessKeyId)) if (string.IsNullOrWhiteSpace(globalSettings.Amazon?.AccessKeyId))
{ {
@ -28,8 +39,8 @@ namespace Bit.Core.Services
{ {
throw new ArgumentNullException(nameof(globalSettings.Amazon.Region)); throw new ArgumentNullException(nameof(globalSettings.Amazon.Region));
} }
_client = new AmazonSQSClient(globalSettings.Amazon.AccessKeyId,
globalSettings.Amazon.AccessKeySecret, RegionEndpoint.GetBySystemName(globalSettings.Amazon.Region)); _client = amazonSqs;
} }
public void Dispose() public void Dispose()