DX: Cleanup tests using Ruff (#45)

Author: pfouque

tests/testapp/models.py

@@ -40,13 +40,26 @@ def dean_rejected(self):
         pass
 
 
+class DbState(models.Model):
+    """
+    States in DB
+    """
+
+    id = models.CharField(primary_key=True, max_length=50)
+
+    label = models.CharField(max_length=255)
+
+    def __str__(self):
+        return self.label
+
+
 class FKApplication(models.Model):
     """
     Student application need to be approved by dept chair and dean.
     Test workflow for FSMKeyField
     """
 
-    state = FSMKeyField("testapp.DbState", default="new", on_delete=models.CASCADE)
+    state = FSMKeyField(DbState, default="new", on_delete=models.CASCADE)
 
     @transition(field=state, source="new", target="draft")
     def draft(self):
@@ -73,19 +86,6 @@ def dean_rejected(self):
         pass
 
 
-class DbState(models.Model):
-    """
-    States in DB
-    """
-
-    id = models.CharField(primary_key=True, max_length=50)
-
-    label = models.CharField(max_length=255)
-
-    def __str__(self):
-        return self.label
-
-
 class BlogPost(models.Model):
     """
     Test workflow

tests/testapp/tests/test_access_deferred_fsm_field.py

@@ -11,9 +11,6 @@
 class DeferrableModel(models.Model):
     state = FSMField(default="new")
 
-    class Meta:
-        app_label = "testapp"
-
     @transition(field=state, source="new", target="published")
     def publish(self):
         pass

tests/testapp/tests/test_custom_data.py

@@ -10,9 +10,6 @@
 class BlogPostWithCustomData(models.Model):
     state = FSMField(default="new")
 
-    class Meta:
-        app_label = "testapp"
-
     @transition(field=state, source="new", target="published", conditions=[], custom={"label": "Publish", "type": "*"})
     def publish(self):
         pass

tests/testapp/tests/test_exception_transitions.py

@@ -13,9 +13,6 @@
 class ExceptionalBlogPost(models.Model):
     state = FSMField(default="new")
 
-    class Meta:
-        app_label = "testapp"
-
     @transition(field=state, source="new", target="published", on_error="crashed")
     def publish(self):
         raise Exception("Upss")

tests/testapp/tests/test_key_field.py

@@ -50,8 +50,7 @@ def moderate(self):
 
 class FSMKeyFieldTest(TestCase):
     def setUp(self):
-        for item in FK_AVAILABLE_STATES:
-            DbState.objects.create(pk=item[0], label=item[1])
+        DbState.objects.bulk_create(DbState(pk=item[0], label=item[1]) for item in FK_AVAILABLE_STATES)
         self.model = FKBlogPost()
 
     def test_initial_state_instantiated(self):
@@ -126,9 +125,11 @@ def hide(self):
 
 class BlogPostWithFKStateTest(TestCase):
     def setUp(self):
-        BlogPostStatus.objects.create(name="new")
-        BlogPostStatus.objects.create(name="published")
-        BlogPostStatus.objects.create(name="hidden")
+        BlogPostStatus.objects.bulk_create([
+            BlogPostStatus(name="new")
+            BlogPostStatus(name="published")
+            BlogPostStatus(name="hidden")
+        ])
         self.model = BlogPostWithFKState()
 
     def test_known_transition_should_succeed(self):

tests/testapp/tests/test_lock_mixin.py

@@ -14,9 +14,6 @@ class LockedBlogPost(ConcurrentTransitionMixin, models.Model):
     state = FSMField(default="new")
     text = models.CharField(max_length=50)
 
-    class Meta:
-        app_label = "testapp"
-
     @transition(field=state, source="new", target="published")
     def publish(self):
         pass
@@ -30,9 +27,6 @@ class ExtendedBlogPost(LockedBlogPost):
     review_state = FSMField(default="waiting", protected=True)
     notes = models.CharField(max_length=50)
 
-    class Meta:
-        app_label = "testapp"
-
     @transition(field=review_state, source="waiting", target="rejected")
     def reject(self):
         pass

tests/testapp/tests/test_mixin_support.py

@@ -8,9 +8,6 @@
 
 
 class WorkflowMixin:
-    class Meta:
-        app_label = "testapp"
-
     @transition(field="state", source="*", target="draft")
     def draft(self):
         pass

tests/testapp/tests/test_model_create_with_generic.py

@@ -1,10 +1,6 @@
 from __future__ import annotations
 
-try:
-    from django.contrib.contenttypes.fields import GenericForeignKey
-except ImportError:
-    # Django 1.6
-    from django.contrib.contenttypes.generic import GenericForeignKey
+from django.contrib.contenttypes.fields import GenericForeignKey
 from django.contrib.contenttypes.models import ContentType
 from django.db import models
 from django.test import TestCase
@@ -13,9 +9,7 @@
 from django_fsm import transition
 
 
-class Ticket(models.Model):
-    class Meta:
-        app_label = "testapp"
+class Ticket(models.Model): ...
 
 
 class TaskState(models.TextChoices):
@@ -29,9 +23,6 @@ class Task(models.Model):
     causality = GenericForeignKey("content_type", "object_id")
     state = FSMField(default=TaskState.NEW)
 
-    class Meta:
-        app_label = "testapp"
-
     @transition(field=state, source=TaskState.NEW, target=TaskState.DONE)
     def do(self):
         pass

tests/testapp/tests/test_multi_resultstate.py

@@ -14,9 +14,6 @@
 class MultiResultTest(models.Model):
     state = FSMField(default="new")
 
-    class Meta:
-        app_label = "testapp"
-
     @transition(field=state, source="new", target=RETURN_VALUE("for_moderators", "published"))
     def publish(self, *, is_public=False):
         return "published" if is_public else "for_moderators"

tests/testapp/tests/test_multidecorators.py

@@ -13,9 +13,6 @@ class TestModel(models.Model):
     signal_counter = models.IntegerField(default=0)
     state = FSMField(default="SUBMITTED_BY_USER")
 
-    class Meta:
-        app_label = "testapp"
-
     @transition(field=state, source="SUBMITTED_BY_USER", target="REVIEW_USER")
     @transition(field=state, source="SUBMITTED_BY_ADMIN", target="REVIEW_ADMIN")
     @transition(field=state, source="SUBMITTED_BY_ANONYMOUS", target="REVIEW_ANONYMOUS")