diff --git a/tests/postgres_tests/migrations/0001_setup_extensions.py b/tests/postgres_tests/migrations/0001_setup_extensions.py index 6342ba7732..d090ff7fd6 100644 --- a/tests/postgres_tests/migrations/0001_setup_extensions.py +++ b/tests/postgres_tests/migrations/0001_setup_extensions.py @@ -9,22 +9,22 @@ try: ) except ImportError: BtreeGinExtension = mock.Mock() + CITextExtension = mock.Mock() CreateExtension = mock.Mock() HStoreExtension = mock.Mock() TrigramExtension = mock.Mock() UnaccentExtension = mock.Mock() - CITextExtension = mock.Mock() class Migration(migrations.Migration): operations = [ BtreeGinExtension(), + CITextExtension(), # Ensure CreateExtension quotes extension names by creating one with a # dash in its name. CreateExtension('uuid-ossp'), HStoreExtension(), TrigramExtension(), UnaccentExtension(), - CITextExtension(), ]