From f958c5e78a7d5665073483d79926e91709a355fe Mon Sep 17 00:00:00 2001 From: Patrick Uiterwijk Date: Thu, 3 Sep 2015 23:01:56 +0200 Subject: [PATCH] Add openid_extensions table to be created Signed-off-by: Patrick Uiterwijk Reviewed-by: Rob Crittenden --- ipsilon/providers/openid/store.py | 10 +++++++++- tests/blobs/old_dbs/v1/openid.sqlite.dump | 5 +++++ 2 files changed, 14 insertions(+), 1 deletion(-) diff --git a/ipsilon/providers/openid/store.py b/ipsilon/providers/openid/store.py index 40e8b8e..7c637d4 100644 --- a/ipsilon/providers/openid/store.py +++ b/ipsilon/providers/openid/store.py @@ -1,6 +1,6 @@ # Copyright (C) 2014 Ipsilon project Contributors, for license see COPYING -from ipsilon.util.data import Store, UNIQUE_DATA_TABLE +from ipsilon.util.data import Store, UNIQUE_DATA_TABLE, OPTIONS_TABLE from openid import oidutil from openid.association import Association @@ -82,6 +82,9 @@ class OpenIDStore(Store, OpenIDStoreInterface): q = self._query(self._db, 'association', UNIQUE_DATA_TABLE, trans=False) q.create() + q = self._query(self._db, 'openid_extensions', OPTIONS_TABLE, + trans=False) + q.create() def _upgrade_schema(self, old_version): if old_version == 1: @@ -90,6 +93,11 @@ class OpenIDStore(Store, OpenIDStoreInterface): table = self._query(self._db, 'association', UNIQUE_DATA_TABLE, trans=False)._table self._db.add_constraint(table.primary_key) + for index in table.indexes: + self._db.add_index(index) + table = self._query(self._db, 'openid_extensions', OPTIONS_TABLE, + trans=False)._table + self._db.add_constraint(table.primary_key) for index in table.indexes: self._db.add_index(index) return 2 diff --git a/tests/blobs/old_dbs/v1/openid.sqlite.dump b/tests/blobs/old_dbs/v1/openid.sqlite.dump index 4618785..2ffef8e 100644 --- a/tests/blobs/old_dbs/v1/openid.sqlite.dump +++ b/tests/blobs/old_dbs/v1/openid.sqlite.dump @@ -11,4 +11,9 @@ CREATE TABLE association ( name TEXT NOT NULL, value TEXT ); +CREATE TABLE openid_extensions ( + name TEXT NOT NULL, + option TEXT NOT NULL, + value TEXT +); COMMIT; -- 2.20.1