from unittest import TestCase
from mainapp import app
from models import db
from models.user_model import UserModel, CardModel, Role


class TestUser(TestCase):
    def test_add(self):
        app.app_context().push()
        user=UserModel()
        user.name='li'
        user.auth_key='dsadas'
        user.phone='31231233131'
        db.session.add(user)
        print(user.name)
        db.session.commit()
    def test_waijianguanxi(self):
        app.app_context().push()
        user=UserModel.query.get(1)
        print(user.name)
        print(user.cards)
        card=CardModel()
        card.number='131230293023903902390923021'
        card.money=0
        card.publisher='健身房'
        db.session.add(card)
        user.cards.append(card)
        db.session.commit()
    def test_addmany(self):
        app.app_context().push()
        li=[]
        for i in range(10):
            user=UserModel()
            user.phone='123123213'+str(i)
            user.name='dsd'+str(i)
            user.auth_key='dsafafjiaj'+str(i)
            li.append(user)
        db.session.add_all(li)
        db.session.commit()

    def test_query1(self):
        app.app_context().push()

        user=UserModel.query.get(1)
        print(user.name)

        # user2=UserModel.query.filter(user.name==UserModel.name.like('dsd%'))
        user2=UserModel.query.filter(UserModel.name.like('dsd%'))
        # print(user2[0].name)
        for i in user2:
            print(i.id,i.name)


    def test_query2(self):
        app.app_context().push()
        user=db.session.query(UserModel).all()
        for i in user:
            print(i.name,i.id)

        user=db.session.query(UserModel.name,UserModel.phone).order_by(UserModel.phone).limit(3)
        for i in user:
            print(i.name,i.phone)


    def test_update(self):

        app.app_context().push()

        user=UserModel.query.get(1)
        user.name='AAA'
        user.phone='213231312'
        user.auth_key='asdsdjasidnio'
        db.session.commit()
    def testremove(self):

        app.app_context().push()

        user=UserModel.query.get(2)
        print(user.name,user.phone)
        db.session.delete(user)
        db.session.commit()


    #N:M

    def test_addmodel2(self):
        app.app_context().push()
        user=UserModel.query.get(1)
        role=Role()
        role.name='超级管理员'
        db.session.add(role)
        user.roles.append(role)
        db.session.commit()

    def test_query(self):
        app.app_context().push()
        user=UserModel.query.get(1)
        print(user.roles)

        role=Role.query.all()[0]
        print(role.users)

    def test_update2(self):
        app.app_context().push()

        user=UserModel.query.get(1)
        user.roles[0].name='普通管理员'
        db.session.commit()


    def test_del2(self):
        app.app_context().push()
        role=Role()
        role.name='实习工'
        db.session.add(role)
        users=UserModel.query.filter(UserModel.name.like('dsd%'))
        for user in users:
            user.roles.append(role)
        db.session.commit()

    def test_del3(self):
        app.app_context().push()

        roles=Role.query.filter(Role.name=='实习工').all()
        for role in roles:
            users=role.users
            for user in users:
                db.session.delete(user)

        db.session.commit()