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)