def setup_class(cls):
from factories.user import UserFactory
super().setup_class()
session = cls.session_factory()
cls.user_password = hash_password('12345678')
active_user = UserFactory.create(email='{}@example.com'.format(
time.time()),
status='ACTIVE',
password=cls.user_password)
session.refresh(active_user)
cls.user_info = to_dict(active_user)
def _create_token(self, obj, payload, **token_extra):
now = datetime.now()
expiry = datetime.now() + timedelta(days=AuthService.DAY_EXPIRE)
payload['iat'] = datetime.timestamp(now)
payload['exp'] = datetime.timestamp(expiry)
token = generate_token(payload)
tokens = obj.tokens or {}
tokens[payload['client_id']] = {
'token': hash_password(token),
'expiry': payload['exp']
}
obj.tokens = tokens
return token
def seed_data():
engine = db_engine()
db_session.configure(bind=engine)
session = db_session()
user_service = UserService(session)
password = '******'
password_hashed = hash_password(password)
with session_scope(session):
user_service.create(email='*****@*****.**',
name='Example',
password=password_hashed,
status='ACTIVE')
def create_user_with_token(self, ui_data):
_user_fields = self._user_fields(ui_data)
user = self.user_service.get_user_by_domain(_user_fields['email'])
if user:
raise UBadRequest('User already exists !!')
_user_fields['password'] = hash_password(_user_fields['password'])
user = self.user_service.create(**_user_fields)
token = self._create_token(
user, {
'sub': user.email,
'client_id': user.id,
'client_type': ClientType.User.value
})
return token, user
def create_admin_am(cls):
password = hash_password('12345678')
session = cls.session_factory
from factories.admin import AdminFactory
am_admin = AdminFactory.create(
email='am_admin_{}@example.com'.format(time.time()),
password=password,
status='ACTIVE',
admin_roles=['AM']
)
session.refresh(am_admin)
cls.am_admin = to_dict(am_admin)
cls.am_admin_token = generate_token({
'admin_id': am_admin.id,
'client_type': 'ADMIN',
'login_at': time.time()
})