Fixed test about warnings

lorinkoz · Jun 26, 2023 · d0d777d · d0d777d
1 parent 67faa99
commit d0d777d
Showing 1 changed file with 6 additions and 6 deletions.
diff --git a/dpgs_sandbox/tests/test_schema.py b/dpgs_sandbox/tests/test_schema.py
@@ -1,16 +1,16 @@
+import warnings
+
 from django.test import SimpleTestCase
 
 
 class TestDeprecatedImports(SimpleTestCase):
     def test_schema_descriptor_warning(self):
-        import warnings
-
-        warnings.simplefilter("error")
-
-        with self.assertRaises(DeprecationWarning) as ctx:
+        with warnings.catch_warnings(record=True) as w:
+            warnings.simplefilter("always")
             from django_pgschemas.schema import SchemaDescriptor  # noqa
 
-            self.assertEqual(str(ctx.exception), "'SchemaDescriptor' is deprecated, use 'Schema' instead")
+            self.assertEqual(len(w), 1)
+            self.assertEqual(str(w[0].message), "'SchemaDescriptor' is deprecated, use 'Schema' instead")
 
     def test_schema_descriptor_accessible(self):
         from django_pgschemas.schema import SchemaDescriptor