def factory(params=None, files=None, user=None):
ctx = rest.Context(method=None,
url=None,
headers={},
params=params or {},
files=files or {})
ctx.session = session
ctx.user = user or model.User()
return ctx
def factory(params=None, files=None, user=None, headers=None):
ctx = rest.Context(env={'HTTP_ORIGIN': 'http://example.com'},
method=None,
url=None,
headers=headers or {},
params=params or {},
files=files or {})
ctx.session = session
ctx.user = user or model.User()
return ctx
def factory(name=None, rank=model.User.RANK_REGULAR, email='dummy'):
user = model.User()
user.name = name or get_unique_name()
user.password_salt = 'dummy'
user.password_hash = 'dummy'
user.email = email
user.rank = rank
user.creation_time = datetime(1997, 1, 1)
user.avatar_style = model.User.AVATAR_GRAVATAR
return user
def test_get_avatar_url(
user_name, user_email, avatar_style, expected_url, config_injector):
config_injector({
'data_url': 'http://example.com/',
'thumbnails': {'avatar_width': 100},
})
user = model.User()
user.name = user_name
user.email = user_email
user.avatar_style = avatar_style
assert users.get_avatar_url(user) == expected_url
def create_user(name: str, password: str, email: str) -> model.User:
user = model.User()
update_user_name(user, name)
update_user_password(user, password)
update_user_email(user, email)
if get_user_count() > 0:
user.rank = util.flip(auth.RANK_MAP)[config.config['default_rank']]
else:
user.rank = model.User.RANK_ADMINISTRATOR
user.creation_time = datetime.utcnow()
user.avatar_style = model.User.AVATAR_GRAVATAR
return user
def test_get_avatar_url(user_name, user_email, avatar_style, expected_url,
config_injector):
config_injector({
"data_url": "http://example.com/",
"thumbnails": {
"avatar_width": 100
},
})
user = model.User()
user.name = user_name
user.email = user_email
user.avatar_style = avatar_style
assert users.get_avatar_url(user) == expected_url
def __init__(self,
method: str,
url: str,
headers: Dict[str, str] = None,
params: Request = None,
files: Dict[str, bytes] = None) -> None:
self.method = method
self.url = url
self._headers = headers or {}
self._params = params or {}
self._files = files or {}
self.user = model.User()
self.user.name = None
self.user.rank = 'anonymous'
self.session = None # type: Any
def test_saving_user():
user = model.User()
user.name = "name"
user.password_salt = "salt"
user.password_hash = "hash"
user.email = "email"
user.rank = "rank"
user.creation_time = datetime(1997, 1, 1)
user.avatar_style = model.User.AVATAR_GRAVATAR
db.session.add(user)
db.session.flush()
db.session.refresh(user)
assert not db.session.dirty
assert user.name == "name"
assert user.password_salt == "salt"
assert user.password_hash == "hash"
assert user.email == "email"
assert user.rank == "rank"
assert user.creation_time == datetime(1997, 1, 1)
assert user.avatar_style == model.User.AVATAR_GRAVATAR
def test_saving_user():
user = model.User()
user.name = 'name'
user.password_salt = 'salt'
user.password_hash = 'hash'
user.email = 'email'
user.rank = 'rank'
user.creation_time = datetime(1997, 1, 1)
user.avatar_style = model.User.AVATAR_GRAVATAR
db.session.add(user)
db.session.flush()
db.session.refresh(user)
assert not db.session.dirty
assert user.name == 'name'
assert user.password_salt == 'salt'
assert user.password_hash == 'hash'
assert user.email == 'email'
assert user.rank == 'rank'
assert user.creation_time == datetime(1997, 1, 1)
assert user.avatar_style == model.User.AVATAR_GRAVATAR
def _post_to_webhooks(snapshot: model.Snapshot) -> None:
webhook_user = model.User()
webhook_user.name = None
webhook_user.rank = "anonymous"
net.post_to_webhooks(serialize_snapshot(snapshot, webhook_user))
