import pytest
from buyercall.blueprints.user.models import User
from buyercall.blueprints.block_numbers.models import Block


class TestBlockModel(object):

    @pytest.mark.skip(reason="method not currently used in system")
    def test_block_search(self, blocked_numbers):
        test = ''

    def test_block_create(self, blocked_numbers):
        user = User.find_by_identity('admin@localhost.com')

        block_params = {
            'phonenumber': '+11115551113',
            'status': 'blocked',
            'partnership_accounts_id': user.partnership_account.id,
            'global_exclusion': True
        }

        result = Block.create(block_params)

        if result:
            blocked_number = Block.query\
                .filter(Block.phonenumber == '+11115551113')\
                .first()

            if blocked_number and blocked_number.status == 'blocked' and blocked_number.global_exclusion == True:
                assert True
            else:
                assert False
        else:
            assert False

    def test_block_update_ok(self, blocked_numbers):
        blocked_number = Block.query \
            .filter(Block.phonenumber == '+11115551112') \
            .first()

        result = Block.update(blocked_number.id, blocked_number.partnership_accounts_id, 'test', '+22115551112', False)

        if result:
            blocked_number_updated = Block.query \
                .filter(Block.phonenumber == '+22115551112') \
                .first()

            blocked_number = Block.query \
                .filter(Block.phonenumber == '+11115551112') \
                .first()

            if blocked_number_updated.status == 'blocked' \
                    and blocked_number_updated.global_exclusion == False \
                    and blocked_number is None:
                assert True
            else:
                assert False

    def test_block_update_nok(self, blocked_numbers):
        blocked_number = Block.query \
            .filter(Block.phonenumber == '+11115551112') \
            .first()

        result = Block.update(777, blocked_number.partnership_accounts_id, 'test', '+22115551112', False)

        if not result:
            assert True
        else:
            assert False