def createuser(self, **kwargs): """ Create user without admin status """ user = User(**kwargs) user.password = kwargs['password'] user.create()
def test_create_user(db): user = User.create("test", "password") assert user.name == "test" assert User.exists("test") with pytest.raises(User.UserExistsError) as excinfo: user = User.create("test", "password") assert "test" in str(excinfo)
def arguments_wrapper(*args, **kwargs): try: changed = False user = User.get(telegram_user_id=args[0].from_user.id) if not user.has_messages_after_notification: user.has_messages_after_notification = True changed = True if user.username != args[0].from_user.username: user.username = args[0].from_user.username changed = True if user.first_name != args[0].from_user.first_name: user.first_name = args[0].from_user.first_name changed = True if user.surname != args[0].from_user.last_name: user.surname = args[0].from_user.last_name changed = True if changed: user.save() except Exception as e: user = User.create( username=args[0].from_user.username, telegram_chat_id=args[0].chat.id, first_name=args[0].from_user.first_name, surname=args[0].from_user.last_name, # phone=args[0].contact.phone_number, telegram_user_id=args[0].from_user.id) obj = tele_func(*args, **kwargs, user=user) return obj
def test_user_id_is_null(self): try: self.user = User.create(user_id=None) except IntegrityError: assert True else: assert False
def test_user_id_is_not_unique(self): try: self.user = User.create(user_id='vk_1234') except IntegrityError: assert True else: assert False
def add_user_(nickname): """ Добавить пользователя :param nickname: никнейм :return: json ответ с полями нового пользователя """ try: with db.atomic(): user = User.create(nickname=nickname, active=True) user.save() user_json = __create_json_auth_user(user) except Exception as e: error_message = str(e).split('\n')[0] user_json = {'error': error_message} return user_json
def setup_class(self): self.user = User.create(user_id='vk_1234')
def setup_class(self): self.user = User.create(user_id='id_1234') self.theory = Theory.create(text='lala', theme='lala') self.readed_theory = ReadedTheory(user=self.user, theory=self.theory)