inventree/InvenTree
2
0
Fork 0
mirror of https://github.com/inventree/InvenTree.git synced 2025-10-31 05:05:42 +00:00
Code Activity

Add more unit testing for BOM file upload

Browse Source 
- Test "levels" functionality
- Test part guessing / introspection
This commit is contained in:
Oliver
2022-02-10 00:13:37 +11:00
parent ffb319e136
commit f460b14014
1 changed files with 83 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

83
InvenTree/part/test_bom_import.py
View File

@@ -35,6 +35,7 @@ class BomUploadTest(InvenTreeAPITestCase):
for i in range(10):
Part.objects.create(
name=f"Component {i}",
IPN=f"CMP_{i}",
description="A subcomponent that can be used in a BOM",
component=True,
assembly=False,
@@ -213,3 +214,85 @@ class BomUploadTest(InvenTreeAPITestCase):
for idx, row in enumerate(response.data['rows'][:-1]):
self.assertEqual(str(row['part']), str(components[idx].pk))
def test_part_guess(self):
"""
Test part 'guessing' when PK values are not supplied
"""
dataset = tablib.Dataset()
# Should be able to 'guess' the part from the name
dataset.headers = ['part_name', 'quantity']
components = Part.objects.filter(component=True)
for idx, cmp in enumerate(components):
dataset.append([
f"Component {idx}",
10,
])
response = self.post_bom(
'test.csv',
bytes(dataset.csv, 'utf8'),
expected_code=201,
)
rows = response.data['rows']
self.assertEqual(len(rows), 10)
for idx in range(10):
self.assertEqual(rows[idx]['part'], components[idx].pk)
# Should also be able to 'guess' part by the IPN value
dataset = tablib.Dataset()
dataset.headers = ['part_ipn', 'quantity']
for idx, cmp in enumerate(components):
dataset.append([
f"CMP_{idx}",
10,
])
response = self.post_bom(
'test.csv',
bytes(dataset.csv, 'utf8'),
expected_code=201,
)
rows = response.data['rows']
self.assertEqual(len(rows), 10)
for idx in range(10):
self.assertEqual(rows[idx]['part'], components[idx].pk)
def test_levels(self):
"""
Test that multi-level BOMs are correctly handled during upload
"""
dataset = tablib.Dataset()
dataset.headers = ['level', 'part', 'quantity']
components = Part.objects.filter(component=True)
for idx, cmp in enumerate(components):
dataset.append([
idx % 3,
cmp.pk,
2,
])
response = self.post_bom(
'test.csv',
bytes(dataset.csv, 'utf8'),
expected_code=201,
)
# Only parts at index 1, 4, 7 should have been returned
self.assertEqual(len(response.data['rows']), 3)