feature: factories

djangoldp_notification/management/commands/mock_notification.py

@@ -0,0 +1,14 @@
+from django.core.management.base import BaseCommand, CommandError
+from djangoldp_notification.factories import NotificationFactory
+
+class Command(BaseCommand):
+    help = 'Mock data'
+
+    def add_arguments(self, parser):
+        parser.add_argument('--size', type=int, default=0, help='Number of notifications to create')
+
+    def handle(self, *args, **options):
+        for i in range(0, options['size']):
+            notif = NotificationFactory.create()
+
+        self.stdout.write(self.style.SUCCESS('Successful data mock install'))