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)