def test_list_extensions_complex(self): """ List all extensions """ idx = [{'name': 'compo1', 'table': 'editors', 'columns': ['id', 'name'], 'tablespace': 'ssd1', 'extension': 'unaccent'}, {'name': 'compo2', 'table': 'editors', 'columns': ['id', 'name'], 'tablespace': 'ssd1', 'extension': 'unaccent'}, {'name': 'compo3', 'table': 'editors', 'columns': ['id', 'name'], 'tablespace': 'ssd1', 'extension': 'pg_trgm'}, {'name': 'compo4', 'table': 'editors', 'columns': ['id', 'name'], 'tablespace': 'ssd1'}] extensions = util.list_extensions(idx) self.assertEqual(len(extensions), 1) self.assertEqual(len(extensions['default']), 2) self.assertTrue('unaccent' in extensions['default']) self.assertTrue('pg_trgm' in extensions['default'])
def test_list_extensions_multidb(self): """ List all extensions """ idx = [{'name': 'compo1', 'table': 'editors', 'columns': ['id', 'name'], 'tablespace': 'ssd1', 'extension': 'unaccent'}, {'name': 'compo2', 'table': 'editors', 'columns': ['id', 'name'], 'tablespace': 'ssd1', 'extension': 'unaccent', 'database': 'slave'}] extensions = util.list_extensions(idx) self.assertEqual(len(extensions), 2) self.assertTrue('default' in extensions.keys()) self.assertTrue('slave' in extensions.keys()) self.assertEqual(extensions['default'], ['unaccent']) self.assertEqual(extensions['slave'], ['unaccent'])