def test_company_index(self):
avatar = User.get(3)
self.config.add_route('company', '/company/{cid}')
self.config.add_static_view('static', 'autonomie:static')
request = self.get_csrf_request()
request._user = avatar
request.user = avatar
request.context = avatar.companies[0]
response = company_index(request)
self.assertEqual(avatar.companies[0].name, response['company'].name )
def submit_success(self, appstruct):
"""
Edit the database entry and return redirect
"""
user_id = appstruct.get('user_id')
company = Company()
company = merge_session_with_post(company, appstruct)
if user_id is not None:
user_account = User.get(user_id)
if user_account is not None:
company.employees.append(user_account)
self.dbsession.add(company)
self.dbsession.flush()
message = u"L'entreprise '{0}' a bien été ajoutée".format(company.name)
self.session.flash(message)
return HTTPFound(self.request.route_path("company", id=company.id))
def submit_success(self, appstruct):
"""
Edit the database entry and return redirect
"""
user_id = appstruct.get('user_id')
company = Company()
company.activities = fetch_activities_objects(appstruct)
company = merge_session_with_post(company, appstruct)
if user_id is not None:
user_account = User.get(user_id)
if user_account is not None:
company.employees.append(user_account)
self.dbsession.add(company)
self.dbsession.flush()
message = u"L'entreprise '{0}' a bien été ajoutée".format(company.name)
self.session.flash(message)
return HTTPFound(self.request.route_path("company", id=company.id))
def company_remove_employee_view(context, request):
"""
Enlève un employé de l'entreprise courante
"""
uid = request.params.get("uid")
if not uid:
request.session.flash("Missing uid parameter", "error")
user = User.get(uid)
if not user:
request.session.flash("User not found", "error")
if user in context.employees:
context.employees = [employee for employee in context.employees if employee != user]
request.session.flash(
u"L'utilisateur {0} ne fait plus partie de l'entreprise {1}".format(format_account(user), context.name)
)
url = request.referer
if url is None:
url = request.route_path("company", id=context.id)
return HTTPFound(url)
def company_remove_employee_view(context, request):
"""
Enlève un employé de l'entreprise courante
"""
uid = request.params.get('uid')
if not uid:
request.session.flash('Missing uid parameter', 'error')
user = User.get(uid)
if not user:
request.session.flash('User not found', 'error')
if user in context.employees:
context.employees = [
employee for employee in context.employees if employee != user
]
request.session.flash(
u"L'utilisateur {0} ne fait plus partie de l'entreprise {1}".
format(format_account(user), context.name))
url = request.referer
if url is None:
url = request.route_path('company', id=context.id)
return HTTPFound(url)
def test_get_company(self):
user1 = User.get(3)
cid = 1
company = user1.get_company(cid)
self.assertEqual(company.name, u'Laveur de K-ro')
self.assertRaises(KeyError, user1.get_company, 3000)
