def test_get_filtered_configs_es_error(self): table_manager = ConfigurableReportTableManagerMixin( MockDataSourceProvider(), filter_missing_domains=True) ds1 = get_sample_data_source() ds1.domain = 'domain1' ds2 = DataSourceConfiguration.wrap(ds1.to_json()) ds2.domain = 'domain2' with patch('corehq.apps.es.es_query.run_query') as run_query: run_query.side_effect = ESError filtered_configs = table_manager.get_filtered_configs([ds1, ds2]) self.assertEqual(filtered_configs, [ds1, ds2])
def test_get_filtered_configs(self): table_manager = ConfigurableReportTableManagerMixin( MockDataSourceProvider(), filter_missing_domains=True) ds1 = get_sample_data_source() ds1.domain = 'domain1' ds2 = DataSourceConfiguration.wrap(ds1.to_json()) ds2.domain = 'domain2' with patch('corehq.apps.es.es_query.run_query') as run_query: run_query.return_value = { 'hits': { 'hits': [{ '_id': 'd1', '_source': { 'name': 'domain1' } }] } } filtered_configs = table_manager.get_filtered_configs([ds1, ds2]) self.assertEqual(filtered_configs, [ds1])