mirror of
https://github.com/inventree/InvenTree.git
synced 2025-08-10 05:40:55 +00:00
173 lines
4.3 KiB
Python
173 lines
4.3 KiB
Python
# -*- coding: utf-8 -*-
|
|
from __future__ import unicode_literals
|
|
|
|
from django.test import TestCase
|
|
from django.urls import reverse
|
|
|
|
from django.contrib.auth import get_user_model
|
|
from django.contrib.auth.models import Group
|
|
|
|
from datetime import datetime, timedelta
|
|
|
|
from .models import Build
|
|
from stock.models import StockItem
|
|
|
|
from InvenTree.status_codes import BuildStatus
|
|
|
|
|
|
class BuildTestSimple(TestCase):
|
|
|
|
fixtures = [
|
|
'category',
|
|
'part',
|
|
'location',
|
|
'build',
|
|
]
|
|
|
|
def setUp(self):
|
|
# Create a user for auth
|
|
user = get_user_model()
|
|
user.objects.create_user('testuser', 'test@testing.com', 'password')
|
|
|
|
self.user = user.objects.get(username='testuser')
|
|
|
|
g = Group.objects.create(name='builders')
|
|
self.user.groups.add(g)
|
|
|
|
for rule in g.rule_sets.all():
|
|
if rule.name == 'build':
|
|
rule.can_change = True
|
|
rule.can_add = True
|
|
rule.can_delete = True
|
|
|
|
rule.save()
|
|
|
|
g.save()
|
|
|
|
self.client.login(username='testuser', password='password')
|
|
|
|
def test_build_objects(self):
|
|
# Ensure the Build objects were correctly created
|
|
self.assertEqual(Build.objects.count(), 5)
|
|
b = Build.objects.get(pk=2)
|
|
self.assertEqual(b.batch, 'B2')
|
|
self.assertEqual(b.quantity, 21)
|
|
|
|
self.assertEqual(str(b), 'BO0002')
|
|
|
|
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)
|
|
|
|
self.assertEqual(b2.status, BuildStatus.COMPLETE)
|
|
|
|
def test_overdue(self):
|
|
"""
|
|
Test overdue status functionality
|
|
"""
|
|
|
|
today = datetime.now().date()
|
|
|
|
build = Build.objects.get(pk=1)
|
|
self.assertFalse(build.is_overdue)
|
|
|
|
build.target_date = today - timedelta(days=1)
|
|
build.save()
|
|
self.assertTrue(build.is_overdue)
|
|
|
|
build.target_date = today + timedelta(days=80)
|
|
build.save()
|
|
self.assertFalse(build.is_overdue)
|
|
|
|
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
|
|
|
|
def test_cancel_build(self):
|
|
""" Test build cancellation function """
|
|
|
|
build = Build.objects.get(id=1)
|
|
|
|
self.assertEqual(build.status, BuildStatus.PENDING)
|
|
|
|
build.cancel_build(self.user)
|
|
|
|
self.assertEqual(build.status, BuildStatus.CANCELLED)
|
|
|
|
|
|
class TestBuildViews(TestCase):
|
|
""" Tests for Build app views """
|
|
|
|
fixtures = [
|
|
'category',
|
|
'part',
|
|
'location',
|
|
'build',
|
|
]
|
|
|
|
def setUp(self):
|
|
super().setUp()
|
|
|
|
# Create a user
|
|
user = get_user_model()
|
|
self.user = user.objects.create_user('username', 'user@email.com', 'password')
|
|
|
|
g = Group.objects.create(name='builders')
|
|
self.user.groups.add(g)
|
|
|
|
for rule in g.rule_sets.all():
|
|
if rule.name == 'build':
|
|
rule.can_change = True
|
|
rule.can_add = True
|
|
rule.can_delete = True
|
|
|
|
rule.save()
|
|
|
|
g.save()
|
|
|
|
self.client.login(username='username', password='password')
|
|
|
|
# Create a build output for build # 1
|
|
self.build = Build.objects.get(pk=1)
|
|
|
|
self.output = StockItem.objects.create(
|
|
part=self.build.part,
|
|
quantity=self.build.quantity,
|
|
build=self.build,
|
|
is_building=True,
|
|
)
|
|
|
|
def test_build_index(self):
|
|
""" test build index view """
|
|
|
|
response = self.client.get(reverse('build-index'))
|
|
self.assertEqual(response.status_code, 200)
|
|
|
|
def test_build_detail(self):
|
|
""" Test the detail view for a Build object """
|
|
|
|
pk = 1
|
|
|
|
response = self.client.get(reverse('build-detail', args=(pk,)))
|
|
self.assertEqual(response.status_code, 200)
|
|
|
|
build = Build.objects.get(pk=pk)
|
|
|
|
content = str(response.content)
|
|
|
|
self.assertIn(build.title, content)
|