summaryrefslogtreecommitdiffstats
path: root/tests/test_sql_upgrade.py
diff options
context:
space:
mode:
authorAdam Young <ayoung@redhat.com>2012-11-06 17:15:50 -0500
committerAdam Young <ayoung@redhat.com>2012-11-07 22:29:11 -0500
commit0478276993afdb82b55606b47fae010fec32c640 (patch)
treed7b5cc38136e0cd56d2e01ecb0bdba6b5c796dfa /tests/test_sql_upgrade.py
parent126dd9c9bdab46074d812f4a16358357d364e789 (diff)
downloadkeystone-0478276993afdb82b55606b47fae010fec32c640.tar.gz
keystone-0478276993afdb82b55606b47fae010fec32c640.tar.xz
keystone-0478276993afdb82b55606b47fae010fec32c640.zip
SQL upgrade test.
Tests upgrade to version 1. Confirms all of the identity tables layout. blueprint: normalize-sql Change-Id: If66250af068b396fc55f38c66f789b9447353bda
Diffstat (limited to 'tests/test_sql_upgrade.py')
-rw-r--r--tests/test_sql_upgrade.py104
1 files changed, 104 insertions, 0 deletions
diff --git a/tests/test_sql_upgrade.py b/tests/test_sql_upgrade.py
new file mode 100644
index 00000000..833a73e7
--- /dev/null
+++ b/tests/test_sql_upgrade.py
@@ -0,0 +1,104 @@
+# vim: tabstop=4 shiftwidth=4 softtabstop=4
+
+# Copyright 2012 OpenStack LLC
+#
+# Licensed under the Apache License, Version 2.0 (the "License"); you may
+# not use this file except in compliance with the License. You may obtain
+# a copy of the License at
+#
+# http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
+# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
+# License for the specific language governing permissions and limitations
+# under the License.
+
+from migrate.versioning import api as versioning_api
+import sqlalchemy
+
+from keystone.common import sql
+from keystone import config
+from keystone import test
+from keystone.common.sql import migration
+
+
+CONF = config.CONF
+
+
+class SqlUpgradeTests(test.TestCase):
+ def setUp(self):
+ super(SqlUpgradeTests, self).setUp()
+ self.config([test.etcdir('keystone.conf.sample'),
+ test.testsdir('test_overrides.conf'),
+ test.testsdir('backend_sql.conf')])
+
+ # create and share a single sqlalchemy engine for testing
+ self.engine = sql.Base().get_engine()
+ self.metadata = sqlalchemy.MetaData()
+ # populate the engine with tables & fixtures
+ self.metadata.bind = self.engine
+ self.repo_path = migration._find_migrate_repo()
+ self.schema = versioning_api.ControlledSchema.create(self.engine,
+ self.repo_path, 0)
+
+ def tearDown(self):
+ super(SqlUpgradeTests, self).tearDown()
+
+ def test_blank_db_to_start(self):
+ self.assertFalse(self.is_user_table_created(),
+ "User should not be defined yet")
+
+ def test_start_version_0(self):
+ version = migration.db_version()
+ self.assertEqual(version, 0, "DB is at version 0")
+
+ def assertTableColumns(self, name, expected_cols):
+ """
+ Confirms that the set of columns expected match the actual based on
+ name and number of columns
+ """
+ table = self.select_table(name)
+ self.assertEqual(len(table.columns), len(expected_cols))
+ col_names = []
+ for col in table.columns:
+ col_names.append(col.name)
+ for expected_col in expected_cols:
+ self.assertIn(expected_col, col_names,
+ "column %s not defined" % expected_col)
+
+ def test_upgrade_0_to_1(self):
+ self.assertEqual(self.schema.version, 0, "DB is at version 0")
+ self._migrate(self.repo_path, 1)
+ self.assertEqual(self.schema.version, 1, "DB is at version 0")
+ self.assertTableColumns("user", ["id", "name", "extra"])
+ self.assertTableColumns("tenant", ["id", "name", "extra"])
+ self.assertTableColumns("role", ["id", "name"])
+ self.assertTableColumns("user_tenant_membership",
+ ["user_id", "tenant_id"])
+ self.assertTableColumns("metadata", ["user_id", "tenant_id", "data"])
+
+ def select_table(self, name):
+ table = sqlalchemy.Table(name,
+ self.metadata,
+ autoload=True)
+ s = sqlalchemy.select([table])
+ return s
+
+ def is_user_table_created(self):
+ try:
+ self.select_table("user")
+ return True
+ except sqlalchemy.exc.NoSuchTableError:
+ return False
+
+ def _migrate(self, repository, version):
+ upgrade = True
+ err = ""
+ version = versioning_api._migrate_version(self.schema,
+ version,
+ upgrade,
+ err)
+ changeset = self.schema.changeset(version)
+ for ver, change in changeset:
+ self.schema.runchange(ver, change, changeset.step)