mirror of
https://github.com/django/django.git
synced 2024-12-22 17:16:24 +00:00
f48f671223
It is now a method instead of a property and returns a context manager that yields a cursor on entry and closes the cursor and connection upon exit.
43 lines
1.3 KiB
Python
43 lines
1.3 KiB
Python
from django.db import connection
|
|
|
|
from . import PostgreSQLTestCase
|
|
|
|
try:
|
|
from django.contrib.postgres.signals import (
|
|
get_citext_oids, get_hstore_oids, register_type_handlers,
|
|
)
|
|
except ImportError:
|
|
pass # pyscogp2 isn't installed.
|
|
|
|
|
|
class OIDTests(PostgreSQLTestCase):
|
|
|
|
def assertOIDs(self, oids):
|
|
self.assertIsInstance(oids, tuple)
|
|
self.assertGreater(len(oids), 0)
|
|
self.assertTrue(all(isinstance(oid, int) for oid in oids))
|
|
|
|
def test_hstore_cache(self):
|
|
get_hstore_oids(connection.alias)
|
|
with self.assertNumQueries(0):
|
|
get_hstore_oids(connection.alias)
|
|
|
|
def test_citext_cache(self):
|
|
get_citext_oids(connection.alias)
|
|
with self.assertNumQueries(0):
|
|
get_citext_oids(connection.alias)
|
|
|
|
def test_hstore_values(self):
|
|
oids, array_oids = get_hstore_oids(connection.alias)
|
|
self.assertOIDs(oids)
|
|
self.assertOIDs(array_oids)
|
|
|
|
def test_citext_values(self):
|
|
oids = get_citext_oids(connection.alias)
|
|
self.assertOIDs(oids)
|
|
|
|
def test_register_type_handlers_no_db(self):
|
|
"""Registering type handlers for the nodb connection does nothing."""
|
|
with connection._nodb_cursor() as cursor:
|
|
register_type_handlers(cursor.db)
|