Merge pull request #14 from Styria-Digital/feature/test_command

Feature/test command

setup.py

@@ -5,7 +5,7 @@
 
 setup(
     name='django-unload',
-    version='0.2',
+    version='0.3',
     url="https://github.com/Styria-Digital/django-unload",
     author='Styria Digital Services',
     description='Remove unused custom Django template tags and filters',

unload/management/commands/find_unnecessary_loads.py

@@ -28,4 +28,5 @@ def add_arguments(self, parser):
     def handle(self, *args, **options):
         # Find the app
         app_label = options.get('app', None)
-        list_unnecessary_loads(app_label)
+        self.stdout.write(
+            'Has issues: {}'.format(str(list_unnecessary_loads(app_label))))

unload/tests/test_commands.py

@@ -0,0 +1,35 @@
+# -*- coding: utf-8 -*-
+
+from __future__ import unicode_literals
+
+import sys
+
+from django.core.management import call_command
+from django.test import TestCase
+
+PYTHON_VERSION = sys.version_info
+
+if PYTHON_VERSION.major == 2:
+    from StringIO import StringIO
+elif PYTHON_VERSION.major == 3:
+    from io import StringIO
+
+
+class TestCommand(TestCase):
+
+    def test_find_unnecessary_loads(self):
+        output = StringIO()
+        call_command('find_unnecessary_loads', stdout=output)
+        self.assertEqual('Has issues: True', output.getvalue().strip())
+
+        output = StringIO()
+        call_command('find_unnecessary_loads', app='app', stdout=output)
+        self.assertEqual('Has issues: True', output.getvalue().strip())
+
+        output = StringIO()
+        call_command('find_unnecessary_loads', app='clean', stdout=output)
+        self.assertEqual('Has issues: False', output.getvalue().strip())
+
+        output = StringIO()
+        call_command('find_unnecessary_loads', app='empty', stdout=output)
+        self.assertEqual('Has issues: False', output.getvalue().strip())