def test_user_services(self): company = CompanyEntity(name='alese.it', domain='alese.it') user = UserEntity(email='*****@*****.**', company_id=1) db_session.add(company) db_session.add(user) db_session.commit() company = services.company.get_for_domain('alese.it') assert company.domain == 'alese.it' user = services.user.get_by_email('*****@*****.**') assert user.email == '*****@*****.**' token = services.user.generate_auth_token(user) assert services.user.verify_token(token)
def company(database): company_entity = CompanyEntity(name='TestCompany', domain='alpha-i.co') database.add(company_entity) database.commit() return company_entity
def get_for_email(email): company = CompanyEntity.get_for_email(email) return Company.from_model(company)
def get_for_domain(domain): company = CompanyEntity.get_for_domain(domain) return Company.from_model(company)
def get_by_id(company_id): company = CompanyEntity.get_by_id(company_id) return Company.from_model(company)