def tearDown(self): # get any current run adapter and clean up its connections before we # reset them. It'll probably be different from ours because # handle_and_check() calls reset_adapters(). register_adapter(self.config) adapter = get_adapter(self.config) if adapter is not self.adapter: adapter.cleanup_connections() if not hasattr(self, 'adapter'): self.adapter = adapter self._drop_schemas() self.adapter.cleanup_connections() reset_adapters() os.chdir(INITIAL_ROOT) try: shutil.rmtree(self.test_root_dir) except EnvironmentError: logger.exception('Could not clean up after test - {} not removable' .format(self.test_root_dir))
def setUp(self): self.dbt_core_install_root = os.path.dirname(dbt.__file__) log_manager.reset_handlers() self.initial_dir = INITIAL_ROOT os.chdir(self.initial_dir) # before we go anywhere, collect the initial path info self._logs_dir = os.path.join(self.initial_dir, 'logs', self.prefix) _really_makedirs(self._logs_dir) self.test_original_source_path = _pytest_get_test_root() self.test_root_dir = self._generate_test_root_dir() os.chdir(self.test_root_dir) try: self._symlink_test_folders() except Exception as exc: msg = '\n\t'.join(( 'Failed to symlink test folders!', 'initial_dir={0.initial_dir}', 'test_original_source_path={0.test_original_source_path}', 'test_root_dir={0.test_root_dir}' )).format(self) logger.exception(msg) # if logging isn't set up, I still really want this message. print(msg) traceback.print_exc() raise self._created_schemas = set() reset_deprecations() flags.reset() template_cache.clear() self.use_profile(self._pick_profile()) self.use_default_project() self.set_packages() self.set_selectors() self.load_config()
def setUp(self): self.initial_dir = INITIAL_ROOT os.chdir(self.initial_dir) # before we go anywhere, collect the initial path info self._logs_dir = os.path.join(self.initial_dir, 'logs', self.prefix) print('initial_dir={}'.format(self.initial_dir)) _really_makedirs(self._logs_dir) self.test_original_source_path = _pytest_get_test_root() print('test_original_source_path={}'.format( self.test_original_source_path)) self.test_root_dir = tempfile.mkdtemp(prefix='dbt-int-test-') print('test_root_dir={}'.format(self.test_root_dir)) os.chdir(self.test_root_dir) try: self._symlink_test_folders() except Exception as exc: msg = '\n\t'.join( ('Failed to symlink test folders!', 'initial_dir={0.initial_dir}', 'test_original_source_path={0.test_original_source_path}', 'test_root_dir={0.test_root_dir}')).format(self) logger.exception(msg) # if logging isn't set up, I still really want this message. print(msg) traceback.print_exc() raise self._created_schemas = set() flags.reset() template_cache.clear() # disable capturing warnings logging.captureWarnings(False) self.use_profile(self._pick_profile()) self.use_default_project() self.set_packages() self.load_config()