add get_absolute_url method to User model

This commit is contained in:
Andreas Zweili 2020-01-01 19:54:37 +01:00
parent 1e406621ca
commit b3af7881b2
2 changed files with 14 additions and 0 deletions

View File

@ -19,6 +19,10 @@ class User(models.Model):
def __str__(self):
return self.name + " " + self.first_name
def get_absolute_url(self):
from django.urls import reverse
return reverse('user', args=[str(self.id)])
class UserInAdGroup(models.Model):
user = models.ForeignKey(User, on_delete=models.CASCADE)

View File

@ -0,0 +1,10 @@
import pytest
from mixer.backend.django import mixer
pytestmark = pytest.mark.django_db
def test_user_reverse_url():
user = mixer.blend('inventory.User')
assert (user.get_absolute_url()
== "/user/" + str(user.id) + "/")