Add openid_extensions table to be created

Signed-off-by: Patrick Uiterwijk <puiterwijk@redhat.com>
Reviewed-by: Rob Crittenden <rcritten@redhat.com>

diff --git a/ipsilon/providers/openid/store.py b/ipsilon/providers/openid/store.py
index 40e8b8e..7c637d4 100644 (file)
--- 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
 
 # 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
 
 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, '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:
 
     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)
             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
             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 (file)
--- 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
 );
        name TEXT NOT NULL, 
        value TEXT
 );

+CREATE TABLE openid_extensions (
+    name TEXT NOT NULL,
+    option TEXT NOT NULL,
+    value TEXT
+);

 COMMIT;
 COMMIT;