1
0
mirror of https://github.com/bitwarden/server.git synced 2025-07-03 17:12:49 -05:00

Add EnableFeatureFlag extension method for tests

This commit is contained in:
Thomas Rittson
2023-10-10 14:15:43 +10:00
parent d9b0f01b72
commit 05192a9a8d
2 changed files with 13 additions and 5 deletions

View File

@ -1,4 +1,6 @@
using AutoFixture;
using Bit.Core;
using Bit.Core.Context;
using Bit.Core.Services;
using Bit.Core.Settings;
using NSubstitute;
@ -47,4 +49,14 @@ public static class SutProviderExtensions
.SetDependency(mockHttpClientFactory)
.Create();
}
public static SutProvider<T> EnableFeatureFlag<T>(this SutProvider<T> sutProvider,
string featureFlagKey)
{
sutProvider.GetDependency<IFeatureService>()
.IsEnabled(featureFlagKey, Arg.Any<ICurrentContext>(), Arg.Any<bool>())
.Returns(true);
return sutProvider;
}
}