Add basic test of graph_transitions (and centralize tests) (#43)

Author: pfouque

django_fsm/management/commands/graph_transitions.py

@@ -114,7 +114,7 @@ def get_graphviz_layouts():
     except ModuleNotFoundError:
         return {"sfdp", "circo", "twopi", "dot", "neato", "fdp", "osage", "patchwork"}
     else:
-        return graphviz.backend.ENGINES
+        return graphviz.ENGINES
 
 
 class Command(BaseCommand):

django_fsm/tests/__init__.py

@@ -12,7 +12,7 @@
 from django_fsm.signals import pre_transition
 
 
-class BlogPost(models.Model):
+class SimpleBlogPost(models.Model):
     state = FSMField(default="new")
 
     @transition(field=state, source="new", target="published")
@@ -50,7 +50,7 @@ def empty(self):
 
 class FSMFieldTest(TestCase):
     def setUp(self):
-        self.model = BlogPost()
+        self.model = SimpleBlogPost()
 
     def test_initial_state_instantiated(self):
         self.assertEqual(self.model.state, "new")
@@ -121,11 +121,11 @@ def test_empty_string_target(self):
 
 class StateSignalsTests(TestCase):
     def setUp(self):
-        self.model = BlogPost()
+        self.model = SimpleBlogPost()
         self.pre_transition_called = False
         self.post_transition_called = False
-        pre_transition.connect(self.on_pre_transition, sender=BlogPost)
-        post_transition.connect(self.on_post_transition, sender=BlogPost)
+        pre_transition.connect(self.on_pre_transition, sender=SimpleBlogPost)
+        post_transition.connect(self.on_post_transition, sender=SimpleBlogPost)
 
     def on_pre_transition(self, sender, instance, name, source, target, **kwargs):
         self.assertEqual(instance.state, source)
@@ -148,7 +148,7 @@ def test_signals_not_called_on_invalid_transition(self):
 
 class TestFieldTransitionsInspect(TestCase):
     def setUp(self):
-        self.model = BlogPost()
+        self.model = SimpleBlogPost()
 
     def test_in_operator_for_available_transitions(self):
         # store the generator in a list, so we can reuse the generator and do multiple asserts

django_fsm/tests/test_abstract_inheritance.py renamed to tests/testapp/tests/test_abstract_inheritance.py

@@ -0,0 +1,49 @@
+from __future__ import annotations
+
+from django.core.management import call_command
+from django.db import models
+from django.test import TestCase
+
+from django_fsm import FSMField
+from django_fsm import transition
+
+
+class VisualBlogPost(models.Model):
+    state = FSMField(default="new")
+
+    @transition(field=state, source="new", target="published")
+    def publish(self):
+        pass
+
+    @transition(source="published", field=state)
+    def notify_all(self):
+        pass
+
+    @transition(source="published", target="hidden", field=state)
+    def hide(self):
+        pass
+
+    @transition(source="new", target="removed", field=state)
+    def remove(self):
+        raise Exception("Upss")
+
+    @transition(source=["published", "hidden"], target="stolen", field=state)
+    def steal(self):
+        pass
+
+    @transition(source="*", target="moderated", field=state)
+    def moderate(self):
+        pass
+
+    @transition(source="+", target="blocked", field=state)
+    def block(self):
+        pass
+
+    @transition(source="*", target="", field=state)
+    def empty(self):
+        pass
+
+
+class GraphTransitionsCommandTest(TestCase):
+    def test_dummy(self):
+        call_command("graph_transitions", "testapp.VisualBlogPost")