Handle case where Stripe IDs do not relate to Stripe entities (#4021)

src/Core/Billing/Queries/Implementations/SubscriberQueries.cs

@@ -24,16 +24,27 @@ public class SubscriberQueries(
             return null;
         }
 
-        var customer = await stripeAdapter.CustomerGetAsync(subscriber.GatewayCustomerId, customerGetOptions);
-
-        if (customer != null)
+        try
         {
-            return customer;
-        }
+            var customer = await stripeAdapter.CustomerGetAsync(subscriber.GatewayCustomerId, customerGetOptions);
+
+            if (customer != null)
+            {
+                return customer;
+            }
+
+            logger.LogError("Could not find Stripe customer ({CustomerID}) for subscriber ({SubscriberID})",
+                subscriber.GatewayCustomerId, subscriber.Id);
 
-        logger.LogError("Could not find Stripe customer ({CustomerID}) for subscriber ({SubscriberID})", subscriber.GatewayCustomerId, subscriber.Id);
+            return null;
+        }
+        catch (StripeException exception)
+        {
+            logger.LogError("An error occurred while trying to retrieve Stripe customer ({CustomerID}) for subscriber ({SubscriberID}): {Error}",
+                subscriber.GatewayCustomerId, subscriber.Id, exception.Message);
 
-        return null;
+            return null;
+        }
     }
 
     public async Task<Subscription> GetSubscription(
@@ -49,65 +60,100 @@ public class SubscriberQueries(
             return null;
         }
 
-        var subscription = await stripeAdapter.SubscriptionGetAsync(subscriber.GatewaySubscriptionId, subscriptionGetOptions);
-
-        if (subscription != null)
+        try
         {
-            return subscription;
-        }
+            var subscription =
+                await stripeAdapter.SubscriptionGetAsync(subscriber.GatewaySubscriptionId, subscriptionGetOptions);
+
+            if (subscription != null)
+            {
+                return subscription;
+            }
 
-        logger.LogError("Could not find Stripe subscription ({SubscriptionID}) for subscriber ({SubscriberID})", subscriber.GatewaySubscriptionId, subscriber.Id);
+            logger.LogError("Could not find Stripe subscription ({SubscriptionID}) for subscriber ({SubscriberID})",
+                subscriber.GatewaySubscriptionId, subscriber.Id);
 
-        return null;
+            return null;
+        }
+        catch (StripeException exception)
+        {
+            logger.LogError("An error occurred while trying to retrieve Stripe subscription ({SubscriptionID}) for subscriber ({SubscriberID}): {Error}",
+                subscriber.GatewaySubscriptionId, subscriber.Id, exception.Message);
+
+            return null;
+        }
     }
 
-    public async Task<Subscription> GetSubscriptionOrThrow(
+    public async Task<Customer> GetCustomerOrThrow(
         ISubscriber subscriber,
-        SubscriptionGetOptions subscriptionGetOptions = null)
+        CustomerGetOptions customerGetOptions = null)
     {
         ArgumentNullException.ThrowIfNull(subscriber);
 
-        if (string.IsNullOrEmpty(subscriber.GatewaySubscriptionId))
+        if (string.IsNullOrEmpty(subscriber.GatewayCustomerId))
         {
-            logger.LogError("Cannot retrieve subscription for subscriber ({SubscriberID}) with no {FieldName}", subscriber.Id, nameof(subscriber.GatewaySubscriptionId));
+            logger.LogError("Cannot retrieve customer for subscriber ({SubscriberID}) with no {FieldName}", subscriber.Id, nameof(subscriber.GatewayCustomerId));
 
             throw ContactSupport();
         }
 
-        var subscription = await stripeAdapter.SubscriptionGetAsync(subscriber.GatewaySubscriptionId, subscriptionGetOptions);
-
-        if (subscription != null)
+        try
         {
-            return subscription;
-        }
+            var customer = await stripeAdapter.CustomerGetAsync(subscriber.GatewayCustomerId, customerGetOptions);
+
+            if (customer != null)
+            {
+                return customer;
+            }
 
-        logger.LogError("Could not find Stripe subscription ({SubscriptionID}) for subscriber ({SubscriberID})", subscriber.GatewaySubscriptionId, subscriber.Id);
+            logger.LogError("Could not find Stripe customer ({CustomerID}) for subscriber ({SubscriberID})",
+                subscriber.GatewayCustomerId, subscriber.Id);
 
-        throw ContactSupport();
+            throw ContactSupport();
+        }
+        catch (StripeException exception)
+        {
+            logger.LogError("An error occurred while trying to retrieve Stripe customer ({CustomerID}) for subscriber ({SubscriberID}): {Error}",
+                subscriber.GatewayCustomerId, subscriber.Id, exception.Message);
+
+            throw ContactSupport("An error occurred while trying to retrieve a Stripe Customer", exception);
+        }
     }
 
-    public async Task<Customer> GetCustomerOrThrow(
+    public async Task<Subscription> GetSubscriptionOrThrow(
         ISubscriber subscriber,
-        CustomerGetOptions customerGetOptions = null)
+        SubscriptionGetOptions subscriptionGetOptions = null)
     {
         ArgumentNullException.ThrowIfNull(subscriber);
 
-        if (string.IsNullOrEmpty(subscriber.GatewayCustomerId))
+        if (string.IsNullOrEmpty(subscriber.GatewaySubscriptionId))
         {
-            logger.LogError("Cannot retrieve customer for subscriber ({SubscriberID}) with no {FieldName}", subscriber.Id, nameof(subscriber.GatewayCustomerId));
+            logger.LogError("Cannot retrieve subscription for subscriber ({SubscriberID}) with no {FieldName}", subscriber.Id, nameof(subscriber.GatewaySubscriptionId));
 
             throw ContactSupport();
         }
 
-        var customer = await stripeAdapter.CustomerGetAsync(subscriber.GatewayCustomerId, customerGetOptions);
-
-        if (customer != null)
+        try
         {
-            return customer;
-        }
+            var subscription =
+                await stripeAdapter.SubscriptionGetAsync(subscriber.GatewaySubscriptionId, subscriptionGetOptions);
 
-        logger.LogError("Could not find Stripe customer ({CustomerID}) for subscriber ({SubscriberID})", subscriber.GatewayCustomerId, subscriber.Id);
+            if (subscription != null)
+            {
+                return subscription;
+            }
 
-        throw ContactSupport();
+            logger.LogError("Could not find Stripe subscription ({SubscriptionID}) for subscriber ({SubscriberID})",
+                subscriber.GatewaySubscriptionId, subscriber.Id);
+
+            throw ContactSupport();
+        }
+        catch (StripeException exception)
+        {
+            logger.LogError("An error occurred while trying to retrieve Stripe subscription ({SubscriptionID}) for subscriber ({SubscriberID}): {Error}",
+                subscriber.GatewaySubscriptionId, subscriber.Id, exception.Message);
+
+            throw ContactSupport("An error occurred while trying to retrieve a Stripe Subscription", exception);
+        }
     }
 }

test/Core.Test/Billing/Queries/SubscriberQueriesTests.cs

@@ -4,6 +4,7 @@ using Bit.Core.Services;
 using Bit.Test.Common.AutoFixture;
 using Bit.Test.Common.AutoFixture.Attributes;
 using NSubstitute;
+using NSubstitute.ExceptionExtensions;
 using NSubstitute.ReturnsExtensions;
 using Stripe;
 using Xunit;
@@ -48,6 +49,20 @@ public class SubscriberQueriesTests
         Assert.Null(customer);
     }
 
+    [Theory, BitAutoData]
+    public async Task GetCustomer_StripeException_ReturnsNull(
+        Organization organization,
+        SutProvider<SubscriberQueries> sutProvider)
+    {
+        sutProvider.GetDependency<IStripeAdapter>()
+            .CustomerGetAsync(organization.GatewayCustomerId)
+            .ThrowsAsync<StripeException>();
+
+        var customer = await sutProvider.Sut.GetCustomer(organization);
+
+        Assert.Null(customer);
+    }
+
     [Theory, BitAutoData]
     public async Task GetCustomer_Succeeds(
         Organization organization,
@@ -98,6 +113,20 @@ public class SubscriberQueriesTests
         Assert.Null(subscription);
     }
 
+    [Theory, BitAutoData]
+    public async Task GetSubscription_StripeException_ReturnsNull(
+        Organization organization,
+        SutProvider<SubscriberQueries> sutProvider)
+    {
+        sutProvider.GetDependency<IStripeAdapter>()
+            .SubscriptionGetAsync(organization.GatewaySubscriptionId)
+            .ThrowsAsync<StripeException>();
+
+        var subscription = await sutProvider.Sut.GetSubscription(organization);
+
+        Assert.Null(subscription);
+    }
+
     [Theory, BitAutoData]
     public async Task GetSubscription_Succeeds(
         Organization organization,
@@ -123,7 +152,7 @@ public class SubscriberQueriesTests
             async () => await sutProvider.Sut.GetCustomerOrThrow(null));
 
     [Theory, BitAutoData]
-    public async Task GetCustomerOrThrow_NoGatewaySubscriptionId_ThrowsGatewayException(
+    public async Task GetCustomerOrThrow_NoGatewayCustomerId_ContactSupport(
         Organization organization,
         SutProvider<SubscriberQueries> sutProvider)
     {
@@ -133,7 +162,7 @@ public class SubscriberQueriesTests
     }
 
     [Theory, BitAutoData]
-    public async Task GetSubscriptionOrThrow_NoCustomer_ThrowsGatewayException(
+    public async Task GetCustomerOrThrow_NoCustomer_ContactSupport(
         Organization organization,
         SutProvider<SubscriberQueries> sutProvider)
     {
@@ -144,6 +173,23 @@ public class SubscriberQueriesTests
         await ThrowsContactSupportAsync(async () => await sutProvider.Sut.GetCustomerOrThrow(organization));
     }
 
+    [Theory, BitAutoData]
+    public async Task GetCustomerOrThrow_StripeException_ContactSupport(
+        Organization organization,
+        SutProvider<SubscriberQueries> sutProvider)
+    {
+        var stripeException = new StripeException();
+
+        sutProvider.GetDependency<IStripeAdapter>()
+            .CustomerGetAsync(organization.GatewayCustomerId)
+            .ThrowsAsync(stripeException);
+
+        await ThrowsContactSupportAsync(
+            async () => await sutProvider.Sut.GetCustomerOrThrow(organization),
+            "An error occurred while trying to retrieve a Stripe Customer",
+            stripeException);
+    }
+
     [Theory, BitAutoData]
     public async Task GetCustomerOrThrow_Succeeds(
         Organization organization,
@@ -169,7 +215,7 @@ public class SubscriberQueriesTests
             async () => await sutProvider.Sut.GetSubscriptionOrThrow(null));
 
     [Theory, BitAutoData]
-    public async Task GetSubscriptionOrThrow_NoGatewaySubscriptionId_ThrowsGatewayException(
+    public async Task GetSubscriptionOrThrow_NoGatewaySubscriptionId_ContactSupport(
         Organization organization,
         SutProvider<SubscriberQueries> sutProvider)
     {
@@ -179,7 +225,7 @@ public class SubscriberQueriesTests
     }
 
     [Theory, BitAutoData]
-    public async Task GetSubscriptionOrThrow_NoSubscription_ThrowsGatewayException(
+    public async Task GetSubscriptionOrThrow_NoSubscription_ContactSupport(
         Organization organization,
         SutProvider<SubscriberQueries> sutProvider)
     {
@@ -190,6 +236,23 @@ public class SubscriberQueriesTests
         await ThrowsContactSupportAsync(async () => await sutProvider.Sut.GetSubscriptionOrThrow(organization));
     }
 
+    [Theory, BitAutoData]
+    public async Task GetSubscriptionOrThrow_StripeException_ContactSupport(
+        Organization organization,
+        SutProvider<SubscriberQueries> sutProvider)
+    {
+        var stripeException = new StripeException();
+
+        sutProvider.GetDependency<IStripeAdapter>()
+            .SubscriptionGetAsync(organization.GatewaySubscriptionId)
+            .ThrowsAsync(stripeException);
+
+        await ThrowsContactSupportAsync(
+            async () => await sutProvider.Sut.GetSubscriptionOrThrow(organization),
+            "An error occurred while trying to retrieve a Stripe Subscription",
+            stripeException);
+    }
+
     [Theory, BitAutoData]
     public async Task GetSubscriptionOrThrow_Succeeds(
         Organization organization,

test/Core.Test/Billing/Utilities.cs

@@ -7,12 +7,17 @@ namespace Bit.Core.Test.Billing;
 
 public static class Utilities
 {
-    public static async Task ThrowsContactSupportAsync(Func<Task> function)
+    public static async Task ThrowsContactSupportAsync(
+        Func<Task> function,
+        string internalMessage = null,
+        Exception innerException = null)
     {
-        var contactSupport = ContactSupport();
+        var contactSupport = ContactSupport(internalMessage, innerException);
 
         var exception = await Assert.ThrowsAsync<BillingException>(function);
 
+        Assert.Equal(contactSupport.ClientFriendlyMessage, exception.ClientFriendlyMessage);
         Assert.Equal(contactSupport.Message, exception.Message);
+        Assert.Equal(contactSupport.InnerException, exception.InnerException);
     }
 }