def setUp(self):
self.johndoe = get_user_model().objects.create_user('johndoe')
self.hansmueller = get_user_model().objects.create_user('hansmueller')
self.uk = Country.objects.create(iso_3166_1_a2='GB',
name="UNITED KINGDOM")
self.at = Country.objects.create(iso_3166_1_a2='AT', name="AUSTRIA")
self.de = Country.objects.create(iso_3166_1_a2='DE', name="GERMANY")
def setUp(self):
self.johndoe = get_user_model().objects.create_user('johndoe')
self.hansmueller = get_user_model().objects.create_user('hansmueller')
self.uk = Country.objects.create(
iso_3166_1_a2='GB', name="UNITED KINGDOM")
self.at = Country.objects.create(
iso_3166_1_a2='AT', name="AUSTRIA")
self.de = Country.objects.create(
iso_3166_1_a2='DE', name="GERMANY")
def step(context):
from django_dynamic_fixture import G
from oscar.core.compat import get_user_model
User = get_user_model()
context.user = G(User)
from django_webtest import DjangoTestApp
def test_bankcard_card_correct_save(self):
# issue #1486
user_klass = get_user_model()
user = user_klass.objects.create_user('_', '*****@*****.**', 'pwd')
end = datetime.date(day=1, month=1, year=2010)
bankcard = Bankcard.objects.create(
user=user, number="5500000000000004", expiry_date=end)
saved_bankcard = Bankcard.objects.get(id=bankcard.id)
self.assertEqual('Mastercard', saved_bankcard.card_type)
class UserFactory(factory.DjangoModelFactory):
FACTORY_FOR = get_user_model()
email = '*****@*****.**'
first_name = 'joseph'
last_name = 'winterbottom'
username = '******'
password = factory.PostGenerationMethodCall('set_password', 'skelebrain')
is_active = True
is_superuser = False
is_staff = False
def setUp(self):
self.johndoe = get_user_model().objects.create_user('johndoe')
self.uk = Country.objects.create(
iso_3166_1_a2='GB', name="UNITED KINGDOM")
self.address = self.model()
self.address.user = self.johndoe
self.address.first_name = "John"
self.address.last_name = "Doe"
self.address.line1 = "123 No Such Street"
self.address.line4 = "Brighton"
self.address.postcode = "BN1 6XX"
self.address.country = self.uk
class ApproverManagementView(generic.ListView):
template_name = 'oscar_approval/dashboard/approver_management.html'
model = get_user_model()
form_class = forms.UserSearchForm
def get_queryset(self):
qs = self.model.objects.none()
if self.form.is_bound:
cd = self.form.cleaned_data
if not cd['email'] and not cd['name']:
return self.model.objects.none()
qs = (self.model.objects.all().select_related('user'))
if cd['email']:
qs = qs.filter(user__email__startswith=cd['email'])
if cd['name']:
parts = cd['name'].split()
if len(parts) == 2:
qs = qs.filter(
Q(user__first_name__istartswith=parts[0])
| Q(user__last_name__istartswith=parts[1])).distinct()
else:
qs = qs.filter(
Q(user__first_name__istartswith=cd['name'])
| Q(user__last_name__istartswith=cd['name'])).distinct(
)
return qs
def get_context_data(self, *args, **kwargs):
ctx = super(ApproverManagementView,
self).get_context_data(*args, **kwargs)
ctx['form'] = getattr(self, 'form', None) or self.form_class()
ctx['approvers'] = (self.model.objects.filter(
is_order_approver=True).select_related('user'))
return ctx
def get(self, request, *args, **kwargs):
self.form = self.form_class(request.GET or None)
if not self.form.is_valid() and self.form.is_bound:
ctx = self.get_context_data()
return self.render_to_response(ctx)
return super(ApproverManagementView, self).get(request, *args,
**kwargs)
class ApproverUpdateView(generic.UpdateView):
model = get_user_model()
def post(self, request, *args, **kwargs):
is_approver = bool(kwargs['is_approver'])
profile = self.get_object()
profile.is_order_approver = is_approver
profile.save()
messages.success(request, _(self.get_success_message(is_approver)))
return HttpResponseRedirect(self.get_success_url())
def get_success_message(self, is_approver):
if is_approver:
msg = 'Approver has been added.'
else:
msg = 'Approver has been removed.'
return msg
def get_success_url(self):
return reverse('dashboard:approver-management')
def setUp(self):
self.user_klass = get_user_model()
class Meta:
model = get_user_model()
class UserFactory(factory.DjangoModelFactory):
FACTORY_FOR = get_user_model()
username = factory.Sequence(
lambda n: "Peter The {0}".format(humanize.apnumber(n)))
class Meta:
model = get_user_model()
fields = ('id', 'email', 'first_name', 'last_name', 'display_text')
def setUp(self):
self.user_klass = get_user_model()
def test_can_be_created_without_error(self):
klass = compat.get_user_model()
try:
klass.objects.create_user('_', '*****@*****.**', 'pa55w0rd')
except Exception, e:
self.fail("Unable to create user model: %s" % e)
from django import forms
from django.utils.translation import gettext_lazy as _
from django.utils.translation import pgettext_lazy
from oscar.core.compat import get_user_model
from oscar.core.loading import get_model
User = get_user_model()
ProductAlert = get_model('customer', 'ProductAlert')
class UserSearchForm(forms.Form):
email = forms.CharField(required=False, label=_("Email"))
name = forms.CharField(
required=False, label=pgettext_lazy("User's name", "Name"))
class ProductAlertUpdateForm(forms.ModelForm):
def __init__(self, *args, **kwargs):
super().__init__(*args, **kwargs)
alert = kwargs['instance']
if alert.user:
# Remove 'unconfirmed' from list of available choices when editing
# an alert for a real user
choices = self.fields['status'].choices
del choices[0]
self.fields['status'].choices = choices
class Meta:
model = ProductAlert
from http import client as http_client
from django.contrib.auth.models import Permission
from django.urls import reverse
from django_webtest import WebTest
from purl import URL
from oscar.core.compat import get_user_model
User = get_user_model()
def add_permissions(user, permissions):
"""
Grant permissions to the passed user
:param permissions: e.g. ['partner.dashboard_access']
"""
for permission in permissions:
app_label, __, codename = permission.partition('.')
perm = Permission.objects.get(content_type__app_label=app_label,
codename=codename)
user.user_permissions.add(perm)
class WebTestCase(WebTest):
is_staff = False
is_anonymous = False
is_superuser = False
username = '******'
def test_can_be_created_without_error(self):
klass = compat.get_user_model()
try:
klass.objects.create_user('_', '*****@*****.**', 'pa55w0rd')
except Exception, e:
self.fail("Unable to create user model: %s" % e)
