mirror of
https://github.com/inventree/InvenTree
synced 2024-08-30 18:33:04 +00:00
Use reverse to get Build URL
- Add test cases to Build
This commit is contained in:
parent
d46afcdfe8
commit
5098712d9c
@ -3,6 +3,7 @@ from __future__ import unicode_literals
|
|||||||
|
|
||||||
from django.utils.translation import ugettext as _
|
from django.utils.translation import ugettext as _
|
||||||
|
|
||||||
|
from django.urls import reverse
|
||||||
from django.db import models
|
from django.db import models
|
||||||
from django.core.validators import MinValueValidator
|
from django.core.validators import MinValueValidator
|
||||||
|
|
||||||
@ -14,7 +15,7 @@ class Build(models.Model):
|
|||||||
"""
|
"""
|
||||||
|
|
||||||
def get_absolute_url(self):
|
def get_absolute_url(self):
|
||||||
return '/build/{pk}/'.format(pk=self.id)
|
return reverse('build-detail', kwargs={'pk': self.id})
|
||||||
|
|
||||||
# Build status codes
|
# Build status codes
|
||||||
PENDING = 10 # Build is pending / active
|
PENDING = 10 # Build is pending / active
|
||||||
|
@ -1,6 +1,59 @@
|
|||||||
# -*- coding: utf-8 -*-
|
# -*- coding: utf-8 -*-
|
||||||
from __future__ import unicode_literals
|
from __future__ import unicode_literals
|
||||||
|
|
||||||
# from django.test import TestCase
|
from django.test import TestCase
|
||||||
|
|
||||||
|
from .models import Build
|
||||||
|
from part.models import Part
|
||||||
|
|
||||||
# Create your tests here.
|
# Create your tests here.
|
||||||
|
|
||||||
|
|
||||||
|
class BuildTestSimple(TestCase):
|
||||||
|
|
||||||
|
def setUp(self):
|
||||||
|
part = Part.objects.create(name='Test part',
|
||||||
|
description='Simple description')
|
||||||
|
Build.objects.create(part=part,
|
||||||
|
batch='B1',
|
||||||
|
status=Build.PENDING,
|
||||||
|
title='Building 7 parts',
|
||||||
|
quantity=7,
|
||||||
|
notes='Some simple notes')
|
||||||
|
|
||||||
|
Build.objects.create(part=part,
|
||||||
|
batch='B2',
|
||||||
|
status=Build.COMPLETE,
|
||||||
|
title='Building 21 parts',
|
||||||
|
quantity=21,
|
||||||
|
notes='Some simple notes')
|
||||||
|
|
||||||
|
def test_build_objects(self):
|
||||||
|
# Ensure the Build objects were correctly created
|
||||||
|
self.assertEqual(Build.objects.count(), 2)
|
||||||
|
b = Build.objects.get(pk=2)
|
||||||
|
self.assertEqual(b.batch, 'B2')
|
||||||
|
self.assertEqual(b.quantity, 21)
|
||||||
|
|
||||||
|
def test_url(self):
|
||||||
|
b1 = Build.objects.get(pk=1)
|
||||||
|
self.assertEqual(b1.get_absolute_url(), '/build/1/')
|
||||||
|
|
||||||
|
def test_is_complete(self):
|
||||||
|
b1 = Build.objects.get(pk=1)
|
||||||
|
b2 = Build.objects.get(pk=2)
|
||||||
|
|
||||||
|
self.assertEqual(b1.is_complete, False)
|
||||||
|
self.assertEqual(b2.is_complete, True)
|
||||||
|
|
||||||
|
def test_is_active(self):
|
||||||
|
b1 = Build.objects.get(pk=1)
|
||||||
|
b2 = Build.objects.get(pk=2)
|
||||||
|
|
||||||
|
self.assertEqual(b1.is_active, True)
|
||||||
|
self.assertEqual(b2.is_active, False)
|
||||||
|
|
||||||
|
def test_required_parts(self):
|
||||||
|
# TODO - Generate BOM for test part
|
||||||
|
pass
|
||||||
|
|
Loading…
Reference in New Issue
Block a user