summaryrefslogtreecommitdiff
path: root/tests/postgres_tests/__init__.py
blob: dbb913dbba88ea39851c6999b24ca388af6e041e (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
import unittest

from django.db import connection
from django.db.backends.signals import connection_created
from django.test import TestCase


@unittest.skipUnless(connection.vendor == 'postgresql', "PostgreSQL specific tests")
class PostgreSQLTestCase(TestCase):
    @classmethod
    def tearDownClass(cls):
        # No need to keep that signal overhead for non PostgreSQL-related tests.
        from django.contrib.postgres.signals import register_hstore_handler

        connection_created.disconnect(register_hstore_handler)
        super(PostgreSQLTestCase, cls).tearDownClass()