def setup_method(self, _):
# Always see the diff
# https://docs.python.org/2/library/unittest.html#unittest.TestCase.maxDiff
self.maxDiff = None
from c2cgeoportal_commons.models import DBSession
from c2cgeoportal_commons.models.main import Interface, LayerWMS, RestrictionArea, Role
from c2cgeoportal_commons.models.static import User
setup_db()
user1 = User(username="******", password="******")
role1 = Role(name="__test_role1", description="__test_role1")
user1.roles = [role1]
user1.email = "Tarenpion"
user2 = User(username="******", password="******")
role2 = Role(name="__test_role2", description="__test_role2")
user2.roles = [role2]
user2.email = "Tarenpion"
ogc_server_internal = create_default_ogcserver()
main = Interface(name="main")
layer1 = LayerWMS("layer_1", public=False)
layer1.layer = "layer_1"
layer1.ogc_server = ogc_server_internal
layer1.interfaces = [main]
layer2 = LayerWMS("layer_2", public=False)
layer2.layer = "layer_2"
layer2.ogc_server = ogc_server_internal
layer2.interfaces = [main]
layer3 = LayerWMS("layer_3", public=False)
layer3.layer = "layer_3"
layer3.ogc_server = ogc_server_internal
layer3.interfaces = [main]
area = "POLYGON((-100 30, -100 50, 100 50, 100 30, -100 30))"
area = WKTElement(area, srid=21781)
restricted_area1 = RestrictionArea("__test_ra1",
"", [layer1, layer2], [role1],
area,
readwrite=True)
area = "POLYGON((-100 0, -100 20, 100 20, 100 0, -100 0))"
area = WKTElement(area, srid=21781)
restricted_area2 = RestrictionArea("__test_ra2", "",
[layer1, layer2, layer3], [role2],
area)
DBSession.add_all(
[user1, user2, role1, role2, restricted_area1, restricted_area2])
transaction.commit()
def insert_users_test_data(dbsession):
from c2cgeoportal_commons.models.static import User
from c2cgeoportal_commons.models.main import Role
user = User("babar")
user.roles = [Role(name='Role1'), Role(name='Role2')]
t = dbsession.begin_nested()
dbsession.add(user)
dbsession.flush()
yield
t.rollback()
def setter(request):
del request # Unused
class User:
pass
class Role:
pass
user = User()
user.username = "******"
role = Role()
role.name = "__bar"
user.settings_role = role
user.roles = [role]
return user
def test_add(self, dbsession):
from c2cgeoportal_commons.models.main import Role
from c2cgeoportal_commons.models.static import User, user_role
user = User(username="******")
user.roles = [Role(name="Role3")]
dbsession.begin_nested()
dbsession.add(user)
dbsession.commit()
assert dbsession.query(User).count() == 2, "added a user"
dbsession.expire(user)
assert user.username == "momo", "added user is momo"
assert 1 == len(user.roles)
assert user.roles[0].name == "Role3"
assert 1 == dbsession.query(user_role).filter(
user_role.c.user_id == user.id).count()
