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 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 = '******'