123456789101112131415161718192021222324252627282930313233 |
- import datetime
- from django.contrib.auth import get_user_model
- from django.test import TestCase
- from ..models import Cost, Document, Good, Service, GoodUse, CostUse
- User = get_user_model()
- class ViewsTests(TestCase):
- def setUp(self):
- user = User(username='user', is_staff=True, is_active=True)
- user.set_password('password')
- user.save()
- self.doc = Document.objects.create(name='budget')
- rent = Cost.objects.create(name='Rent', price=100, document=self.doc)
- server = Good.objects.create(
- name='Server',
- price=1000,
- provisioning_duration=datetime.timedelta(days=36*20.6),
- document=self.doc,
- )
- vpn = Service.objects.create(name='VPN', document=self.doc)
- GoodUse.objects.create(resource=server, service=vpn, share=0.5)
- CostUse.objects.create(resource=rent, service=vpn, share=0.5)
- def test_detail_service(self):
- assert self.client.login(username='user', password='password')
- response = self.client.get('/costs/services/1')
- self.assertEqual(response.status_code, 200)
|