diff options
| author | Nick Pope <nick.pope@flightdataservices.com> | 2020-07-29 12:59:29 +0100 |
|---|---|---|
| committer | Mariusz Felisiak <felisiak.mariusz@gmail.com> | 2020-10-16 12:59:09 +0200 |
| commit | be183fc94ffa6d9dbefb4faac915f8aad34fc4b7 (patch) | |
| tree | ef22fee2067630389c7bf49c9aa17cb1c62f39ca /tests/inspectdb | |
| parent | 1fb97fb965786a4398a2236e4e619b78a25d5875 (diff) | |
Optimized inspectdb tests by specifying database tables/views.
Diffstat (limited to 'tests/inspectdb')
| -rw-r--r-- | tests/inspectdb/tests.py | 39 |
1 files changed, 34 insertions, 5 deletions
diff --git a/tests/inspectdb/tests.py b/tests/inspectdb/tests.py index b2da1c0a26..a0c4aff2e5 100644 --- a/tests/inspectdb/tests.py +++ b/tests/inspectdb/tests.py @@ -19,6 +19,13 @@ def inspectdb_tables_only(table_name): return table_name.startswith('inspectdb_') +def inspectdb_views_only(table_name): + return ( + table_name.startswith('inspectdb_') and + table_name.endswith(('_materialized', '_view')) + ) + + def special_table_only(table_name): return table_name.startswith('inspectdb_special') @@ -337,11 +344,20 @@ class InspectDBTransactionalTests(TransactionTestCase): view_model = 'class InspectdbPeopleView(models.Model):' view_managed = 'managed = False # Created from a view.' try: - call_command('inspectdb', table_name_filter=inspectdb_tables_only, stdout=out) + call_command( + 'inspectdb', + table_name_filter=inspectdb_views_only, + stdout=out, + ) no_views_output = out.getvalue() self.assertNotIn(view_model, no_views_output) self.assertNotIn(view_managed, no_views_output) - call_command('inspectdb', table_name_filter=inspectdb_tables_only, include_views=True, stdout=out) + call_command( + 'inspectdb', + table_name_filter=inspectdb_views_only, + include_views=True, + stdout=out, + ) with_views_output = out.getvalue() self.assertIn(view_model, with_views_output) self.assertIn(view_managed, with_views_output) @@ -361,11 +377,20 @@ class InspectDBTransactionalTests(TransactionTestCase): view_model = 'class InspectdbPeopleMaterialized(models.Model):' view_managed = 'managed = False # Created from a view.' try: - call_command('inspectdb', table_name_filter=inspectdb_tables_only, stdout=out) + call_command( + 'inspectdb', + table_name_filter=inspectdb_views_only, + stdout=out, + ) no_views_output = out.getvalue() self.assertNotIn(view_model, no_views_output) self.assertNotIn(view_managed, no_views_output) - call_command('inspectdb', table_name_filter=inspectdb_tables_only, include_views=True, stdout=out) + call_command( + 'inspectdb', + table_name_filter=inspectdb_views_only, + include_views=True, + stdout=out, + ) with_views_output = out.getvalue() self.assertIn(view_model, with_views_output) self.assertIn(view_managed, with_views_output) @@ -426,7 +451,11 @@ class InspectDBTransactionalTests(TransactionTestCase): foreign_table_model = 'class InspectdbIrisForeignTable(models.Model):' foreign_table_managed = 'managed = False' try: - call_command('inspectdb', stdout=out) + call_command( + 'inspectdb', + table_name_filter=inspectdb_tables_only, + stdout=out, + ) output = out.getvalue() self.assertIn(foreign_table_model, output) self.assertIn(foreign_table_managed, output) |