def populate_db():
import geoalchemy2
import shapely
config = ConfigParser.ConfigParser()
config.read(local_settings_path)
db_url = config.get('app:main', 'sqlalchemy.url')
engine = create_engine(db_url)
Base.metadata.drop_all(engine)
Base.metadata.create_all(engine)
DBSession.configure(bind=engine)
# those users are immutables ie. they're not suppose to change during tests
user = User(USER1_ID, u'user1')
DBSession.add(user)
user = User(USER2_ID, u'user2')
DBSession.add(user)
user = User(ADMIN_USER_ID, u'admin_user')
user.role = User.role_admin
DBSession.add(user)
user = User(PROJECT_MANAGER_USER_ID, u'project_manager_user')
user.role = User.role_project_manager
DBSession.add(user)
license = License()
license.name = u'LicenseBar'
license.description = u'the_description_for_license_bar'
license.plain_text = u'the_plain_text_for_license_bar'
DBSession.add(license)
shape = shapely.geometry.Polygon(
[(7.23, 41.25), (7.23, 41.12), (7.41, 41.20)])
geometry = geoalchemy2.shape.from_shape(shape, 4326)
area = Area(geometry)
project = Project(u'test project')
project.area = area
project.auto_fill(12)
project.status = Project.status_published
DBSession.add(project)
transaction.commit()
DBSession.remove()
def populate_db():
import geoalchemy2
import shapely
testing.setUp()
config = ConfigParser.ConfigParser()
config.read(local_settings_path)
db_url = config.get('app:main', 'sqlalchemy.url')
engine = create_engine(db_url)
Base.metadata.drop_all(engine)
Base.metadata.create_all(engine)
DBSession.configure(bind=engine)
# those users are immutables ie. they're not suppose to change during tests
user = User(USER1_ID, u'user1')
DBSession.add(user)
user = User(USER2_ID, u'user2')
DBSession.add(user)
user = User(ADMIN_USER_ID, u'admin_user')
user.role = User.role_admin
DBSession.add(user)
user = User(PROJECT_MANAGER_USER_ID, u'project_manager_user')
user.role = User.role_project_manager
DBSession.add(user)
license = License()
license.name = u'LicenseBar'
license.description = u'the_description_for_license_bar'
license.plain_text = u'the_plain_text_for_license_bar'
DBSession.add(license)
shape = shapely.geometry.Polygon([(7.23, 41.25), (7.23, 41.12),
(7.41, 41.20)])
geometry = geoalchemy2.shape.from_shape(shape, 4326)
area = Area(geometry)
project = Project(u'test project')
project.area = area
project.auto_fill(12)
project.status = Project.status_published
DBSession.add(project)
transaction.commit()
DBSession.remove()
def populate_db():
import geoalchemy2
import shapely
engine = create_engine(db_url)
Base.metadata.drop_all(engine)
Base.metadata.create_all(engine)
DBSession.configure(bind=engine)
# those users are immutables ie. they're not suppose to change during tests
user = User(USER1_ID, u'user1', False)
DBSession.add(user)
user = User(USER2_ID, u'user2', False)
DBSession.add(user)
user = User(ADMIN_USER_ID, u'admin_user', True)
DBSession.add(user)
license = License()
license.name = u'LicenseBar'
license.description = u'the_description_for_license_bar'
license.plain_text = u'the_plain_text_for_license_bar'
DBSession.add(license)
shape = shapely.geometry.Polygon(
[(7.23, 41.25), (7.23, 41.12), (7.41, 41.20)])
geometry = geoalchemy2.shape.from_shape(shape, 4326)
area = Area(geometry)
project = Project(u'test project')
project.area = area
project.auto_fill(12)
DBSession.add(project)
transaction.commit()
DBSession.remove()
def test_license_delete(self):
import transaction
from osmtm.models import DBSession, License
license = License()
DBSession.add(license)
DBSession.flush()
license_id = license.id
transaction.commit()
headers = self.login_as_admin()
self.testapp.get('/license/%d/delete' % license_id,
headers=headers,
status=302)
self.assertEqual(DBSession.query(License).count(), 1)
v1_user = session_v1.query(users_table).filter(
users_table.c.username == username).one()
if v1_user.admin:
user.role = user.role_project_manager
session_v2.add(user)
users_count += 1
session_v2.flush()
success("%d users - successfully imported" % (users_count))
header("Importing licenses")
with transaction.manager:
query = session_v1.query(licenses).all()
for l in query:
license = License()
license.id = l.id
license.name = l.name
license.description = l.description
license.plain_text = l.plain_text
session_v2.add(license)
success("License %s - \"%s\" successfully imported" % (l.id, l.name))
session_v2.flush()
header("Importing jobs and tasks")
for job in session_v1.query(jobs):
with transaction.manager:
geometry = shapely.wkt.loads(job.geometry)
geometry = ST_Transform(shape.from_shape(geometry, 3857), 4326)
user = User(id, username)
v1_user = session_v1.query(users_table).filter(users_table.c.username == username).one()
if v1_user.admin:
user.role = user.role_project_manager
session_v2.add(user)
users_count += 1
session_v2.flush()
success("%d users - successfully imported" % (users_count))
header("Importing licenses")
with transaction.manager:
query = session_v1.query(licenses).all()
for l in query:
license = License()
license.id = l.id
license.name = l.name
license.description = l.description
license.plain_text = l.plain_text
session_v2.add(license)
success("License %s - \"%s\" successfully imported" % (l.id, l.name))
session_v2.flush()
header("Importing jobs and tasks")
for job in session_v1.query(jobs):
with transaction.manager:
geometry = shapely.wkt.loads(job.geometry)
def populate_db():
import geoalchemy2
import shapely
testing.setUp()
config = ConfigParser.ConfigParser()
config.read(local_settings_path)
db_url = config.get('app:main', 'sqlalchemy.url')
engine = create_engine(db_url)
Base.metadata.drop_all(engine)
Base.metadata.create_all(engine)
DBSession.configure(bind=engine)
# those users are immutables ie. they're not suppose to change during tests
user = User(USER1_ID, u'user1')
DBSession.add(user)
user = User(USER2_ID, u'user2')
DBSession.add(user)
user = User(ADMIN_USER_ID, u'admin_user')
user.role = User.role_admin
DBSession.add(user)
user = User(PROJECT_MANAGER_USER_ID, u'project_manager_user')
user.role = User.role_project_manager
DBSession.add(user)
user = User(VALIDATOR_ID, u'user_validator')
user.role = User.role_validator
DBSession.add(user)
user = User(EXPERIENCED_MAPPER_ID, u'user_experienced_mapper')
user.role = User.role_experienced_mapper
DBSession.add(user)
license = License()
license.name = u'LicenseBar'
license.description = u'the_description_for_license_bar'
license.plain_text = u'the_plain_text_for_license_bar'
DBSession.add(license)
label0 = Label()
label0.name = u'bar'
label0.color = u'#ff0000'
DBSession.add(label0)
label1 = Label()
label1.name = u'dude label'
label1.color = u'#ff0000'
DBSession.add(label1)
shape = shapely.geometry.Polygon(
[(7.23, 41.25), (7.23, 41.12), (7.41, 41.20)])
geometry = geoalchemy2.shape.from_shape(shape, 4326)
area = Area(geometry)
project = Project(u'test project')
project.short_description = u'lorem ipsum description'
project.area = area
project.auto_fill(12)
project.status = Project.status_published
project.labels = [label0, label1]
DBSession.add(project)
transaction.commit()
DBSession.remove()