test: add tests for flask and django management cli error handling (#23)

* test: add tests for flask and django management cli error handling

* fix: python2 attempts to import management as command

Author: untitaker

tests/integrations/django/myapp/management/__init__.py

@@ -0,0 +1,9 @@
+from django.core.management.base import BaseCommand
+
+
+class Command(BaseCommand):
+    def add_arguments(self, parser):
+        pass
+
+    def handle(self, *args, **options):
+        1 / 0

tests/integrations/django/myapp/management/commands/__init__.py

@@ -48,6 +48,7 @@
     "django.contrib.sessions",
     "django.contrib.messages",
     "django.contrib.staticfiles",
+    "tests.integrations.django.myapp",
 ]
 
 

tests/integrations/django/myapp/management/commands/mycrash.py

@@ -4,6 +4,7 @@
 
 
 from django.test import Client
+from django.core.management import execute_from_command_line
 
 try:
     from django.urls import reverse
@@ -79,3 +80,11 @@ def test_user_captured(client, capture_events):
 def test_404(client):
     response = client.get("/404")
     assert response.status_code == 404
+
+
+def test_management_command_raises():
+    # This just checks for our assumption that Django passes through all
+    # exceptions by default, so our excepthook can be used for management
+    # commands.
+    with pytest.raises(ZeroDivisionError):
+        execute_from_command_line(["manage.py", "mycrash"])

tests/integrations/django/myapp/settings.py

@@ -324,3 +324,19 @@ def index():
 def test_no_errors_without_request(app):
     with app.app_context():
         capture_exception(ValueError())
+
+
+def test_cli_commands_raise(app):
+    if not hasattr(app, "cli"):
+        pytest.skip("Too old flask version")
+
+    from flask.cli import ScriptInfo
+
+    @app.cli.command()
+    def foo():
+        1 / 0
+
+    with pytest.raises(ZeroDivisionError):
+        app.cli.main(
+            args=["foo"], prog_name="myapp", obj=ScriptInfo(create_app=lambda _: app)
+        )