Fix dataset cache invalidation.

Refs coleifer/sqlite-web#58
coleifer · Mar 23, 2019 · 8b42987 · 8b42987
1 parent f1f2cfe
commit 8b42987
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 0 deletions.
diff --git a/playhouse/dataset.py b/playhouse/dataset.py
@@ -98,6 +98,7 @@ def update_cache(self, table=None):
                 dependencies.extend(self.get_table_dependencies(table))
         else:
             dependencies = None  # Update all tables.
+            self._models = {}
         updated = self._introspector.generate_models(
             skip_invalid=True,
             table_names=dependencies,

diff --git a/tests/dataset.py b/tests/dataset.py
@@ -125,6 +125,15 @@ def test_update_cache(self):
         self.assertEqual(sorted(Foo.columns), ['data', 'id'])
         self.assertTrue('foo' in self.dataset._models)
 
+        self.dataset._models['foo'].drop_table()
+        self.dataset.update_cache()
+        self.assertTrue('foo' not in self.database.get_tables())
+
+        # This will create the table again.
+        Foo = self.dataset['foo']
+        self.assertTrue('foo' in self.database.get_tables())
+        self.assertEqual(Foo.columns, ['id'])
+
     def assertQuery(self, query, expected, sort_key='id'):
         key = operator.itemgetter(sort_key)
         self.assertEqual(