def setUpClass(cls): super(ProjectsServicesCommonTestCase, cls).setUpClass() cls.setUpDirs() cls.spatializeDB(cls.getConnection()) cls.con = cls.getConnection() cls.session = cls.getSession(cls.con) cls.refresh_db(bind=cls.session.bind) cls.project = services.create_project(cls.test_project_file, session=cls.session)
def test_delete_project(self): project = services.create_project(project_file=self.test_project_file) dir_attrs = ['data_dir', 'static_dir'] project_dirs = [getattr(project, dir_attr) for dir_attr in dir_attrs] delete_result = services.delete_project(project) self.assertTrue(delete_result) for dir_ in project_dirs: self.assertFalse(os.path.exists(dir_)) num_layers = db.session.query(project_models.MapLayer).count() self.assertEquals(num_layers, 0)
def call(self): self.progress_logger.info(1) self.status = "running" self.msg_logger.info("Starting...") project = create_project( self.input_path, msg_logger=self.msg_logger, progress_logger=self.progress_logger, ingest_kwargs=self.ingest_kwargs, ) self.data["project_id"] = project.id self.msg_logger.info("Completed.") self.progress_logger.info(100) self.status = "resolved"
def main(): argparser = argparse.ArgumentParser() argparser.add_argument('input_path') argparser.add_argument('--db-uri') argparser.add_argument('--commit-interval', default=int(1e5), type=int) args = argparser.parse_args() logger = logging.getLogger() logger.setLevel(logging.INFO) logger.addHandler(logging.StreamHandler()) ingest_kwargs = {'commit_interval': args.commit_interval} p = create_project(args.input_path, logger=logger, db_uri=args.db_uri, ingest_kwargs=ingest_kwargs) print "project id: ", p.id
def test_create_project(self): project = services.create_project(project_file=self.test_project_file) self.assertTrue(getattr(project, 'id', None) is not None)
def test_delete_project_task(self): project = create_project(input_path=self.test_project_file) task = DeleteProjectTask(project_id=project.id) task.call() self.assertEquals(task.status, 'resolved')