django/tests/postgres_tests/migrations/0001_setup_extensions.py

30 lines
835 B
Python
Raw Normal View History

2014-11-05 00:27:35 +08:00
from django.db import migrations
try:
from django.contrib.postgres.operations import (
BtreeGinExtension, CITextExtension, CreateExtension, HStoreExtension,
TrigramExtension, UnaccentExtension,
)
except ImportError:
from django.test import mock
BtreeGinExtension = mock.Mock()
CreateExtension = mock.Mock()
HStoreExtension = mock.Mock()
TrigramExtension = mock.Mock()
UnaccentExtension = mock.Mock()
CITextExtension = mock.Mock()
class Migration(migrations.Migration):
operations = [
BtreeGinExtension(),
# Ensure CreateExtension quotes extension names by creating one with a
# dash in its name.
CreateExtension('uuid-ossp'),
HStoreExtension(),
TrigramExtension(),
UnaccentExtension(),
CITextExtension(),
]