kpsul/kfet/tests/test_views.py

import json
from datetime import datetime, timedelta
from decimal import Decimal
from unittest import mock

from django.contrib.auth.models import Group
from django.core.urlresolvers import reverse
from django.test import Client, TestCase
from django.utils import timezone

from ..config import kfet_config
from ..models import (
    Account, Article, ArticleCategory, Checkout, CheckoutStatement, Inventory,
    InventoryArticle, Operation, OperationGroup, Order, OrderArticle, Supplier,
    SupplierArticle, TransferGroup,
)
from .testcases import ViewTestCaseMixin
from .utils import create_team, create_user, get_perms


class AccountListViewTests(ViewTestCaseMixin, TestCase):
    url_name = 'kfet.account'
    url_expected = '/k-fet/accounts/'

    auth_user = 'team'
    auth_forbidden = [None, 'user']

    def test_ok(self):
        r = self.client.get(self.url)
        self.assertEqual(r.status_code, 200)


class AccountValidFreeTests(ViewTestCaseMixin, TestCase):
    url_name = 'kfet.account.is_validandfree.ajax'
    url_expected = '/k-fet/accounts/is_validandfree'

    auth_user = 'team'
    auth_forbidden = [None, 'user']

    def test_ok_isvalid_isfree(self):
        """Upper case trigramme not taken is valid and free."""
        r = self.client.get(self.url, {'trigramme': 'AAA'})
        self.assertDictEqual(json.loads(r.content.decode('utf-8')), {
            'is_valid': True,
            'is_free': True,
        })

    def test_ok_isvalid_notfree(self):
        """Already taken trigramme is not free, but valid."""
        r = self.client.get(self.url, {'trigramme': '000'})
        self.assertDictEqual(json.loads(r.content.decode('utf-8')), {
            'is_valid': True,
            'is_free': False,
        })

    def test_ok_notvalid_isfree(self):
        """Lower case if forbidden but free."""
        r = self.client.get(self.url, {'trigramme': 'aaa'})
        self.assertDictEqual(json.loads(r.content.decode('utf-8')), {
            'is_valid': False,
            'is_free': True,
        })


class AccountCreateViewTests(ViewTestCaseMixin, TestCase):
    url_name = 'kfet.account.create'
    url_expected = '/k-fet/accounts/new'

    http_methods = ['GET', 'POST']

    auth_user = 'team'
    auth_forbidden = [None, 'user']

    post_data = {
        'trigramme': 'AAA',
        'username': 'plopplopplop',
        'first_name': 'first',
        'last_name': 'last',
        'email': 'email@domain.net',
    }

    def get_users_extra(self):
        return {
            'team1': create_team('team1', '101', perms=['kfet.add_account']),
        }

    def test_get_ok(self):
        r = self.client.get(self.url)
        self.assertEqual(r.status_code, 200)

    def test_post_ok(self):
        client = Client()
        client.login(username='team1', password='team1')

        r = client.post(self.url, self.post_data)
        self.assertRedirects(r, reverse('kfet.account.create'))

        account = Account.objects.get(trigramme='AAA')

        self.assertInstanceExpected(account, {
            'username': 'plopplopplop',
            'first_name': 'first',
            'last_name': 'last',
        })

    def test_post_forbidden(self):
        r = self.client.post(self.url, self.post_data)
        self.assertForbiddenKfet(r)


class AccountCreateAjaxViewTests(ViewTestCaseMixin, TestCase):
    urls_conf = [{
        'name': 'kfet.account.create.fromuser',
        'kwargs': {'username': 'user'},
        'expected': '/k-fet/accounts/new/user/user',
    }, {
        'name': 'kfet.account.create.fromclipper',
        'kwargs': {
            'login_clipper': 'myclipper',
            'fullname': 'first last1 last2',
        },
        'expected': (
            '/k-fet/accounts/new/clipper/myclipper/first%20last1%20last2'
        ),
    }, {
        'name': 'kfet.account.create.empty',
        'expected': '/k-fet/accounts/new/empty',
    }]

    auth_user = 'team'
    auth_forbidden = [None, 'user']

    def test_fromuser(self):
        r = self.client.get(self.t_urls[0])
        self.assertEqual(r.status_code, 200)

        user = self.users['user']

        self.assertEqual(r.context['user_form'].instance, user)
        self.assertEqual(r.context['cof_form'].instance, user.profile)
        self.assertIn('account_form', r.context)

    def test_fromclipper(self):
        r = self.client.get(self.t_urls[1])
        self.assertEqual(r.status_code, 200)

        self.assertIn('user_form', r.context)
        self.assertIn('cof_form', r.context)
        self.assertIn('account_form', r.context)

    def test_empty(self):
        r = self.client.get(self.t_urls[2])
        self.assertEqual(r.status_code, 200)

        self.assertIn('user_form', r.context)
        self.assertIn('cof_form', r.context)
        self.assertIn('account_form', r.context)


class AccountCreateAutocompleteViewTests(ViewTestCaseMixin, TestCase):
    url_name = 'kfet.account.create.autocomplete'
    url_expected = '/k-fet/autocomplete/account_new'

    auth_user = 'team'
    auth_forbidden = [None, 'user']

    def test_ok(self):
        r = self.client.get(self.url, {'q': 'first'})
        self.assertEqual(r.status_code, 200)
        self.assertEqual(len(r.context['users_notcof']), 0)
        self.assertEqual(len(r.context['users_cof']), 0)
        self.assertSetEqual(set(r.context['kfet']), set([
            (self.accounts['user'], self.users['user']),
        ]))


class AccountSearchViewTests(ViewTestCaseMixin, TestCase):
    url_name = 'kfet.account.search.autocomplete'
    url_expected = '/k-fet/autocomplete/account_search'

    auth_user = 'team'
    auth_forbidden = [None, 'user']

    def test_ok(self):
        r = self.client.get(self.url, {'q': 'first'})
        self.assertEqual(r.status_code, 200)
        self.assertSetEqual(set(r.context['accounts']), set([
            ('000', 'first last'),
        ]))


class AccountReadViewTests(ViewTestCaseMixin, TestCase):
    url_name = 'kfet.account.read'
    url_kwargs = {'trigramme': '001'}
    url_expected = '/k-fet/accounts/001'

    auth_user = 'team'
    auth_forbidden = [None, 'user']

    def get_users_extra(self):
        return {
            'user1': create_user('user1', '001'),
        }

    def setUp(self):
        super().setUp()

        user1_acc = self.accounts['user1']
        team_acc = self.accounts['team']

        # Dummy operations and operation groups
        checkout = Checkout.objects.create(
            created_by=team_acc, name="checkout",
            valid_from=timezone.now(),
            valid_to=timezone.now() + timezone.timedelta(days=365)
        )
        opeg_data = [
            (timezone.now(), Decimal('10')),
            (timezone.now() - timezone.timedelta(days=3), Decimal('3')),
        ]
        OperationGroup.objects.bulk_create([
            OperationGroup(
                on_acc=user1_acc, checkout=checkout, at=at, is_cof=False,
                amount=amount
            )
            for (at, amount) in opeg_data
        ])
        self.operation_groups = OperationGroup.objects.order_by("-amount")
        Operation.objects.create(
            group=self.operation_groups[0],
            type=Operation.PURCHASE,
            amount=Decimal('10')
        )
        Operation.objects.create(
            group=self.operation_groups[1],
            type=Operation.PURCHASE,
            amount=Decimal('3')
        )

    def test_ok(self):
        """We can query the "Account - Read" page."""
        r = self.client.get(self.url)
        self.assertEqual(r.status_code, 200)

    def test_ok_self(self):
        client = Client()
        client.login(username='user1', password='user1')
        r = client.get(self.url)
        self.assertEqual(r.status_code, 200)

    def test_json(self):
        r = self.client.get(self.url, {'format': 'json'})

        self.assertEqual(r.status_code, 200)

        content = json.loads(r.content.decode('utf-8'))

        expected = {
            'name': 'first last',
            'trigramme': '001',
            'balance': '0.00',
        }
        self.assertDictContainsSubset(expected, content)

        self.assertSetEqual(set(content.keys()), set([
            'id', 'trigramme', 'first_name', 'last_name', 'name', 'email',
            'is_cof', 'promo', 'balance', 'is_frozen', 'departement',
            'nickname',
        ]))


class AccountUpdateViewTests(ViewTestCaseMixin, TestCase):
    url_name = 'kfet.account.update'
    url_kwargs = {'trigramme': '001'}
    url_expected = '/k-fet/accounts/001/edit'

    http_methods = ['GET', 'POST']

    auth_user = 'team'
    auth_forbidden = [None, 'user']

    post_data = {
        # User
        'first_name': 'The first',
        'last_name': 'The last',
        'email': '',
        # Group
        'groups[]': [],
        # Account
        'trigramme': '051',
        'nickname': '',
        'promo': '',
        # 'is_frozen': not checked
        # Account password
        'pwd1': '',
        'pwd2': '',
    }

    def get_users_extra(self):
        return {
            'user1': create_user('user1', '001'),
            'team1': create_team('team1', '101', perms=[
                'kfet.change_account',
            ]),
        }

    def test_get_ok(self):
        r = self.client.get(self.url)
        self.assertEqual(r.status_code, 200)

    def test_get_ok_self(self):
        client = Client()
        client.login(username='user1', password='user1')
        r = client.get(self.url)
        self.assertEqual(r.status_code, 200)

    def test_post_ok(self):
        client = Client()
        client.login(username='team1', password='team1')

        r = client.post(self.url, self.post_data)
        self.assertRedirects(r, reverse('kfet.account.read', args=['051']))

        self.accounts['user1'].refresh_from_db()
        self.users['user1'].refresh_from_db()

        self.assertInstanceExpected(self.accounts['user1'], {
            'first_name': 'The first',
            'last_name': 'The last',
            'trigramme': '051',
        })

    def test_post_ok_self(self):
        client = Client()
        client.login(username='user1', password='user1')

        post_data = {
            'first_name': 'The first',
            'last_name': 'The last',
        }

        r = client.post(self.url, post_data)
        self.assertRedirects(r, reverse('kfet.account.read', args=['001']))

        self.accounts['user1'].refresh_from_db()
        self.users['user1'].refresh_from_db()

        self.assertInstanceExpected(self.accounts['user1'], {
            'first_name': 'The first',
            'last_name': 'The last',
        })

    def test_post_forbidden(self):
        r = self.client.post(self.url, self.post_data)
        self.assertForbiddenKfet(r)


class AccountGroupListViewTests(ViewTestCaseMixin, TestCase):
    url_name = 'kfet.account.group'
    url_expected = '/k-fet/accounts/groups'

    auth_user = 'team1'
    auth_forbidden = [None, 'user', 'team']

    def get_users_extra(self):
        return {
            'team1': create_team('team1', '101', perms=['kfet.manage_perms']),
        }

    def setUp(self):
        super().setUp()
        self.group1 = Group.objects.create(name='K-Fêt - Group1')
        self.group2 = Group.objects.create(name='K-Fêt - Group2')

    def test_ok(self):
        r = self.client.get(self.url)
        self.assertEqual(r.status_code, 200)

        self.assertQuerysetEqual(
            r.context['groups'],
            map(repr, [self.group1, self.group2]),
            ordered=False,
        )


class AccountGroupCreateViewTests(ViewTestCaseMixin, TestCase):
    url_name = 'kfet.account.group.create'
    url_expected = '/k-fet/accounts/groups/new'

    http_methods = ['GET', 'POST']

    auth_user = 'team1'
    auth_forbidden = [None, 'user', 'team']

    def get_users_extra(self):
        return {
            'team1': create_team('team1', '101', perms=['kfet.manage_perms']),
        }

    @property
    def post_data(self):
        return {
            'name': 'The Group',
            'permissions': [
                str(self.perms['kfet.is_team'].pk),
                str(self.perms['kfet.manage_perms'].pk),
            ],
        }

    def setUp(self):
        super().setUp()
        self.perms = get_perms(
            'kfet.is_team',
            'kfet.manage_perms',
        )

    def test_get_ok(self):
        r = self.client.get(self.url)
        self.assertEqual(r.status_code, 200)

    def test_post_ok(self):
        r = self.client.post(self.url, self.post_data)
        self.assertRedirects(r, reverse('kfet.account.group'))

        group = Group.objects.get(name='K-Fêt The Group')

        self.assertQuerysetEqual(
            group.permissions.all(),
            map(repr, [
                self.perms['kfet.is_team'],
                self.perms['kfet.manage_perms'],
            ]),
            ordered=False,
        )


class AccountGroupUpdateViewTests(ViewTestCaseMixin, TestCase):
    url_name = 'kfet.account.group.update'

    http_methods = ['GET', 'POST']

    auth_user = 'team1'
    auth_forbidden = [None, 'user', 'team']

    @property
    def url_kwargs(self):
        return {'pk': self.group.pk}

    @property
    def url_expected(self):
        return '/k-fet/accounts/groups/{}/edit'.format(self.group.pk)

    def get_users_extra(self):
        return {
            'team1': create_team('team1', '101', perms=['kfet.manage_perms']),
        }

    @property
    def post_data(self):
        return {
            'name': 'The Group',
            'permissions': [
                str(self.perms['kfet.is_team'].pk),
                str(self.perms['kfet.manage_perms'].pk),
            ],
        }

    def setUp(self):
        super().setUp()
        self.perms = get_perms(
            'kfet.is_team',
            'kfet.manage_perms',
        )
        self.group = Group.objects.create(name='K-Fêt - Group')
        self.group.permissions.set(self.perms.values())

    def test_get_ok(self):
        r = self.client.get(self.url)
        self.assertEqual(r.status_code, 200)

    def test_post_ok(self):
        r = self.client.post(self.url, self.post_data)
        self.assertRedirects(r, reverse('kfet.account.group'))

        self.group.refresh_from_db()

        self.assertEqual(self.group.name, 'K-Fêt The Group')
        self.assertQuerysetEqual(
            self.group.permissions.all(),
            map(repr, [
                self.perms['kfet.is_team'],
                self.perms['kfet.manage_perms'],
            ]),
            ordered=False,
        )


class AccountNegativeListViewTests(ViewTestCaseMixin, TestCase):
    url_name = 'kfet.account.negative'
    url_expected = '/k-fet/accounts/negatives'

    auth_user = 'team1'
    auth_forbidden = [None, 'user', 'team']

    def get_users_extra(self):
        return {
            'team1': create_team('team1', '101', perms=['kfet.view_negs']),
        }

    def setUp(self):
        super().setUp()
        account = self.accounts['user']
        account.balance = -5
        account.save()
        account.update_negative()

    def test_ok(self):
        r = self.client.get(self.url)
        self.assertEqual(r.status_code, 200)
        self.assertQuerysetEqual(
            r.context['negatives'],
            map(repr, [self.accounts['user'].negative]),
            ordered=False,
        )


class AccountStatOperationListViewTests(ViewTestCaseMixin, TestCase):
    url_name = 'kfet.account.stat.operation.list'
    url_kwargs = {'trigramme': '001'}
    url_expected = '/k-fet/accounts/001/stat/operations/list'

    auth_user = 'user1'
    auth_forbidden = [None, 'user', 'team']

    def get_users_extra(self):
        return {'user1': create_user('user1', '001')}

    def test_ok(self):
        r = self.client.get(self.url)
        self.assertEqual(r.status_code, 200)

        content = json.loads(r.content.decode('utf-8'))

        base_url = reverse('kfet.account.stat.operation', args=['001'])

        expected_stats = [{
            'label': 'Derniers mois',
            'url': {
                'path': base_url,
                'query': {
                    'scale_n_steps': ['7'],
                    'scale_name': ['month'],
                    'types': ["['purchase']"],
                    'scale_last': ['True'],
                },
            },
        }, {
            'label': 'Dernières semaines',
            'url': {
                'path': base_url,
                'query': {
                    'scale_n_steps': ['7'],
                    'scale_name': ['week'],
                    'types': ["['purchase']"],
                    'scale_last': ['True'],
                },
            },
        }, {
            'label': 'Derniers jours',
            'url': {
                'path': base_url,
                'query': {
                    'scale_n_steps': ['7'],
                    'scale_name': ['day'],
                    'types': ["['purchase']"],
                    'scale_last': ['True'],
                },
            },
        }]

        for stat, expected in zip(content['stats'], expected_stats):
            expected_url = expected.pop('url')
            self.assertUrlsEqual(stat['url'], expected_url)
            self.assertDictContainsSubset(expected, stat)


class AccountStatOperationViewTests(ViewTestCaseMixin, TestCase):
    url_name = 'kfet.account.stat.operation'
    url_kwargs = {'trigramme': '001'}
    url_expected = '/k-fet/accounts/001/stat/operations'

    auth_user = 'user1'
    auth_forbidden = [None, 'user', 'team']

    def get_users_extra(self):
        return {'user1': create_user('user1', '001')}

    def test_ok(self):
        r = self.client.get(self.url)
        self.assertEqual(r.status_code, 200)


class AccountStatBalanceListViewTests(ViewTestCaseMixin, TestCase):
    url_name = 'kfet.account.stat.balance.list'
    url_kwargs = {'trigramme': '001'}
    url_expected = '/k-fet/accounts/001/stat/balance/list'

    auth_user = 'user1'
    auth_forbidden = [None, 'user', 'team']

    def get_users_extra(self):
        return {'user1': create_user('user1', '001')}

    def test_ok(self):
        r = self.client.get(self.url)
        self.assertEqual(r.status_code, 200)

        content = json.loads(r.content.decode('utf-8'))

        base_url = reverse('kfet.account.stat.balance', args=['001'])

        expected_stats = [{
            'label': 'Tout le temps',
            'url': base_url,
        }, {
            'label': '1 an',
            'url': {
                'path': base_url,
                'query': {'last_days': ['365']},
            },
        }, {
            'label': '6 mois',
            'url': {
                'path': base_url,
                'query': {'last_days': ['183']},
            },
        }, {
            'label': '3 mois',
            'url': {
                'path': base_url,
                'query': {'last_days': ['90']},
            },
        }, {
            'label': '30 jours',
            'url': {
                'path': base_url,
                'query': {'last_days': ['30']},
            },
        }]

        for stat, expected in zip(content['stats'], expected_stats):
            expected_url = expected.pop('url')
            self.assertUrlsEqual(stat['url'], expected_url)
            self.assertDictContainsSubset(expected, stat)


class AccountStatBalanceViewTests(ViewTestCaseMixin, TestCase):
    url_name = 'kfet.account.stat.balance'
    url_kwargs = {'trigramme': '001'}
    url_expected = '/k-fet/accounts/001/stat/balance'

    auth_user = 'user1'
    auth_forbidden = [None, 'user', 'team']

    def get_users_extra(self):
        return {'user1': create_user('user1', '001')}

    def test_ok(self):
        r = self.client.get(self.url)
        self.assertEqual(r.status_code, 200)


class CheckoutListViewTests(ViewTestCaseMixin, TestCase):
    url_name = 'kfet.checkout'
    url_expected = '/k-fet/checkouts/'

    auth_user = 'team'
    auth_forbidden = [None, 'user']

    def setUp(self):
        super().setUp()
        self.checkout1 = Checkout.objects.create(
            name='Checkout 1',
            created_by=self.accounts['team'],
            valid_from=self.now,
            valid_to=self.now + timedelta(days=5),
        )
        self.checkout2 = Checkout.objects.create(
            name='Checkout 2',
            created_by=self.accounts['team'],
            valid_from=self.now + timedelta(days=10),
            valid_to=self.now + timedelta(days=15),
        )

    def test_ok(self):
        r = self.client.get(self.url)
        self.assertEqual(r.status_code, 200)
        self.assertQuerysetEqual(
            r.context['checkouts'],
            map(repr, [self.checkout1, self.checkout2]),
            ordered=False,
        )


class CheckoutCreateViewTests(ViewTestCaseMixin, TestCase):
    url_name = 'kfet.checkout.create'
    url_expected = '/k-fet/checkouts/new'

    http_methods = ['GET', 'POST']

    auth_user = 'team'
    auth_forbidden = [None, 'user']

    post_data = {
        'name': 'Checkout',
        'valid_from': '2017-10-08 17:45:00',
        'valid_to': '2017-11-08 16:00:00',
        'balance': '3.14',
        # 'is_protected': not checked
    }

    def get_users_extra(self):
        return {
            'team1': create_team('team1', '101', perms=['kfet.add_checkout']),
        }

    def test_get_ok(self):
        r = self.client.get(self.url)
        self.assertEqual(r.status_code, 200)

    def test_post_ok(self):
        client = Client()
        client.login(username='team1', password='team1')

        r = client.post(self.url, self.post_data)

        checkout = Checkout.objects.get(name='Checkout')
        self.assertRedirects(r, checkout.get_absolute_url())

        self.assertInstanceExpected(checkout, {
            'name': 'Checkout',
            'valid_from': timezone.make_aware(datetime(2017, 10, 8, 17, 45)),
            'valid_to': timezone.make_aware(datetime(2017, 11, 8, 16, 00)),
            'balance': Decimal('3.14'),
            'is_protected': False,
        })

    def test_post_forbidden(self):
        r = self.client.post(self.url, self.post_data)
        self.assertForbiddenKfet(r)


class CheckoutReadViewTests(ViewTestCaseMixin, TestCase):
    url_name = 'kfet.checkout.read'

    auth_user = 'team'
    auth_forbidden = [None, 'user']

    @property
    def url_kwargs(self):
        return {'pk': self.checkout.pk}

    @property
    def url_expected(self):
        return '/k-fet/checkouts/{}'.format(self.checkout.pk)

    def setUp(self):
        super().setUp()

        self.checkout = Checkout.objects.create(
            name='Checkout',
            created_by=self.accounts['team'],
            valid_from=datetime(2018, 1, 12, 16, 20, tzinfo=timezone.utc),
            valid_to=datetime(2018, 1, 17, 16, 20, tzinfo=timezone.utc),
        )

        # TODO: This statement is currently created by CheckoutCreate view, it
        # should rather be done in Checkout.save().
        with mock.patch('django.utils.timezone.now') as mock_now:
            mock_now.return_value = (
                datetime(2018, 1, 12, 16, 20, tzinfo=timezone.utc))
            self.checkout.statements.create(
                balance_new=0, balance_old=0, amount_taken=0,
                by=self.accounts['team'],
            )

    def test_ok(self):
        r = self.client.get(self.url)
        self.assertEqual(r.status_code, 200)
        self.assertEqual(r.context['checkout'], self.checkout)

    def test_json(self):
        r = self.client.get(self.url, {
            'format': 'json',
            'last_statement': '1',
        })

        self.assertEqual(r.status_code, 200)
        content = json.loads(r.content.decode('utf-8'))

        self.assertEqual(content, {
            'id': self.checkout.pk,
            'name': 'Checkout',
            'balance': '0.00',
            'valid_from': '2018-01-12T16:20:00Z',
            'valid_to': '2018-01-17T16:20:00Z',
            'laststatement': {
                'id': self.checkout.statements.all()[0].pk,
                'at': '2018-01-12T16:20:00Z',
                'balance_new': '0.00',
                'balance_old': '0.00',
                'by': str(self.accounts['team']),
            },
        })


class CheckoutUpdateViewTests(ViewTestCaseMixin, TestCase):
    url_name = 'kfet.checkout.update'

    http_methods = ['GET', 'POST']

    auth_user = 'team'
    auth_forbidden = [None, 'user']

    post_data = {
        'name': 'Checkout updated',
        'valid_from': '2018-01-01 08:00:00',
        'valid_to': '2018-07-01 16:00:00',
    }

    @property
    def url_kwargs(self):
        return {'pk': self.checkout.pk}

    @property
    def url_expected(self):
        return '/k-fet/checkouts/{}/edit'.format(self.checkout.pk)

    def get_users_extra(self):
        return {
            'team1': create_team('team1', '101', perms=[
                'kfet.change_checkout',
            ]),
        }

    def setUp(self):
        super().setUp()
        self.checkout = Checkout.objects.create(
            name='Checkout',
            valid_from=self.now,
            valid_to=self.now + timedelta(days=5),
            balance='3.14',
            is_protected=False,
            created_by=self.accounts['team'],
        )

    def test_get_ok(self):
        r = self.client.get(self.url)
        self.assertEqual(r.status_code, 200)

    def test_post_ok(self):
        client = Client()
        client.login(username='team1', password='team1')

        r = client.post(self.url, self.post_data)
        self.assertRedirects(r, self.checkout.get_absolute_url())

        self.checkout.refresh_from_db()

        self.assertInstanceExpected(self.checkout, {
            'name': 'Checkout updated',
            'valid_from': timezone.make_aware(datetime(2018, 1, 1, 8, 0, 0)),
            'valid_to': timezone.make_aware(datetime(2018, 7, 1, 16, 0, 0)),
        })

    def test_post_forbidden(self):
        r = self.client.post(self.url, self.post_data)
        self.assertForbiddenKfet(r)


class CheckoutStatementListViewTests(ViewTestCaseMixin, TestCase):
    url_name = 'kfet.checkoutstatement'
    url_expected = '/k-fet/checkouts/statements/'

    auth_user = 'team'
    auth_forbidden = [None, 'user']

    def setUp(self):
        super().setUp()
        self.checkout1 = Checkout.objects.create(
            created_by=self.accounts['team'],
            name='Checkout 1',
            valid_from=self.now,
            valid_to=self.now + timedelta(days=5),
        )
        self.checkout2 = Checkout.objects.create(
            created_by=self.accounts['team'],
            name='Checkout 2',
            valid_from=self.now + timedelta(days=10),
            valid_to=self.now + timedelta(days=15),
        )
        self.statement1 = CheckoutStatement.objects.create(
            checkout=self.checkout1,
            by=self.accounts['team'],
            balance_old=5,
            balance_new=0,
            amount_taken=5,
        )

    def test_ok(self):
        r = self.client.get(self.url)
        self.assertEqual(r.status_code, 200)

        expected_statements = (
            list(self.checkout1.statements.all()) +
            list(self.checkout2.statements.all())
        )

        self.assertQuerysetEqual(
            r.context['checkoutstatements'],
            map(repr, expected_statements),
        )


class CheckoutStatementCreateViewTests(ViewTestCaseMixin, TestCase):
    url_name = 'kfet.checkoutstatement.create'

    http_methods = ['GET', 'POST']

    auth_user = 'team'
    auth_forbidden = [None, 'user']

    post_data = {
        # Let
        'balance_001': 0, 'balance_002': 0, 'balance_005': 0,
        'balance_01': 0, 'balance_02': 0, 'balance_05': 0,
        'balance_1': 1, 'balance_2': 0, 'balance_5': 0,
        'balance_10': 1, 'balance_20': 0, 'balance_50': 0,
        'balance_100': 1, 'balance_200': 0, 'balance_500': 0,
        # Taken
        'taken_001': 0, 'taken_002': 0, 'taken_005': 0,
        'taken_01': 0, 'taken_02': 0, 'taken_05': 0,
        'taken_1': 2, 'taken_2': 0, 'taken_5': 0,
        'taken_10': 2, 'taken_20': 0, 'taken_50': 0,
        'taken_100': 2, 'taken_200': 0, 'taken_500': 0,
        'taken_cheque': 0,
        # 'not_count': not checked
    }

    @property
    def url_kwargs(self):
        return {'pk_checkout': self.checkout.pk}

    @property
    def url_expected(self):
        return '/k-fet/checkouts/{}/statements/add'.format(self.checkout.pk)

    def get_users_extra(self):
        return {
            'team1': create_team('team1', '001', perms=[
                'kfet.add_checkoutstatement',
            ]),
        }

    def setUp(self):
        super().setUp()
        self.checkout = Checkout.objects.create(
            name='Checkout',
            created_by=self.accounts['team'],
            balance=5,
            valid_from=self.now,
            valid_to=self.now + timedelta(days=5),
        )

    def test_get_ok(self):
        r = self.client.get(self.url)
        self.assertEqual(r.status_code, 200)

    @mock.patch('django.utils.timezone.now')
    def test_post_ok(self, mock_now):
        self.now += timedelta(days=2)
        mock_now.return_value = self.now

        client = Client()
        client.login(username='team1', password='team1')

        r = client.post(self.url, self.post_data)
        self.assertRedirects(r, self.checkout.get_absolute_url())

        statement = CheckoutStatement.objects.get(at=self.now)

        self.assertInstanceExpected(statement, {
            'by': self.accounts['team1'],
            'checkout': self.checkout,
            'balance_old': Decimal('5'),
            'balance_new': Decimal('111'),
            'amount_taken': Decimal('222'),
            'amount_error': Decimal('328'),
            'at': self.now,
            'not_count': False,
        })

    def test_post_forbidden(self):
        r = self.client.post(self.url, self.post_data)
        self.assertForbiddenKfet(r)


class CheckoutStatementUpdateViewTests(ViewTestCaseMixin, TestCase):
    url_name = 'kfet.checkoutstatement.update'

    http_methods = ['GET', 'POST']

    auth_user = 'team'
    auth_forbidden = [None, 'user']

    post_data = {
        'amount_taken': 3,
        'amount_error': 2,
        'balance_old': 8,
        'balance_new': 5,
        # Taken
        'taken_001': 0, 'taken_002': 0, 'taken_005': 0,
        'taken_01': 0, 'taken_02': 0, 'taken_05': 0,
        'taken_1': 1, 'taken_2': 1, 'taken_5': 0,
        'taken_10': 0, 'taken_20': 0, 'taken_50': 0,
        'taken_100': 0, 'taken_200': 0, 'taken_500': 0,
        'taken_cheque': 0,
    }

    @property
    def url_kwargs(self):
        return {
            'pk_checkout': self.checkout.pk,
            'pk': self.statement.pk,
        }

    @property
    def url_expected(self):
        return '/k-fet/checkouts/{pk_checkout}/statements/{pk}/edit'.format(
            pk_checkout=self.checkout.pk,
            pk=self.statement.pk,
        )

    def get_users_extra(self):
        return {
            'team1': create_team('team1', '101', perms=[
                'kfet.change_checkoutstatement',
            ]),
        }

    def setUp(self):
        super().setUp()
        self.checkout = Checkout.objects.create(
            name='Checkout',
            created_by=self.accounts['team'],
            balance=5,
            valid_from=self.now,
            valid_to=self.now + timedelta(days=5),
        )
        self.statement = CheckoutStatement.objects.create(
            by=self.accounts['team'],
            checkout=self.checkout,
            balance_new=5,
            balance_old=8,
            amount_error=2,
            amount_taken=5,
        )

    def test_get_ok(self):
        r = self.client.get(self.url)
        self.assertEqual(r.status_code, 200)

    @mock.patch('django.utils.timezone.now')
    def test_post_ok(self, mock_now):
        self.now += timedelta(days=2)
        mock_now.return_value = self.now

        client = Client()
        client.login(username='team1', password='team1')

        r = client.post(self.url, self.post_data)
        self.assertRedirects(r, self.checkout.get_absolute_url())

        self.statement.refresh_from_db()

        self.assertInstanceExpected(self.statement, {
            'taken_1': 1,
            'taken_2': 1,
            'balance_new': 5,
            'balance_old': 8,
            'amount_error': 0,
            'amount_taken': 3,
        })

    def test_post_forbidden(self):
        r = self.client.post(self.url, self.post_data)
        self.assertForbiddenKfet(r)


class ArticleCategoryListViewTests(ViewTestCaseMixin, TestCase):
    url_name = 'kfet.category'
    url_expected = '/k-fet/categories/'

    auth_user = 'team'
    auth_forbidden = [None, 'user']

    def setUp(self):
        super().setUp()
        self.category1 = ArticleCategory.objects.create(name='Category 1')
        self.category2 = ArticleCategory.objects.create(name='Category 2')

    def test_ok(self):
        r = self.client.get(self.url)
        self.assertEqual(r.status_code, 200)

        self.assertQuerysetEqual(
            r.context['categories'],
            map(repr, [self.category1, self.category2]),
        )


class ArticleCategoryUpdateViewTests(ViewTestCaseMixin, TestCase):
    url_name = 'kfet.category.update'

    http_methods = ['GET', 'POST']

    auth_user = 'team'
    auth_forbidden = [None, 'user']

    @property
    def url_kwargs(self):
        return {'pk': self.category.pk}

    @property
    def url_expected(self):
        return '/k-fet/categories/{}/edit'.format(self.category.pk)

    def get_users_extra(self):
        return {
            'team1': create_team('team1', '101', perms=[
                'kfet.change_articlecategory',
            ]),
        }

    @property
    def post_data(self):
        return {
            'name': 'The Category',
            # 'has_addcost': not checked
        }

    def setUp(self):
        super().setUp()
        self.category = ArticleCategory.objects.create(name='Category')

    def test_get_ok(self):
        r = self.client.get(self.url)
        self.assertEqual(r.status_code, 200)

    def test_post_ok(self):
        client = Client()
        client.login(username='team1', password='team1')

        r = client.post(self.url, self.post_data)
        self.assertRedirects(r, reverse('kfet.category'))

        self.category.refresh_from_db()

        self.assertInstanceExpected(self.category, {
            'name': 'The Category',
            'has_addcost': False,
        })

    def test_post_forbidden(self):
        r = self.client.post(self.url, self.post_data)
        self.assertForbiddenKfet(r)


class ArticleListViewTests(ViewTestCaseMixin, TestCase):
    url_name = 'kfet.article'
    url_expected = '/k-fet/articles/'

    auth_user = 'team'
    auth_forbidden = [None, 'user']

    def setUp(self):
        super().setUp()
        category = ArticleCategory.objects.create(name='Category')
        self.article1 = Article.objects.create(
            name='Article 1',
            category=category,
        )
        self.article2 = Article.objects.create(
            name='Article 2',
            category=category,
        )

    def test_ok(self):
        r = self.client.get(self.url)
        self.assertEqual(r.status_code, 200)
        self.assertQuerysetEqual(
            r.context['articles'],
            map(repr, [self.article1, self.article2]),
        )


class ArticleCreateViewTests(ViewTestCaseMixin, TestCase):
    url_name = 'kfet.article.create'
    url_expected = '/k-fet/articles/new'

    http_methods = ['GET', 'POST']

    auth_user = 'team'
    auth_forbidden = [None, 'user']

    def get_users_extra(self):
        return {
            'team1': create_team('team1', '101', perms=['kfet.add_article']),
        }

    @property
    def post_data(self):
        return {
            'name': 'Article',
            'category': self.category.pk,
            'stock': 5,
            'price': '2.5',
        }

    def setUp(self):
        super().setUp()
        self.category = ArticleCategory.objects.create(name='Category')

    def test_get_ok(self):
        r = self.client.get(self.url)
        self.assertEqual(r.status_code, 200)

    def test_post_ok(self):
        client = Client()
        client.login(username='team1', password='team1')

        r = client.post(self.url, self.post_data)

        article = Article.objects.get(name='Article')

        self.assertRedirects(r, article.get_absolute_url())

        self.assertInstanceExpected(article, {
            'name': 'Article',
            'category': self.category,
        })

    def test_post_forbidden(self):
        r = self.client.post(self.url, self.post_data)
        self.assertForbiddenKfet(r)


class ArticleReadViewTests(ViewTestCaseMixin, TestCase):
    url_name = 'kfet.article.read'

    auth_user = 'team'
    auth_forbidden = [None, 'user']

    @property
    def url_kwargs(self):
        return {'pk': self.article.pk}

    @property
    def url_expected(self):
        return '/k-fet/articles/{}'.format(self.article.pk)

    def setUp(self):
        super().setUp()
        self.article = Article.objects.create(
            name='Article',
            category=ArticleCategory.objects.create(name='Category'),
            stock=5,
            price=Decimal('2.5'),
        )

    def test_ok(self):
        r = self.client.get(self.url)
        self.assertEqual(r.status_code, 200)
        self.assertEqual(r.context['article'], self.article)


class ArticleUpdateViewTests(ViewTestCaseMixin, TestCase):
    url_name = 'kfet.article.update'

    http_methods = ['GET', 'POST']

    auth_user = 'team'
    auth_forbidden = [None, 'user']

    @property
    def url_kwargs(self):
        return {'pk': self.article.pk}

    @property
    def url_expected(self):
        return '/k-fet/articles/{}/edit'.format(self.article.pk)

    def get_users_extra(self):
        return {
            'team1': create_team('team1', '101', perms=[
                'kfet.change_article',
            ]),
        }

    @property
    def post_data(self):
        return {
            'name': 'The Article',
            'category': self.article.category.pk,
            'is_sold': '1',
            'price': '3.5',
            'box_type': 'carton',
            # 'hidden': not checked
        }

    def setUp(self):
        super().setUp()
        self.category = ArticleCategory.objects.create(name='Category')
        self.article = Article.objects.create(
            name='Article',
            category=self.category,
            stock=5,
            price=Decimal('2.5'),
        )

    def test_get_ok(self):
        r = self.client.get(self.url)
        self.assertEqual(r.status_code, 200)

    def test_post_ok(self):
        client = Client()
        client.login(username='team1', password='team1')

        r = client.post(self.url, self.post_data)

        self.assertRedirects(r, self.article.get_absolute_url())

        self.article.refresh_from_db()

        self.assertInstanceExpected(self.article, {
            'name': 'The Article',
            'price': Decimal('3.5'),
        })

    def test_post_forbidden(self):
        r = self.client.post(self.url, self.post_data)
        self.assertForbiddenKfet(r)


class ArticleStatSalesListViewTests(ViewTestCaseMixin, TestCase):
    url_name = 'kfet.article.stat.sales.list'

    auth_user = 'team'
    auth_forbidden = [None, 'user']

    @property
    def url_kwargs(self):
        return {'pk': self.article.pk}

    @property
    def url_expected(self):
        return '/k-fet/articles/{}/stat/sales/list'.format(self.article.pk)

    def setUp(self):
        super().setUp()
        self.article = Article.objects.create(
            name='Article',
            category=ArticleCategory.objects.create(name='Category'),
        )

    def test_ok(self):
        r = self.client.get(self.url)
        self.assertEqual(r.status_code, 200)

        content = json.loads(r.content.decode('utf-8'))

        base_url = reverse('kfet.article.stat.sales', args=[self.article.pk])

        expected_stats = [
            {
                'label': 'Derniers mois',
                'url': {
                    'path': base_url,
                    'query': {
                        'scale_n_steps': ['7'],
                        'scale_name': ['month'],
                        'scale_last': ['True'],
                    },
                },
            },
            {
                'label': 'Dernières semaines',
                'url': {
                    'path': base_url,
                    'query': {
                        'scale_n_steps': ['7'],
                        'scale_name': ['week'],
                        'scale_last': ['True'],
                    },
                },
            },
            {
                'label': 'Derniers jours',
                'url': {
                    'path': base_url,
                    'query': {
                        'scale_n_steps': ['7'],
                        'scale_name': ['day'],
                        'scale_last': ['True'],
                    },
                },
            },
        ]

        for stat, expected in zip(content['stats'], expected_stats):
            expected_url = expected.pop('url')
            self.assertUrlsEqual(stat['url'], expected_url)
            self.assertDictContainsSubset(expected, stat)


class ArticleStatSalesViewTests(ViewTestCaseMixin, TestCase):
    url_name = 'kfet.article.stat.sales'

    auth_user = 'team'
    auth_forbidden = [None, 'user']

    @property
    def url_kwargs(self):
        return {'pk': self.article.pk}

    @property
    def url_expected(self):
        return '/k-fet/articles/{}/stat/sales'.format(self.article.pk)

    def setUp(self):
        super().setUp()
        self.article = Article.objects.create(
            name='Article',
            category=ArticleCategory.objects.create(name='Category'),
        )

    def test_ok(self):
        r = self.client.get(self.url)
        self.assertEqual(r.status_code, 200)


class KPsulViewTests(ViewTestCaseMixin, TestCase):
    url_name = 'kfet.kpsul'
    url_expected = '/k-fet/k-psul/'

    auth_user = 'team'
    auth_forbidden = [None, 'user']

    def test_ok(self):
        r = self.client.get(self.url)
        self.assertEqual(r.status_code, 200)


class KPsulPerformOperationsViewTests(ViewTestCaseMixin, TestCase):
    url_name = 'kfet.kpsul.perform_operations'
    url_expected = '/k-fet/k-psul/perform_operations'

    http_methods = ['POST']

    auth_user = 'team'
    auth_forbidden = [None, 'user']

    def test_ok(self):
        pass


class KPsulCancelOperationsViewTests(ViewTestCaseMixin, TestCase):
    url_name = 'kfet.kpsul.cancel_operations'
    url_expected = '/k-fet/k-psul/cancel_operations'

    http_methods = ['POST']

    auth_user = 'team1'
    auth_forbidden = [None, 'user']

    def get_users_extra(self):
        return {
            'team1': create_team('team1', '101', perms=[
                'kfet.perform_negative_operations',
            ]),
            'u1': create_user('user1', '001'),
            'u2': create_user('user2', '002'),
            'u3': create_user('user3', '003'),
        }

    def setUp(self):
        super().setUp()

        trg1 = TransferGroup.objects.create()
        self.trg1_1 = trg1.transfers.create(
            from_acc=self.accounts['u1'],
            to_acc=self.accounts['u2'],
            amount='3.5',
        )
        self.trg1_2 = trg1.transfers.create(
            from_acc=self.accounts['u2'],
            to_acc=self.accounts['u3'],
            amount='2.4',
        )

        trg2 = TransferGroup.objects.create(
            at=(
                timezone.now() -
                (kfet_config.cancel_duration + timedelta(seconds=1))
            ),
        )
        self.trg2_1 = trg2.transfers.create(
            from_acc=self.accounts['u1'],
            to_acc=self.accounts['u2'],
            amount='5',
        )

    def test_ok(self):
        data = {
            'transfers[]': [str(self.trg1_1.pk), str(self.trg1_2.pk)],
        }

        r = self.client.post(self.url, data)

        self.assertEqual(r.status_code, 200)

        u1 = self.accounts['u1']
        u1.refresh_from_db()
        self.assertEqual(u1.balance, Decimal('3.5'))

        u2 = self.accounts['u2']
        u2.refresh_from_db()
        self.assertEqual(u2.balance, Decimal('-1.1'))

        u3 = self.accounts['u3']
        u3.refresh_from_db()
        self.assertEqual(u3.balance, Decimal('-2.4'))

    def test_error_tooold(self):
        data = {
                'transfers[]': [str(self.trg2_1.pk)],
        }

        r = self.client.post(self.url, data)

        self.assertEqual(r.status_code, 403)
        self.assertDictEqual(json.loads(r.content.decode('utf-8')), {
            'canceled': {},
            'warnings': {},
            'errors': {
                'missing_perms': ['Annuler des commandes non récentes'],
            },
        })


class KPsulArticlesData(ViewTestCaseMixin, TestCase):
    url_name = 'kfet.kpsul.articles_data'
    url_expected = '/k-fet/k-psul/articles_data'

    auth_user = 'team'
    auth_forbidden = [None, 'user']

    def setUp(self):
        super().setUp()
        self.ac = ArticleCategory.objects.create(name='Catégorie')

        self.a1 = self.ac.articles.create(name='Article 1')
        self.a2 = self.ac.articles.create(name='Article 2', price='2.5')

    def test_ok(self):
        r = self.client.get(self.url)
        self.assertEqual(r.status_code, 200)

        content = json.loads(r.content.decode('utf-8'))

        self.assertEqual(content, {
            'objects': {
                'article': [
                    {
                        'id': self.a1.pk,
                        'name': 'Article 1',
                        'price': '0.00',
                        'stock': 0,
                        'category__id': self.a1.category.pk,
                    },
                    {
                        'id': self.a2.pk,
                        'name': 'Article 2',
                        'price': '2.50',
                        'stock': 0,
                        'category__id': self.a2.category.pk,
                    },
                ],
            },
            'related': {
                'category': [
                    {
                        'id': self.ac.pk,
                        'name': 'Catégorie',
                        'has_addcost': True,
                    },
                ],
            },
        })


class KPsulUpdateAddcost(ViewTestCaseMixin, TestCase):
    url_name = 'kfet.kpsul.update_addcost'
    url_expected = '/k-fet/k-psul/update_addcost'

    http_methods = ['POST']

    auth_user = 'team'
    auth_forbidden = [None, 'user']

    post_data = {
        'trigramme': '000',
        'amount': '0.5',
    }

    def get_users_extra(self):
        return {
            'team1': create_team('team1', '101', perms=[
                'kfet.manage_addcosts',
            ]),
        }

    def test_ok(self):
        client = Client()
        client.login(username='team1', password='team1')

        r = client.post(self.url, self.post_data)
        self.assertEqual(r.status_code, 200)

        self.assertEqual(
            kfet_config.addcost_for,
            Account.objects.get(trigramme='000'),
        )
        self.assertEqual(kfet_config.addcost_amount, Decimal('0.5'))

    def test_post_forbidden(self):
        r = self.client.post(self.url, self.post_data)
        self.assertForbidden(r)


class KPsulGetSettings(ViewTestCaseMixin, TestCase):
    url_name = 'kfet.kpsul.get_settings'
    url_expected = '/k-fet/k-psul/get_settings'

    auth_user = 'team'
    auth_forbidden = [None, 'user']

    def test_ok(self):
        r = self.client.get(self.url)
        self.assertEqual(r.status_code, 200)


class HistoryJSONViewTests(ViewTestCaseMixin, TestCase):
    url_name = 'kfet.history.json'
    url_expected = '/k-fet/history.json'

    auth_user = 'user'
    auth_forbidden = [None]

    def test_ok(self):
        r = self.client.post(self.url)
        self.assertEqual(r.status_code, 200)


class SettingsListViewTests(ViewTestCaseMixin, TestCase):
    url_name = 'kfet.settings'
    url_expected = '/k-fet/settings/'

    auth_user = 'team1'
    auth_forbidden = [None, 'user', 'team']

    def get_users_extra(self):
        return {
            'team1': create_team('team1', '101', perms=[
                'kfet.see_config',
            ]),
        }

    def test_ok(self):
        r = self.client.get(self.url)
        self.assertEqual(r.status_code, 200)


class SettingsUpdateViewTests(ViewTestCaseMixin, TestCase):
    url_name = 'kfet.settings.update'
    url_expected = '/k-fet/settings/edit'

    http_methods = ['GET', 'POST']

    auth_user = 'team1'
    auth_forbidden = [None, 'user', 'team']

    @property
    def post_data(self):
        return {
            'kfet_reduction_cof': '25',
            'kfet_addcost_amount': '0.5',
            'kfet_addcost_for': self.accounts['user'].pk,
            'kfet_overdraft_duration': '2 00:00:00',
            'kfet_overdraft_amount': '25',
            'kfet_cancel_duration': '00:20:00',
        }

    def get_users_extra(self):
        return {
            'team1': create_team('team1', '101', perms=[
                'kfet.change_config',
            ]),
        }

    def test_get_ok(self):
        r = self.client.get(self.url)
        self.assertEqual(r.status_code, 200)

    def test_post_ok(self):
        r = self.client.post(self.url, self.post_data)
        # Redirect is skipped because client may lack permissions.
        self.assertRedirects(
            r,
            reverse('kfet.settings'),
            fetch_redirect_response=False,
        )

        expected_config = {
            'reduction_cof': Decimal('25'),
            'addcost_amount': Decimal('0.5'),
            'addcost_for': self.accounts['user'],
            'overdraft_duration': timedelta(days=2),
            'overdraft_amount': Decimal('25'),
            'cancel_duration': timedelta(minutes=20),
        }

        for key, expected in expected_config.items():
            self.assertEqual(getattr(kfet_config, key), expected)


class TransferListViewTests(ViewTestCaseMixin, TestCase):
    url_name = 'kfet.transfers'
    url_expected = '/k-fet/transfers/'

    auth_user = 'team'
    auth_forbidden = [None, 'user']

    def test_ok(self):
        r = self.client.get(self.url)
        self.assertEqual(r.status_code, 200)


class TransferCreateViewTests(ViewTestCaseMixin, TestCase):
    url_name = 'kfet.transfers.create'
    url_expected = '/k-fet/transfers/new'

    auth_user = 'team'
    auth_forbidden = [None, 'user']

    def test_ok(self):
        r = self.client.get(self.url)
        self.assertEqual(r.status_code, 200)


class TransferPerformViewTests(ViewTestCaseMixin, TestCase):
    url_name = 'kfet.transfers.perform'
    url_expected = '/k-fet/transfers/perform'

    http_methods = ['POST']

    auth_user = 'team1'
    auth_forbidden = [None, 'user', 'team']

    def get_users_extra(self):
        return {
            'team1': create_team('team1', '101', perms=[
                # Required
                'kfet.add_transfer',
                # Convenience
                'kfet.perform_negative_operations',
            ]),
        }

    @property
    def post_data(self):
        return {
            # General
            'comment': '',
            # Formset management
            'form-TOTAL_FORMS': '10',
            'form-INITIAL_FORMS': '0',
            'form-MIN_NUM_FORMS': '1',
            'form-MAX_NUM_FORMS': '1000',
            # Transfer 1
            'form-0-from_acc': str(self.accounts['user'].pk),
            'form-0-to_acc': str(self.accounts['team'].pk),
            'form-0-amount': '3.5',
            # Transfer 2
            'form-1-from_acc': str(self.accounts['team'].pk),
            'form-1-to_acc': str(self.accounts['team1'].pk),
            'form-1-amount': '2.4',
        }

    def test_ok(self):
        r = self.client.post(self.url, self.post_data)
        self.assertEqual(r.status_code, 200)

        user = self.accounts['user']
        user.refresh_from_db()
        self.assertEqual(user.balance, Decimal('-3.5'))

        team = self.accounts['team']
        team.refresh_from_db()
        self.assertEqual(team.balance, Decimal('1.1'))

        team1 = self.accounts['team1']
        team1.refresh_from_db()
        self.assertEqual(team1.balance, Decimal('2.4'))


class InventoryListViewTests(ViewTestCaseMixin, TestCase):
    url_name = 'kfet.inventory'
    url_expected = '/k-fet/inventaires/'

    auth_user = 'team'
    auth_forbidden = [None, 'user']

    def setUp(self):
        super().setUp()
        self.inventory = Inventory.objects.create(
            by=self.accounts['team'],
        )
        category = ArticleCategory.objects.create(name='Category')
        article = Article.objects.create(
            name='Article',
            category=category,
        )
        InventoryArticle.objects.create(
            inventory=self.inventory,
            article=article,
            stock_old=5,
            stock_new=0,
        )

    def test_ok(self):
        r = self.client.get(self.url)
        self.assertEqual(r.status_code, 200)

        inventories = r.context['inventories']
        self.assertQuerysetEqual(
            inventories,
            map(repr, [self.inventory]),
        )


class InventoryCreateViewTests(ViewTestCaseMixin, TestCase):
    url_name = 'kfet.inventory.create'
    url_expected = '/k-fet/inventaires/new'

    http_methods = ['GET', 'POST']

    auth_user = 'team'
    auth_forbidden = [None, 'user']

    def get_users_extra(self):
        return {
            'team1': create_team('team1', '101', perms=[
                'kfet.add_inventory',
            ]),
        }

    @property
    def post_data(self):
        return {
            # Formset management
            'form-TOTAL_FORMS': '2',
            'form-INITIAL_FORMS': '2',
            'form-MIN_NUM_FORMS': '0',
            'form-MAX_NUM_FORMS': '1000',
            # Article 1
            'form-0-article': str(self.article1.pk),
            'form-0-stock_new': '5',
            # Article 2
            'form-1-article': str(self.article2.pk),
            'form-1-stock_new': '10',
        }

    def setUp(self):
        super().setUp()
        category = ArticleCategory.objects.create(name='Category')
        self.article1 = Article.objects.create(
            category=category,
            name='Article 1',
        )
        self.article2 = Article.objects.create(
            category=category,
            name='Article 2',
        )

    def test_get_ok(self):
        r = self.client.get(self.url)
        self.assertEqual(r.status_code, 200)

    def test_post_ok(self):
        client = Client()
        client.login(username='team1', password='team1')

        r = client.post(self.url, self.post_data)
        self.assertRedirects(r, reverse('kfet.inventory'))

    def test_post_forbidden(self):
        r = self.client.post(self.url, self.post_data)
        self.assertForbiddenKfet(r)


class InventoryReadViewTests(ViewTestCaseMixin, TestCase):
    url_name = 'kfet.inventory.read'

    auth_user = 'team'
    auth_forbidden = [None, 'user']

    @property
    def url_kwargs(self):
        return {'pk': self.inventory.pk}

    @property
    def url_expected(self):
        return '/k-fet/inventaires/{}'.format(self.inventory.pk)

    def setUp(self):
        super().setUp()
        self.inventory = Inventory.objects.create(
            by=self.accounts['team'],
        )
        category = ArticleCategory.objects.create(name='Category')
        article = Article.objects.create(
            name='Article',
            category=category,
        )
        InventoryArticle.objects.create(
            inventory=self.inventory,
            article=article,
            stock_old=5,
            stock_new=0,
        )

    def test_ok(self):
        r = self.client.get(self.url)
        self.assertEqual(r.status_code, 200)


class OrderListViewTests(ViewTestCaseMixin, TestCase):
    url_name = 'kfet.order'
    url_expected = '/k-fet/orders/'

    auth_user = 'team'
    auth_forbidden = [None, 'user']

    def setUp(self):
        super().setUp()
        category = ArticleCategory.objects.create(name='Category')
        article = Article.objects.create(name='Article', category=category)

        supplier = Supplier.objects.create(name='Supplier')
        SupplierArticle.objects.create(supplier=supplier, article=article)

        self.order = Order.objects.create(supplier=supplier)
        OrderArticle.objects.create(
            order=self.order,
            article=article,
            quantity_ordered=24,
        )

    def test_ok(self):
        r = self.client.get(self.url)
        self.assertEqual(r.status_code, 200)

        orders = r.context['orders']
        self.assertQuerysetEqual(
            orders,
            map(repr, [self.order]),
        )


class OrderReadViewTests(ViewTestCaseMixin, TestCase):
    url_name = 'kfet.order.read'

    auth_user = 'team'
    auth_forbidden = [None, 'user']

    @property
    def url_kwargs(self):
        return {'pk': self.order.pk}

    @property
    def url_expected(self):
        return '/k-fet/orders/{}'.format(self.order.pk)

    def setUp(self):
        super().setUp()
        category = ArticleCategory.objects.create(name='Category')
        article = Article.objects.create(name='Article', category=category)

        supplier = Supplier.objects.create(name='Supplier')
        SupplierArticle.objects.create(supplier=supplier, article=article)

        self.order = Order.objects.create(supplier=supplier)
        OrderArticle.objects.create(
            order=self.order,
            article=article,
            quantity_ordered=24,
        )

    def test_ok(self):
        r = self.client.get(self.url)
        self.assertEqual(r.status_code, 200)


class SupplierUpdateViewTests(ViewTestCaseMixin, TestCase):
    url_name = 'kfet.order.supplier.update'

    http_methods = ['GET', 'POST']

    auth_user = 'team'
    auth_forbidden = [None, 'user']

    @property
    def url_kwargs(self):
        return {'pk': self.supplier.pk}

    @property
    def url_expected(self):
        return '/k-fet/orders/suppliers/{}/edit'.format(self.supplier.pk)

    def get_users_extra(self):
        return {
            'team1': create_team('team1', '101', perms=[
                'kfet.change_supplier',
            ]),
        }

    @property
    def post_data(self):
        return {
            'name': 'The Supplier',
            'phone': '',
            'comment': '',
            'address': '',
            'email': '',
        }

    def setUp(self):
        super().setUp()
        self.supplier = Supplier.objects.create(name='Supplier')

    def test_get_ok(self):
        r = self.client.get(self.url)
        self.assertEqual(r.status_code, 200)

    def test_post_ok(self):
        client = Client()
        client.login(username='team1', password='team1')

        r = client.post(self.url, self.post_data)
        self.assertRedirects(r, reverse('kfet.order'))

        self.supplier.refresh_from_db()
        self.assertEqual(self.supplier.name, 'The Supplier')

    def test_post_forbidden(self):
        r = self.client.post(self.url, self.post_data)
        self.assertForbiddenKfet(r)


class OrderCreateViewTests(ViewTestCaseMixin, TestCase):
    url_name = 'kfet.order.new'

    auth_user = 'team'
    auth_forbidden = [None, 'user']

    @property
    def url_kwargs(self):
        return {'pk': self.supplier.pk}

    @property
    def url_expected(self):
        return '/k-fet/orders/suppliers/{}/new-order'.format(self.supplier.pk)

    def get_users_extra(self):
        return {
            'team1': create_team('team1', '101', perms=['kfet.add_order']),
        }

    @property
    def post_data(self):
        return {
            # Formset management
            'form-TOTAL_FORMS': '1',
            'form-INITIAL_FORMS': '1',
            'form-MIN_NUM_FORMS': '0',
            'form-MAX_NUM_FORMS': '1000',
            # Article
            'form-0-article': self.article.pk,
            'form-0-quantity_ordered': '20',
        }

    def setUp(self):
        super().setUp()
        category = ArticleCategory.objects.create(name='Category')
        self.article = Article.objects.create(
            name='Article',
            category=category,
        )

        self.supplier = Supplier.objects.create(name='Supplier')
        SupplierArticle.objects.create(
            supplier=self.supplier,
            article=self.article,
        )

    def test_get_ok(self):
        r = self.client.get(self.url)
        self.assertEqual(r.status_code, 200)

    @mock.patch('django.utils.timezone.now')
    def test_post_ok(self, mock_now):
        mock_now.return_value = self.now

        client = Client()
        client.login(username='team1', password='team1')

        r = client.post(self.url, self.post_data)

        order = Order.objects.get(at=self.now)

        self.assertRedirects(r, reverse('kfet.order.read', args=[order.pk]))

    def test_post_forbidden(self):
        r = self.client.post(self.url, self.post_data)
        self.assertForbiddenKfet(r)


class OrderToInventoryViewTests(ViewTestCaseMixin, TestCase):
    url_name = 'kfet.order.to_inventory'

    http_methods = ['GET', 'POST']

    auth_user = 'team'
    auth_forbidden = [None, 'user']

    @property
    def url_kwargs(self):
        return {'pk': self.order.pk}

    @property
    def url_expected(self):
        return '/k-fet/orders/{}/to_inventory'.format(self.order.pk)

    def get_users_extra(self):
        return {
            'team1': create_team('team1', '101', perms=[
                'kfet.order_to_inventory',
            ]),
        }

    @property
    def post_data(self):
        return {
            # Formset mangaement
            'form-TOTAL_FORMS': '1',
            'form-INITIAL_FORMS': '1',
            'form-MIN_NUM_FORMS': '0',
            'form-MAX_NUM_FORMS': '1000',
            # Article 1
            'form-0-article': self.article.pk,
            'form-0-quantity_received': '20',
            'form-0-price_HT': '',
            'form-0-TVA': '',
            'form-0-rights': '',
        }

    def setUp(self):
        super().setUp()
        category = ArticleCategory.objects.create(name='Category')
        self.article = Article.objects.create(
            name='Article',
            category=category,
        )

        supplier = Supplier.objects.create(name='Supplier')
        SupplierArticle.objects.create(supplier=supplier, article=self.article)

        self.order = Order.objects.create(supplier=supplier)
        OrderArticle.objects.create(
            order=self.order,
            article=self.article,
            quantity_ordered=24,
        )

    def test_get_ok(self):
        r = self.client.get(self.url)
        self.assertEqual(r.status_code, 200)

    @mock.patch('django.utils.timezone.now')
    def test_post_ok(self, mock_now):
        mock_now.return_value = self.now

        client = Client()
        client.login(username='team1', password='team1')

        r = client.post(self.url, self.post_data)
        self.assertRedirects(r, reverse('kfet.order'))

        inventory = Inventory.objects.first()

        self.assertInstanceExpected(inventory, {
            'by': self.accounts['team1'],
            'at': self.now,
            'order': self.order,
        })
        self.assertQuerysetEqual(
            inventory.articles.all(),
            map(repr, [self.article]),
        )

        compte = InventoryArticle.objects.get(article=self.article)

        self.assertInstanceExpected(compte, {
            'stock_old': 0,
            'stock_new': 20,
            'stock_error': 0,
        })

    def test_post_forbidden(self):
        r = self.client.post(self.url, self.post_data)
        self.assertForbiddenKfet(r)