def setUpClass(cls): super(UCRMultiDBTest, cls).setUpClass() cls.db2_name = 'cchq_ucr_tests' default_db_url = connections.connection_manager.get_connection_string(DEFAULT_DB_ALIAS) db_conn_parts = default_db_url.split('/') db_conn_parts[-1] = cls.db2_name cls.db2_url = '/'.join(db_conn_parts) cls.context_managers = ExitStack() cls.context_managers.enter_context(connections.override_engine('engine-1', default_db_url, 'default')) cls.context_managers.enter_context(connections.override_engine('engine-2', cls.db2_url, cls.db2_name)) # setup data sources data_source_template = get_sample_data_source() cls.ds_1 = DataSourceConfiguration.wrap(data_source_template.to_json()) cls.ds_1.engine_id = 'engine-1' cls.ds_1.save() cls.ds_2 = DataSourceConfiguration.wrap(data_source_template.to_json()) cls.ds_2.engine_id = 'engine-2' cls.ds_2.save() cls.context_managers.enter_context(temporary_database(cls.db2_name)) cls.ds1_adapter = get_indicator_adapter(cls.ds_1) cls.ds2_adapter = get_indicator_adapter(cls.ds_2)
def setUpClass(cls): cls.db2_name = 'cchq_ucr_tests' db_conn_parts = settings.SQL_REPORTING_DATABASE_URL.split('/') db_conn_parts[-1] = cls.db2_name cls.db2_url = '/'.join(db_conn_parts) # setup patches cls.connection_string_patch = patch('corehq.sql_db.connections.connection_manager.get_connection_string') def connection_string_for_engine(engine_id): if engine_id == 'engine-1': return settings.SQL_REPORTING_DATABASE_URL else: return cls.db2_url mock_manager = cls.connection_string_patch.start() mock_manager.side_effect = connection_string_for_engine # setup data sources data_source_template = get_sample_data_source() cls.ds_1 = DataSourceConfiguration.wrap(data_source_template.to_json()) cls.ds_1.engine_id = 'engine-1' cls.ds_1.save() cls.ds_2 = DataSourceConfiguration.wrap(data_source_template.to_json()) cls.ds_2.engine_id = 'engine-2' cls.ds_2.save() cls.db_context = temporary_database(cls.db2_name) cls.db_context.__enter__() cls.ds1_adapter = IndicatorSqlAdapter(cls.ds_1) cls.ds2_adapter = IndicatorSqlAdapter(cls.ds_2)
def setUp(self): super(TestSavingToUCRDatabase, self).setUp() self.cc_domain, self.cc_user = create_domain_and_user(self.domain_name, 'user_ucr') self.ucr_db_name = 'cchq_ucr_tests' db_conn_parts = connection_manager.get_connection_string('default').split('/') db_conn_parts[-1] = self.ucr_db_name self.ucr_db_url = '/'.join(db_conn_parts) self.db_context = temporary_database(self.ucr_db_name) self.db_context.__enter__()
def setUp(self): super(TestSavingToUCRDatabase, self).setUp() self.cc_domain, self.cc_user = create_domain_and_user(self.domain_name, 'user_ucr') self.ucr_db_name = 'cchq_ucr_tests' db_conn_parts = connection_manager.get_connection_string(DEFAULT_DB_ALIAS).split('/') db_conn_parts[-1] = self.ucr_db_name self.ucr_db_url = '/'.join(db_conn_parts) self.db_context = temporary_database(self.ucr_db_name) self.db_context.__enter__()
def setUp(self): super(TestSavingToUCRDatabase, self).setUp() self.cc_domain, self.cc_user = create_domain_and_user(self.domain_name, 'user_ucr') create_cases_for_types(self.cc_domain.name, ['person', 'dog']) self.ucr_db_name = 'cchq_ucr_tests' db_conn_parts = settings.SQL_REPORTING_DATABASE_URL.split('/') db_conn_parts[-1] = self.ucr_db_name self.ucr_db_url = '/'.join(db_conn_parts) self.db_context = temporary_database(self.ucr_db_name) self.db_context.__enter__()
def setUp(self): super(TestSavingToUCRDatabase, self).setUp() self.cc_domain, self.cc_user = create_domain_and_user( self.domain_name, 'user_ucr') self.ucr_db_name = 'cchq_ucr_tests' db_conn_parts = settings.SQL_REPORTING_DATABASE_URL.split('/') db_conn_parts[-1] = self.ucr_db_name self.ucr_db_url = '/'.join(db_conn_parts) self.db_context = temporary_database(self.ucr_db_name) self.db_context.__enter__()
def setUpClass(cls): super(UCRMultiDBTest, cls).setUpClass() cls.db2_name = 'cchq_ucr_tests' db_conn_parts = connections.connection_manager.get_connection_string('default').split('/') db_conn_parts[-1] = cls.db2_name cls.db2_url = '/'.join(db_conn_parts) cls.context_manager = connections.override_engine('engine-2', cls.db2_url, cls.db2_name) cls.context_manager.__enter__() # setup data sources data_source_template = get_sample_data_source() cls.ds_1 = DataSourceConfiguration.wrap(data_source_template.to_json()) cls.ds_1.engine_id = 'engine-1' cls.ds_1.save() cls.ds_2 = DataSourceConfiguration.wrap(data_source_template.to_json()) cls.ds_2.engine_id = 'engine-2' cls.ds_2.save() cls.db_context = temporary_database(cls.db2_name) cls.db_context.__enter__() cls.ds1_adapter = get_indicator_adapter(cls.ds_1) cls.ds2_adapter = get_indicator_adapter(cls.ds_2)