add get_absolute_url method to User model
This commit is contained in:
parent
1e406621ca
commit
b3af7881b2
|
@ -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)
|
||||
|
|
|
@ -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) + "/")
|
Loading…
Reference in New Issue