|
@@ -1,12 +1,20 @@
|
|
import datetime
|
|
import datetime
|
|
|
|
|
|
|
|
+from django.contrib.auth import get_user_model
|
|
from django.test import TestCase
|
|
from django.test import TestCase
|
|
|
|
|
|
from ..models import Cost, Document, Good, Service, GoodUse, CostUse
|
|
from ..models import Cost, Document, Good, Service, GoodUse, CostUse
|
|
|
|
|
|
|
|
|
|
|
|
+User = get_user_model()
|
|
|
|
+
|
|
|
|
+
|
|
class ViewsTests(TestCase):
|
|
class ViewsTests(TestCase):
|
|
def setUp(self):
|
|
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')
|
|
self.doc = Document.objects.create(name='budget')
|
|
rent = Cost.objects.create(name='Rent', price=100, document=self.doc)
|
|
rent = Cost.objects.create(name='Rent', price=100, document=self.doc)
|
|
server = Good.objects.create(
|
|
server = Good.objects.create(
|
|
@@ -20,5 +28,6 @@ class ViewsTests(TestCase):
|
|
CostUse.objects.create(resource=rent, service=vpn, share=0.5)
|
|
CostUse.objects.create(resource=rent, service=vpn, share=0.5)
|
|
|
|
|
|
def test_detail_service(self):
|
|
def test_detail_service(self):
|
|
- response = self.client.get('/transparency/services/1')
|
|
|
|
|
|
+ assert self.client.login(username='user', password='password')
|
|
|
|
+ response = self.client.get('/costs/services/1')
|
|
self.assertEqual(response.status_code, 200)
|
|
self.assertEqual(response.status_code, 200)
|