1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28
|
from django import test
from django.contrib.auth.models import User
from . import forms
class TestUserKwargModelFormMixin(test.TestCase):
"""
Tests for UserKwargModelFormMixin.
"""
def test_without_user_kwarg(self):
"""
It should be possible to create form without 'user' kwarg.
In that case 'user' attribute should be set to None.
"""
form = forms.FormWithUserKwarg()
assert form.user is None
def test_with_user_kwarg(self):
"""
Form's 'user' attribute should be set to value passed as 'user'
argument.
"""
user = User(username="test")
form = forms.FormWithUserKwarg(user=user)
assert form.user is user
|