aboutsummaryrefslogtreecommitdiff
path: root/server/szurubooru/tests/api/util.py
blob: 26a1ad4be11ab24fef793cb4a6757b2fb6333830 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
from datetime import datetime
from szurubooru import db

def mock_user(name, rank='admin'):
    user = db.User()
    user.name = name
    user.password = 'dummy'
    user.password_salt = 'dummy'
    user.password_hash = 'dummy'
    user.email = 'dummy'
    user.rank = rank
    user.creation_time = datetime(1997, 1, 1)
    user.avatar_style = db.User.AVATAR_GRAVATAR
    return user

def mock_params(context, params):
    def get_param_as_string(key, default=None):
        if key not in params:
            return default
        return params[key]
    def get_param_as_int(key, default=None):
        if key not in params:
            return default
        return int(params[key])
    context.get_param_as_string = get_param_as_string
    context.get_param_as_int = get_param_as_int

© 2015 - 2026 Jakob L. Kreuze