fix: #328 sign-up-returns-validation-error-on-fresh-installation-with…

…out-env-variables

packages/backend/apps/finances/signals.py

@@ -1,6 +1,7 @@
 from django.contrib.auth import get_user_model
 from django.db.models.signals import post_save
 from django.dispatch import receiver
+from stripe.error import AuthenticationError
 
 from .services import subscriptions
 
@@ -10,4 +11,9 @@
 @receiver(post_save, sender=User)
 def create_free_plan_subscription(sender, instance, created, **kwargs):
     if created:
-        subscriptions.initialize_user(user=instance)
+        try:
+            subscriptions.initialize_user(user=instance)
+        except AuthenticationError:
+            return
+        except Exception as e:
+            raise e

packages/backend/apps/users/management/commands/init_customers_plans.py

@@ -0,0 +1,20 @@
+from apps.finances.services import subscriptions
+from django.core.management.base import BaseCommand
+
+from config import settings
+from ...models import User
+
+
+class Command(BaseCommand):
+    help = 'Creates stripe customer and schedule subscription plan'
+
+    def handle(self, *args, **options):
+        if not settings.STRIPE_LIVE_MODE and not settings.STRIPE_TEST_SECRET_KEY:
+            return
+
+        if settings.STRIPE_LIVE_MODE and not settings.STRIPE_LIVE_SECRET_KEY:
+            return
+
+        users = User.objects.filter(djstripe_customers__isnull=True, is_superuser=False)
+        for user in users:
+            subscriptions.initialize_user(user=user)

packages/backend/apps/users/tests/factories.py

@@ -5,6 +5,7 @@
 from django.contrib.auth import hashers
 from django.contrib.auth.models import Group
 from django.core.files.uploadedfile import SimpleUploadedFile, UploadedFile
+from djstripe.models import Customer
 
 from common.acl.helpers import CommonGroups
 
@@ -91,3 +92,10 @@ def image_factory(name: str, params: Optional[ImageFactoryParams] = None) -> Upl
         content=image_field._make_data(params or {}),
         content_type=mimetypes.guess_type(name)[0],
     )
+
+
+class StripeCustomerFactory(factory.django.DjangoModelFactory):
+    class Meta:
+        model = Customer
+
+    subscriber = factory.SubFactory(UserFactory, profile=None)

packages/backend/apps/users/tests/fixtures.py

@@ -6,6 +6,7 @@
 pytest_factoryboy.register(factories.UserFactory)
 pytest_factoryboy.register(factories.UserProfileFactory)
 pytest_factoryboy.register(factories.UserAvatarFactory)
+pytest_factoryboy.register(factories.StripeCustomerFactory)
 
 
 @pytest.fixture()

packages/backend/apps/users/tests/test_commands.py

@@ -0,0 +1,28 @@
+import pytest
+import factory
+from django.db.models import signals
+
+from apps.users.management.commands.init_customers_plans import Command
+
+pytestmark = pytest.mark.django_db
+
+
+class TestUserPostSaveSignal:
+    @factory.django.mute_signals(signals.post_save)
+    def test_command_run_for_users_without_customer(self, stripe_customer_factory, user_factory, mocker):
+        mock = mocker.patch("apps.finances.services.subscriptions.initialize_user")
+        user = user_factory.create()
+        stripe_customer_factory.create()
+
+        Command().handle()
+
+        mock.assert_called_once_with(user=user)
+
+    @factory.django.mute_signals(signals.post_save)
+    def test_command_do_not_run_for_superusers(self, user_factory, mocker):
+        mock = mocker.patch("apps.finances.services.subscriptions.initialize_user")
+        user_factory.create(is_superuser=True)
+
+        Command().handle()
+
+        mock.assert_not_called()

packages/backend/apps/users/tests/test_signals.py

@@ -0,0 +1,22 @@
+import pytest
+from stripe.error import AuthenticationError
+
+pytestmark = pytest.mark.django_db
+
+
+class TestUserPostSaveSignal:
+    def test_signal_is_not_raising_exception_on_auth_error(self, user_factory, mocker):
+        mock = mocker.patch("apps.finances.services.subscriptions.initialize_user", side_effect=AuthenticationError())
+
+        user_factory.create()
+
+        mock.assert_called_once()
+
+    def test_reraise_stripe_error(self, user_factory, totp_mock, mocker):
+        initial_error = Exception
+        mocker.patch("apps.finances.services.subscriptions.initialize_user", side_effect=initial_error())
+
+        with pytest.raises(initial_error) as error:
+            user_factory.create()
+
+        assert initial_error == error.type

packages/backend/scripts/run_local.sh

@@ -28,6 +28,7 @@ if (echo "$STRIPE_LIVE_SECRET_KEY" | grep -q "<CHANGE_ME>") && (echo "$STRIPE_TE
 else
     python manage.py djstripe_sync_models Product Price
     python manage.py init_subscriptions
+    python manage.py init_customers_plans
     echo "Stripe initialized"
 fi
 

packages/backend/scripts/run_migrations.sh

@@ -2,3 +2,4 @@
 
 /bin/chamber exec $CHAMBER_SERVICE_NAME -- ./manage.py migrate
 /bin/chamber exec $CHAMBER_SERVICE_NAME -- ./manage.py init_subscriptions
+/bin/chamber exec $CHAMBER_SERVICE_NAME -- ./manage.py init_customers_plans