arona/PyRIGS
1
0
Fork 0
mirror of https://github.com/nottinghamtec/PyRIGS.git synced 2026-01-23 16:32:15 +00:00
Code Issues Projects Releases Wiki Activity

Fix model tests for vat rate fixture

Browse Source
This commit is contained in:
FreneticScribbler
2021-02-04 13:28:37 +00:00
parent 9279131edf
commit 6cb3d1855a
1 changed files with 13 additions and 23 deletions
Download Patch File Download Diff File Expand all files Collapse all files

36
RIGS/tests/test_models.py
View File

@@ -2,6 +2,7 @@ from datetime import date, timedelta, datetime, time
from decimal import * from decimal import *
import pytz import pytz
import pytest
from django.conf import settings from django.conf import settings
from django.test import TestCase from django.test import TestCase
from reversion import revisions as reversion from reversion import revisions as reversion
@@ -9,30 +10,22 @@ from reversion import revisions as reversion
from RIGS import models from RIGS import models
class ProfileTestCase(TestCase): def test_str():
def test_str(self): profile = models.Profile(first_name='Test', last_name='Case')
profile = models.Profile(first_name='Test', last_name='Case') assert str(profile) == 'Test Case'
self.assertEqual(str(profile), 'Test Case') profile.initials = 'TC'
profile.initials = 'TC' assert str(profile) == 'Test Case "TC"'
self.assertEqual(str(profile), 'Test Case "TC"')
class VatRateTestCase(TestCase): @pytest.mark.django_db
@classmethod def test_find_correct(vat_rate):
def setUpTestData(cls): new_rate = models.VatRate.objects.create(start_at='2016-03-01', rate=0.15, comment='test2')
cls.rates = { assert str(models.VatRate.objects.find_rate('2015-03-01')) == vat_rate
0: models.VatRate.objects.create(start_at='2014-03-01', rate=0.20, comment='test1'), assert str(models.VatRate.objects.find_rate('2016-03-01')) == new_rate
1: models.VatRate.objects.create(start_at='2016-03-01', rate=0.15, comment='test2'),
}
def test_find_correct(self):
r = models.VatRate.objects.find_rate('2015-03-01')
self.assertEqual(r, self.rates[0])
r = models.VatRate.objects.find_rate('2016-03-01')
self.assertEqual(r, self.rates[1])
def test_percent_correct(self): def test_percent_correct(vat_rate):
self.assertEqual(self.rates[0].as_percent, 20) assert vat_rate.as_percent == 20
class EventTestCase(TestCase): class EventTestCase(TestCase):
@@ -41,7 +34,6 @@ class EventTestCase(TestCase):
cls.all_events = set(range(1, 18)) cls.all_events = set(range(1, 18))
cls.current_events = (1, 2, 3, 6, 7, 8, 10, 11, 12, 14, 15, 16, 18) cls.current_events = (1, 2, 3, 6, 7, 8, 10, 11, 12, 14, 15, 16, 18)
cls.not_current_events = set(cls.all_events) - set(cls.current_events) cls.not_current_events = set(cls.all_events) - set(cls.current_events)
cls.vatrate = models.VatRate.objects.create(start_at='2014-03-05', rate=0.20, comment='test1') cls.vatrate = models.VatRate.objects.create(start_at='2014-03-05', rate=0.20, comment='test1')
cls.profile = models.Profile.objects.create(username="testuser1", email="1@test.com") cls.profile = models.Profile.objects.create(username="testuser1", email="1@test.com")
@@ -331,7 +323,6 @@ class EventItemTestCase(TestCase):
class EventPricingTestCase(TestCase): class EventPricingTestCase(TestCase):
def setUp(self): def setUp(self):
models.VatRate.objects.create(rate=0.20, comment="TP V1", start_at='2013-01-01')
models.VatRate.objects.create(rate=0.10, comment="TP V2", start_at=date.today() - timedelta(days=1)) models.VatRate.objects.create(rate=0.10, comment="TP V2", start_at=date.today() - timedelta(days=1))
self.e1 = models.Event.objects.create(name="TP E1", start_date=date.today() - timedelta(days=2)) self.e1 = models.Event.objects.create(name="TP E1", start_date=date.today() - timedelta(days=2))
self.e2 = models.Event.objects.create(name="TP E2", start_date=date.today()) self.e2 = models.Event.objects.create(name="TP E2", start_date=date.today())
@@ -364,7 +355,6 @@ class EventPricingTestCase(TestCase):
class EventAuthorisationTestCase(TestCase): class EventAuthorisationTestCase(TestCase):
def setUp(self): def setUp(self):
models.VatRate.objects.create(rate=0.20, comment="TP V1", start_at='2013-01-01')
self.profile = models.Profile.objects.get_or_create( self.profile = models.Profile.objects.get_or_create(
first_name='Test', first_name='Test',
last_name='TEC User', last_name='TEC User',