Fix unittest: migration from 0158 to 0160 (DefectDojo#6234)

kiblik · web-flow · commit 4441a3255f94 · 2022-05-03T12:11:18.000-05:00
diff --git a/unittests/test_migrations.py b/unittests/test_migrations.py
@@ -359,7 +359,7 @@ def prepare(self):
     def test_broken_eps(self):
         Finding = self.new_state.apps.get_model('dojo', 'Finding')
         Endpoint = self.new_state.apps.get_model('dojo', 'Endpoint')
-        Endpoint_Status = self.old_state.apps.get_model('dojo', 'Endpoint_Status')
+        Endpoint_Status = self.new_state.apps.get_model('dojo', 'Endpoint_Status')
 
         with self.subTest('Stadnard eps for finding'):
             f = Finding.objects.filter(id=self.finding)
