kpsul/kfet/tests/test_models.py

59 lines
1.5 KiB
Python
Raw Permalink Normal View History

import datetime
2017-09-22 23:31:46 +02:00
from django.contrib.auth import get_user_model
from django.test import TestCase
from django.utils import timezone
from kfet.models import Account, Checkout
2017-09-22 23:31:46 +02:00
from .utils import create_user
2017-09-22 23:31:46 +02:00
User = get_user_model()
class AccountTests(TestCase):
def setUp(self):
self.account = Account(trigramme="000")
self.account.save({"username": "user"})
2017-09-22 23:31:46 +02:00
def test_password(self):
self.account.change_pwd("anna")
2017-09-22 23:31:46 +02:00
self.account.save()
self.assertEqual(Account.objects.get_by_password("anna"), self.account)
2017-09-22 23:31:46 +02:00
with self.assertRaises(Account.DoesNotExist):
Account.objects.get_by_password(None)
with self.assertRaises(Account.DoesNotExist):
Account.objects.get_by_password("bernard")
class CheckoutTests(TestCase):
def setUp(self):
self.now = timezone.now()
self.u = create_user()
self.u_acc = self.u.profile.account_kfet
self.c = Checkout(
created_by=self.u_acc,
valid_from=self.now,
valid_to=self.now + datetime.timedelta(days=1),
)
def test_initial_statement(self):
"""A statement is added with initial balance on creation."""
self.c.balance = 10
self.c.save()
st = self.c.statements.get()
self.assertEqual(st.balance_new, 10)
self.assertEqual(st.amount_taken, 0)
self.assertEqual(st.amount_error, 0)
# Saving again doesn't create a new statement.
self.c.save()
self.assertEqual(self.c.statements.count(), 1)