示例#1
0
 def test_create_extension_newerthan(self):
     '''
     scenario of creating upgrading extensions with possible schema and
     version specifications
     '''
     self.assertTrue(postgres.create_extension('foo'))
     self.assertTrue(
         re.match('CREATE EXTENSION IF NOT EXISTS "foo" ;',
                  postgres._psql_prepare_and_run.call_args[0][0][1]))
     self.assertTrue(
         postgres.create_extension('foo',
                                   schema='a',
                                   ext_version='b',
                                   from_version='c'))
     self.assertTrue(
         re.match(
             'CREATE EXTENSION IF NOT EXISTS "foo" '
             'WITH SCHEMA "a" VERSION b FROM c ;',
             postgres._psql_prepare_and_run.call_args[0][0][1]))
     self.assertFalse(postgres.create_extension('foo'))
     ret = postgres.create_extension('foo', ext_version='a', schema='b')
     self.assertTrue(ret)
     self.assertTrue(
         re.match(
             'ALTER EXTENSION "foo" SET SCHEMA "b";'
             ' ALTER EXTENSION "foo" UPDATE TO a;',
             postgres._psql_prepare_and_run.call_args[0][0][1]))
     ret = postgres.create_extension('foo', ext_version='a', schema='b')
     self.assertTrue(ret)
     self.assertTrue(
         re.match('ALTER EXTENSION "foo" SET SCHEMA "b";',
                  postgres._psql_prepare_and_run.call_args[0][0][1]))
     ret = postgres.create_extension('foo', ext_version='a', schema='b')
     self.assertTrue(ret)
     self.assertTrue(
         re.match('ALTER EXTENSION "foo" UPDATE TO a;',
                  postgres._psql_prepare_and_run.call_args[0][0][1]))
     self.assertFalse(
         postgres.create_extension('foo', ext_version='a', schema='b'))
     self.assertFalse(
         postgres.create_extension('foo', ext_version='a', schema='b'))
示例#2
0
 def test_create_extension_newerthan(self):
     '''
     scenario of creating upgrading extensions with possible schema and
     version specifications
     '''
     self.assertTrue(postgres.create_extension('foo'))
     self.assertTrue(re.match(
         'CREATE EXTENSION IF NOT EXISTS "foo" ;',
         postgres._psql_prepare_and_run.call_args[0][0][1]))
     self.assertTrue(postgres.create_extension(
         'foo', schema='a', ext_version='b', from_version='c'))
     self.assertTrue(re.match(
         'CREATE EXTENSION IF NOT EXISTS "foo" '
         'WITH SCHEMA a VERSION b FROM c ;',
         postgres._psql_prepare_and_run.call_args[0][0][1]))
     self.assertFalse(postgres.create_extension('foo'))
     ret = postgres.create_extension('foo', ext_version='a', schema='b')
     self.assertTrue(ret)
     self.assertTrue(re.match(
         'ALTER EXTENSION "foo" SET SCHEMA b;'
         ' ALTER EXTENSION "foo" UPDATE TO a;',
         postgres._psql_prepare_and_run.call_args[0][0][1]))
     ret = postgres.create_extension('foo', ext_version='a', schema='b')
     self.assertTrue(ret)
     self.assertTrue(re.match(
         'ALTER EXTENSION "foo" SET SCHEMA b;',
         postgres._psql_prepare_and_run.call_args[0][0][1]))
     ret = postgres.create_extension('foo', ext_version='a', schema='b')
     self.assertTrue(ret)
     self.assertTrue(re.match(
         'ALTER EXTENSION "foo" UPDATE TO a;',
         postgres._psql_prepare_and_run.call_args[0][0][1]))
     self.assertFalse(postgres.create_extension(
         'foo', ext_version='a', schema='b'))
     self.assertFalse(postgres.create_extension(
         'foo', ext_version='a', schema='b'))