add diamond inheritance test #131

Author: bckohan

tests/apps/test_app/management/commands/inheritance1.py

@@ -0,0 +1,36 @@
+"""
+Test that multi-level and multiple inheritance resolves correctly.
+"""
+
+from django_typer.management import TyperCommand, command, group, initialize
+
+
+class Command(TyperCommand):
+    """
+    Inheritance1
+    """
+
+    @initialize(invoke_without_command=True)
+    def init(self):
+        assert isinstance(self, Command)
+        return "inheritance1::init()"
+
+    @command()
+    def a(self):
+        assert isinstance(self, Command)
+        return "inheritance1::a()"
+
+    @group(invoke_without_command=True)
+    def g(self):
+        assert isinstance(self, Command)
+        return "inheritance1::g()"
+
+    @g.command()
+    def ga(self):
+        assert isinstance(self, Command)
+        return "inheritance1::g::ga()"
+
+    @g.command()
+    def gb(self):
+        assert isinstance(self, Command)
+        return "inheritance1::g::gb()"

tests/apps/test_app/management/commands/inheritance2_1.py

@@ -0,0 +1,43 @@
+from django_typer.management import TyperCommand, command, group, initialize
+from .inheritance1 import Command as Inheritance1Command
+
+
+class Command(Inheritance1Command):
+    """
+    Inheritance2_1
+    """
+
+    @initialize(invoke_without_command=True)
+    def init(self):
+        assert isinstance(self, Command)
+        return "inheritance2_1::init()"
+
+    @command()
+    def a(self):
+        assert isinstance(self, Command)
+        return "inheritance2_1::a()"
+
+    @command()
+    def b(self):
+        assert isinstance(self, Command)
+        return "inheritance2_1::b()"
+
+    @group()
+    def g(self):
+        assert isinstance(self, Command)
+        return "inheritance2_1::g()"
+
+    @g.command()
+    def ga(self):
+        assert isinstance(self, Command)
+        return "inheritance2_1::g::ga()"
+
+    @g.command()
+    def gb(self):
+        assert isinstance(self, Command)
+        return "inheritance2_1::g::gb()"
+
+    @g.command()
+    def gc(self):
+        assert isinstance(self, Command)
+        return "inheritance2_1::g::gc()"

tests/apps/test_app/management/commands/inheritance2_2.py

@@ -0,0 +1,43 @@
+from django_typer.management import TyperCommand, command, group, initialize
+from .inheritance1 import Command as Inheritance1Command
+
+
+class Command(Inheritance1Command):
+    """
+    Inheritance2_2
+    """
+
+    @initialize(invoke_without_command=True)
+    def init(self):
+        assert isinstance(self, Command)
+        return "inheritance2_2::init()"
+
+    @command()
+    def a(self):
+        assert isinstance(self, Command)
+        return "inheritance2_2::a()"
+
+    @command()
+    def b(self):
+        assert isinstance(self, Command)
+        return "inheritance2_2::b()"
+
+    @command()
+    def c(self):
+        assert isinstance(self, Command)
+        return "inheritance2_2::c()"
+
+    @group()
+    def g(self):
+        assert isinstance(self, Command)
+        return "inheritance2_2::g()"
+
+    @g.command()
+    def ga(self):
+        assert isinstance(self, Command)
+        return "inheritance2_2::g::ga()"
+
+    @group(invoke_without_command=True)
+    def g2(self):
+        assert isinstance(self, Command)
+        return "inheritance2_2::g2()"

tests/apps/test_app/management/commands/inheritance3.py

@@ -0,0 +1,20 @@
+from django_typer.management import TyperCommand, command, group, initialize
+from .inheritance2_1 import Command as Inheritance21Command
+from .inheritance2_2 import Command as Inheritance22Command
+
+
+class Command(Inheritance21Command, Inheritance22Command):
+    @command()
+    def b(self):
+        assert isinstance(self, Command)
+        return "inheritance3::b()"
+
+    @command()
+    def d(self):
+        assert isinstance(self, Command)
+        return "inheritance3::d()"
+
+    @Inheritance22Command.g2.command()
+    def g2a(self):
+        assert isinstance(self, Command)
+        return "inheritance3::g2::g2a()"

tests/test_inheritance.py

@@ -29,3 +29,64 @@ def test_handle_override(self):
                 "class": "<class 'tests.apps.test_app.management.commands.help_precedence16.Command'>",
             },
         )
+
+
+class TestDiamondInheritance(TestCase):
+    """
+    Multiple inheritance works!! Even diamond inheritance!
+    """
+
+    def test_diamond_inheritance_run(self):
+        stdout, _, retcode = run_command("inheritance3")
+        self.assertEqual(retcode, 0)
+        self.assertTrue("inheritance2_1::init()" in stdout)
+
+        stdout, _, retcode = run_command("inheritance3", "a")
+        self.assertEqual(retcode, 0)
+        self.assertTrue("inheritance2_1::a()" in stdout)
+
+        stdout, _, retcode = run_command("inheritance2_2", "a")
+        self.assertEqual(retcode, 0)
+        self.assertTrue("inheritance2_2::a()" in stdout)
+
+        stdout, _, retcode = run_command("inheritance3", "b")
+        self.assertEqual(retcode, 0)
+        self.assertTrue("inheritance3::b()" in stdout)
+
+        stdout, _, retcode = run_command("inheritance3", "c")
+        self.assertEqual(retcode, 0)
+        self.assertTrue("inheritance2_2::c()" in stdout)
+
+        stdout, _, retcode = run_command("inheritance3", "g2", "g2a")
+        self.assertEqual(retcode, 0)
+        self.assertTrue("inheritance3::g2::g2a()" in stdout)
+
+        stdout, _, retcode = run_command("inheritance1", "g")
+        self.assertEqual(retcode, 0)
+        self.assertTrue("inheritance1::g()" in stdout)
+
+        stdout, _, retcode = run_command("inheritance3", "g")
+        self.assertGreater(retcode, 0)
+
+        stdout, _, retcode = run_command("inheritance3", "g", "ga")
+        self.assertEqual(retcode, 0)
+        self.assertTrue("inheritance2_1::g::ga()" in stdout)
+
+        stdout, _, retcode = run_command("inheritance3", "g", "gb")
+        self.assertEqual(retcode, 0)
+        self.assertTrue("inheritance2_1::g::gb()" in stdout)
+
+        stdout, _, retcode = run_command("inheritance3", "g", "gc")
+        self.assertEqual(retcode, 0)
+        self.assertTrue("inheritance2_1::g::gc()" in stdout)
+
+        stdout, _, retcode = run_command("inheritance1", "g", "gc")
+        self.assertGreater(retcode, 0)
+
+        stdout, _, retcode = run_command("inheritance1", "g", "ga")
+        self.assertEqual(retcode, 0)
+        self.assertTrue("inheritance1::g::ga()" in stdout)
+
+        stdout, _, retcode = run_command("inheritance1", "g", "gb")
+        self.assertEqual(retcode, 0)
+        self.assertTrue("inheritance1::g::gb()" in stdout)