[PM-20452] - Offloading Stripe Update (#6034)

* Adding job to update stripe subscriptions and increment seat count when inviting a user. * Updating name * Added ef migrations * Fixing script * Fixing procedures. Added repo tests. * Fixed set stored procedure. Fixed parameter name. * Added tests for database calls and updated stored procedures * Fixed build for sql file. * fixing sproc * File is nullsafe * Adding view to select from instead of table. * Updating UpdateSubscriptionStatus to use a CTE and do all the updates in 1 statement. * Setting revision date when incrementing seat count * Added feature flag check for the background job. * Fixing nullable property. * Removing new table and just adding the column to org. Updating to query and command. Updated tests. * Adding migration script rename * Add SyncSeats to Org.sql def * Adding contraint name * Removing old table files. * Added tests * Upped the frequency to be at the top of every 3rd hour. * Updating error message. * Removing extension method * Changed to GuidIdArray * Added xml doc and switched class to record
2025-12-16 00:03:54 +00:00 · 2025-07-31 07:54:51 -05:00
parent 88dd977848
commit 86ce3a86e9
38 changed files with 10968 additions and 43 deletions
--- a/test/Api.Test/AdminConsole/Jobs/OrganizationSubscriptionUpdateJobTests.cs
+++ b/test/Api.Test/AdminConsole/Jobs/OrganizationSubscriptionUpdateJobTests.cs
@@ -0,0 +1,60 @@
+using Bit.Api.AdminConsole.Jobs;
+using Bit.Core;
+using Bit.Core.AdminConsole.Models.Data.Organizations;
+using Bit.Core.AdminConsole.OrganizationFeatures.Organizations.Interfaces;
+using Bit.Core.Services;
+using Bit.Test.Common.AutoFixture;
+using Bit.Test.Common.AutoFixture.Attributes;
+using NSubstitute;
+using Quartz;
+using Xunit;
+
+namespace Bit.Api.Test.AdminConsole.Jobs;
+
+[SutProviderCustomize]
+public class OrganizationSubscriptionUpdateJobTests
+{
+    [Theory]
+    [BitAutoData]
+    public async Task ExecuteJobAsync_WhenScimInviteUserIsDisabled_ThenQueryAndCommandAreNotExecuted(
+        SutProvider<OrganizationSubscriptionUpdateJob> sutProvider)
+    {
+        sutProvider.GetDependency<IFeatureService>()
+            .IsEnabled(FeatureFlagKeys.ScimInviteUserOptimization)
+            .Returns(false);
+
+        var contextMock = Substitute.For<IJobExecutionContext>();
+
+        await sutProvider.Sut.Execute(contextMock);
+
+        await sutProvider.GetDependency<IGetOrganizationSubscriptionsToUpdateQuery>()
+            .DidNotReceive()
+            .GetOrganizationSubscriptionsToUpdateAsync();
+
+        await sutProvider.GetDependency<IUpdateOrganizationSubscriptionCommand>()
+            .DidNotReceive()
+            .UpdateOrganizationSubscriptionAsync(Arg.Any<IEnumerable<OrganizationSubscriptionUpdate>>());
+    }
+
+    [Theory]
+    [BitAutoData]
+    public async Task ExecuteJobAsync_WhenScimInviteUserIsEnabled_ThenQueryAndCommandAreExecuted(
+        SutProvider<OrganizationSubscriptionUpdateJob> sutProvider)
+    {
+        sutProvider.GetDependency<IFeatureService>()
+            .IsEnabled(FeatureFlagKeys.ScimInviteUserOptimization)
+            .Returns(true);
+
+        var contextMock = Substitute.For<IJobExecutionContext>();
+
+        await sutProvider.Sut.Execute(contextMock);
+
+        await sutProvider.GetDependency<IGetOrganizationSubscriptionsToUpdateQuery>()
+            .Received(1)
+            .GetOrganizationSubscriptionsToUpdateAsync();
+
+        await sutProvider.GetDependency<IUpdateOrganizationSubscriptionCommand>()
+            .Received(1)
+            .UpdateOrganizationSubscriptionAsync(Arg.Any<IEnumerable<OrganizationSubscriptionUpdate>>());
+    }
+}