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')
