mirror of
https://github.com/jazzband/django-eav2.git
synced 2026-03-16 22:40:26 +00:00
96 lines
2.9 KiB
Python
96 lines
2.9 KiB
Python
from django.test import TestCase
|
|
from django.contrib.admin.sites import AdminSite
|
|
|
|
import eav
|
|
import sys
|
|
from eav.admin import *
|
|
from .models import Patient, M2MModel, ExampleModel
|
|
from eav.models import Attribute
|
|
from eav.forms import BaseDynamicEntityForm
|
|
from django.contrib import admin
|
|
from django.core.handlers.base import BaseHandler
|
|
from django.test.client import RequestFactory
|
|
from django.forms import ModelForm
|
|
|
|
|
|
class MockRequest(RequestFactory):
|
|
def request(self, **request):
|
|
"Construct a generic request object."
|
|
request = RequestFactory.request(self, **request)
|
|
handler = BaseHandler()
|
|
handler.load_middleware()
|
|
# BaseHandler_request_middleware is not set in Django2.0
|
|
# and removed in Django2.1
|
|
if sys.version_info[0] < 2:
|
|
for middleware_method in handler._request_middleware:
|
|
if middleware_method(request):
|
|
raise Exception("Couldn't create request mock object - "
|
|
"request middleware returned a response")
|
|
return request
|
|
|
|
|
|
class MockSuperUser:
|
|
def __init__(self):
|
|
self.is_active = True
|
|
self.is_staff = True
|
|
|
|
def has_perm(self, perm):
|
|
return True
|
|
|
|
|
|
request = MockRequest().request()
|
|
request.user = MockSuperUser()
|
|
|
|
|
|
class PatientForm(ModelForm):
|
|
class Meta:
|
|
model = Patient
|
|
fields = '__all__'
|
|
|
|
|
|
class M2MModelForm(ModelForm):
|
|
class Meta:
|
|
model = M2MModel
|
|
fields = '__all__'
|
|
|
|
|
|
class Forms(TestCase):
|
|
def setUp(self):
|
|
eav.register(Patient)
|
|
Attribute.objects.create(name='weight', datatype=Attribute.TYPE_FLOAT)
|
|
Attribute.objects.create(name='color', datatype=Attribute.TYPE_TEXT)
|
|
self.instance = Patient.objects.create(name='Jim Morrison')
|
|
self.site = AdminSite()
|
|
|
|
def test_fields(self):
|
|
admin = BaseEntityAdmin(Patient, self.site)
|
|
admin.form = BaseDynamicEntityForm
|
|
view = admin.change_view(request, str(self.instance.pk))
|
|
|
|
own_fields = 1
|
|
adminform = view.context_data['adminform']
|
|
|
|
self.assertEqual(
|
|
len(adminform.form.fields), Attribute.objects.count() + own_fields
|
|
)
|
|
|
|
def test_valid_submit(self):
|
|
self.instance.eav.color = 'Blue'
|
|
form = PatientForm(self.instance.__dict__, instance=self.instance)
|
|
jim = form.save()
|
|
|
|
self.assertEqual(jim.eav.color, 'Blue')
|
|
|
|
|
|
def test_invalid_submit(self):
|
|
form = PatientForm(dict(color='Blue'), instance=self.instance)
|
|
with self.assertRaises(ValueError):
|
|
jim = form.save()
|
|
|
|
|
|
def test_m2m(self):
|
|
m2mmodel = M2MModel.objects.create(name='name')
|
|
model = ExampleModel.objects.create(name='name')
|
|
form = M2MModelForm(dict(name='Lorem', models=[model.pk]), instance=m2mmodel)
|
|
form.save()
|
|
self.assertEqual(len(m2mmodel.models.all()), 1)
|