From 0bc561782570636f8de0c817839dc26ef000b92e Mon Sep 17 00:00:00 2001 From: Oliver Walters Date: Thu, 25 Apr 2019 00:59:34 +1000 Subject: [PATCH] Reverse url lookup for Part model --- InvenTree/part/models.py | 2 +- InvenTree/part/test_part.py | 22 ++++++++++++++++++++++ InvenTree/part/tests.py | 3 --- 3 files changed, 23 insertions(+), 4 deletions(-) create mode 100644 InvenTree/part/test_part.py delete mode 100644 InvenTree/part/tests.py diff --git a/InvenTree/part/models.py b/InvenTree/part/models.py index c9e4ae0500..81e05567bd 100644 --- a/InvenTree/part/models.py +++ b/InvenTree/part/models.py @@ -89,7 +89,7 @@ class Part(models.Model): """ def get_absolute_url(self): - return '/part/{id}/'.format(id=self.id) + return reverse('part-detail', kwargs={'pk': self.id}) # Short name of the part name = models.CharField(max_length=100, unique=True, help_text='Part name (must be unique)') diff --git a/InvenTree/part/test_part.py b/InvenTree/part/test_part.py new file mode 100644 index 0000000000..540c98bfa4 --- /dev/null +++ b/InvenTree/part/test_part.py @@ -0,0 +1,22 @@ +from django.test import TestCase + +from .models import Part, PartCategory + +class SimplePartTest(TestCase): + + def setUp(self): + + cat = PartCategory.objects.create(name='TLC', description='Top level category') + + self.px = Part.objects.create(name='x', description='A part called x', buildable=True) + self.py = Part.objects.create(name='y', description='A part called y', consumable=False) + self.pz = Part.objects.create(name='z', description='A part called z', category=cat) + + def test_metadata(self): + self.assertEqual(self.px.name, 'x') + self.assertEqual(self.py.get_absolute_url(), '/part/2/') + self.assertEqual(str(self.pz), 'z - A part called z') + + def test_category(self): + self.assertEqual(self.px.category_path, '') + self.assertEqual(self.pz.category_path, 'TLC') \ No newline at end of file diff --git a/InvenTree/part/tests.py b/InvenTree/part/tests.py deleted file mode 100644 index a79ca8be56..0000000000 --- a/InvenTree/part/tests.py +++ /dev/null @@ -1,3 +0,0 @@ -# from django.test import TestCase - -# Create your tests here.