[chore] Fixed failing test for sample app

openwisp · Jul 5, 2021 · 4d036d1 · 4d036d1
1 parent db7bc6a
commit 4d036d1
Showing 1 changed file with 9 additions and 2 deletions.
diff --git a/tests/openwisp2/sample_config/migrations/0002_default_groups_permissions.py b/tests/openwisp2/sample_config/migrations/0002_default_groups_permissions.py
@@ -1,6 +1,9 @@
 from django.db import migrations
 
-from openwisp_controller.config.migrations import assign_permissions_to_groups
+from openwisp_controller.config.migrations import (
+    assign_devicegroup_permissions_to_groups,
+    assign_permissions_to_groups,
+)
 
 
 class Migration(migrations.Migration):
@@ -9,5 +12,9 @@ class Migration(migrations.Migration):
     operations = [
         migrations.RunPython(
             assign_permissions_to_groups, reverse_code=migrations.RunPython.noop
-        )
+        ),
+        migrations.RunPython(
+            assign_devicegroup_permissions_to_groups,
+            reverse_code=migrations.RunPython.noop,
+        ),
     ]